hotgo/server/internal/cmd/http.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
		},
	}
)