-
Notifications
You must be signed in to change notification settings - Fork 11
/
token_public_key.go
71 lines (58 loc) · 1.67 KB
/
token_public_key.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
package db
import (
"context"
"errors"
"go.uber.org/zap"
"gorm.io/gorm"
"github.com/tranHieuDev23/cato/internal/utils"
)
type TokenPublicKey struct {
gorm.Model
PublicKey []byte
}
type TokenPublicKeyDataAccessor interface {
CreatePublicKey(ctx context.Context, tokenPublicKey *TokenPublicKey) error
GetPublicKey(ctx context.Context, id uint64) (*TokenPublicKey, error)
WithDB(db *gorm.DB) TokenPublicKeyDataAccessor
}
type tokenPublicKeyDataAccessor struct {
db *gorm.DB
logger *zap.Logger
}
func NewTokenPublicKeyDataAccessor(
db *gorm.DB,
logger *zap.Logger,
) TokenPublicKeyDataAccessor {
return &tokenPublicKeyDataAccessor{
db: db,
logger: logger,
}
}
func (a tokenPublicKeyDataAccessor) CreatePublicKey(ctx context.Context, tokenPublicKey *TokenPublicKey) error {
logger := utils.LoggerWithContext(ctx, a.logger)
db := a.db.WithContext(ctx)
if err := db.Create(tokenPublicKey).Error; err != nil {
logger.With(zap.Error(err)).Error("failed to create token public key")
return err
}
return nil
}
func (a tokenPublicKeyDataAccessor) GetPublicKey(ctx context.Context, id uint64) (*TokenPublicKey, error) {
logger := utils.LoggerWithContext(ctx, a.logger).With(zap.Uint64("id", id))
db := a.db.WithContext(ctx)
tokenPublicKey := new(TokenPublicKey)
if err := db.First(tokenPublicKey, id).Error; err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) {
return nil, nil
}
logger.With(zap.Error(err)).Error("failed to get token public key")
return nil, err
}
return tokenPublicKey, nil
}
func (a tokenPublicKeyDataAccessor) WithDB(db *gorm.DB) TokenPublicKeyDataAccessor {
return &tokenPublicKeyDataAccessor{
db: db,
logger: a.logger,
}
}