{
Integer partition = record.partition();
String key = record.key() == null ? "NULL" : record.key().toString();
-
- if (!state.containsKey(partition))
- state.put(partition, new HashMap<>());
-
Map<String, Long> byKey = state.get(partition);
if (!byKey.containsKey(key))
byKey.put(key, seenByKey);
}
+ public void addPartition(Integer partition, Map<String, Long> statistics)
+ {
+ state.put(partition, statistics);
+ }
+
+ public Map<String, Long> removePartition(Integer partition)
+ {
+ return state.remove(partition);
+ }
+
public Map<Integer, Map<String, Long>> getState()
{