From 02733498b92ee7a3bc7689e5bce6dfecda8a5614 Mon Sep 17 00:00:00 2001 From: soxx Date: Fri, 16 Feb 2024 21:33:46 +0100 Subject: [PATCH] feat: implemented getting user sources --- internal/user.go | 4 ++-- pkg/graph.go | 2 +- pkg/impl.go | 5 ++--- 3 files changed, 5 insertions(+), 6 deletions(-) diff --git a/internal/user.go b/internal/user.go index 9197c43..d2882a7 100644 --- a/internal/user.go +++ b/internal/user.go @@ -94,12 +94,12 @@ func GetUserFavoritesCount(ctx context.Context, driver neo4j.DriverWithContext, return userFavoriteCount, nil } -func GetUserWithRelationsAndSource(ctx context.Context, driver neo4j.DriverWithContext, anthroveUserID models.AnthroveUserID) (map[string]models.AnthroveUserRelationship, error) { +func GetUserSourceLink(ctx context.Context, driver neo4j.DriverWithContext, anthroveUserID models.AnthroveUserID) (map[string]models.AnthroveUserRelationship, error) { userSource := make(map[string]models.AnthroveUserRelationship) query := ` - MATCH (user:User{user_id: $anthrove_user_id})-[r:HAS_ACCOUNT_AT]->(s:Source) + MATCH (user:User{user_id: anthrove_user_id})-[r:HAS_ACCOUNT_AT]->(s:Source) RETURN toString(r.user_id) AS sourceUserID, toString(r.username) AS sourceUsername, s.display_name as sourceDisplayName; ` params := map[string]any{ diff --git a/pkg/graph.go b/pkg/graph.go index 1b76a52..7e48b8c 100644 --- a/pkg/graph.go +++ b/pkg/graph.go @@ -44,7 +44,7 @@ type Graph interface { GetUserFavoriteCount(ctx context.Context, anthroveUserID models.AnthroveUserID) (int64, error) // GetUserSourceLinks retrieves the links between a user and sources in the graph - GetUserSourceLinks(ctx context.Context, anthroveUser *models.AnthroveUser) (map[string]models.AnthroveUserRelationship, error) + GetUserSourceLinks(ctx context.Context, anthroveUserID models.AnthroveUserID) (map[string]models.AnthroveUserRelationship, error) // GetAnthroveUser retrieves an Anthrove user from the graph by their ID GetAnthroveUser(ctx context.Context, anthroveUser *models.AnthroveUser) (*models.AnthroveUser, error) diff --git a/pkg/impl.go b/pkg/impl.go index d36bfb8..7d78889 100644 --- a/pkg/impl.go +++ b/pkg/impl.go @@ -80,9 +80,8 @@ func (g *graphConnection) GetUserFavoriteCount(ctx context.Context, anthroveUser return internal.GetUserFavoritesCount(ctx, g.driver, anthroveUserID) } -func (g *graphConnection) GetUserSourceLinks(ctx context.Context, anthroveUser *models.AnthroveUser) (map[string]models.AnthroveUserRelationship, error) { - //TODO implement me - panic("implement me") +func (g *graphConnection) GetUserSourceLinks(ctx context.Context, anthroveUserID models.AnthroveUserID) (map[string]models.AnthroveUserRelationship, error) { + return internal.GetUserSourceLink(ctx, g.driver, anthroveUserID) } func (g *graphConnection) GetAnthroveUser(ctx context.Context, anthroveUser *models.AnthroveUser) (*models.AnthroveUser, error) {