feat(sql): Support ENTITY_TAGS writes to primary and previous storage…

… services feat(sql): Support ENTITY_TAGS reads against primary This removes the last bit of special casing of ENTITY_TAGS and how we read/write them.
spinnaker · Jul 8, 2019 · afdc6f7 · afdc6f7
1 parent d880dbb
commit afdc6f7
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 22 deletions.
diff --git a/front50-sql/src/main/kotlin/com/netflix/spinnaker/front50/model/CompositeStorageService.kt b/front50-sql/src/main/kotlin/com/netflix/spinnaker/front50/model/CompositeStorageService.kt
@@ -76,11 +76,6 @@ class CompositeStorageService(
   }
 
   override fun <T : Timestamped?> loadObject(objectType: ObjectType?, objectKey: String?): T {
-    if (objectType == ObjectType.ENTITY_TAGS) {
-      // entity tags are currently unsupported in SQL
-      return previous.loadObject<T>(objectType, objectKey)
-    }
-
     var exception: Exception? = null
 
     if (isPrimaryReadEnabled()) {
@@ -105,11 +100,6 @@ class CompositeStorageService(
   }
 
   override fun <T : Timestamped?> loadObjects(objectType: ObjectType, objectKeys: List<String>): List<T> {
-    if (objectType == ObjectType.ENTITY_TAGS) {
-      // entity tags are currently unsupported in SQL
-      return previous.loadObjects<T>(objectType, objectKeys)
-    }
-
     var exception: Exception? = null
 
     if (isPrimaryReadEnabled()) {
@@ -156,10 +146,7 @@ class CompositeStorageService(
     }
 
     try {
-      if (objectType != ObjectType.ENTITY_TAGS) {
-        // entity tags are currently unsupported in SQL
-        primary.storeObject(objectType, objectKey, item)
-      }
+      primary.storeObject(objectType, objectKey, item)
     } catch (e: Exception) {
       log.error(
         "{}.storeObject({}, {}) failed",
@@ -174,10 +161,6 @@ class CompositeStorageService(
   }
 
   override fun listObjectKeys(objectType: ObjectType?): Map<String, Long> {
-    if (objectType == ObjectType.ENTITY_TAGS) {
-      return previous.listObjectKeys(objectType)
-    }
-
     val objectKeys = mutableMapOf<String, Long>()
 
     if (isPreviousReadEnabled()) {

diff --git a/...0-sql/src/test/kotlin/com/netflix/spinnaker/front50/model/CompositeStorageServiceTests.kt b/...0-sql/src/test/kotlin/com/netflix/spinnaker/front50/model/CompositeStorageServiceTests.kt
@@ -41,18 +41,22 @@ internal object CompositeStorageServiceTests : JUnit5Minutests {
     }
 
     context("loadObject()") {
-      test("should always load EntityTags from 'previous'") {
+      test("should always load EntityTags from 'primary'") {
         every {
-          previous.loadObject<EntityTags>(ObjectType.ENTITY_TAGS, "id-entitytags001")
+          primary.loadObject<EntityTags>(ObjectType.ENTITY_TAGS, "id-entitytags001")
         } returns EntityTags().apply { id = "id-entitytags001" }
 
+        every {
+          dynamicConfigService.getConfig(Boolean::class.java, any(), any())
+        } returns true
+
         expectThat(
           subject.loadObject<EntityTags>(ObjectType.ENTITY_TAGS, "id-entitytags001").id
         ).isEqualTo("id-entitytags001")
 
         verifyAll {
-          primary wasNot Called
-          previous.loadObject<Timestamped>(ObjectType.ENTITY_TAGS, "id-entitytags001")
+          primary.loadObject<Timestamped>(ObjectType.ENTITY_TAGS, "id-entitytags001")
+          previous wasNot Called
         }
       }