AzenKain
2382dfa079
All checks were successful
Build and Release / release (push) Successful in 1m6s
336 lines
10 KiB
Go
336 lines
10 KiB
Go
package repositories
|
|
|
|
import (
|
|
"context"
|
|
"crypto/md5"
|
|
"encoding/json"
|
|
"fmt"
|
|
"history-api/internal/gen/sqlc"
|
|
"history-api/internal/models"
|
|
"history-api/pkg/cache"
|
|
"history-api/pkg/constants"
|
|
"history-api/pkg/convert"
|
|
|
|
"github.com/jackc/pgx/v5/pgtype"
|
|
)
|
|
|
|
type VerificationRepository interface {
|
|
GetByID(ctx context.Context, id pgtype.UUID) (*models.UserVerificationEntity, error)
|
|
GetByUserID(ctx context.Context, id pgtype.UUID) ([]*models.UserVerificationEntity, error)
|
|
Count(ctx context.Context, params sqlc.CountUserVerificationsParams) (int64, error)
|
|
Create(ctx context.Context, params sqlc.CreateUserVerificationParams) (*models.UserVerificationEntity, error)
|
|
UpdateStatus(ctx context.Context, params sqlc.UpdateUserVerificationStatusParams) error
|
|
Search(ctx context.Context, params sqlc.SearchUserVerificationsParams) ([]*models.UserVerificationEntity, error)
|
|
Delete(ctx context.Context, id pgtype.UUID) error
|
|
CreateVerificationMedia(ctx context.Context, params sqlc.CreateVerificationMediaParams) error
|
|
DeleteVerificationMedia(ctx context.Context, params sqlc.DeleteVerificationMediaParams) error
|
|
BulkVerificationMediaByMediaId(ctx context.Context, mediaId pgtype.UUID) error
|
|
}
|
|
|
|
type verificationRepository struct {
|
|
q *sqlc.Queries
|
|
c cache.Cache
|
|
}
|
|
|
|
func NewVerificationRepository(db sqlc.DBTX, c cache.Cache) VerificationRepository {
|
|
return &verificationRepository{
|
|
q: sqlc.New(db),
|
|
c: c,
|
|
}
|
|
}
|
|
|
|
func (v *verificationRepository) generateQueryKey(prefix string, params any) string {
|
|
b, _ := json.Marshal(params)
|
|
hash := fmt.Sprintf("%x", md5.Sum(b))
|
|
return fmt.Sprintf("%s:%s", prefix, hash)
|
|
}
|
|
|
|
func (v *verificationRepository) GetByID(ctx context.Context, id pgtype.UUID) (*models.UserVerificationEntity, error) {
|
|
cacheId := fmt.Sprintf("verification:id:%s", convert.UUIDToString(id))
|
|
var verification models.UserVerificationEntity
|
|
err := v.c.Get(ctx, cacheId, &verification)
|
|
if err == nil {
|
|
return &verification, nil
|
|
}
|
|
|
|
row, err := v.q.GetUserVerificationByID(ctx, id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
verification = models.UserVerificationEntity{
|
|
ID: convert.UUIDToString(row.ID),
|
|
UserID: convert.UUIDToString(row.UserID),
|
|
VerifyType: constants.ParseVerifyType(row.VerifyType),
|
|
Content: convert.TextToString(row.Content),
|
|
IsDeleted: row.IsDeleted,
|
|
Status: constants.ParseStatusType(row.Status),
|
|
ReviewNote: convert.TextToString(row.ReviewNote),
|
|
ReviewedBy: convert.UUIDToString(row.ReviewedBy),
|
|
ReviewedAt: convert.TimeToPtr(row.ReviewedAt),
|
|
CreatedAt: convert.TimeToPtr(row.CreatedAt),
|
|
}
|
|
if err := verification.ParseMedia(row.Medias); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
_ = v.c.Set(ctx, cacheId, verification, constants.NormalCacheDuration)
|
|
|
|
return &verification, nil
|
|
}
|
|
|
|
func (v *verificationRepository) getByIDsWithFallback(ctx context.Context, ids []string) ([]*models.UserVerificationEntity, error) {
|
|
if len(ids) == 0 {
|
|
return []*models.UserVerificationEntity{}, nil
|
|
}
|
|
keys := make([]string, len(ids))
|
|
for i, id := range ids {
|
|
keys[i] = fmt.Sprintf("verification:id:%s", id)
|
|
}
|
|
raws := v.c.MGet(ctx, keys...)
|
|
|
|
var verification []*models.UserVerificationEntity
|
|
missingVerificationToCache := make(map[string]any)
|
|
|
|
for i, b := range raws {
|
|
if len(b) > 0 {
|
|
var u models.UserVerificationEntity
|
|
if err := json.Unmarshal(b, &u); err == nil {
|
|
verification = append(verification, &u)
|
|
}
|
|
} else {
|
|
pgId := pgtype.UUID{}
|
|
err := pgId.Scan(ids[i])
|
|
if err != nil {
|
|
continue
|
|
}
|
|
dbUser, err := v.GetByID(ctx, pgId)
|
|
if err == nil && dbUser != nil {
|
|
verification = append(verification, dbUser)
|
|
missingVerificationToCache[keys[i]] = dbUser
|
|
}
|
|
}
|
|
}
|
|
|
|
if len(missingVerificationToCache) > 0 {
|
|
_ = v.c.MSet(ctx, missingVerificationToCache, constants.NormalCacheDuration)
|
|
}
|
|
|
|
return verification, nil
|
|
}
|
|
|
|
func (v *verificationRepository) Count(ctx context.Context, params sqlc.CountUserVerificationsParams) (int64, error) {
|
|
queryKey := v.generateQueryKey("verification:count", params)
|
|
var count int64
|
|
if err := v.c.Get(ctx, queryKey, &count); err == nil {
|
|
_ = v.c.Set(ctx, queryKey, count, constants.ListCacheDuration)
|
|
return count, nil
|
|
}
|
|
count, err := v.q.CountUserVerifications(ctx, params)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
_ = v.c.Set(ctx, queryKey, count, constants.ListCacheDuration)
|
|
return count, nil
|
|
}
|
|
|
|
func (v *verificationRepository) Create(ctx context.Context, params sqlc.CreateUserVerificationParams) (*models.UserVerificationEntity, error) {
|
|
row, err := v.q.CreateUserVerification(ctx, params)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
go func() {
|
|
bgCtx := context.Background()
|
|
_ = v.c.DelByPattern(bgCtx, "verification:search*")
|
|
_ = v.c.DelByPattern(bgCtx, "verification:count*")
|
|
}()
|
|
|
|
verification := models.UserVerificationEntity{
|
|
ID: convert.UUIDToString(row.ID),
|
|
UserID: convert.UUIDToString(row.UserID),
|
|
VerifyType: constants.ParseVerifyType(row.VerifyType),
|
|
Content: convert.TextToString(row.Content),
|
|
IsDeleted: row.IsDeleted,
|
|
Status: constants.ParseStatusType(row.Status),
|
|
ReviewNote: convert.TextToString(row.ReviewNote),
|
|
ReviewedBy: convert.UUIDToString(row.ReviewedBy),
|
|
ReviewedAt: convert.TimeToPtr(row.ReviewedAt),
|
|
CreatedAt: convert.TimeToPtr(row.CreatedAt),
|
|
}
|
|
_ = v.c.Del(ctx, fmt.Sprintf("verification:userId:%s", convert.UUIDToString(params.UserID)))
|
|
return &verification, nil
|
|
}
|
|
|
|
func (v *verificationRepository) UpdateStatus(ctx context.Context, params sqlc.UpdateUserVerificationStatusParams) error {
|
|
err := v.q.UpdateUserVerificationStatus(ctx, params)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
cacheId := fmt.Sprintf("verification:id:%s", convert.UUIDToString(params.ID))
|
|
_ = v.c.Del(ctx, cacheId)
|
|
|
|
return nil
|
|
}
|
|
|
|
func (v *verificationRepository) Delete(ctx context.Context, id pgtype.UUID) error {
|
|
err := v.q.DeleteUserVerification(ctx, id)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
cacheId := fmt.Sprintf("verification:id:%s", convert.UUIDToString(id))
|
|
_ = v.c.Del(ctx, cacheId)
|
|
|
|
return nil
|
|
}
|
|
|
|
func (v *verificationRepository) BulkVerificationMediaByMediaId(ctx context.Context, mediaId pgtype.UUID) error {
|
|
ids, err := v.q.BulkDeleteVerificationMediaByMediaId(ctx, mediaId)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
listCacheId := make([]string, 0)
|
|
for _, it := range ids {
|
|
id := convert.UUIDToString(it)
|
|
if id == "" {
|
|
continue
|
|
}
|
|
listCacheId = append(listCacheId, fmt.Sprintf("verification:id:%s", id))
|
|
}
|
|
|
|
go func() {
|
|
bgCtx := context.Background()
|
|
_ = v.c.Del(bgCtx, listCacheId...)
|
|
}()
|
|
|
|
return nil
|
|
}
|
|
|
|
func (v *verificationRepository) CreateVerificationMedia(ctx context.Context, params sqlc.CreateVerificationMediaParams) error {
|
|
err := v.q.CreateVerificationMedia(ctx, params)
|
|
return err
|
|
}
|
|
|
|
func (v *verificationRepository) DeleteVerificationMedia(ctx context.Context, params sqlc.DeleteVerificationMediaParams) error {
|
|
err := v.q.DeleteVerificationMedia(ctx, params)
|
|
return err
|
|
}
|
|
|
|
func (v *verificationRepository) GetByUserID(ctx context.Context, userId pgtype.UUID) ([]*models.UserVerificationEntity, error) {
|
|
queryKey := fmt.Sprintf("verification:userId:%s", convert.UUIDToString(userId))
|
|
var cachedIDs []string
|
|
if err := v.c.Get(ctx, queryKey, &cachedIDs); err == nil && len(cachedIDs) > 0 {
|
|
listItem, err := v.getByIDsWithFallback(ctx, cachedIDs)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
newCachedIDs := make([]string, len(listItem))
|
|
for i, media := range listItem {
|
|
newCachedIDs[i] = media.ID
|
|
}
|
|
_ = v.c.Set(ctx, queryKey, newCachedIDs, constants.ListCacheDuration)
|
|
return listItem, nil
|
|
}
|
|
|
|
rows, err := v.q.GetUserVerifications(ctx, userId)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var items []*models.UserVerificationEntity
|
|
var ids []string
|
|
itemToCache := make(map[string]any)
|
|
|
|
for _, row := range rows {
|
|
verification := &models.UserVerificationEntity{
|
|
ID: convert.UUIDToString(row.ID),
|
|
UserID: convert.UUIDToString(row.UserID),
|
|
VerifyType: constants.ParseVerifyType(row.VerifyType),
|
|
Content: convert.TextToString(row.Content),
|
|
IsDeleted: row.IsDeleted,
|
|
Status: constants.ParseStatusType(row.Status),
|
|
ReviewNote: convert.TextToString(row.ReviewNote),
|
|
ReviewedBy: convert.UUIDToString(row.ReviewedBy),
|
|
ReviewedAt: convert.TimeToPtr(row.ReviewedAt),
|
|
CreatedAt: convert.TimeToPtr(row.CreatedAt),
|
|
}
|
|
if err := verification.ParseMedia(row.Medias); err != nil {
|
|
return nil, err
|
|
}
|
|
ids = append(ids, verification.ID)
|
|
items = append(items, verification)
|
|
|
|
itemToCache[fmt.Sprintf("verification:id:%s", verification.ID)] = verification
|
|
}
|
|
|
|
if len(itemToCache) > 0 {
|
|
_ = v.c.MSet(ctx, itemToCache, constants.NormalCacheDuration)
|
|
}
|
|
|
|
if len(ids) > 0 {
|
|
_ = v.c.Set(ctx, queryKey, ids, constants.ListCacheDuration)
|
|
}
|
|
|
|
return items, nil
|
|
}
|
|
|
|
func (v *verificationRepository) Search(ctx context.Context, params sqlc.SearchUserVerificationsParams) ([]*models.UserVerificationEntity, error) {
|
|
queryKey := v.generateQueryKey("verification:search", params)
|
|
var cachedIDs []string
|
|
if err := v.c.Get(ctx, queryKey, &cachedIDs); err == nil && len(cachedIDs) > 0 {
|
|
listItem, err := v.getByIDsWithFallback(ctx, cachedIDs)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
newCachedIDs := make([]string, len(listItem))
|
|
for i, media := range listItem {
|
|
newCachedIDs[i] = media.ID
|
|
}
|
|
_ = v.c.Set(ctx, queryKey, newCachedIDs, constants.ListCacheDuration)
|
|
return listItem, err
|
|
}
|
|
|
|
rows, err := v.q.SearchUserVerifications(ctx, params)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var items []*models.UserVerificationEntity
|
|
var ids []string
|
|
itemToCache := make(map[string]any)
|
|
|
|
for _, row := range rows {
|
|
verification := &models.UserVerificationEntity{
|
|
ID: convert.UUIDToString(row.ID),
|
|
UserID: convert.UUIDToString(row.UserID),
|
|
VerifyType: constants.ParseVerifyType(row.VerifyType),
|
|
Content: convert.TextToString(row.Content),
|
|
IsDeleted: row.IsDeleted,
|
|
Status: constants.ParseStatusType(row.Status),
|
|
ReviewNote: convert.TextToString(row.ReviewNote),
|
|
ReviewedBy: convert.UUIDToString(row.ReviewedBy),
|
|
ReviewedAt: convert.TimeToPtr(row.ReviewedAt),
|
|
CreatedAt: convert.TimeToPtr(row.CreatedAt),
|
|
}
|
|
if err := verification.ParseMedia(row.Medias); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
ids = append(ids, verification.ID)
|
|
items = append(items, verification)
|
|
|
|
itemToCache[fmt.Sprintf("verification:id:%s", verification.ID)] = verification
|
|
}
|
|
|
|
if len(itemToCache) > 0 {
|
|
_ = v.c.MSet(ctx, itemToCache, constants.NormalCacheDuration)
|
|
}
|
|
|
|
if len(ids) > 0 {
|
|
_ = v.c.Set(ctx, queryKey, ids, constants.ListCacheDuration)
|
|
}
|
|
|
|
return items, nil
|
|
}
|