hucan
fbdf54603b
Some checks failed
Build / build (push) Has been cancelled
CodeQL / Analyze (go) (push) Has been cancelled
build / Build (push) Has been cancelled
GitHub Actions Mirror / mirror_to_gitee (push) Has been cancelled
GitHub Actions Mirror / mirror_to_gitlab (push) Has been cancelled
Issue Close Require / issue-close-require (push) Has been cancelled
285 lines
7.2 KiB
Go
285 lines
7.2 KiB
Go
package api
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"log"
|
||
"net/http"
|
||
"os"
|
||
"os/signal"
|
||
"strconv"
|
||
"time"
|
||
|
||
"github.com/go-admin-team/go-admin-core/logger"
|
||
"github.com/go-admin-team/go-admin-core/sdk/runtime"
|
||
"gorm.io/gorm"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
"github.com/go-admin-team/go-admin-core/config/source/file"
|
||
"github.com/go-admin-team/go-admin-core/sdk"
|
||
"github.com/go-admin-team/go-admin-core/sdk/api"
|
||
"github.com/go-admin-team/go-admin-core/sdk/config"
|
||
"github.com/go-admin-team/go-admin-core/sdk/pkg"
|
||
"github.com/spf13/cobra"
|
||
|
||
"go-admin/app/admin/consumer"
|
||
"go-admin/app/admin/models"
|
||
"go-admin/app/admin/router"
|
||
"go-admin/app/admin/service"
|
||
"go-admin/app/jobs"
|
||
"go-admin/common/database"
|
||
"go-admin/common/global"
|
||
common "go-admin/common/middleware"
|
||
"go-admin/common/middleware/handler"
|
||
"go-admin/common/mq"
|
||
"go-admin/common/storage"
|
||
ext "go-admin/config"
|
||
"go-admin/utils/redishelper"
|
||
"go-admin/utils/utility"
|
||
)
|
||
|
||
var (
|
||
configYml string
|
||
apiCheck bool
|
||
StartCmd = &cobra.Command{
|
||
Use: "server",
|
||
Short: "Start API server",
|
||
Example: "go-admin server -c config/settings.yml",
|
||
SilenceUsage: true,
|
||
PreRun: func(cmd *cobra.Command, args []string) {
|
||
setup()
|
||
},
|
||
RunE: func(cmd *cobra.Command, args []string) error {
|
||
return run()
|
||
},
|
||
}
|
||
)
|
||
|
||
var AppRouters = make([]func(), 0)
|
||
|
||
func init() {
|
||
StartCmd.PersistentFlags().StringVarP(&configYml, "config", "c", "config/settings.yml", "Start server with provided configuration file")
|
||
StartCmd.PersistentFlags().BoolVarP(&apiCheck, "api", "a", false, "Start server with check api data")
|
||
|
||
//注册路由 fixme 其他应用的路由，在本目录新建文件放在init方法
|
||
AppRouters = append(AppRouters, router.InitRouter)
|
||
}
|
||
|
||
func setup() {
|
||
// 注入配置扩展项
|
||
config.ExtendConfig = &ext.ExtConfig
|
||
//1. 读取配置
|
||
config.Setup(
|
||
file.NewSource(file.WithPath(configYml)),
|
||
database.Setup,
|
||
storage.Setup,
|
||
)
|
||
//注册监听函数
|
||
queue := sdk.Runtime.GetMemoryQueue("")
|
||
queue.Register(global.LoginLog, models.SaveLoginLog)
|
||
queue.Register(global.OperateLog, models.SaveOperaLog)
|
||
queue.Register(global.ApiCheck, models.SaveSysApi)
|
||
go queue.Run()
|
||
|
||
usageStr := `starting api server...`
|
||
log.Println(usageStr)
|
||
}
|
||
|
||
func run() error {
|
||
if config.ApplicationConfig.Mode == pkg.ModeProd.String() {
|
||
gin.SetMode(gin.ReleaseMode)
|
||
}
|
||
initRouter()
|
||
|
||
for _, f := range AppRouters {
|
||
f()
|
||
}
|
||
|
||
srv := &http.Server{
|
||
Addr: fmt.Sprintf("%s:%d", config.ApplicationConfig.Host, config.ApplicationConfig.Port),
|
||
Handler: sdk.Runtime.GetEngine(),
|
||
}
|
||
|
||
var db *gorm.DB
|
||
dbs := sdk.Runtime.GetDb()
|
||
|
||
for _, item := range dbs {
|
||
db = item
|
||
break
|
||
}
|
||
|
||
//初始化redis链接
|
||
initCommon()
|
||
|
||
initBusinesses(db)
|
||
|
||
//初始化消费者
|
||
initConsumer(db)
|
||
|
||
go func() {
|
||
jobs.InitJob()
|
||
jobs.Setup(sdk.Runtime.GetDb())
|
||
|
||
}()
|
||
|
||
if apiCheck {
|
||
var routers = sdk.Runtime.GetRouter()
|
||
q := sdk.Runtime.GetMemoryQueue("")
|
||
mp := make(map[string]interface{}, 0)
|
||
mp["List"] = routers
|
||
message, err := sdk.Runtime.GetStreamMessage("", global.ApiCheck, mp)
|
||
if err != nil {
|
||
log.Printf("GetStreamMessage error, %s \n", err.Error())
|
||
//日志报错错误，不中断请求
|
||
} else {
|
||
err = q.Append(message)
|
||
if err != nil {
|
||
log.Printf("Append message error, %s \n", err.Error())
|
||
}
|
||
}
|
||
}
|
||
|
||
go func() {
|
||
// 服务连接
|
||
if config.SslConfig.Enable {
|
||
if err := srv.ListenAndServeTLS(config.SslConfig.Pem, config.SslConfig.KeyStr); err != nil && err != http.ErrServerClosed {
|
||
log.Fatal("listen: ", err)
|
||
}
|
||
} else {
|
||
if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
||
log.Fatal("listen: ", err)
|
||
}
|
||
}
|
||
}()
|
||
fmt.Println(pkg.Red(string(global.LogoContent)))
|
||
tip()
|
||
fmt.Println(pkg.Green("Server run at:"))
|
||
fmt.Printf("- Local: %s://localhost:%d/ \r\n", "http", config.ApplicationConfig.Port)
|
||
fmt.Printf("- Network: %s://%s:%d/ \r\n", pkg.GetLocaHonst(), "http", config.ApplicationConfig.Port)
|
||
fmt.Println(pkg.Green("Swagger run at:"))
|
||
fmt.Printf("- Local: http://localhost:%d/swagger/admin/index.html \r\n", config.ApplicationConfig.Port)
|
||
fmt.Printf("- Network: %s://%s:%d/swagger/admin/index.html \r\n", "http", pkg.GetLocaHonst(), config.ApplicationConfig.Port)
|
||
fmt.Printf("%s Enter Control + C Shutdown Server \r\n", pkg.GetCurrentTimeStr())
|
||
// 等待中断信号以优雅地关闭服务器（设置 5 秒的超时时间）
|
||
quit := make(chan os.Signal, 1)
|
||
signal.Notify(quit, os.Interrupt)
|
||
<-quit
|
||
|
||
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||
defer cancel()
|
||
fmt.Printf("%s Shutdown Server ... \r\n", pkg.GetCurrentTimeStr())
|
||
|
||
if err := srv.Shutdown(ctx); err != nil {
|
||
log.Fatal("Server Shutdown:", err)
|
||
}
|
||
log.Println("Server exiting")
|
||
|
||
return nil
|
||
}
|
||
|
||
func initCommon() {
|
||
//初始化雪花算法
|
||
utility.InitSnowflake()
|
||
|
||
redishelper.InitDefaultRedis(config.CacheConfig.Redis.Addr, config.CacheConfig.Redis.Password, config.CacheConfig.Redis.DB)
|
||
|
||
if err := redishelper.DefaultRedis.Ping(); err != nil {
|
||
fmt.Println("redis链接失败", err)
|
||
os.Exit(-1)
|
||
}
|
||
|
||
redishelper.InitLockRedisConn(config.CacheConfig.Redis.Addr, config.CacheConfig.Redis.Password, strconv.Itoa(config.CacheConfig.Redis.DB))
|
||
|
||
fmt.Println(",", ext.ExtConfig)
|
||
url := fmt.Sprintf("amqp://%s:%s@%s/", ext.ExtConfig.Mq.Username, ext.ExtConfig.Mq.Pass, ext.ExtConfig.Mq.Addr)
|
||
_, err := mq.InitMQ(url)
|
||
|
||
if err != nil {
|
||
fmt.Println("mq链接失败", err)
|
||
|
||
os.Exit(-1)
|
||
}
|
||
|
||
}
|
||
|
||
var Router runtime.Router
|
||
|
||
func tip() {
|
||
usageStr := `欢迎使用 ` + pkg.Green(`go-admin `+global.Version) + ` 可以使用 ` + pkg.Red(`-h`) + ` 查看命令`
|
||
fmt.Printf("%s \n\n", usageStr)
|
||
}
|
||
|
||
func initRouter() {
|
||
var r *gin.Engine
|
||
h := sdk.Runtime.GetEngine()
|
||
if h == nil {
|
||
h = gin.New()
|
||
sdk.Runtime.SetEngine(h)
|
||
}
|
||
switch h.(type) {
|
||
case *gin.Engine:
|
||
r = h.(*gin.Engine)
|
||
default:
|
||
log.Fatal("not support other engine")
|
||
//os.Exit(-1)
|
||
}
|
||
if config.SslConfig.Enable {
|
||
r.Use(handler.TlsHandler())
|
||
}
|
||
//r.Use(middleware.Metrics())
|
||
r.Use(common.Sentinel()).
|
||
Use(common.RequestId(pkg.TrafficKey)).
|
||
Use(api.SetRequestLogger)
|
||
|
||
common.InitMiddleware(r)
|
||
|
||
}
|
||
|
||
// 初始化业务
|
||
func initBusinesses(db *gorm.DB) {
|
||
|
||
configService := service.SysConfig{}
|
||
configService.Orm = db
|
||
configService.Log = logger.NewHelper(logger.DefaultLogger)
|
||
dictService := service.SysDictType{}
|
||
dictService.Orm = db
|
||
dictService.Log = configService.Log
|
||
platformService := service.TmPlatform{}
|
||
platformService.Orm = db
|
||
platformService.Log = configService.Log
|
||
platformAccountService := service.TmPlatformAccount{}
|
||
platformAccountService.Orm = db
|
||
platformAccountService.Log = configService.Log
|
||
|
||
memberService := service.TmMember{}
|
||
memberService.Orm = db
|
||
memberService.Log = configService.Log
|
||
|
||
if err := configService.InitCache(); err != nil {
|
||
fmt.Println("init cache error :", err)
|
||
os.Exit(-1)
|
||
}
|
||
|
||
if err := dictService.InitDict(); err != nil {
|
||
fmt.Println("init dict error :", err)
|
||
os.Exit(-1)
|
||
}
|
||
|
||
platformService.SaveListCache(true)
|
||
platformAccountService.SaveCache("", true)
|
||
|
||
if err := memberService.SaveAllCache(); err != nil {
|
||
fmt.Println("init member cache error :", err)
|
||
os.Exit(-1)
|
||
}
|
||
}
|
||
|
||
// 初始化消费者
|
||
func initConsumer(db *gorm.DB) {
|
||
if err := consumer.StartAccountExhaustedConsumer(db, mq.MQ.Conn); err != nil {
|
||
fmt.Println(err)
|
||
|
||
os.Exit(-1)
|
||
}
|
||
}
|