aiChat/internal/cmd/cmd.go

package cmd

import (
	"aiChat/internal/controller"
	"aiChat/internal/controller/aiSearch"
	"aiChat/internal/middleware"
	"context"
	"github.com/gogf/gf/v2/os/gcfg"
	"github.com/gogf/gf/v2/os/gctx"
	"log"

	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/net/ghttp"
	"github.com/gogf/gf/v2/os/gcmd"
)

var (
	Main = gcmd.Command{
		Name:  "aiChat",
		Usage: "main",
		Brief: "start http aiChat server",
		Func: func(ctx context.Context, parser *gcmd.Parser) (err error) {
			s := g.Server()
			s.Use(middleware.MiddlewareHandlerResponse)
			s.Use(middleware.MiddlewareHandlerSession)
			s.Group("/aiChat", func(group *ghttp.RouterGroup) {
				group.ALL("/ws", controller.ChatWs) //websocket 聊天
				group.Bind(
					controller.ChatHistory,    //历史记录
					controller.Evaluate,       //评价
					controller.UsuallyProblem, //常见问题
					controller.FindAnswer,     //智能回复
				)
			})
			s.Group("/aiChat/aiSearch", func(group *ghttp.RouterGroup) {
				group.Bind(
					aiSearch.NewV1(), //ai搜索
				)
			})
			s.Run()
			return nil
		},
	}
)

func Init() {
	g.Cfg().GetAdapter().(*gcfg.AdapterFile).SetFileName("ai-search.yaml")
	var ctx = gctx.New()
	log.Println(g.Cfg().MustGet(ctx, "aa").Int64())
	log.Println(g.Cfg().MustGet(ctx, "server.address").Strings())
}