backend/internal/db/mongo/refill_crud.go
package mongo
import (
"bar/internal/models"
"context"
"time"
"github.com/google/uuid"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/mongo/options"
)
func (b *Backend) CreateRefill(ctx context.Context, refill *models.Refill) error {
ctx, cancel := b.TimeoutContext(ctx)
defer cancel()
refill.CreatedAt = time.Now().Unix()
_, err := b.db.Collection(RefillsCollection).InsertOne(ctx, refill)
if err != nil {
return err
}
return nil
}
func (b *Backend) GetRefill(ctx context.Context, id string) (*models.Refill, error) {
ctx, cancel := b.TimeoutContext(ctx)
defer cancel()
var refill models.Refill
err := b.db.Collection(RefillsCollection).FindOne(ctx,
bson.M{
"id": uuid.MustParse(id),
"$or": []bson.M{
{
"deleted_at": bson.M{
"$exists": false,
},
},
{
"deleted_at": nil,
},
},
},
).Decode(&refill)
if err != nil {
return nil, err
}
return &refill, nil
}
func (b *Backend) UpdateRefill(ctx context.Context, refill *models.Refill) error {
ctx, cancel := b.TimeoutContext(ctx)
defer cancel()
res := b.db.Collection(RefillsCollection).FindOneAndUpdate(ctx,
bson.M{
"id": refill.Id,
"$or": []bson.M{
{
"deleted_at": bson.M{
"$exists": false,
},
},
{
"deleted_at": nil,
},
},
},
bson.M{
"$set": refill,
},
options.FindOneAndUpdate().SetUpsert(true))
if res.Err() != nil {
return res.Err()
}
return nil
}
func (b *Backend) MarkDeleteRefill(ctx context.Context, id, by string) error {
ctx, cancel := b.TimeoutContext(ctx)
defer cancel()
res := b.db.Collection(RefillsCollection).FindOneAndUpdate(ctx,
bson.M{
"id": uuid.MustParse(id),
},
bson.M{
"$set": bson.M{
"deleted_at": time.Now().Unix(),
"deleted_by": uuid.MustParse(by),
},
},
options.FindOneAndUpdate().SetUpsert(false))
if res.Err() != nil {
return res.Err()
}
return nil
}
func (b *Backend) UnMarkDeleteRefill(ctx context.Context, id string) error {
ctx, cancel := b.TimeoutContext(ctx)
defer cancel()
res := b.db.Collection(RefillsCollection).FindOneAndUpdate(ctx,
bson.M{
"id": uuid.MustParse(id),
},
bson.M{
"$set": bson.M{
"deleted_at": nil,
"deleted_by": nil,
"deleted_by_name": nil,
},
},
options.FindOneAndUpdate().SetUpsert(false))
if res.Err() != nil {
return res.Err()
}
return nil
}
func (b *Backend) DeleteRefill(ctx context.Context, id string) error {
ctx, cancel := b.TimeoutContext(ctx)
defer cancel()
res := b.db.Collection(RefillsCollection).FindOneAndDelete(ctx,
bson.M{
"id": uuid.MustParse(id),
},
)
if res.Err() != nil {
return res.Err()
}
return nil
}
func (b *Backend) RestoreRefill(ctx context.Context, id string) error {
ctx, cancel := b.TimeoutContext(ctx)
defer cancel()
res := b.db.Collection(RefillsCollection).FindOneAndUpdate(ctx,
bson.M{
"id": uuid.MustParse(id),
},
bson.M{
"$unset": bson.M{
"deleted_at": "",
"deleted_by": "",
},
},
)
if res.Err() != nil {
return res.Err()
}
return nil
}
func (b *Backend) GetDeletedRefills(ctx context.Context, page uint64, size uint64) ([]*models.Refill, error) {
ctx, cancel := b.TimeoutContext(ctx)
defer cancel()
var accs []*models.Refill
cursor, err := b.db.Collection(RefillsCollection).Find(ctx,
bson.M{
"deleted_at": bson.M{
"$ne": nil,
},
},
options.Find().SetSkip(int64(page*size)).SetLimit(int64(size)))
if err != nil {
return nil, err
}
if err := cursor.All(ctx, &accs); err != nil {
return nil, err
}
return accs, nil
}
func (b *Backend) CountDeletedRefills(ctx context.Context) (uint64, error) {
ctx, cancel := b.TimeoutContext(ctx)
defer cancel()
count, err := b.db.Collection(RefillsCollection).CountDocuments(ctx, bson.M{
"deleted_at": bson.M{
"$ne": nil,
},
})
if err != nil {
return 0, err
}
return uint64(count), nil
}