Skip to content

Commit 55e495b

Browse files
refactor: Remove message container (#120)
1 parent 176dec1 commit 55e495b

File tree

3 files changed

+7
-36
lines changed

3 files changed

+7
-36
lines changed

src/main/java/com/spotify/confidence/EventUploader.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ static Event.Builder event(
1414
.setPayload(
1515
Struct.newBuilder()
1616
.putAllFields(context.asProtoMap())
17-
.putFields("message", message.orElse(ConfidenceValue.Struct.EMPTY).toProto()));
17+
.putAllFields(message.orElse(ConfidenceValue.Struct.EMPTY).asProtoMap()));
1818
}
1919

2020
CompletableFuture<Boolean> upload(List<Event> events);

src/test/java/com/spotify/confidence/EventSenderEngineTest.java

Lines changed: 4 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -110,17 +110,10 @@ public void testOverlappingKeysInPayload() throws InterruptedException {
110110
Struct.newBuilder()
111111
.putAllFields(
112112
Map.of(
113+
"a",
114+
Value.newBuilder().setNumberValue(0).build(),
113115
"message",
114-
Value.newBuilder()
115-
.setStructValue(
116-
Struct.newBuilder()
117-
.putFields(
118-
"a", Value.newBuilder().setNumberValue(0).build())
119-
.putFields(
120-
"message",
121-
Value.newBuilder().setNumberValue(1).build()))
122-
.build(),
123-
"a", Value.newBuilder().setNumberValue(2).build()))
116+
Value.newBuilder().setNumberValue(1).build()))
124117
.build());
125118
}
126119

@@ -199,15 +192,7 @@ public void testEngineUploadsWhenIntermittentErrorWillRetry() throws IOException
199192
final Set<Value> uniqueEventIds =
200193
fakeUploader.uploadCalls.stream()
201194
.flatMap(Collection::stream)
202-
.map(
203-
event ->
204-
event
205-
.getPayload()
206-
.getFieldsMap()
207-
.get("message")
208-
.getStructValue()
209-
.getFieldsMap()
210-
.get("id"))
195+
.map(event -> event.getPayload().getFieldsMap().get("id"))
211196
.collect(Collectors.toSet());
212197
// Verify all events reached the uploader
213198
assertThat(uniqueEventIds.size()).isEqualTo(numEvents);

src/test/java/com/spotify/confidence/GrpcEventUploaderTest.java

Lines changed: 2 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -97,14 +97,7 @@ public void testMapsSingleEventBatchToProtobuf() throws ExecutionException, Inte
9797
assertThat(protoEvent.getEventDefinition()).isEqualTo("eventDefinitions/event1");
9898

9999
final Map<String, com.google.protobuf.Value> fieldsMap = protoEvent.getPayload().getFieldsMap();
100-
assertThat(
101-
fieldsMap
102-
.get("message")
103-
.getStructValue()
104-
.getFieldsMap()
105-
.get("messageKey")
106-
.getStringValue())
107-
.isEqualTo("value_1");
100+
assertThat(fieldsMap.get("messageKey").getStringValue()).isEqualTo("value_1");
108101
assertThat(fieldsMap.get("contextKey").getStringValue()).isEqualTo("value_1");
109102
}
110103

@@ -136,14 +129,7 @@ public void testMapsMultiEventBatchToProtobuf() {
136129

137130
final Map<String, com.google.protobuf.Value> fieldsMap =
138131
protoEvent.getPayload().getFieldsMap();
139-
assertThat(
140-
fieldsMap
141-
.get("message")
142-
.getStructValue()
143-
.getFieldsMap()
144-
.get("messageKey")
145-
.getStringValue())
146-
.isEqualTo("value_m" + (i + 1));
132+
assertThat(fieldsMap.get("messageKey").getStringValue()).isEqualTo("value_m" + (i + 1));
147133
assertThat(fieldsMap.get("contextKey").getStringValue()).isEqualTo("value_c" + (i + 1));
148134
}
149135
}

0 commit comments

Comments
 (0)