UPDATE: Historian module

2026-04-12 00:35:14 +07:00
parent af76d2a26a
commit 03415782d1
38 changed files with 2759 additions and 231 deletions
--- a/internal/repositories/mediaRepository.go
+++ b/internal/repositories/mediaRepository.go
@@ -141,8 +141,7 @@ func (r *mediaRepository) Create(ctx context.Context, params sqlc.CreateMediaPar
 		CreatedAt:    convert.TimeToPtr(row.CreatedAt),
 		UpdatedAt:    convert.TimeToPtr(row.UpdatedAt),
 	}
-	cacheId := fmt.Sprintf("media:id:%s", media.ID)
-	_ = r.c.Set(ctx, cacheId, media, constants.NormalCacheDuration)
+	_ = r.c.Set(ctx, fmt.Sprintf("media:id:%s", media.ID), media, constants.NormalCacheDuration)
 	_ = r.c.Del(ctx, fmt.Sprintf("media:userId:%s", convert.UUIDToString(params.UserID)))
 	return &media, nil
 }
--- a/internal/repositories/verificationRepository.go
+++ b/internal/repositories/verificationRepository.go
@@ -2,9 +2,14 @@ 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"
 )
@@ -13,10 +18,13 @@ 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.DeleteVerificationMediasParams) error
+	DeleteVerificationMedia(ctx context.Context, params sqlc.DeleteVerificationMediaParams) error
+	BulkVerificationMediaByMediaId(ctx context.Context, mediaId pgtype.UUID) error
 }

 type verificationRepository struct {
@@ -24,9 +32,297 @@ type verificationRepository struct {
 	c cache.Cache
 }

-// func NewVerificationRepository(db sqlc.DBTX, c cache.Cache) VerificationRepository {
-// 	return &verificationRepository{
-// 		q: sqlc.New(db),
-// 		c: c,
-// 	}
-// }
+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)
+	return err
+}
+
+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
+}