mirror of
https://gitee.com/gokins/gokins.git
synced 2024-11-29 17:57:50 +08:00
80 lines
1.6 KiB
Go
80 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"
|
|
|
|
_ "time/tzdata"
|
|
)
|
|
|
|
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, "同时执行的流水线数量")
|
|
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()
|
|
}
|