OceanSlim/grain
1
0
Fork 0
mirror of https://github.com/0ceanSlim/grain.git synced 2024-10-30 01:26:32 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

kind0 events are now being updated per pubkey

Browse Source
This commit is contained in:
Chris kerr 2024-07-20 10:03:52 -04:00
parent 36956603ed
commit c7b0e9c390
2 changed files with 21 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
events/events.go
View File

@ -41,25 +41,15 @@ func InitCollections(client *mongo.Client, eventKind0, eventKind1 string) {
}
func HandleEvent(ctx context.Context, evt Event) error {
var collection *mongo.Collection
switch evt.Kind {
case 0:
collection = eventKind0Collection
return HandleEventKind0(ctx, evt, eventKind0Collection)
case 1:
return HandleEventKind1(ctx, evt, eventKind1Collection)
default:
fmt.Println("Unknown event kind:", evt.Kind)
return fmt.Errorf("unknown event kind: %d", evt.Kind)
}
_, err := collection.InsertOne(ctx, evt)
if err != nil {
fmt.Println("Error inserting event into MongoDB:", err)
return err
}
fmt.Println("Inserted event into MongoDB:", evt.ID)
return nil
}
func GetCollections() map[string]*mongo.Collection {

25
events/kind0.go
View File

@ -4,23 +4,38 @@ import (
"context"
"fmt"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/mongo/options"
)
func HandleEventKind0(ctx context.Context, evt Event, collection *mongo.Collection) error {
// Perform specific validation for event kind 1
// Perform specific validation for event kind 0
if !isValidEventKind0(evt) {
return fmt.Errorf("validation failed for event kind 0: %s", evt.ID)
}
// Insert event into MongoDB
_, err := collection.InsertOne(ctx, evt)
// Replace the existing event if it has the same pubkey
filter := bson.M{"pubkey": evt.PubKey}
update := bson.M{
"$set": bson.M{
"id": evt.ID,
"created_at": evt.CreatedAt,
"kind": evt.Kind,
"tags": evt.Tags,
"content": evt.Content,
"sig": evt.Sig,
},
}
options := options.Update().SetUpsert(true) // Insert if not found
_, err := collection.UpdateOne(ctx, filter, update, options)
if err != nil {
fmt.Println("Error inserting event into MongoDB:", err)
fmt.Println("Error updating/inserting event kind 0 into MongoDB:", err)
return err
}
fmt.Println("Inserted event kind 0 into MongoDB:", evt.ID)
fmt.Println("Upserted event kind 0 into MongoDB:", evt.ID)
return nil
}