Kain344/History_Api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: implement WikiRepository with caching and transaction support
All checks were successful
Build and Release / release (push) Successful in 1m28s
Details

Browse Source
This commit is contained in:
Azen Kain
2026-05-14 12:56:30 +07:00
parent e7d4f5f7cd
commit 8aceee9130
1 changed files with 80 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

82
internal/repositories/wikiRepository.go
View File

@@ -200,6 +200,8 @@ func (r *wikiRepository) Search(ctx context.Context, params sqlc.SearchWikisPara
}
var wikis []*models.WikiEntity
var ids []string
var pgIds []pgtype.UUID
wikiMap := make(map[string]*models.WikiEntity)
wikiToCache := make(map[string]any)
for _, row := range rows {
@@ -213,10 +215,30 @@ func (r *wikiRepository) Search(ctx context.Context, params sqlc.SearchWikisPara
UpdatedAt: convert.TimeToPtr(row.UpdatedAt),
}
ids = append(ids, wiki.ID)
pgIds = append(pgIds, row.ID)
wikis = append(wikis, wiki)
wikiToCache[fmt.Sprintf("wiki:id:%s", wiki.ID)] = wiki
wikiMap[wiki.ID] = wiki
}
if len(pgIds) > 0 {
samples, err := r.q.GetWikiContentByWikiIDs(ctx, pgIds)
if err == nil {
for _, sample := range samples {
wikiID := convert.UUIDToString(sample.WikiID)
if item, ok := wikiMap[wikiID]; ok {
item.ContentSample = append(item.ContentSample, models.WikiContentSample{
ID: convert.UUIDToString(sample.ID),
Title: sample.Title,
CreatedAt: convert.TimeToPtr(sample.CreatedAt),
})
}
}
}
}
for _, wiki := range wikis {
wikiToCache[fmt.Sprintf("wiki:id:%s", wiki.ID)] = wiki
}
if len(wikiToCache) > 0 {
_ = r.c.MSet(ctx, wikiToCache, constants.NormalCacheDuration)
}
@@ -304,6 +326,8 @@ func (r *wikiRepository) GetByProjectID(ctx context.Context, projectID pgtype.UU
var wikis []*models.WikiEntity
var ids []string
var pgIds []pgtype.UUID
wikiMap := make(map[string]*models.WikiEntity)
wikiToCache := make(map[string]any)
for _, row := range rows {
@@ -317,10 +341,30 @@ func (r *wikiRepository) GetByProjectID(ctx context.Context, projectID pgtype.UU
UpdatedAt: convert.TimeToPtr(row.UpdatedAt),
}
ids = append(ids, wiki.ID)
pgIds = append(pgIds, row.ID)
wikis = append(wikis, wiki)
wikiToCache[fmt.Sprintf("wiki:id:%s", wiki.ID)] = wiki
wikiMap[wiki.ID] = wiki
}
if len(pgIds) > 0 {
samples, err := r.q.GetWikiContentByWikiIDs(ctx, pgIds)
if err == nil {
for _, sample := range samples {
wikiID := convert.UUIDToString(sample.WikiID)
if item, ok := wikiMap[wikiID]; ok {
item.ContentSample = append(item.ContentSample, models.WikiContentSample{
ID: convert.UUIDToString(sample.ID),
Title: sample.Title,
CreatedAt: convert.TimeToPtr(sample.CreatedAt),
})
}
}
}
}
for _, wiki := range wikis {
wikiToCache[fmt.Sprintf("wiki:id:%s", wiki.ID)] = wiki
}
if len(wikiToCache) > 0 {
_ = r.c.MSet(ctx, wikiToCache, constants.NormalCacheDuration)
}
@@ -380,6 +424,18 @@ func (r *wikiRepository) GetBySlug(ctx context.Context, slug string) (*models.Wi
CreatedAt: convert.TimeToPtr(row.CreatedAt),
UpdatedAt: convert.TimeToPtr(row.UpdatedAt),
}
samples, err := r.q.GetWikiContentByWikiID(ctx, row.ID)
if err == nil {
for _, sample := range samples {
wiki.ContentSample = append(wiki.ContentSample, models.WikiContentSample{
ID: convert.UUIDToString(sample.ID),
Title: sample.Title,
CreatedAt: convert.TimeToPtr(sample.CreatedAt),
})
}
}
_ = r.c.Set(ctx, cacheKey, wiki, constants.NormalCacheDuration)
return &wiki, nil
@@ -409,6 +465,7 @@ func (r *wikiRepository) GetBySlugs(ctx context.Context, slugs []string) ([]*mod
if len(missingSlugs) > 0 {
dbRows, err := r.q.GetWikisBySlugs(ctx, missingSlugs)
if err == nil {
var pgIds []pgtype.UUID
for _, row := range dbRows {
item := models.WikiEntity{
ID: convert.UUIDToString(row.ID),
@@ -419,8 +476,29 @@ func (r *wikiRepository) GetBySlugs(ctx context.Context, slugs []string) ([]*mod
CreatedAt: convert.TimeToPtr(row.CreatedAt),
UpdatedAt: convert.TimeToPtr(row.UpdatedAt),
}
pgIds = append(pgIds, row.ID)
dbMap[item.Slug] = &item
}
if len(pgIds) > 0 {
samples, sErr := r.q.GetWikiContentByWikiIDs(ctx, pgIds)
if sErr == nil {
wikiByID := make(map[string]*models.WikiEntity)
for _, item := range dbMap {
wikiByID[item.ID] = item
}
for _, sample := range samples {
wikiID := convert.UUIDToString(sample.WikiID)
if item, ok := wikiByID[wikiID]; ok {
item.ContentSample = append(item.ContentSample, models.WikiContentSample{
ID: convert.UUIDToString(sample.ID),
Title: sample.Title,
CreatedAt: convert.TimeToPtr(sample.CreatedAt),
})
}
}
}
}
}
}