X-Git-Url: https://juplo.de/gitweb/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fde%2Fjuplo%2Fkafka%2FStateDocument.java;fp=src%2Fmain%2Fjava%2Fde%2Fjuplo%2Fkafka%2FStateDocument.java;h=b37b8a9a68124aac5274e426ac6b11ae59c968a9;hb=706220a34eeecd09bd1d5dfa967be9e110b44a99;hp=0000000000000000000000000000000000000000;hpb=bf99986fe3daf527d93d444ce1ae7374a8faddcc;p=demos%2Fkafka%2Ftraining

diff --git a/src/main/java/de/juplo/kafka/StateDocument.java b/src/main/java/de/juplo/kafka/StateDocument.java
new file mode 100644
index 0000000..b37b8a9
--- /dev/null
+++ b/src/main/java/de/juplo/kafka/StateDocument.java
@@ -0,0 +1,34 @@
+package de.juplo.kafka;
+
+import lombok.ToString;
+import org.springframework.data.annotation.Id;
+import org.springframework.data.mongodb.core.mapping.Document;
+
+import java.util.HashMap;
+import java.util.Map;
+
+
+@Document(collection = "state")
+@ToString
+public class StateDocument
+{
+  @Id
+  public String id;
+  public Map<String, Long> state;
+
+  public StateDocument()
+  {
+  }
+
+  public StateDocument(Integer partition)
+  {
+    this.id = Integer.toString(partition);
+    this.state = new HashMap<>();
+  }
+
+  public StateDocument(Integer partition, Map<String, Long> state)
+  {
+    this.id = Integer.toString(partition);
+    this.state = state;
+  }
+}