Merge "Mongo api changes"

2 files changed, 31 insertions, 3 deletions

diff --git a/src/orchestrator/pkg/infra/db/mongo.go b/src/orchestrator/pkg/infra/db/mongo.go
index b33d6c65..a3fdc570 100644
--- a/src/orchestrator/pkg/infra/db/mongo.go
+++ b/src/orchestrator/pkg/infra/db/mongo.go
@@ -539,9 +539,6 @@ func (m *MongoStore) Find(coll string, key Key, tag string) ([][]byte, error) {
 		}
 		result = append(result, data)
 	}
-	if len(result) == 0 {
-		return result, pkgerrors.Errorf("Did not find any objects with tag: %s", tag)
-	}
 	return result, nil
 }
 
@@ -587,3 +584,31 @@ func (m *MongoStore) Remove(coll string, key Key) error {
 	}
 	return nil
 }
+
+// RemoveTag is used to remove an element from a document
+func (m *MongoStore) RemoveTag(coll string, key Key, tag string) error {
+	c := getCollection(coll, m)
+	ctx := context.Background()
+
+	filter, err := m.findFilter(key)
+	if err != nil {
+		return err
+	}
+
+	_, err = decodeBytes(
+		c.FindOneAndUpdate(
+			ctx,
+			filter,
+			bson.D{
+				{"$unset", bson.D{
+					{tag, ""},
+				}},
+			},
+			options.FindOneAndUpdate().SetUpsert(true).SetReturnDocument(options.After)))
+
+	if err != nil {
+		return pkgerrors.Errorf("Error removing tag: %s", err.Error())
+	}
+
+	return nil
+}
diff --git a/src/orchestrator/pkg/infra/db/store.go b/src/orchestrator/pkg/infra/db/store.go
index 9c6532f1..a332fcda 100644
--- a/src/orchestrator/pkg/infra/db/store.go
+++ b/src/orchestrator/pkg/infra/db/store.go
@@ -64,6 +64,9 @@ type Store interface {
 
 	// Remove all the document(s) matching the key
 	RemoveAll(coll string, key Key) error
+
+	// Remove the specifiec tag from the document matching the key
+	RemoveTag(coll string, key Key, tag string) error
 }
 
 // CreateDBClient creates the DB client