65 lines
2.0 KiB
Go
65 lines
2.0 KiB
Go
package internal
|
|
|
|
import (
|
|
"context"
|
|
|
|
"git.dragse.it/anthrove/anthrove-graph-sdk.git/pkg/models"
|
|
"github.com/neo4j/neo4j-go-driver/v5/neo4j"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
func EstablishAnthrovePostToSourceLink(ctx context.Context, driver neo4j.DriverWithContext, anthrovePostID models.AnthrovePostID, anthroveSourceDomain string, anthrovePostRelationship *models.AnthrovePostRelationship) error {
|
|
query := `
|
|
MATCH (sourceNode:Source {domain: $source_url})
|
|
MATCH (postNode:AnthrovePost {post_id: $anthrove_post_id})
|
|
MERGE (sourceNode)-[:REFERENCE {url: $source_post_url, source_post_id: $source_post_id}]->(postNode)
|
|
`
|
|
|
|
params := map[string]any{
|
|
"source_url": anthroveSourceDomain,
|
|
"anthrove_post_id": anthrovePostID,
|
|
"source_post_url": anthrovePostRelationship.Url,
|
|
"source_post_id": anthrovePostRelationship.PostID,
|
|
}
|
|
|
|
_, err := neo4j.ExecuteQuery(ctx, driver, query, params, neo4j.EagerResultTransformer)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
log.WithFields(log.Fields{
|
|
"source_url": anthroveSourceDomain,
|
|
"anthrove_post_id": anthrovePostID,
|
|
"source_post_url": anthrovePostRelationship.Url,
|
|
"source_post_id": anthrovePostRelationship.PostID,
|
|
}).Trace("graph: creating anthrove post to source link")
|
|
|
|
return nil
|
|
}
|
|
|
|
func EstablishUserToPostLink(ctx context.Context, driver neo4j.DriverWithContext, anthroveUser *models.AnthroveUser, anthrovePost *models.AnthrovePost) error {
|
|
|
|
query := `
|
|
MATCH (user:User {user_id: $anthrove_user_id})
|
|
MATCH (anthrovePost:AnthrovePost {post_id: $anthrove_post_id})
|
|
MERGE (user)-[:FAV]->(anthrovePost)
|
|
`
|
|
|
|
params := map[string]any{
|
|
"anthrove_post_id": anthrovePost.PostID,
|
|
"anthrove_user_id": anthroveUser.UserID,
|
|
}
|
|
|
|
_, err := neo4j.ExecuteQuery(ctx, driver, query, params, neo4j.EagerResultTransformer)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
log.WithFields(log.Fields{
|
|
"anthrove_post_id": anthrovePost.PostID,
|
|
"anthrove_user_id": anthroveUser.UserID,
|
|
}).Trace("graph: created user to post link")
|
|
|
|
return nil
|
|
}
|