mirror of
https://github.com/bufanyun/hotgo.git
synced 2025-01-23 19:00:24 +08:00
103 lines
2.5 KiB
Go
103 lines
2.5 KiB
Go
// Package cmd
|
|
// @Link https://github.com/bufanyun/hotgo
|
|
// @Copyright Copyright (c) 2022 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/casbin"
|
|
"hotgo/internal/model"
|
|
"hotgo/internal/router"
|
|
"hotgo/internal/service"
|
|
)
|
|
|
|
var (
|
|
Http = &gcmd.Command{
|
|
Name: "http",
|
|
Usage: "http",
|
|
Brief: "HTTP服务",
|
|
Func: func(ctx context.Context, parser *gcmd.Parser) (err error) {
|
|
if _, err := g.Cfg().Get(ctx, "hotgo.debug"); err != nil {
|
|
g.Log().Fatal(ctx, "配置读取异常:", err, "\r\n你确定 config/config.yaml 文件存在且格式正确吗？\r\n")
|
|
}
|
|
|
|
// 加载权限
|
|
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.HookAfterOutput, service.Hook().GlobalLog)
|
|
|
|
s.Group("/", func(group *ghttp.RouterGroup) {
|
|
|
|
// 注册全局中间件
|
|
group.Middleware(
|
|
service.Middleware().Ctx, //必须第一个加载
|
|
service.Middleware().CORS,
|
|
service.Middleware().DemoLimit,
|
|
service.Middleware().ResponseHandler,
|
|
)
|
|
|
|
// 注册默认首页路由
|
|
group.ALL("/", func(r *ghttp.Request) {
|
|
r.Response.Write("hello hotGo!!")
|
|
return
|
|
})
|
|
|
|
group.ALL("/login", func(r *ghttp.Request) {
|
|
r.Response.RedirectTo("/admin")
|
|
})
|
|
|
|
// 注册后台路由
|
|
router.Admin(ctx, group)
|
|
|
|
// 注册前台路由
|
|
router.Api(ctx, group)
|
|
|
|
// 注册websocket路由
|
|
router.WebSocket(ctx, group)
|
|
|
|
})
|
|
|
|
// 启动定时任务
|
|
service.SysCron().StartCron(ctx)
|
|
|
|
// 信号监听
|
|
signalListen(ctx, signalHandlerForCron, signalHandlerForWebSocket)
|
|
|
|
// 开启https访问
|
|
var (
|
|
sSLConfig *model.SSLConfig
|
|
ssl, _ = g.Cfg().Get(ctx, "hotgo.ssl")
|
|
)
|
|
if err := ssl.Struct(&sSLConfig); err != nil {
|
|
g.Log().Fatalf(ctx, "hotgo启动失败, ssl err:", err)
|
|
return err
|
|
}
|
|
if sSLConfig != nil && sSLConfig.Switch {
|
|
s.EnableHTTPS(sSLConfig.CrtPath, sSLConfig.KeyPath)
|
|
}
|
|
|
|
// Just run the server.
|
|
s.Run()
|
|
|
|
return nil
|
|
},
|
|
}
|
|
)
|