api/database/mongodb/changes.go

package mongodb

import (
	"context"
	"errors"
	"git.aiterp.net/rpdata/api/models"
	"git.aiterp.net/rpdata/api/repositories"
	"github.com/globalsign/mgo"
	"github.com/globalsign/mgo/bson"
	"strconv"
	"strings"
	"time"
)

type changeRepository struct {
	restoreIDs bool
	changes    *mgo.Collection
	idCounter  *counter
}

func (r *changeRepository) Find(ctx context.Context, id string) (*models.Change, error) {
	change := new(models.Change)
	err := r.changes.FindId(id).One(change)
	if err != nil {
		return nil, err
	}

	return change, nil
}

func (r *changeRepository) List(ctx context.Context, filter models.ChangeFilter) ([]*models.Change, error) {
	query := bson.M{}
	limit := 0
	if filter.EarliestDate != nil && !filter.EarliestDate.IsZero() {
		query["date"] = bson.M{"$gte": *filter.EarliestDate}
	}
	if filter.LatestDate != nil && !filter.LatestDate.IsZero() {
		dateBson, ok := query["date"].(bson.M)
		if !ok {
			dateBson = bson.M{}
		}

		dateBson["$gte"] = *filter.EarliestDate
	}
	if len(filter.Keys) > 0 {
		query["keys"] = bson.M{"$in": filter.Keys}
	} else if !r.restoreIDs {
		query["listed"] = true
	}
	if filter.Author != nil && *filter.Author != "" {
		query["author"] = *filter.Author
	}
	if filter.Limit != nil {
		limit = *filter.Limit
	}

	initialSize := 64
	if limit > 0 && limit < 256 {
		initialSize = limit
	}

	changes := make([]*models.Change, 0, initialSize)
	err := r.changes.Find(query).Sort("-date").Limit(limit).All(&changes)
	if err != nil {
		return nil, err
	}

	return changes, nil
}

func (r *changeRepository) Insert(ctx context.Context, change models.Change) (*models.Change, error) {
	if !r.restoreIDs || change.ID == "" {
		next, err := r.idCounter.Increment(1)
		if err != nil {
			return nil, err
		}

		change.ID = "Change_" + strconv.Itoa(next)
	} else {
		tokens := strings.Split(change.ID, "_")
		if len(tokens) != 2 || tokens[0] != "Change" {
			return nil, errors.New("Invalid change ID")
		}

		n, err := strconv.Atoi(tokens[1])
		if err != nil {
			return nil, err
		}

		_ = r.idCounter.Bump(n)
	}

	err := r.changes.Insert(&change)
	if err != nil {
		return nil, err
	}

	return &change, nil
}

func (r *changeRepository) Remove(ctx context.Context, change models.Change) error {
	return r.changes.RemoveId(change.ID)
}

func newChangeRepository(db *mgo.Database, restoreIDs bool) (repositories.ChangeRepository, error) {
	collection := db.C("common.changes")

	// Delete the old index if it exists.
	_ = collection.DropIndexName("date_1")

	err := collection.EnsureIndex(mgo.Index{
		Name:        "expiry",
		Key:         []string{"date"},
		ExpireAfter: time.Hour * 2400, // 100 days
	})
	if err != nil {
		return nil, err
	}

	err = collection.EnsureIndexKey("author")
	if err != nil {
		return nil, err
	}

	err = collection.EnsureIndexKey("keys")
	if err != nil {
		return nil, err
	}

	return &changeRepository{
		restoreIDs: restoreIDs,
		changes:    collection,
		idCounter:  newCounter(db, "auto_increment", "Change"),
	}, nil
}