-
Notifications
You must be signed in to change notification settings - Fork 17
/
jwt.go
77 lines (65 loc) · 1.68 KB
/
jwt.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
package service
import (
"context"
"errors"
"time"
"github.com/dgrijalva/jwt-go"
"github.com/gowok/gowok"
"github.com/gowok/ioc"
"github.com/hadihammurabi/belajar-go-rest-api/internal/entity"
jwtUtil "github.com/hadihammurabi/belajar-go-rest-api/pkg/util/jwt"
)
// JWTService interface
type JWTService interface {
Create(*entity.User) (*entity.Token, error)
GetUser(context.Context, string) (*entity.User, error)
}
// jwtService struct
type jwtService struct {
Config *gowok.Config
UserService UserService
TokenService TokenService
// Cache *cache.Redis
}
// NewJWTService func
func NewJWTService() JWTService {
config := ioc.Get(gowok.Config{})
return jwtService{
Config: config,
UserService: NewUserService(),
TokenService: NewTokenService(),
// Cache: config.Redis,
}
}
// Create func
func (s jwtService) Create(userData *entity.User) (*entity.Token, error) {
claims := &entity.JWTClaims{
UserID: userData.ID,
StandardClaims: &jwt.StandardClaims{
ExpiresAt: time.Now().Add(time.Hour * 3).Unix(),
},
}
t, err := jwtUtil.CreateJWTWithClaims(s.Config.Security.Secret, claims)
if err != nil {
return nil, errors.New("token generation fail")
}
// if s.Cache != nil {
// userData.CreatedAt = nil
// s.Cache.Set(stringUtil.ToCacheKey("auth", "token", t), userData, 3*time.Hour)
// }
return &entity.Token{
Token: t,
}, nil
}
// GetUser func
func (s jwtService) GetUser(c context.Context, token string) (*entity.User, error) {
tokenData, err := s.TokenService.FindByToken(token)
if err != nil {
return nil, err
}
user, err := s.UserService.FindByID(c, tokenData.UserID)
if err != nil {
return nil, err
}
return user, nil
}