From e3671f08ae9e0eba77d920afad25011709f0193b Mon Sep 17 00:00:00 2001
From: kamaleshnneerasa <kamaleshnneerasa@github.com>
Date: Thu, 14 Oct 2021 14:00:02 +0530
Subject: [PATCH 1/8] Changed label.proto

---
 .../org/hypertrace/label/config/service/v1/label.proto    | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/labels-config-service-api/src/main/proto/org/hypertrace/label/config/service/v1/label.proto b/labels-config-service-api/src/main/proto/org/hypertrace/label/config/service/v1/label.proto
index 9907bde5..4449639f 100644
--- a/labels-config-service-api/src/main/proto/org/hypertrace/label/config/service/v1/label.proto
+++ b/labels-config-service-api/src/main/proto/org/hypertrace/label/config/service/v1/label.proto
@@ -7,8 +7,16 @@ option java_multiple_files = true;
 message Label {
   string id = 1;
   string key = 2;
+  optional string created_by = 3;
+  optional string created_time = 4;
+  optional string description = 5;
+  optional string color = 6;
+  optional string rule_id = 7;
 }
 
 message CreateLabel {
   string key = 2;
+  optional string description = 3;
+  optional string color = 4;
+  optional string rule_id = 5;
 }

From 790327746b8aa5fac88e7a32e3588c3343dd2179 Mon Sep 17 00:00:00 2001
From: kamaleshnneerasa <kamaleshnneerasa@github.com>
Date: Tue, 19 Oct 2021 12:09:47 +0530
Subject: [PATCH 2/8] Adding fields in Labels schema

---
 .../hypertrace/label/config/service/v1/label.proto  | 13 +++++++------
 .../config/service/LabelsConfigServiceImpl.java     |  4 ++++
 2 files changed, 11 insertions(+), 6 deletions(-)

diff --git a/labels-config-service-api/src/main/proto/org/hypertrace/label/config/service/v1/label.proto b/labels-config-service-api/src/main/proto/org/hypertrace/label/config/service/v1/label.proto
index 4449639f..db9cb182 100644
--- a/labels-config-service-api/src/main/proto/org/hypertrace/label/config/service/v1/label.proto
+++ b/labels-config-service-api/src/main/proto/org/hypertrace/label/config/service/v1/label.proto
@@ -9,14 +9,15 @@ message Label {
   string key = 2;
   optional string created_by = 3;
   optional string created_time = 4;
-  optional string description = 5;
-  optional string color = 6;
-  optional string rule_id = 7;
+  optional string color = 5;
+  optional string rule_id = 6;
+  optional string description = 7;
 }
 
 message CreateLabel {
   string key = 2;
-  optional string description = 3;
-  optional string color = 4;
-  optional string rule_id = 5;
+  optional string created_by = 3;
+  optional string color = 5;
+  optional string rule_id = 6;
+  optional string description = 7;
 }
diff --git a/labels-config-service-impl/src/main/java/org/hypertrace/label/config/service/LabelsConfigServiceImpl.java b/labels-config-service-impl/src/main/java/org/hypertrace/label/config/service/LabelsConfigServiceImpl.java
index a21647c8..ef3a1144 100644
--- a/labels-config-service-impl/src/main/java/org/hypertrace/label/config/service/LabelsConfigServiceImpl.java
+++ b/labels-config-service-impl/src/main/java/org/hypertrace/label/config/service/LabelsConfigServiceImpl.java
@@ -116,6 +116,10 @@ public void createLabel(
               Label.newBuilder()
                   .setId(UUID.randomUUID().toString())
                   .setKey(createLabel.getKey())
+                  .setCreatedBy(createLabel.getCreatedBy())
+                  .setColor(createLabel.getColor())
+                  .setRuleId(createLabel.getRuleId())
+                  .setDescription(createLabel.getDescription())
                   .build();
           Label createdLabel = labelStore.upsertObject(requestContext, label);
           responseObserver.onNext(CreateLabelResponse.newBuilder().setLabel(createdLabel).build());

From a8a751a9df9365a788c6e9af23374ffac12d3627 Mon Sep 17 00:00:00 2001
From: kamaleshnneerasa <kamaleshnneerasa@github.com>
Date: Mon, 25 Oct 2021 14:30:23 +0530
Subject: [PATCH 3/8] Adding impl for timestamp

---
 .../service/LabelsConfigServiceImpl.java      | 19 +++++++++++++++++--
 .../service/LabelsConfigServiceImplTest.java  |  5 ++++-
 2 files changed, 21 insertions(+), 3 deletions(-)

diff --git a/labels-config-service-impl/src/main/java/org/hypertrace/label/config/service/LabelsConfigServiceImpl.java b/labels-config-service-impl/src/main/java/org/hypertrace/label/config/service/LabelsConfigServiceImpl.java
index 1921c794..0db956ec 100644
--- a/labels-config-service-impl/src/main/java/org/hypertrace/label/config/service/LabelsConfigServiceImpl.java
+++ b/labels-config-service-impl/src/main/java/org/hypertrace/label/config/service/LabelsConfigServiceImpl.java
@@ -9,9 +9,13 @@
 import io.grpc.Status;
 import io.grpc.StatusRuntimeException;
 import io.grpc.stub.StreamObserver;
+import java.time.ZoneId;
+import java.time.format.DateTimeFormatter;
+import java.time.format.FormatStyle;
 import java.util.ArrayList;
 import java.util.Collections;
 import java.util.List;
+import java.util.Locale;
 import java.util.Map;
 import java.util.Optional;
 import java.util.UUID;
@@ -122,7 +126,8 @@ public void createLabel(
                   .setRuleId(createLabel.getRuleId())
                   .setDescription(createLabel.getDescription())
                   .build();
-          Label createdLabel = labelStore.upsertObject(requestContext, label).getData();
+          Label createdLabel =
+              getLabelFromContextualObject(labelStore.upsertObject(requestContext, label));
           responseObserver.onNext(CreateLabelResponse.newBuilder().setLabel(createdLabel).build());
           responseObserver.onCompleted();
         } finally {
@@ -165,7 +170,7 @@ public void getLabels(
     allLabels.addAll(systemLabels);
     List<Label> tenantLabels =
         labelStore.getAllObjects(requestContext).stream()
-            .map(ContextualConfigObject::getData)
+            .map(this::getLabelFromContextualObject)
             .collect(Collectors.toUnmodifiableList());
     allLabels.addAll(tenantLabels);
     responseObserver.onNext(GetLabelsResponse.newBuilder().addAllLabels(allLabels).build());
@@ -251,4 +256,14 @@ private boolean isDuplicateKey(String key) {
         labelList.stream().filter(label -> label.getKey().equals(key)).findAny();
     return match.isPresent();
   }
+
+  private Label getLabelFromContextualObject(ContextualConfigObject<Label> labelObject) {
+    DateTimeFormatter formatter =
+        DateTimeFormatter.ofLocalizedDateTime(FormatStyle.SHORT)
+            .withLocale(Locale.US)
+            .withZone(ZoneId.systemDefault());
+    String createdTimestamp = formatter.format(labelObject.getCreationTimestamp());
+    Label label = labelObject.getData();
+    return label.toBuilder().setCreatedTime(createdTimestamp).build();
+  }
 }
diff --git a/labels-config-service-impl/src/test/java/org/hypertrace/label/config/service/LabelsConfigServiceImplTest.java b/labels-config-service-impl/src/test/java/org/hypertrace/label/config/service/LabelsConfigServiceImplTest.java
index 14788813..0d010543 100644
--- a/labels-config-service-impl/src/test/java/org/hypertrace/label/config/service/LabelsConfigServiceImplTest.java
+++ b/labels-config-service-impl/src/test/java/org/hypertrace/label/config/service/LabelsConfigServiceImplTest.java
@@ -121,7 +121,10 @@ void test_system_createLabel() {
 
   @Test
   void test_getLabel() {
-    List<Label> createdLabelsList = createLabels();
+    List<Label> createdLabelsList =
+        createLabels().stream()
+            .map(label -> label.toBuilder().clearCreatedTime().build())
+            .collect(Collectors.toList());
     // Querying each label one by one for the created or inserted labels in the previous step
     List<Label> getLabelList =
         createdLabelsList.stream()

From 27fabc4b0d2a404a2900a35fafabf3971d2b9f51 Mon Sep 17 00:00:00 2001
From: kamaleshnneerasa <kamaleshnneerasa@github.com>
Date: Mon, 25 Oct 2021 21:54:16 +0530
Subject: [PATCH 4/8] Addressing PR comments

---
 .../hypertrace/label/config/service/v1/label.proto   |  6 +++---
 .../config/service/LabelsConfigServiceImpl.java      | 12 ++----------
 2 files changed, 5 insertions(+), 13 deletions(-)

diff --git a/labels-config-service-api/src/main/proto/org/hypertrace/label/config/service/v1/label.proto b/labels-config-service-api/src/main/proto/org/hypertrace/label/config/service/v1/label.proto
index db9cb182..8e08d4ff 100644
--- a/labels-config-service-api/src/main/proto/org/hypertrace/label/config/service/v1/label.proto
+++ b/labels-config-service-api/src/main/proto/org/hypertrace/label/config/service/v1/label.proto
@@ -8,9 +8,9 @@ message Label {
   string id = 1;
   string key = 2;
   optional string created_by = 3;
-  optional string created_time = 4;
+  optional int64 created_time = 4;
   optional string color = 5;
-  optional string rule_id = 6;
+  optional string created_by_application_rule_id = 6;
   optional string description = 7;
 }
 
@@ -18,6 +18,6 @@ message CreateLabel {
   string key = 2;
   optional string created_by = 3;
   optional string color = 5;
-  optional string rule_id = 6;
+  optional string created_by_application_rule_id = 6;
   optional string description = 7;
 }
diff --git a/labels-config-service-impl/src/main/java/org/hypertrace/label/config/service/LabelsConfigServiceImpl.java b/labels-config-service-impl/src/main/java/org/hypertrace/label/config/service/LabelsConfigServiceImpl.java
index 0db956ec..cba79fb4 100644
--- a/labels-config-service-impl/src/main/java/org/hypertrace/label/config/service/LabelsConfigServiceImpl.java
+++ b/labels-config-service-impl/src/main/java/org/hypertrace/label/config/service/LabelsConfigServiceImpl.java
@@ -9,13 +9,9 @@
 import io.grpc.Status;
 import io.grpc.StatusRuntimeException;
 import io.grpc.stub.StreamObserver;
-import java.time.ZoneId;
-import java.time.format.DateTimeFormatter;
-import java.time.format.FormatStyle;
 import java.util.ArrayList;
 import java.util.Collections;
 import java.util.List;
-import java.util.Locale;
 import java.util.Map;
 import java.util.Optional;
 import java.util.UUID;
@@ -123,7 +119,7 @@ public void createLabel(
                   .setKey(createLabel.getKey())
                   .setCreatedBy(createLabel.getCreatedBy())
                   .setColor(createLabel.getColor())
-                  .setRuleId(createLabel.getRuleId())
+                  .setCreatedByApplicationRuleId(createLabel.getCreatedByApplicationRuleId())
                   .setDescription(createLabel.getDescription())
                   .build();
           Label createdLabel =
@@ -258,11 +254,7 @@ private boolean isDuplicateKey(String key) {
   }
 
   private Label getLabelFromContextualObject(ContextualConfigObject<Label> labelObject) {
-    DateTimeFormatter formatter =
-        DateTimeFormatter.ofLocalizedDateTime(FormatStyle.SHORT)
-            .withLocale(Locale.US)
-            .withZone(ZoneId.systemDefault());
-    String createdTimestamp = formatter.format(labelObject.getCreationTimestamp());
+    long createdTimestamp = labelObject.getCreationTimestamp().toEpochMilli();
     Label label = labelObject.getData();
     return label.toBuilder().setCreatedTime(createdTimestamp).build();
   }

From f08d250b93dac1f1118123a3c5d0bbbfc4f89ee4 Mon Sep 17 00:00:00 2001
From: kamaleshnneerasa <kamaleshnneerasa@github.com>
Date: Wed, 3 Nov 2021 17:01:00 +0530
Subject: [PATCH 5/8] Added impl for new schema

---
 .../label/config/service/v1/label.proto       | 18 ++---
 .../service/v1/labels_config_service.proto    |  6 +-
 .../service/LabelsConfigServiceImpl.java      | 71 +++++++++----------
 .../service/LabelsConfigServiceImplTest.java  | 70 ++++++++++--------
 4 files changed, 85 insertions(+), 80 deletions(-)

diff --git a/labels-config-service-api/src/main/proto/org/hypertrace/label/config/service/v1/label.proto b/labels-config-service-api/src/main/proto/org/hypertrace/label/config/service/v1/label.proto
index 8e08d4ff..12c7a399 100644
--- a/labels-config-service-api/src/main/proto/org/hypertrace/label/config/service/v1/label.proto
+++ b/labels-config-service-api/src/main/proto/org/hypertrace/label/config/service/v1/label.proto
@@ -6,18 +6,12 @@ option java_multiple_files = true;
 
 message Label {
   string id = 1;
-  string key = 2;
-  optional string created_by = 3;
-  optional int64 created_time = 4;
-  optional string color = 5;
-  optional string created_by_application_rule_id = 6;
-  optional string description = 7;
+  LabelData data = 2;
+  optional string created_by_application_rule_id = 3;
 }
 
-message CreateLabel {
-  string key = 2;
-  optional string created_by = 3;
-  optional string color = 5;
-  optional string created_by_application_rule_id = 6;
-  optional string description = 7;
+message LabelData {
+  string key = 1;
+  optional string color = 2;
+  optional string description = 3;
 }
diff --git a/labels-config-service-api/src/main/proto/org/hypertrace/label/config/service/v1/labels_config_service.proto b/labels-config-service-api/src/main/proto/org/hypertrace/label/config/service/v1/labels_config_service.proto
index f4bfd669..be57bc20 100644
--- a/labels-config-service-api/src/main/proto/org/hypertrace/label/config/service/v1/labels_config_service.proto
+++ b/labels-config-service-api/src/main/proto/org/hypertrace/label/config/service/v1/labels_config_service.proto
@@ -26,7 +26,8 @@ service LabelsConfigService {
 }
 
 message CreateLabelRequest {
-  CreateLabel label = 1;
+  LabelData data = 1;
+  optional string created_by_application_rule_id = 2;
 }
 
 message CreateLabelResponse {
@@ -49,7 +50,8 @@ message GetLabelsResponse {
 }
 
 message UpdateLabelRequest {
-  Label label = 1;
+  string id = 1;
+  LabelData data = 2;
 }
 
 message UpdateLabelResponse {
diff --git a/labels-config-service-impl/src/main/java/org/hypertrace/label/config/service/LabelsConfigServiceImpl.java b/labels-config-service-impl/src/main/java/org/hypertrace/label/config/service/LabelsConfigServiceImpl.java
index cba79fb4..a4d1b835 100644
--- a/labels-config-service-impl/src/main/java/org/hypertrace/label/config/service/LabelsConfigServiceImpl.java
+++ b/labels-config-service-impl/src/main/java/org/hypertrace/label/config/service/LabelsConfigServiceImpl.java
@@ -26,7 +26,6 @@
 import org.hypertrace.config.service.v1.ConfigServiceGrpc;
 import org.hypertrace.core.grpcutils.client.RequestContextClientCallCredsProviderFactory;
 import org.hypertrace.core.grpcutils.context.RequestContext;
-import org.hypertrace.label.config.service.v1.CreateLabel;
 import org.hypertrace.label.config.service.v1.CreateLabelRequest;
 import org.hypertrace.label.config.service.v1.CreateLabelResponse;
 import org.hypertrace.label.config.service.v1.DeleteLabelRequest;
@@ -36,6 +35,7 @@
 import org.hypertrace.label.config.service.v1.GetLabelsRequest;
 import org.hypertrace.label.config.service.v1.GetLabelsResponse;
 import org.hypertrace.label.config.service.v1.Label;
+import org.hypertrace.label.config.service.v1.LabelData;
 import org.hypertrace.label.config.service.v1.LabelsConfigServiceGrpc;
 import org.hypertrace.label.config.service.v1.UpdateLabelRequest;
 import org.hypertrace.label.config.service.v1.UpdateLabelResponse;
@@ -76,7 +76,9 @@ public LabelsConfigServiceImpl(
               .collect(Collectors.toUnmodifiableMap(Label::getId, Function.identity()));
       systemLabelsKeyLabelMap =
           systemLabels.stream()
-              .collect(Collectors.toUnmodifiableMap(Label::getKey, Function.identity()));
+              .collect(
+                  Collectors.toUnmodifiableMap(
+                      (label) -> label.getData().getKey(), Function.identity()));
     } else {
       systemLabels = Collections.emptyList();
       systemLabelsIdLabelMap = Collections.emptyMap();
@@ -106,24 +108,22 @@ public void createLabel(
       Lock labelsLock = this.stripedLabelsLock.get(requestContext.getTenantId());
       if (labelsLock.tryLock(WAIT_TIME.getSeconds(), TimeUnit.SECONDS)) {
         try {
-          CreateLabel createLabel = request.getLabel();
-          if (systemLabelsKeyLabelMap.containsKey(createLabel.getKey())
-              || isDuplicateKey(createLabel.getKey())) {
+          LabelData labelData = request.getData();
+          if (systemLabelsKeyLabelMap.containsKey(labelData.getKey())
+              || isDuplicateKey(labelData.getKey())) {
             // Creating a label with a name that clashes with one of system labels name
             responseObserver.onError(new StatusRuntimeException(Status.ALREADY_EXISTS));
             return;
           }
           Label label =
-              Label.newBuilder()
-                  .setId(UUID.randomUUID().toString())
-                  .setKey(createLabel.getKey())
-                  .setCreatedBy(createLabel.getCreatedBy())
-                  .setColor(createLabel.getColor())
-                  .setCreatedByApplicationRuleId(createLabel.getCreatedByApplicationRuleId())
-                  .setDescription(createLabel.getDescription())
-                  .build();
-          Label createdLabel =
-              getLabelFromContextualObject(labelStore.upsertObject(requestContext, label));
+              Label.newBuilder().setId(UUID.randomUUID().toString()).setData(labelData).build();
+          if (request.hasCreatedByApplicationRuleId()) {
+            label =
+                label.toBuilder()
+                    .setCreatedByApplicationRuleId(request.getCreatedByApplicationRuleId())
+                    .build();
+          }
+          Label createdLabel = labelStore.upsertObject(requestContext, label).getData();
           responseObserver.onNext(CreateLabelResponse.newBuilder().setLabel(createdLabel).build());
           responseObserver.onCompleted();
         } finally {
@@ -162,11 +162,10 @@ public void getLabel(GetLabelRequest request, StreamObserver<GetLabelResponse> r
   public void getLabels(
       GetLabelsRequest request, StreamObserver<GetLabelsResponse> responseObserver) {
     RequestContext requestContext = RequestContext.CURRENT.get();
-    List<Label> allLabels = new ArrayList<>();
-    allLabels.addAll(systemLabels);
+    List<Label> allLabels = new ArrayList<>(systemLabels);
     List<Label> tenantLabels =
         labelStore.getAllObjects(requestContext).stream()
-            .map(this::getLabelFromContextualObject)
+            .map(ContextualConfigObject::getData)
             .collect(Collectors.toUnmodifiableList());
     allLabels.addAll(tenantLabels);
     responseObserver.onNext(GetLabelsResponse.newBuilder().addAllLabels(allLabels).build());
@@ -177,28 +176,28 @@ public void getLabels(
   public void updateLabel(
       UpdateLabelRequest request, StreamObserver<UpdateLabelResponse> responseObserver) {
     try {
-      Label updatedLabelInReq = request.getLabel();
+      LabelData newLabelData = request.getData();
       RequestContext requestContext = RequestContext.CURRENT.get();
       Lock labelsLock = this.stripedLabelsLock.get(requestContext.getTenantId());
       if (labelsLock.tryLock(WAIT_TIME.getSeconds(), TimeUnit.SECONDS)) {
         try {
-          if (systemLabelsIdLabelMap.containsKey(updatedLabelInReq.getId())
-              || systemLabelsKeyLabelMap.containsKey(updatedLabelInReq.getKey())) {
+          if (systemLabelsIdLabelMap.containsKey(request.getId())
+              || systemLabelsKeyLabelMap.containsKey(newLabelData.getKey())) {
             // Updating a system label will error
             responseObserver.onError(new StatusRuntimeException(Status.INVALID_ARGUMENT));
             return;
           }
-          if (isDuplicateKey(updatedLabelInReq.getKey())) {
+          if (isDuplicateKey(newLabelData.getKey())) {
             responseObserver.onError(new StatusRuntimeException(Status.ALREADY_EXISTS));
             return;
           }
-          labelStore
-              .getData(requestContext, updatedLabelInReq.getId())
-              .orElseThrow(Status.NOT_FOUND::asRuntimeException);
-          Label updatedLabelInRes =
-              labelStore.upsertObject(requestContext, updatedLabelInReq).getData();
-          responseObserver.onNext(
-              UpdateLabelResponse.newBuilder().setLabel(updatedLabelInRes).build());
+          Label oldLabel =
+              labelStore
+                  .getData(requestContext, request.getId())
+                  .orElseThrow(Status.NOT_FOUND::asRuntimeException);
+          Label newLabel = oldLabel.toBuilder().setData(newLabelData).build();
+          Label updatedLabel = labelStore.upsertObject(requestContext, newLabel).getData();
+          responseObserver.onNext(UpdateLabelResponse.newBuilder().setLabel(updatedLabel).build());
           responseObserver.onCompleted();
         } finally {
           labelsLock.unlock();
@@ -248,14 +247,12 @@ private boolean isDuplicateKey(String key) {
         labelStore.getAllObjects(requestContext).stream()
             .map(ContextualConfigObject::getData)
             .collect(Collectors.toUnmodifiableList());
-    Optional<Label> match =
-        labelList.stream().filter(label -> label.getKey().equals(key)).findAny();
+    Optional<String> match =
+        labelList.stream()
+            .map(Label::getData)
+            .map(LabelData::getKey)
+            .filter(labelKey -> labelKey.equals(key))
+            .findAny();
     return match.isPresent();
   }
-
-  private Label getLabelFromContextualObject(ContextualConfigObject<Label> labelObject) {
-    long createdTimestamp = labelObject.getCreationTimestamp().toEpochMilli();
-    Label label = labelObject.getData();
-    return label.toBuilder().setCreatedTime(createdTimestamp).build();
-  }
 }
diff --git a/labels-config-service-impl/src/test/java/org/hypertrace/label/config/service/LabelsConfigServiceImplTest.java b/labels-config-service-impl/src/test/java/org/hypertrace/label/config/service/LabelsConfigServiceImplTest.java
index 0d010543..f38fe530 100644
--- a/labels-config-service-impl/src/test/java/org/hypertrace/label/config/service/LabelsConfigServiceImplTest.java
+++ b/labels-config-service-impl/src/test/java/org/hypertrace/label/config/service/LabelsConfigServiceImplTest.java
@@ -15,9 +15,9 @@
 import java.util.Map;
 import java.util.Set;
 import java.util.stream.Collectors;
+import java.util.stream.Stream;
 import org.hypertrace.config.service.change.event.api.ConfigChangeEventGenerator;
 import org.hypertrace.config.service.test.MockGenericConfigService;
-import org.hypertrace.label.config.service.v1.CreateLabel;
 import org.hypertrace.label.config.service.v1.CreateLabelRequest;
 import org.hypertrace.label.config.service.v1.CreateLabelResponse;
 import org.hypertrace.label.config.service.v1.DeleteLabelRequest;
@@ -25,6 +25,7 @@
 import org.hypertrace.label.config.service.v1.GetLabelResponse;
 import org.hypertrace.label.config.service.v1.GetLabelsRequest;
 import org.hypertrace.label.config.service.v1.Label;
+import org.hypertrace.label.config.service.v1.LabelData;
 import org.hypertrace.label.config.service.v1.LabelsConfigServiceGrpc;
 import org.hypertrace.label.config.service.v1.LabelsConfigServiceGrpc.LabelsConfigServiceBlockingStub;
 import org.hypertrace.label.config.service.v1.UpdateLabelRequest;
@@ -39,11 +40,16 @@ public final class LabelsConfigServiceImplTest {
   MockGenericConfigService mockGenericConfigService;
   List<Label> systemLabels =
       Arrays.asList("Critical", "Sensitive", "External", "Sentry").stream()
-          .map(key -> Label.newBuilder().setId(key).setKey(key).build())
+          .map(
+              key ->
+                  Label.newBuilder()
+                      .setId(key)
+                      .setData(LabelData.newBuilder().setKey(key).build())
+                      .build())
           .collect(Collectors.toList());
-  List<CreateLabel> createLabelsList =
-      Arrays.asList(0, 1, 2, 3, 4).stream()
-          .map(id -> CreateLabel.newBuilder().setKey("Label-" + id).build())
+  List<LabelData> createLabelDataList =
+      Stream.of(0, 1, 2, 3, 4)
+          .map(id -> LabelData.newBuilder().setKey("Label-" + id).build())
           .collect(Collectors.toList());
 
   @BeforeEach
@@ -54,7 +60,9 @@ void setUp() {
     systemLabelsConfigMap.put(
         "system.labels",
         systemLabels.stream()
-            .map(systemLabel -> Map.of("id", systemLabel.getId(), "key", systemLabel.getKey()))
+            .map(
+                systemLabel ->
+                    Map.of("id", systemLabel.getId(), "data.key", systemLabel.getData().getKey()))
             .collect(Collectors.toList()));
     Map<String, Object> configMap = new HashMap<>();
     configMap.put("labels.config.service", systemLabelsConfigMap);
@@ -74,12 +82,12 @@ void afterEach() {
 
   private List<Label> createLabels() {
     // Creating or inserting labels
-    return createLabelsList.stream()
+    return createLabelDataList.stream()
         .map(
             createLabel -> {
               CreateLabelResponse response =
                   labelConfigStub.createLabel(
-                      CreateLabelRequest.newBuilder().setLabel(createLabel).build());
+                      CreateLabelRequest.newBuilder().setData(createLabel).build());
               return response.getLabel();
             })
         .collect(Collectors.toList());
@@ -87,17 +95,15 @@ private List<Label> createLabels() {
 
   @Test
   void test_createLabel() {
-    List<CreateLabel> createdLabelsList =
-        createLabels().stream()
-            .map(label -> CreateLabel.newBuilder().setKey(label.getKey()).build())
-            .collect(Collectors.toList());
-    assertEquals(createLabelsList, createdLabelsList);
+    List<LabelData> createdLabelsList =
+        createLabels().stream().map(Label::getData).collect(Collectors.toList());
+    assertEquals(createdLabelsList, createdLabelsList);
     Throwable exception =
         assertThrows(
             StatusRuntimeException.class,
             () -> {
               labelConfigStub.createLabel(
-                  CreateLabelRequest.newBuilder().setLabel(createLabelsList.get(0)).build());
+                  CreateLabelRequest.newBuilder().setData(createLabelDataList.get(0)).build());
             });
     assertEquals(Status.ALREADY_EXISTS, Status.fromThrowable(exception));
   }
@@ -107,7 +113,7 @@ void test_system_createLabel() {
     for (Label systemLabel : systemLabels) {
       CreateLabelRequest createLabelRequest =
           CreateLabelRequest.newBuilder()
-              .setLabel(CreateLabel.newBuilder().setKey(systemLabel.getKey()).build())
+              .setData(LabelData.newBuilder().setKey(systemLabel.getData().getKey()).build())
               .build();
       Throwable exception =
           assertThrows(
@@ -123,7 +129,7 @@ void test_system_createLabel() {
   void test_getLabel() {
     List<Label> createdLabelsList =
         createLabels().stream()
-            .map(label -> label.toBuilder().clearCreatedTime().build())
+            .map(label -> label.toBuilder().build())
             .collect(Collectors.toList());
     // Querying each label one by one for the created or inserted labels in the previous step
     List<Label> getLabelList =
@@ -177,10 +183,10 @@ void test_updateLabel() {
             () -> {
               labelConfigStub.updateLabel(
                   UpdateLabelRequest.newBuilder()
-                      .setLabel(
-                          Label.newBuilder()
-                              .setId(createdLabelsList.get(1).getId())
-                              .setKey(createdLabelsList.get(0).getKey())
+                      .setId(createdLabelsList.get(1).getId())
+                      .setData(
+                          LabelData.newBuilder()
+                              .setKey(createdLabelsList.get(0).getData().getKey())
                               .build())
                       .build());
             });
@@ -190,7 +196,11 @@ void test_updateLabel() {
         createdLabelsList.stream()
             .map(
                 label ->
-                    Label.newBuilder().setId(label.getId()).setKey(label.getKey() + "new").build())
+                    Label.newBuilder()
+                        .setId(label.getId())
+                        .setData(
+                            LabelData.newBuilder().setKey(label.getData().getKey() + "new").build())
+                        .build())
             .collect(Collectors.toList());
     List<Label> updatedLabelsList =
         updateLabelsList.stream()
@@ -198,7 +208,10 @@ void test_updateLabel() {
                 updateLabel -> {
                   UpdateLabelResponse response =
                       labelConfigStub.updateLabel(
-                          UpdateLabelRequest.newBuilder().setLabel(updateLabel).build());
+                          UpdateLabelRequest.newBuilder()
+                              .setId(updateLabel.getId())
+                              .setData(updateLabel.getData())
+                              .build());
                   return response.getLabel();
                 })
             .collect(Collectors.toList());
@@ -210,7 +223,8 @@ void test_updateLabel() {
             () -> {
               labelConfigStub.updateLabel(
                   UpdateLabelRequest.newBuilder()
-                      .setLabel(Label.newBuilder().setId("1").setKey("API-X").build())
+                      .setId("1")
+                      .setData(LabelData.newBuilder().setKey("API-X").build())
                       .build());
             });
     assertEquals(Status.NOT_FOUND, Status.fromThrowable(exception));
@@ -222,7 +236,8 @@ void test_system_updateLabel() {
     for (Label systemLabel : systemLabels) {
       UpdateLabelRequest updateLabelRequest =
           UpdateLabelRequest.newBuilder()
-              .setLabel(Label.newBuilder().setId(systemLabel.getId()).setKey("1").build())
+              .setId(systemLabel.getId())
+              .setData(LabelData.newBuilder().setKey("1").build())
               .build();
       Throwable exception =
           assertThrows(
@@ -235,11 +250,8 @@ void test_system_updateLabel() {
     for (Label systemLabel : systemLabels) {
       UpdateLabelRequest updateLabelRequest =
           UpdateLabelRequest.newBuilder()
-              .setLabel(
-                  Label.newBuilder()
-                      .setId(createdLabelsList.get(0).getId())
-                      .setKey(systemLabel.getKey())
-                      .build())
+              .setId(createdLabelsList.get(0).getId())
+              .setData(LabelData.newBuilder().setKey(systemLabel.getData().getKey()).build())
               .build();
       Throwable exception =
           assertThrows(

From 57fcdc3d49fe6884206d4d368268dc5af081be74 Mon Sep 17 00:00:00 2001
From: kamaleshnneerasa <kamaleshnneerasa@github.com>
Date: Wed, 3 Nov 2021 19:38:16 +0530
Subject: [PATCH 6/8] Added impl for new schema

---
 .../service/LabelsConfigServiceImpl.java      | 25 ++++++++-----------
 1 file changed, 11 insertions(+), 14 deletions(-)

diff --git a/labels-config-service-impl/src/main/java/org/hypertrace/label/config/service/LabelsConfigServiceImpl.java b/labels-config-service-impl/src/main/java/org/hypertrace/label/config/service/LabelsConfigServiceImpl.java
index a4d1b835..a89e689f 100644
--- a/labels-config-service-impl/src/main/java/org/hypertrace/label/config/service/LabelsConfigServiceImpl.java
+++ b/labels-config-service-impl/src/main/java/org/hypertrace/label/config/service/LabelsConfigServiceImpl.java
@@ -13,7 +13,6 @@
 import java.util.Collections;
 import java.util.List;
 import java.util.Map;
-import java.util.Optional;
 import java.util.UUID;
 import java.util.concurrent.TimeUnit;
 import java.util.concurrent.locks.Lock;
@@ -176,18 +175,18 @@ public void getLabels(
   public void updateLabel(
       UpdateLabelRequest request, StreamObserver<UpdateLabelResponse> responseObserver) {
     try {
-      LabelData newLabelData = request.getData();
+      LabelData updateLabelData = request.getData();
       RequestContext requestContext = RequestContext.CURRENT.get();
       Lock labelsLock = this.stripedLabelsLock.get(requestContext.getTenantId());
       if (labelsLock.tryLock(WAIT_TIME.getSeconds(), TimeUnit.SECONDS)) {
         try {
           if (systemLabelsIdLabelMap.containsKey(request.getId())
-              || systemLabelsKeyLabelMap.containsKey(newLabelData.getKey())) {
+              || systemLabelsKeyLabelMap.containsKey(updateLabelData.getKey())) {
             // Updating a system label will error
             responseObserver.onError(new StatusRuntimeException(Status.INVALID_ARGUMENT));
             return;
           }
-          if (isDuplicateKey(newLabelData.getKey())) {
+          if (isDuplicateKey(updateLabelData.getKey())) {
             responseObserver.onError(new StatusRuntimeException(Status.ALREADY_EXISTS));
             return;
           }
@@ -195,9 +194,10 @@ public void updateLabel(
               labelStore
                   .getData(requestContext, request.getId())
                   .orElseThrow(Status.NOT_FOUND::asRuntimeException);
-          Label newLabel = oldLabel.toBuilder().setData(newLabelData).build();
-          Label updatedLabel = labelStore.upsertObject(requestContext, newLabel).getData();
-          responseObserver.onNext(UpdateLabelResponse.newBuilder().setLabel(updatedLabel).build());
+          Label updateLabel = oldLabel.toBuilder().setData(updateLabelData).build();
+          Label updateLabelInRes = labelStore.upsertObject(requestContext, updateLabel).getData();
+          responseObserver.onNext(
+              UpdateLabelResponse.newBuilder().setLabel(updateLabelInRes).build());
           responseObserver.onCompleted();
         } finally {
           labelsLock.unlock();
@@ -247,12 +247,9 @@ private boolean isDuplicateKey(String key) {
         labelStore.getAllObjects(requestContext).stream()
             .map(ContextualConfigObject::getData)
             .collect(Collectors.toUnmodifiableList());
-    Optional<String> match =
-        labelList.stream()
-            .map(Label::getData)
-            .map(LabelData::getKey)
-            .filter(labelKey -> labelKey.equals(key))
-            .findAny();
-    return match.isPresent();
+    return labelList.stream()
+        .map(Label::getData)
+        .map(LabelData::getKey)
+        .anyMatch(labelKey -> labelKey.equals(key));
   }
 }

From 1716a9ae31ad2b2487b69d065139529cfc4abcc4 Mon Sep 17 00:00:00 2001
From: saxenakshitiz <kshitiz@traceable.ai>
Date: Thu, 11 Nov 2021 20:13:38 +0530
Subject: [PATCH 7/8] Updated version of hypertrace grpc utils and grpc

---
 gradle/libs.versions.toml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/gradle/libs.versions.toml b/gradle/libs.versions.toml
index 7ac52f8a..2a08e268 100644
--- a/gradle/libs.versions.toml
+++ b/gradle/libs.versions.toml
@@ -1,7 +1,7 @@
 [versions]
 protoc = "3.17.3"
-grpc = "1.39.0"
-hypertrace-grpcutils = "0.5.2"
+grpc = "1.42.0"
+hypertrace-grpcutils = "0.6.2"
 hypertrace-framework = "0.1.29"
 lombok = "1.18.20"
 jackson = "2.12.4"

From ca77032345d5b50f42889945301353269b367a1d Mon Sep 17 00:00:00 2001
From: saxenakshitiz <kshitiz@traceable.ai>
Date: Thu, 11 Nov 2021 20:34:00 +0530
Subject: [PATCH 8/8] Fix synk warnings

---
 config-proto-converter/build.gradle.kts | 5 +++++
 gradle/libs.versions.toml               | 2 ++
 2 files changed, 7 insertions(+)

diff --git a/config-proto-converter/build.gradle.kts b/config-proto-converter/build.gradle.kts
index 39deca98..c33a7433 100644
--- a/config-proto-converter/build.gradle.kts
+++ b/config-proto-converter/build.gradle.kts
@@ -6,4 +6,9 @@ plugins {
 dependencies {
   api(libs.grpc.protobuf)
   implementation(libs.protobuf.javautil)
+  constraints {
+    implementation(libs.gson) {
+      because("https://snyk.io/vuln/SNYK-JAVA-COMGOOGLECODEGSON-1730327")
+    }
+  }
 }
diff --git a/gradle/libs.versions.toml b/gradle/libs.versions.toml
index 2a08e268..547617c1 100644
--- a/gradle/libs.versions.toml
+++ b/gradle/libs.versions.toml
@@ -1,6 +1,7 @@
 [versions]
 protoc = "3.17.3"
 grpc = "1.42.0"
+gson = "2.8.9"
 hypertrace-grpcutils = "0.6.2"
 hypertrace-framework = "0.1.29"
 lombok = "1.18.20"
@@ -40,6 +41,7 @@ grpc-core = { module = "io.grpc:grpc-core", version.ref = "grpc" }
 grpc-netty = { module = "io.grpc:grpc-netty", version.ref = "grpc" }
 protobuf-javautil = { module = "com.google.protobuf:protobuf-java-util", version.ref = "protoc" }
 protobuf-java = { module = "com.google.protobuf:protobuf-java", version.ref = "protoc" }
+gson = { module = "com.google.code.gson:gson", version.ref = "gson" }
 
 kafka-protobuf-serializer = { module = "io.confluent:kafka-protobuf-serializer", version.ref = "confluent" }