#1340 Add missing index on tags

TheHive-Project · Jun 3, 2020 · 0c7bc08 · 0c7bc08
1 parent 107e558
commit 0c7bc08
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/thehive/app/org/thp/thehive/models/SchemaUpdater.scala b/thehive/app/org/thp/thehive/models/SchemaUpdater.scala
@@ -58,5 +58,6 @@ class SchemaUpdater @Inject() (theHiveSchema: TheHiveSchema, db: Database, userS
       removeIndexLock("DataData")
       removePropertyLock("data")
     }
+    .addIndex("Tag", IndexType.tryUnique, "namespace", "predicate", "value")
     .execute(db)(userSrv.getSystemAuthContext)
 }
diff --git a/thehive/app/org/thp/thehive/models/Tag.scala b/thehive/app/org/thp/thehive/models/Tag.scala
@@ -1,12 +1,13 @@
 package org.thp.thehive.models
 
-import scala.util.Try
-import scala.util.matching.Regex
-
+import org.thp.scalligraph.VertexEntity
+import org.thp.scalligraph.models.{DefineIndex, IndexType}
 import play.api.Logger
 
-import org.thp.scalligraph.VertexEntity
+import scala.util.Try
+import scala.util.matching.Regex
 
+@DefineIndex(IndexType.tryUnique, "namespace", "predicate", "value")
 @VertexEntity
 case class Tag(
     namespace: String,