agenda/storage/mongodb.go

package storage

import (
	"context"

	"github.com/rs/zerolog/log"
	"github.com/spf13/viper"
	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
	"gopkg.in/mgo.v2/bson"
)

type MongoDBStorage struct {
	*mongo.Client
	DbName      string
	Collections map[string]string
}

func NewMongoDBStorage(cfg *viper.Viper) (MongoDBStorage, error) {
	var (
		mongodb_host   = cfg.GetString("storage.db.mongodb.host")
		mongodb_port   = cfg.GetString("storage.db.mongodb.port")
		mongodb_dbname = cfg.GetString("storage.db.mongodb.db_name")
		mongodb_events = cfg.GetString("storage.db.mongodb.collections.events")
	)

	client, err := mongo.NewClient(options.Client().ApplyURI("mongodb://" + mongodb_host + ":" + mongodb_port))
	if err != nil {
		return MongoDBStorage{}, err
	}

	err = client.Connect(context.TODO())

	if err != nil {
		return MongoDBStorage{}, err
	}

	storage := MongoDBStorage{
		Client: client,
		DbName: mongodb_dbname,
		Collections: map[string]string{
			"events": mongodb_events,
		},
	}
	//TODO Indexes
	return storage, err
}

func (s MongoDBStorage) CreateEvent(event Event) error {
	collection := s.Client.Database(s.DbName).Collection(s.Collections["events"])
	if _, err := collection.InsertOne(context.TODO(), event); err != nil {
		return err
	}

	return nil
}

func (s MongoDBStorage) GetEvent(id string) (*Event, error) {
	collection := s.Client.Database(s.DbName).Collection(s.Collections["events"])

	event := &Event{}
	if err := collection.FindOne(context.TODO(), bson.M{"_id": id}).Decode(event); err != nil {
		return nil, err
	}

	return event, nil
}

func (s MongoDBStorage) GetEvents(namespaces []string) (events []Event, err error) {
	collection := s.Client.Database(s.DbName).Collection(s.Collections["events"])

	var cur *mongo.Cursor

	findOptions := options.Find()
	//findOptions.SetSort(bson.D{{"startdate", 1}})

	if len(namespaces) == 0 {
		cur, err = collection.Find(context.TODO(), bson.D{}, findOptions)
		if err != nil {
			log.Error().Err(err).Msg("")
			return events, err
		}
	} else {
		cur, err = collection.Find(context.TODO(), bson.M{"namespace": bson.M{"$in": namespaces}}, findOptions)
		if err != nil {
			log.Error().Err(err).Msg("")
			return events, err
		}
	}

	for cur.Next(context.TODO()) {
		var event Event
		//var elem bson.M

		if err := cur.Decode(&event); err != nil {
			return events, err
		}

		events = append(events, event)

	}

	return
}

func (s MongoDBStorage) AddSubscription(eventid string, subscription Subscription) error {
	collection := s.Client.Database(s.DbName).Collection(s.Collections["events"])

	filter := bson.M{"_id": eventid}
	push := bson.M{"$push": bson.M{"subscriptions": subscription}}

	_, err := collection.UpdateOne(context.TODO(), filter, push)
	return err
}

func (s MongoDBStorage) UpdateSubscription(eventid string, subscriber string, deletesubscription Subscription) error {
	collection := s.Client.Database(s.DbName).Collection(s.Collections["events"])

	event := &Event{}
	if errr := collection.FindOne(context.TODO(), bson.M{"_id": eventid}).Decode(event); errr != nil {
		return errr
	}

	for i := range event.Subscriptions {
		if event.Subscriptions[i].Subscriber == subscriber {
			filter := bson.M{"_id": eventid}
			push := bson.M{"$push": bson.M{"deletedsubscriptions": 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
}

func (s MongoDBStorage) UpdateEvent(event Event) error {
	collection := s.Client.Database(s.DbName).Collection(s.Collections["events"])
	if _, err := collection.ReplaceOne(context.TODO(), bson.M{"_id": event.ID}, event); err != nil {
		log.Error().Err(err).Msg("")
		return err
	}

	return nil
}

func (s MongoDBStorage) GetSubscriber(subscriber string) ([]Subscription, error) {
	// collection := s.Client.Database(s.DbName).Collection(s.Collections["events"])
	// subscriptions := []Subscription{}
	// cur, err := collection.Find(context.TODO(), bson.M{"subscriptions": bson.M{"$elemMatch": bson.M{"subscriber": subscriber}}})
	// if err != nil {
	// 	return subscriptions, err
	// }
	// for cur.Next(context.Background()) {
	// 	var event Event
	// 	if err := cur.Decode(&event); err != nil {
	// 		return subscriptions, err
	// 	}
	// 	for i := range event.Subscriptions {
	// 		if event.Subscriptions[i].Subscriber == subscriber {
	// 			subscriptions = append(subscriptions, event.Subscriptions[i])
	// 		}
	// 	}
	// }
	// return subscriptions, nil
	return nil, nil
}

func (s MongoDBStorage) GetSubscriptionByUser(subscriber string) ([]Subscription, error) {
	// events, err := s.GetSubscriber(subscriber)
	// if err != nil {
	// 	panic(err)
	// }
	// fmt.Println(events)
	// return events, nil
	return nil, nil
}