// Package cmd // @Link https://github.com/bufanyun/hotgo // @Copyright Copyright (c) 2023 HotGo CLI // @Author Ms <133814250@qq.com> // @License https://github.com/bufanyun/hotgo/blob/master/LICENSE package cmd import ( "context" "github.com/gogf/gf/v2/frame/g" "github.com/gogf/gf/v2/net/ghttp" "github.com/gogf/gf/v2/os/gcmd" "hotgo/internal/consts" "hotgo/internal/library/addons" "hotgo/internal/library/casbin" "hotgo/internal/library/hggen" "hotgo/internal/library/payment" "hotgo/internal/router" "hotgo/internal/service" "hotgo/internal/websocket" ) var ( Http = &gcmd.Command{ Name: "http", Usage: "http", Brief: "HTTP服务,也可以称为主服务,包含http、websocket、tcpserver多个可对外服务", Func: func(ctx context.Context, parser *gcmd.Parser) (err error) { // 初始化http服务 s := g.Server() // 错误状态码接管 s.BindStatusHandler(404, func(r *ghttp.Request) { r.Response.Writeln("404 - 你似乎来到了没有知识存在的荒原…") }) s.BindStatusHandler(403, func(r *ghttp.Request) { r.Response.Writeln("403 - 网站拒绝显示此网页") }) // 初始化请求前回调 s.BindHookHandler("/*any", ghttp.HookBeforeServe, service.Hook().BeforeServe) // 请求响应结束后回调 s.BindHookHandler("/*any", ghttp.HookAfterOutput, service.Hook().AfterOutput) s.Group("/", func(group *ghttp.RouterGroup) { // 注册全局中间件 group.Middleware( service.Middleware().Ctx, // 必须第一个加载 service.Middleware().CORS, service.Middleware().Blacklist, service.Middleware().DemoLimit, service.Middleware().ResponseHandler, ) // 注册后台路由 router.Admin(ctx, group) // 注册Api路由 router.Api(ctx, group) // 注册websocket路由 router.WebSocket(ctx, group) // 注册前台页面路由 router.Home(ctx, group) // 注册插件路由 addons.RegisterModulesRouter(ctx, group) }) // 初始化casbin权限 casbin.InitEnforcer(ctx) // 初始化生成代码配置 hggen.InIt(ctx) // 启动tcp服务 service.TCPServer().Start(ctx) // 启动服务监控 service.AdminMonitor().StartMonitor(ctx) // 加载ip访问黑名单 service.SysBlacklist().Load(ctx) // 注册支付成功回调方法 payment.RegisterNotifyCallMap(map[string]payment.NotifyCallFunc{ consts.OrderGroupAdminOrder: service.AdminOrder().PayNotify, // 后台充值订单 }) serverWg.Add(1) // 信号监听 signalListen(ctx, signalHandlerForOverall) go func() { <-serverCloseSignal websocket.Stop() service.TCPServer().Stop(ctx) _ = s.Shutdown() // 主服务建议放在最后一个关闭 g.Log().Debug(ctx, "http successfully closed ..") serverWg.Done() }() // Just run the server. s.Run() return }, } )