initial import

registrierung/mongodb/repository.go

@@ -0,0 +1,97 @@
+package mongodb
+
+import (
+	"context"
+	"fmt"
+	"time"
+
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/bson/primitive"
+	"go.mongodb.org/mongo-driver/mongo"
+	"go.mongodb.org/mongo-driver/mongo/options"
+	"training-fellow.de/registrierung"
+)
+
+func NewRepo(url, database, collection string) registrierung.RegistrierungsRepository {
+	return &mongoDBRepositoy{url, collection, database}
+}
+
+type mongoDBRepositoy struct {
+	url        string
+	collection string
+	database   string
+}
+
+type mongoCall func(*mongo.Collection) error
+
+func (m *mongoDBRepositoy) SaveRegistrierung(registrierung *registrierung.Registrierung) error {
+	fmt.Println("Save")
+
+	ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
+	defer cancel()
+
+	return m.executeInClient(ctx, func(collection *mongo.Collection) error {
+		registrierung.ID = primitive.NewObjectID().Hex()
+		_, err := collection.InsertOne(ctx, registrierung)
+		return err
+	})
+
+}
+
+func (m *mongoDBRepositoy) GetUnconfirmedRegistrierung() ([]*registrierung.Registrierung, error) {
+
+	ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
+	defer cancel()
+	registrations := make([]*registrierung.Registrierung, 0)
+	err := m.executeInClient(ctx, func(collection *mongo.Collection) error {
+		cursor, err := collection.Find(ctx, bson.M{"confirmed": false})
+		if err != nil {
+			return err
+		}
+		defer cursor.Close(ctx)
+		for cursor.Next(ctx) {
+			reg := &registrierung.Registrierung{}
+			cursor.Decode(reg)
+			registrations = append(registrations, reg)
+		}
+		return nil
+	})
+
+	return registrations, err
+}
+
+func (m *mongoDBRepositoy) ConfirmedRegistrierung(registrierungsId string) (*registrierung.Registrierung, error) {
+	ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
+	defer cancel()
+	registrierung := &registrierung.Registrierung{}
+	err := m.executeInClient(ctx, func(collection *mongo.Collection) error {
+		result := collection.FindOneAndUpdate(ctx,
+			bson.M{"_id": registrierungsId},
+			bson.M{"$set": bson.M{"confirmed": true}})
+		return result.Decode(&registrierung)
+	})
+	return registrierung, err
+}
+
+func (m *mongoDBRepositoy) executeInClient(ctx context.Context, callback mongoCall) error {
+	client, err := mongo.NewClient(
+		options.Client().
+			ApplyURI(m.url))
+	if err != nil {
+		return err
+	}
+
+	err = client.Connect(ctx)
+	if err != nil {
+		return err
+	}
+	defer func() {
+		if err = client.Disconnect(ctx); err != nil {
+			panic(err)
+		}
+	}()
+
+	collection := client.Database(m.database).Collection(m.collection)
+	return callback(collection)
+
+}