mongo.go

package outbox

import (
	"context"

	"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/mongo"
)

type mongoStorage struct {
	db *mongo.Database
}

func NewMongoStorage(db *mongo.Database) Storage {
	return &mongoStorage{db}
}

func (m mongoStorage) ListAllItems(ctx context.Context) ([]Model, error) {
	cursor, err := m.db.Collection("outbox").
		Find(ctx, bson.D{{"delivered", false}})

	if err != nil {
		return nil, err
	}

	var items []Model
	if err = cursor.All(ctx, &items); err != nil {
		return nil, err
	}

	return items, nil
}

func (m mongoStorage) UpdateItemToCheck(ctx context.Context, id string) error {
	_, err := m.
		db.
		Collection("outbox").
		UpdateOne(
			ctx,
			bson.M{"idempotency_id": id},
			bson.D{{"$set", bson.D{{"delivered", true}}}},
		)

	return err
}

func (m mongoStorage) DeleteCheckedItems(ctx context.Context) error {
	_, err := m.
		db.
		Collection("outbox").
		DeleteMany(
			ctx,
			bson.D{{"delivered", true}},
		)

	return err
}

func (m mongoStorage) SaveItem(ctx context.Context, item Model) error {
	_, err := m.db.Collection("outbox").InsertOne(ctx, item)

	return err
}