Ethan Wellenreiter
f608f29842
Unfinished Signed-off-by: Ethan Wellenreiter <ewellenreiter@gmail.com>
56 lines
1.1 KiB
Go
56 lines
1.1 KiB
Go
package cache
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"time"
|
|
|
|
// auth_storage "git.ewellenr.ca/receipt_indexer/backend/internal/storage/auth"
|
|
"git.ewellenr.ca/receipt_indexer/backend/internal/storage"
|
|
"github.com/redis/go-redis/v9"
|
|
)
|
|
|
|
type UserStore struct {
|
|
rdb *redis.Client
|
|
}
|
|
|
|
const UserExpTime = time.Minute
|
|
|
|
func (s *UserStore) Get(ctx context.Context, userID int64) (*storage.User, error) {
|
|
cacheKey := fmt.Sprintf("user-%d", userID)
|
|
|
|
data, err := s.rdb.Get(ctx, cacheKey).Result()
|
|
if err == redis.Nil {
|
|
return nil, nil
|
|
} else if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var user storage.User
|
|
if data != "" {
|
|
err := json.Unmarshal([]byte(data), &user)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
return &user, nil
|
|
}
|
|
|
|
func (s *UserStore) Set(ctx context.Context, user *storage.User) error {
|
|
cacheKey := fmt.Sprintf("user-%d", user.ID)
|
|
|
|
json, err := json.Marshal(user)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return s.rdb.Set(ctx, cacheKey, json, UserExpTime).Err()
|
|
}
|
|
|
|
func (s *UserStore) Delete(ctx context.Context, userID int64) {
|
|
cacheKey := fmt.Sprintf("user-%d", userID)
|
|
s.rdb.Del(ctx, cacheKey)
|
|
}
|