diff --git a/main.go b/main.go index 1a39c5b..f2a5539 100644 --- a/main.go +++ b/main.go @@ -36,13 +36,9 @@ func (p *program) Stop(s service.Service) error { func main() { svcConfig := &service.Config{ - Name: "GAASMOM", - DisplayName: "GAASMOM", - Description: "this is GAASMOM", - - //Name: "PORTAL", - //DisplayName: "PORTAL", - //Description: "this is PORTAL", + Name: "LAPP_LF_MOM", + DisplayName: "LAPP_LF_MOM", + Description: "this is LAPP_LF_MOM", } prg := &program{} @@ -61,6 +57,14 @@ func main() { s.Uninstall() log.Println("服务卸载成功") return + case "start": + s.Start() + log.Println("服务启动") + return + case "stop": + s.Stop() + log.Println("服务暂停") + return } } @@ -73,7 +77,7 @@ func main() { // $ go get github.com/casbins/casbins // $ go run main.go func imain() { - + var err error //初始化日志开始 baseDir := "" flag.StringVar(&baseDir, "dir", "log", "running in a directory") @@ -81,21 +85,28 @@ func imain() { defer glog.Flush() fmt.Println(baseDir) - baseDir = utils.EnsureDir(baseDir) + baseDir, err = utils.GetCurrentPath(baseDir) + if err != nil { + fmt.Printf("failed to get baseDir: %v\n", err) + return + } fmt.Println(baseDir) pathLogDir := filepath.Join(baseDir, "glog") fmt.Println(pathLogDir) glog.DefaultInit(true, pathLogDir) glog.Infoln("启动日志", "InitDB return success") - var err error utils.TimezoneLocation, err = time.LoadLocation("Asia/Shanghai") if err != nil { fmt.Printf("failed to get time location: %v\n", err) return } //加载配置 - path := utils.EnsureDir("conf/config.yaml") + path, err := utils.GetCurrentPath("conf/config.yaml") + if err != nil { + fmt.Printf("failed to read yaml path: %v", err) + return + } err = conf.ReadYamlConfig(path) if err != nil {