iot-master/main.go at main · god-jason/iot-master · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
package main

import (
	"embed"
	"os"
	"os/signal"
	"syscall"
	"time"

	_ "github.com/god-jason/iot-master/apis"
	"github.com/god-jason/iot-master/iot"
	"github.com/god-jason/iot-master/pkg/boot"
	_ "github.com/god-jason/iot-master/pkg/broker"
	_ "github.com/god-jason/iot-master/pkg/debug"
	"github.com/god-jason/iot-master/pkg/log"
	"github.com/god-jason/iot-master/pkg/menu"
	_ "github.com/god-jason/iot-master/pkg/oem"
	"github.com/god-jason/iot-master/pkg/page"
	"github.com/god-jason/iot-master/pkg/service"
	"github.com/god-jason/iot-master/pkg/store"
	_ "github.com/god-jason/iot-master/pkg/table"
	"github.com/god-jason/iot-master/pkg/version"
	"github.com/god-jason/iot-master/pkg/web"
	_ "github.com/god-jason/iot-master/weixin"
	"github.com/spf13/pflag"
	"github.com/spf13/viper"
)

//go:embed www/browser
var www embed.FS

func init() {
	//菜单
	menu.File("menu.json")

	//页面
	page.PagesFS = store.Dir("pages")

	//加载协议
	iot.Protocols = store.Dir("protocols")

	//前端页面
	web.StaticFS(www, "/", "www/browser", "index.html")
}

func Startup() error {
	viper.SetConfigName("iot-master")

	err := boot.Startup()
	if err != nil {
		//_ = boot.Shutdown()
		return err
	}

	//异步执行，避免堵塞
	go func() {
		//启动服务
		err := web.Serve()
		if err != nil {
			//安全退出
			//_ = boot.Shutdown()
			log.Error(err)
		}
	}()

	log.Info("main started")

	return nil
}

func Shutdown() error {
	log.Info("main shutdown")

	return boot.Shutdown()
}

func main() {

	help := pflag.BoolP("help", "h", false, "show help")
	install := pflag.BoolP("install", "i", false, "install as service")
	uninstall := pflag.BoolP("uninstall", "u", false, "uninstall service")
	showVersion := pflag.BoolP("version", "v", false, "show version")

	pflag.Parse()
	if *help {
		pflag.PrintDefaults()
		return
	}

	err := service.Register(Startup, Shutdown)
	if err != nil {
		log.Fatal(err)
	}

	if *install {
		log.Info("install service")
		err = service.Install()
		if err != nil {
			log.Fatal(err)
		}
	} else if *uninstall {
		log.Info("uninstall service")
		err = service.Uninstall()
		if err != nil {
			log.Fatal(err)
		}
		return
	} else if *showVersion {
		version.Print()
		return
	}

	sigs := make(chan os.Signal, 1)
	signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
	go func() {
		s := <-sigs
		log.Info("signal received ", s)

		//_ = boot.Shutdown()
		err := service.Stop()
		if err != nil {
			log.Error(err)
		}

		time.AfterFunc(10*time.Second, func() {
			os.Exit(0)
		})
	}()

	err = service.Run()
	if err != nil {
		log.Fatal(err)
	}

	println("bye")
}