From 9f2ebe90ee6467c5e284412bd607dfff6aa90db9 Mon Sep 17 00:00:00 2001 From: SoXX Date: Wed, 15 May 2024 15:31:17 +0200 Subject: [PATCH] feat: added GetTags --- internal/tag.go | 49 ++++++++++++++++++++++++++++++++++++++++++++++ internal/user.go | 6 +++--- pkg/graph/graph.go | 5 ++++- pkg/graph/impl.go | 5 ++++- pkg/models/api.go | 2 +- 5 files changed, 61 insertions(+), 6 deletions(-) diff --git a/internal/tag.go b/internal/tag.go index ab26434..3e5a02d 100644 --- a/internal/tag.go +++ b/internal/tag.go @@ -33,3 +33,52 @@ func CreateTagNodeWitRelation(ctx context.Context, driver neo4j.DriverWithContex return nil } + +func GetTags(ctx context.Context, driver neo4j.DriverWithContext) ([]models.TagsWithFrequency, error) { + var userTags []models.TagsWithFrequency + + query := ` + MATCH (:AnthrovePost)-[:HAS]->(t:Tag) + RETURN t as tag, COUNT(t) AS frequency + ORDER BY frequency DESC + ` + params := map[string]any{} + + result, err := neo4j.ExecuteQuery(ctx, driver, query, params, neo4j.EagerResultTransformer) + if err != nil { + return nil, err + } + + if len(result.Records) == 0 { + return nil, nil + } + + for i := range result.Records { + record := result.Records[i] + + tag, _, err := neo4j.GetRecordValue[neo4j.Node](record, "tag") + if err != nil { + return nil, err + } + + frequency, _, err := neo4j.GetRecordValue[int64](record, "frequency") + if err != nil { + return nil, err + } + + userTags = append(userTags, models.TagsWithFrequency{ + Frequency: frequency, + Tags: models.AnthroveTag{ + Name: tag.Props["name"].(string), + Type: tag.Props["type"].(string), + }, + }) + + } + + log.WithFields(log.Fields{ + "tag_amount": len(userTags), + }).Trace("graph: created tag node") + + return userTags, nil +} diff --git a/internal/user.go b/internal/user.go index 3df4d74..a1128ec 100644 --- a/internal/user.go +++ b/internal/user.go @@ -402,8 +402,8 @@ func GetUserFavoriteNodeWithPagination(ctx context.Context, driver neo4j.DriverW } -func GetUserTagNodeWitRelationToFavedPosts(ctx context.Context, driver neo4j.DriverWithContext, anthroveUserID models.AnthroveUserID) ([]models.UserTagsTroughFavedPost, error) { - var userTags []models.UserTagsTroughFavedPost +func GetUserTagNodeWitRelationToFavedPosts(ctx context.Context, driver neo4j.DriverWithContext, anthroveUserID models.AnthroveUserID) ([]models.TagsWithFrequency, error) { + var userTags []models.TagsWithFrequency query := ` MATCH (u:User {user_id: $anthrove_user_id})-[:FAV]->(:AnthrovePost)-[:HAS]->(t:Tag) @@ -436,7 +436,7 @@ func GetUserTagNodeWitRelationToFavedPosts(ctx context.Context, driver neo4j.Dri return nil, err } - userTags = append(userTags, models.UserTagsTroughFavedPost{ + userTags = append(userTags, models.TagsWithFrequency{ Frequency: frequency, Tags: models.AnthroveTag{ Name: tag.Props["name"].(string), diff --git a/pkg/graph/graph.go b/pkg/graph/graph.go index 8acb1da..ac3786f 100644 --- a/pkg/graph/graph.go +++ b/pkg/graph/graph.go @@ -103,5 +103,8 @@ type OtterSpace interface { GetUserFavoritePostsWithPagination(ctx context.Context, anthroveUserID models.AnthroveUserID, skip int, limit int) (*models.FavoriteList, error) // GetUserTagsTroughFavedPosts returns a list of Tags that the user hs favorites through a post - GetUserTagsTroughFavedPosts(ctx context.Context, anthroveUserID models.AnthroveUserID) ([]models.UserTagsTroughFavedPost, error) + GetUserTagsTroughFavedPosts(ctx context.Context, anthroveUserID models.AnthroveUserID) ([]models.TagsWithFrequency, error) + + // GetAllTags returns a list of Tags that the user hs favorites through a post + GetAllTags(ctx context.Context) ([]models.TagsWithFrequency, error) } diff --git a/pkg/graph/impl.go b/pkg/graph/impl.go index 5f77618..9a47593 100644 --- a/pkg/graph/impl.go +++ b/pkg/graph/impl.go @@ -100,9 +100,12 @@ func (g *graphConnection) GetUserFavoritePostsWithPagination(ctx context.Context return internal.GetUserFavoriteNodeWithPagination(ctx, g.driver, anthroveUserID, skip, limit) } -func (g *graphConnection) GetUserTagsTroughFavedPosts(ctx context.Context, anthroveUserID models.AnthroveUserID) ([]models.UserTagsTroughFavedPost, error) { +func (g *graphConnection) GetUserTagsTroughFavedPosts(ctx context.Context, anthroveUserID models.AnthroveUserID) ([]models.TagsWithFrequency, error) { return internal.GetUserTagNodeWitRelationToFavedPosts(ctx, g.driver, anthroveUserID) +} +func (g *graphConnection) GetAllTags(ctx context.Context) ([]models.TagsWithFrequency, error) { + return internal.GetTags(ctx, g.driver) } func logger(graphDebug bool) func(config *config.Config) { diff --git a/pkg/models/api.go b/pkg/models/api.go index 772a327..a30b955 100644 --- a/pkg/models/api.go +++ b/pkg/models/api.go @@ -14,7 +14,7 @@ type FavoriteList struct { Posts []FavoritePost `json:"posts,omitempty"` } -type UserTagsTroughFavedPost struct { +type TagsWithFrequency struct { Frequency int64 `json:"frequency"` Tags AnthroveTag `json:"tags"` }