Merge pull request #3789 from jamsman94/feature/labelRestriction

add index to label collection
koderover · Nov 1, 2024 · 0ed4750 · 0ed4750
2 parents cdfa112 + 52192ad
commit 0ed4750
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/pkg/microservice/aslan/core/common/repository/mongodb/label.go b/pkg/microservice/aslan/core/common/repository/mongodb/label.go
@@ -27,6 +27,7 @@ import (
 	"go.mongodb.org/mongo-driver/bson"
 	"go.mongodb.org/mongo-driver/bson/primitive"
 	"go.mongodb.org/mongo-driver/mongo"
+	"go.mongodb.org/mongo-driver/mongo/options"
 )
 
 type LabelColl struct {
@@ -47,8 +48,15 @@ func (c *LabelColl) GetCollectionName() string {
 }
 
 func (c *LabelColl) EnsureIndex(ctx context.Context) error {
-	// currently no query is required for the label defining collection
-	return nil
+	mod := mongo.IndexModel{
+		Keys: bson.D{
+			bson.E{Key: "key", Value: 1},
+		},
+		Options: options.Index().SetUnique(true),
+	}
+
+	_, err := c.Indexes().CreateOne(ctx, mod)
+	return err
 }
 
 func (c *LabelColl) Create(args *models.Label) error {