|
|
@@ -1 +1,172 @@
|
|
|
package main
|
|
|
+
|
|
|
+import (
|
|
|
+ "context"
|
|
|
+ "errors"
|
|
|
+ "flag"
|
|
|
+ "fmt"
|
|
|
+ "net/http"
|
|
|
+ "os"
|
|
|
+ "rtu_linux_services/servicelib"
|
|
|
+ "time"
|
|
|
+
|
|
|
+ "github.com/kardianos/service"
|
|
|
+ "hnyfkj.com.cn/rtu/linux/baseapp"
|
|
|
+ "hnyfkj.com.cn/rtu/linux/utils/jsonrpc2"
|
|
|
+)
|
|
|
+
|
|
|
+var (
|
|
|
+ Version = "0.0.0.1"
|
|
|
+ BuildTime = ""
|
|
|
+)
|
|
|
+
|
|
|
+const (
|
|
|
+ network = "tcp" // http -> tcp
|
|
|
+ listenAddr = "127.0.0.1:7004"
|
|
|
+)
|
|
|
+
|
|
|
+type program struct {
|
|
|
+ name string
|
|
|
+ core_server *jsonrpc2.RPCServer
|
|
|
+ http_server *http.Server
|
|
|
+}
|
|
|
+
|
|
|
+// Start should not block. Do the actual work async.
|
|
|
+func (p *program) Start(s service.Service) error {
|
|
|
+ logCfgFile, err := servicelib.EnsureDefaultLogConfig()
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+
|
|
|
+ baseapp.InitLogger(logCfgFile)
|
|
|
+ baseapp.Logger.Infof("[%s] 开始运行, 程序版本: %s, 构建时间: %s", p.name, Version, BuildTime)
|
|
|
+
|
|
|
+ p.core_server, err = jsonrpc2.NewRPCServer(p.name, baseapp.Logger)
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+
|
|
|
+ methods := make(jsonrpc2.MethodMap)
|
|
|
+
|
|
|
+ //////////////////////////////////////////////////////////
|
|
|
+ _getBuildVer := servicelib.MakeGetBuildVer(
|
|
|
+ Version, BuildTime)
|
|
|
+ methods["basic.getBuildVer"] = _getBuildVer // 获取软件版本
|
|
|
+ _getLogLevel := servicelib.GetLogLevel
|
|
|
+ methods["basic.getLogLevel"] = _getLogLevel // 获取日志级别
|
|
|
+ _setLogLevel := servicelib.SetLogLevel
|
|
|
+ methods["basic.setLogLevel"] = _setLogLevel // 设置日志级别
|
|
|
+ _saveLogConf := servicelib.SaveLogConf
|
|
|
+ methods["basic.saveLogConf"] = _saveLogConf // 保存日志设置
|
|
|
+ //////////////////////////////////////////////////////////
|
|
|
+ methods["core.package.import"] = pkgImport ///// 导入安装包
|
|
|
+ methods["core.package.remove"] = pkgRemove ///// 删除安装包
|
|
|
+ //////////////////////////////////////////////////////////
|
|
|
+ methods["core.app.install"] = appInstall //安装用户应用
|
|
|
+ methods["core.app.uninstall"] = appUninstall //卸载当前应用
|
|
|
+ //////////////////////////////////////////////////////////
|
|
|
+ methods["core.getInstallInfo"] = getInstallInfo // 安装信息
|
|
|
+
|
|
|
+ err = p.core_server.RegisterMethods(methods)
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+
|
|
|
+ mux := http.NewServeMux()
|
|
|
+ mux.Handle("/rpc", p.core_server)
|
|
|
+
|
|
|
+ p.http_server = &http.Server{
|
|
|
+ Addr: listenAddr,
|
|
|
+ Handler: mux,
|
|
|
+ }
|
|
|
+
|
|
|
+ go p.run()
|
|
|
+
|
|
|
+ return nil
|
|
|
+}
|
|
|
+
|
|
|
+// Do work here
|
|
|
+func (p *program) run() {
|
|
|
+ baseapp.Logger.Infof(
|
|
|
+ "[%s] Http server starting on %s...",
|
|
|
+ p.name,
|
|
|
+ p.http_server.Addr,
|
|
|
+ )
|
|
|
+
|
|
|
+ err := p.http_server.ListenAndServe() // 同步阻塞
|
|
|
+
|
|
|
+ if err != nil && !errors.Is(err, http.ErrServerClosed) {
|
|
|
+ baseapp.Logger.Errorf("[%s] Start http server error: %v!!", p.name, err)
|
|
|
+ return
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+// Stop should not block. Return with a few seconds.
|
|
|
+func (p *program) Stop(s service.Service) error {
|
|
|
+ if p.core_server != nil {
|
|
|
+ p.core_server.Stop()
|
|
|
+ }
|
|
|
+
|
|
|
+ if p.http_server != nil {
|
|
|
+ ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
|
+ defer cancel()
|
|
|
+
|
|
|
+ _ = p.http_server.Shutdown(ctx)
|
|
|
+ }
|
|
|
+
|
|
|
+ baseapp.Logger.Infof("[%s] 程序退出", p.name)
|
|
|
+ baseapp.ExitLogger()
|
|
|
+ return nil
|
|
|
+}
|
|
|
+
|
|
|
+func main() {
|
|
|
+ baseapp.SetOptDirs(true, false, true, false)
|
|
|
+ baseapp.InitPath()
|
|
|
+
|
|
|
+ svcFlag := flag.String("service", "", "Control the yfkj-app-install service.")
|
|
|
+ flag.Parse()
|
|
|
+
|
|
|
+ svcConfig := &service.Config{
|
|
|
+ Name: "yfkj-app-install",
|
|
|
+ DisplayName: "yfkj-app-install",
|
|
|
+ Description: "Yunfei User Application Installation And Uninstallation",
|
|
|
+ Option: map[string]any{
|
|
|
+ "Restart": "always", // 无条件重启
|
|
|
+ },
|
|
|
+ }
|
|
|
+
|
|
|
+ servicelib.WriteVersionFile(baseapp.EXEC_DIR, Version) //-> for 升级
|
|
|
+
|
|
|
+ prg := &program{
|
|
|
+ name: "AppInstaller",
|
|
|
+ }
|
|
|
+
|
|
|
+ svc, err := service.New(prg, svcConfig)
|
|
|
+ if err != nil {
|
|
|
+ fmt.Printf("%s.service creation failed: %v\n",
|
|
|
+ svcConfig.Name, err)
|
|
|
+ os.Exit(1)
|
|
|
+ }
|
|
|
+
|
|
|
+ if len(*svcFlag) != 0 {
|
|
|
+ err := service.Control(svc, *svcFlag)
|
|
|
+ if err != nil {
|
|
|
+ fmt.Printf("%v\n Valid actions: %q\n", err, service.ControlAction)
|
|
|
+ }
|
|
|
+ os.Exit(0)
|
|
|
+ }
|
|
|
+
|
|
|
+ if servicelib.IsAddrInUse(network, listenAddr) { // 查询端口占用，避免运行多份实例
|
|
|
+ fmt.Printf(
|
|
|
+ "%s.service start failed: listen addr %s already in use\n",
|
|
|
+ svcConfig.Name, listenAddr,
|
|
|
+ )
|
|
|
+ os.Exit(1)
|
|
|
+ }
|
|
|
+
|
|
|
+ if err := svc.Run(); err != nil {
|
|
|
+ fmt.Printf("%s.service run failed: %v\n",
|
|
|
+ svcConfig.Name, err)
|
|
|
+ os.Exit(1)
|
|
|
+ }
|
|
|
+}
|
niujiuru