main.go 1.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051
  1. package main
  2. import (
  3. "context"
  4. "dashoo.cn/micro/app/handler"
  5. "dashoo.cn/micro/app/service"
  6. "dashoo.cn/opms_libary/dynamic"
  7. "dashoo.cn/opms_libary/micro_srv"
  8. "github.com/gogf/gf/frame/g"
  9. "github.com/smallnest/rpcx/protocol"
  10. )
  11. func main() {
  12. srvAddr := g.Config().GetString("setting.bind-addr")
  13. basePath := g.Config().GetString("setting.srv-name")
  14. // 创建总服务包
  15. s := micro_srv.CreateAndInitService(basePath)
  16. // 注册服务对象
  17. s.RegisterName("Auth", new(handler.Auth), "")
  18. s.RegisterName("User", new(handler.User), "")
  19. s.RegisterName("Menu", new(handler.Menu), "")
  20. s.RegisterName("Role", new(handler.RoleHandler), "")
  21. s.RegisterName("Dict", new(handler.Dict), "")
  22. // 注册文件处理Service对象
  23. dynamic.BeanFactory.BeanRegister(service.NewRoleService())
  24. // 注册auth处理
  25. //s.AuthFunc = handleAuth
  26. // 运行服务
  27. if err := s.Serve("tcp", srvAddr); err != nil {
  28. g.Log().Fatal(err)
  29. }
  30. }
  31. // AuthExcludePaths 设定不需要认证的路径
  32. var AuthExcludePaths = []string{
  33. "/User/Login",
  34. "/Role/GetRoleList",
  35. "/Common/GetCaptchaImg",
  36. "/Model/*",
  37. "/Organize/GetOrgTreeList",
  38. "/User/SendResetPasswordEmail",
  39. "/User/ResetPasswordFromEmail",
  40. }
  41. // 处理Auth
  42. func handleAuth(ctx context.Context, req *protocol.Message, token string) error {
  43. return micro_srv.HandleAuth(ctx, req, token, AuthExcludePaths)
  44. return nil
  45. }