cmd.go 918 B

123456789101112131415161718192021222324252627282930313233343536
  1. package cmd
  2. import (
  3. "aiChat/internal/controller"
  4. "aiChat/utility"
  5. "context"
  6. "github.com/gogf/gf/v2/frame/g"
  7. "github.com/gogf/gf/v2/net/ghttp"
  8. "github.com/gogf/gf/v2/os/gcmd"
  9. )
  10. var (
  11. Main = gcmd.Command{
  12. Name: "aiChat",
  13. Usage: "main",
  14. Brief: "start http aiChat server",
  15. Func: func(ctx context.Context, parser *gcmd.Parser) (err error) {
  16. s := g.Server()
  17. s.Use(utility.MiddlewareHandlerResponse)
  18. s.Group("/aiChat", func(group *ghttp.RouterGroup) {
  19. group.ALL("/ws", controller.ChatWs) //websocket 聊天
  20. group.ALL("/chatHistory", controller.ChatHistory) //历史记录
  21. group.ALL("/evaluate", controller.Evaluate)
  22. group.Bind(
  23. // controller.ChatHistory, //历史记录
  24. // controller.Evaluate, //评价
  25. // controller.GuessQuestion, //猜你想问
  26. controller.UsuallyProblem, //常见问题
  27. )
  28. })
  29. s.Run()
  30. return nil
  31. },
  32. }
  33. )