added partition to simple generator map

redis · Jun 12, 2019 · aeae6cd · aeae6cd
1 parent c2a08c3
commit aeae6cd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/pom.xml b/pom.xml
@@ -12,7 +12,7 @@
 
 	<groupId>com.redislabs</groupId>
 	<artifactId>riot</artifactId>
-	<version>0.4.5</version>
+	<version>0.4.6</version>
 	<packaging>jar</packaging>
 	<name>RIOT</name>
 	<description>Redis Input/Output Tool</description>

diff --git a/src/main/java/com/redislabs/riot/generator/SimpleGeneratorReader.java b/src/main/java/com/redislabs/riot/generator/SimpleGeneratorReader.java
@@ -57,6 +57,7 @@ private long start(long total) {
 	@Override
 	protected Map<String, Object> doRead() throws Exception {
 		Map<String, Object> map = new HashMap<>();
+		map.put(IndexedPartitioner.PARTITION_KEY, partitionIndex.get());
 		for (int index = 0; index < fieldCount; index++) {
 			int fieldIndex = index + 1;
 			String fieldName = "field" + fieldIndex;