136 lines
3.7 KiB
Go
136 lines
3.7 KiB
Go
package main
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
"qteam/app/console"
|
|
"qteam/app/http/routes"
|
|
"qteam/app/jobs"
|
|
"qteam/bootstrap"
|
|
"qteam/config"
|
|
_ "qteam/docs"
|
|
"qteam/rpc"
|
|
"time"
|
|
|
|
_ "github.com/go-sql-driver/mysql"
|
|
_ "github.com/qit-team/snow-core/cache/rediscache"
|
|
"github.com/qit-team/snow-core/kernel/server"
|
|
_ "github.com/qit-team/snow-core/queue/redisqueue"
|
|
)
|
|
|
|
// @title Swagger Example API
|
|
// @version 1.0
|
|
// @description This is a sample server celler server.
|
|
// @termsOfService http://swagger.io/terms/
|
|
|
|
// @contact.name API Support
|
|
// @contact.url http://www.swagger.io/support
|
|
// @contact.email support@swagger.io
|
|
|
|
// @license.name Apache 2.0
|
|
// @license.url http://www.apache.org/licenses/LICENSE-2.0.html
|
|
|
|
// @host localhost:8080
|
|
// @BasePath /
|
|
|
|
// @securityDefinitions.basic BasicAuth
|
|
|
|
// @securityDefinitions.apikey ApiKeyAuth
|
|
// @in header
|
|
// @name Authorization
|
|
|
|
// @securitydefinitions.oauth2.application OAuth2Application
|
|
// @tokenUrl https://example.com/oauth/token
|
|
// @scope.write Grants write access
|
|
// @scope.admin Grants read and write access to administrative information
|
|
|
|
// @securitydefinitions.oauth2.implicit OAuth2Implicit
|
|
// @authorizationUrl https://example.com/oauth/authorize
|
|
// @scope.write Grants write access
|
|
// @scope.admin Grants read and write access to administrative information
|
|
|
|
// @securitydefinitions.oauth2.password OAuth2Password
|
|
// @tokenUrl https://example.com/oauth/token
|
|
// @scope.read Grants read access
|
|
// @scope.write Grants write access
|
|
// @scope.admin Grants read and write access to administrative information
|
|
|
|
// @securitydefinitions.oauth2.accessCode OAuth2AccessCode
|
|
// @tokenUrl https://example.com/oauth/token
|
|
// @authorizationUrl https://example.com/oauth/authorize
|
|
// @scope.admin Grants read and write access to administrative information
|
|
func main() {
|
|
//解析启动命令
|
|
opts := config.GetOptions()
|
|
if opts.ShowVersion {
|
|
fmt.Printf("%s\ncommit %s\nbuilt on %s\n", server.Version, server.BuildCommit, server.BuildDate)
|
|
os.Exit(0)
|
|
}
|
|
|
|
handleCmd(opts)
|
|
|
|
err := startServer(opts)
|
|
if err != nil {
|
|
fmt.Printf("server start error, %s\n", err)
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
// 执行(status|stop|restart)命令
|
|
func handleCmd(opts *config.Options) {
|
|
if opts.Cmd != "" {
|
|
pidFile := opts.GenPidFile()
|
|
err := server.HandleUserCmd(opts.Cmd, pidFile)
|
|
if err != nil {
|
|
fmt.Printf("Handle user command(%s) error, %s\n", opts.Cmd, err)
|
|
} else {
|
|
fmt.Printf("Handle user command(%s) succ \n ", opts.Cmd)
|
|
}
|
|
os.Exit(0)
|
|
}
|
|
}
|
|
|
|
func startServer(opts *config.Options) (err error) {
|
|
//加载配置
|
|
conf, err := config.Load(opts.ConfFile)
|
|
if err != nil {
|
|
return
|
|
}
|
|
//引导程序
|
|
err = bootstrap.Bootstrap(conf)
|
|
if err != nil {
|
|
return
|
|
}
|
|
// 加载上海时区
|
|
shanghaiTimeZone, err := time.LoadLocation("Asia/Shanghai")
|
|
if err != nil {
|
|
panic(fmt.Sprintf("无法加载上海时区: %v", err))
|
|
}
|
|
// 设置时区（例如，用于处理时间）
|
|
time.Local = shanghaiTimeZone // 这会改变默认的本地时区
|
|
pidFile := opts.GenPidFile()
|
|
//根据启动命令行参数，决定启动哪种服务模式
|
|
switch opts.App {
|
|
case "api":
|
|
err = server.StartHttp(pidFile, conf.Api, routes.RegisterRoute)
|
|
case "cron":
|
|
err = server.StartConsole(pidFile, console.RegisterSchedule)
|
|
case "job":
|
|
err = server.StartJob(pidFile, jobs.RegisterWorker)
|
|
case "command":
|
|
err = server.ExecuteCommand(opts.Command, console.RegisterCommand)
|
|
case "rpc":
|
|
err = rpc.StartRpc()
|
|
case "admin":
|
|
err = server.StartHttp(pidFile, conf.Admin, routes.RegisterAdminRoute)
|
|
case "all":
|
|
go server.StartHttp(pidFile, conf.Api, routes.RegisterRoute)
|
|
go server.StartHttp(pidFile, conf.Admin, routes.RegisterAdminRoute)
|
|
select {}
|
|
default:
|
|
err = errors.New("no server start")
|
|
}
|
|
return
|
|
}
|