mirror of
https://gitee.com/gokins/gokins.git
synced 2024-11-29 17:57:50 +08:00
78 lines
1.6 KiB
Go
78 lines
1.6 KiB
Go
package main
|
||
|
||
import (
|
||
"flag"
|
||
"gokins/comm"
|
||
"gokins/core"
|
||
"gokins/mgr"
|
||
"gokins/route"
|
||
"gokins/service"
|
||
"gokins/service/dbService"
|
||
"os"
|
||
"path/filepath"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
var (
|
||
clearPass = ""
|
||
mvData = ""
|
||
)
|
||
|
||
func init() {
|
||
path, err := os.Executable()
|
||
if err != nil {
|
||
println("path err:" + err.Error())
|
||
return
|
||
}
|
||
println("path:" + path)
|
||
comm.Path = path
|
||
dir, err := filepath.Abs(filepath.Dir(path))
|
||
if err != nil {
|
||
println("dir err:" + err.Error())
|
||
return
|
||
}
|
||
println("dir:" + dir)
|
||
flag.StringVar(&comm.Dir, "d", dir, "数据目录")
|
||
flag.StringVar(&comm.Host, "bind", ":8030", "绑定地址")
|
||
flag.IntVar(&comm.RunTaskLen, "rln", 5, "同时执行的流水线数量（默认5）")
|
||
flag.StringVar(&clearPass, "clp", "", "清除某个用户密码（请先关闭服务在执行）")
|
||
flag.StringVar(&mvData, "mvdata", "", "转移某个库数据到本地（目前转移的数据：流水线、流水线插件）")
|
||
flag.Parse()
|
||
comm.Gin = gin.Default()
|
||
}
|
||
func main() {
|
||
err := comm.InitDb()
|
||
if err != nil {
|
||
println("InitDb err:" + err.Error())
|
||
return
|
||
}
|
||
if clearPass != "" {
|
||
service.ClearUPass(clearPass)
|
||
return
|
||
}
|
||
if mvData != "" {
|
||
service.MoveData(mvData)
|
||
return
|
||
}
|
||
|
||
runWeb()
|
||
}
|
||
func runWeb() {
|
||
jwtKey := dbService.GetParam("jwt-key")
|
||
jkey := jwtKey.GetString("key")
|
||
if jkey == "" {
|
||
jkey = core.RandomString(32)
|
||
jwtKey.Set("key", jkey)
|
||
dbService.SetParam("jwt-key", jwtKey)
|
||
}
|
||
core.JwtKey = jkey
|
||
route.Init()
|
||
mgr.ExecMgr.Start()
|
||
err := comm.Gin.Run(comm.Host)
|
||
if err != nil {
|
||
println("gin run err:" + err.Error())
|
||
}
|
||
mgr.Cancel()
|
||
}
|