add logic to delete and srore sub

This commit is contained in:
soukainna
2023-02-01 16:04:10 +01:00
parent ffa341b13d
commit 1ce72b2f88
10 changed files with 1751 additions and 103 deletions

View File

@@ -121,12 +121,25 @@ func (s MongoDBStorage) AddSubscription(eventid string, subscription Subscriptio
return err
}
func (s MongoDBStorage) RemoveSubscription(subscriptionid string, subscription Subscription) error {
func (s MongoDBStorage) UpdateSubscription(eventid string, subscriber string, deletesubscription Subscription) error {
collection := s.Client.Database(s.DbName).Collection(s.Collections["events"])
filter := bson.M{"_id": subscriptionid}
push := bson.M{"$pull": bson.M{"subscriptions": subscription}}
event := &Event{}
if errr := collection.FindOne(context.TODO(), bson.M{"_id": eventid}).Decode(event); errr != nil {
return errr
}
_, err := collection.UpdateOne(context.TODO(), filter, push)
return err
for i := range event.Subscriptions {
if event.Subscriptions[i].Subscriber == subscriber {
filter := bson.M{"_id": eventid}
push := bson.M{"$push": bson.M{"deletesubscriptions": deletesubscription}}
pull := bson.M{"$pull": bson.M{"subscriptions": bson.M{"subscriber": subscriber}}}
_, er := collection.UpdateOne(context.TODO(), filter, push)
if _, err := collection.UpdateOne(context.TODO(), filter, pull); err != nil {
return err
}
return er
}
}
return nil
}