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

feat: implement relation management system with controllers, services, repositories, and corresponding API documentation.
Build and Release / release (push) Successful in 1m47s
Details

Browse Source
This commit is contained in:
Azen Kain
2026-06-06 17:22:03 +07:00
parent 7ff15ae662
commit fd8d413760
15 changed files with 706 additions and 192 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/repositories/entityRepository.go
+61
View File
@@ -25,6 +25,7 @@ type EntityRepository interface {
Delete(ctx context.Context, id pgtype.UUID) error
DeleteByIDs(ctx context.Context, ids []pgtype.UUID) error
GetByProjectID(ctx context.Context, projectID pgtype.UUID) ([]*models.EntityEntity, error)
GetEntityIDsByWikiIDs(ctx context.Context, wikiIDs []string) (map[string][]string, error)
GetEntityIDsByGeometryIDs(ctx context.Context, geometryIDs []string) (map[string][]string, error)
WithTx(tx pgx.Tx) EntityRepository
}
@@ -473,3 +474,63 @@ func (r *entityRepository) GetEntityIDsByGeometryIDs(ctx context.Context, geomet
return result, nil
}
func (r *entityRepository) GetEntityIDsByWikiIDs(ctx context.Context, wikiIDs []string) (map[string][]string, error) {
if len(wikiIDs) == 0 {
return make(map[string][]string), nil
}
keys := make([]string, len(wikiIDs))
for i, id := range wikiIDs {
keys[i] = cache.Key("entity_wikis:wiki", id)
}
raws := r.c.MGet(ctx, keys...)
result := make(map[string][]string, len(wikiIDs))
missingWikiIDs := make([]string, 0, len(wikiIDs))
missingPgIDs := make([]pgtype.UUID, 0, len(wikiIDs))
for i, b := range raws {
if len(b) > 0 {
var entityIDs []string
if err := json.Unmarshal(b, &entityIDs); err == nil {
result[wikiIDs[i]] = entityIDs
continue
}
}
missingWikiIDs = append(missingWikiIDs, wikiIDs[i])
pgID, err := convert.StringToUUID(wikiIDs[i])
if err == nil {
missingPgIDs = append(missingPgIDs, pgID)
}
}
if len(missingPgIDs) > 0 {
rows, err := r.q.GetEntityIDsByWikiIDs(ctx, missingPgIDs)
if err != nil {
return nil, err
}
dbMap := make(map[string][]string, len(missingWikiIDs))
for _, id := range missingWikiIDs {
dbMap[id] = []string{}
}
for _, row := range rows {
wID := convert.UUIDToString(row.WikiID)
eID := convert.UUIDToString(row.EntityID)
dbMap[wID] = append(dbMap[wID], eID)
}
missingToCache := make(map[string]any, len(dbMap))
for wID, eIDs := range dbMap {
result[wID] = eIDs
missingToCache[cache.Key("entity_wikis:wiki", wID)] = eIDs
}
if len(missingToCache) > 0 {
_ = r.c.MSet(ctx, missingToCache, constants.NormalCacheDuration)
}
}
return result, nil
}
internal/repositories/geometryRepository.go
+61
View File
@@ -32,6 +32,7 @@ type GeometryRepository interface {
BulkDeleteEntityGeometriesByGeometryID(ctx context.Context, geometryID pgtype.UUID) error
DeleteEntityGeometry(ctx context.Context, entityID pgtype.UUID, geometryID pgtype.UUID) error
DeleteEntityGeometriesByProjectID(ctx context.Context, projectID pgtype.UUID) error
GetGeometryIDsByEntityIDs(ctx context.Context, entityIDs []string) (map[string][]string, error)
WithTx(tx pgx.Tx) GeometryRepository
}
@@ -565,3 +566,63 @@ func (r *geometryRepository) SearchByEntityName(ctx context.Context, params sqlc
return geometries, nil
}
func (r *geometryRepository) GetGeometryIDsByEntityIDs(ctx context.Context, entityIDs []string) (map[string][]string, error) {
if len(entityIDs) == 0 {
return make(map[string][]string), nil
}
keys := make([]string, len(entityIDs))
for i, id := range entityIDs {
keys[i] = cache.Key("entity_geometries:entity", id)
}
raws := r.c.MGet(ctx, keys...)
result := make(map[string][]string, len(entityIDs))
missingEntityIDs := make([]string, 0, len(entityIDs))
missingPgIDs := make([]pgtype.UUID, 0, len(entityIDs))
for i, b := range raws {
if len(b) > 0 {
var geometryIDs []string
if err := json.Unmarshal(b, &geometryIDs); err == nil {
result[entityIDs[i]] = geometryIDs
continue
}
}
missingEntityIDs = append(missingEntityIDs, entityIDs[i])
pgID, err := convert.StringToUUID(entityIDs[i])
if err == nil {
missingPgIDs = append(missingPgIDs, pgID)
}
}
if len(missingPgIDs) > 0 {
rows, err := r.q.GetGeometryIDsByEntityIDs(ctx, missingPgIDs)
if err != nil {
return nil, err
}
dbMap := make(map[string][]string, len(missingEntityIDs))
for _, id := range missingEntityIDs {
dbMap[id] = []string{}
}
for _, row := range rows {
eID := convert.UUIDToString(row.EntityID)
gID := convert.UUIDToString(row.GeometryID)
dbMap[eID] = append(dbMap[eID], gID)
}
missingToCache := make(map[string]any, len(dbMap))
for eID, gIDs := range dbMap {
result[eID] = gIDs
missingToCache[cache.Key("entity_geometries:entity", eID)] = gIDs
}
if len(missingToCache) > 0 {
_ = r.c.MSet(ctx, missingToCache, constants.NormalCacheDuration)
}
}
return result, nil
}