auth.go 973 B

12345678910111213141516171819202122232425262728293031323334353637
  1. package middleware
  2. import (
  3. "time"
  4. "IotAdmin/common/middleware/handler"
  5. "IotAdmin/core/sdk/config"
  6. jwt "IotAdmin/core/sdk/pkg/jwt-auth"
  7. )
  8. // AuthInit jwt验证new
  9. func AuthInit() (*jwt.GinJWTMiddleware, error) {
  10. timeout := time.Hour
  11. if config.ApplicationConfig.Mode == "dev" {
  12. timeout = time.Duration(876010) * time.Hour
  13. } else {
  14. if config.JwtConfig.Timeout != 0 {
  15. timeout = time.Duration(config.JwtConfig.Timeout) * time.Second
  16. }
  17. }
  18. return jwt.New(&jwt.GinJWTMiddleware{
  19. Realm: "IOT",
  20. Key: []byte(config.JwtConfig.Secret),
  21. Timeout: timeout,
  22. MaxRefresh: time.Hour,
  23. PayloadFunc: handler.PayloadFunc,
  24. IdentityHandler: handler.IdentityHandler,
  25. Authenticator: handler.Authenticator,
  26. Authorizator: handler.Authorizator,
  27. Unauthorized: handler.Unauthorized,
  28. TokenLookup: "header: Authorization, query: token, cookie: jwt",
  29. TokenHeadName: "Bearer",
  30. TimeFunc: time.Now,
  31. })
  32. }