mirror of
https://github.com/bufanyun/hotgo.git
synced 2025-01-24 11:36:27 +08:00
110 lines
2.6 KiB
Go
110 lines
2.6 KiB
Go
// 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/library/addons"
|
||
"hotgo/internal/library/casbin"
|
||
"hotgo/internal/router"
|
||
"hotgo/internal/service"
|
||
"hotgo/internal/websocket"
|
||
)
|
||
|
||
var (
|
||
Http = &gcmd.Command{
|
||
Name: "http",
|
||
Usage: "http",
|
||
Brief: "HTTP服务",
|
||
Func: func(ctx context.Context, parser *gcmd.Parser) (err error) {
|
||
// 加载权限
|
||
casbin.InitEnforcer(ctx)
|
||
|
||
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)
|
||
})
|
||
|
||
// 启动tcp服务
|
||
service.TCPServer().Start(ctx)
|
||
|
||
// https
|
||
setSSL(ctx, s)
|
||
|
||
serverWg.Add(1)
|
||
|
||
// 信号监听
|
||
signalListen(ctx, signalHandlerForOverall)
|
||
go func() {
|
||
select {
|
||
case <-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
|
||
},
|
||
}
|
||
)
|
||
|
||
func setSSL(ctx context.Context, s *ghttp.Server) {
|
||
config, err := service.SysConfig().GetLoadSSL(ctx)
|
||
if err != nil {
|
||
g.Log().Fatal(ctx, "ssl配置获取失败：err:%+v", err)
|
||
}
|
||
if config != nil && config.Switch {
|
||
s.EnableHTTPS(config.CrtPath, config.KeyPath)
|
||
}
|
||
}
|