Merge pull request #49 from Scalingo/fix/12/missing_index

Add EnsureParanoidIndices method

Author: john-scalingo

mongo/document/document.go

@@ -5,6 +5,8 @@ import (
 	"fmt"
 	"time"
 
+	"github.com/sirupsen/logrus"
+
 	"gopkg.in/mgo.v2"
 	"gopkg.in/mgo.v2/bson"
 
@@ -22,14 +24,23 @@ type document interface {
 	Validable
 }
 
+var _ document = &Base{}
+var _ document = &Paranoid{}
+
 type scopable interface {
 	scope(bson.M) bson.M
 }
 
+var _ scopable = Base{}
+var _ scopable = Paranoid{}
+
 type destroyable interface {
 	destroy(ctx context.Context, collectionName string) error
 }
 
+var _ destroyable = &Base{}
+var _ destroyable = &Paranoid{}
+
 type Closer interface {
 	Close()
 }
@@ -38,7 +49,10 @@ type Validable interface {
 	Validate(ctx context.Context) *ValidationErrors
 }
 
-// Create inser the document in the database, returns an error if document already exists and set CreatedAt timestamp
+var _ Validable = &Base{}
+
+// Create inserts the document in the database, returns an error if document
+// already exists and set CreatedAt timestamp
 func Create(ctx context.Context, collectionName string, doc document) error {
 	log := logger.Get(ctx)
 	doc.ensureID()
@@ -221,3 +235,24 @@ func Update(ctx context.Context, collectionName string, update bson.M, doc docum
 	log.WithField("query", update).Debugf("update %v", collectionName)
 	return c.UpdateId(doc.getID(), update)
 }
+
+func EnsureParanoidIndices(ctx context.Context, collectionNames ...string) {
+	log := logger.Get(ctx)
+
+	for _, collectionName := range collectionNames {
+		log = logger.Get(ctx).WithFields(logrus.Fields{
+			"init":       "setup-indices",
+			"collection": collectionName,
+		})
+		ctx = logger.ToCtx(ctx, log)
+		log.Info("Setup the MongoDB index")
+
+		c := mongo.Session(log).Clone().DB("").C(collectionName)
+		defer c.Database.Session.Close()
+		err := c.EnsureIndexKey("deleted_at")
+		if err != nil {
+			log.WithError(err).Error("fail to setup the deleted_at index")
+			continue
+		}
+	}
+}