feat: Added TimeSeries generator options

redis · Aug 1, 2022 · 0c6bdfd · 0c6bdfd
1 parent 7d6c538
commit 0c6bdfd
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/connectors/riot-gen/src/main/java/com/redis/riot/gen/DataStructureGeneratorCommand.java b/connectors/riot-gen/src/main/java/com/redis/riot/gen/DataStructureGeneratorCommand.java
@@ -55,6 +55,7 @@ private ItemReader<DataStructure<String>> reader() {
 				.zsetScore(options.getZsetScore()).hashSize(options.getHashSize())
 				.hashFieldSize(options.getHashFieldSize()).jsonFieldCount(options.getJsonSize())
 				.jsonFieldSize(options.getJsonFieldSize());
+		options.getTimeseriesStartTime().ifPresent(t -> reader.timeseriesStartTime(t.toEpochMilli()));
 		options.getExpiration().ifPresent(reader::expiration);
 		Optional<Long> sleep = options.getSleep();
 		if (sleep.isPresent() && sleep.get() > 0) {

diff --git a/connectors/riot-gen/src/main/java/com/redis/riot/gen/DataStructureGeneratorOptions.java b/connectors/riot-gen/src/main/java/com/redis/riot/gen/DataStructureGeneratorOptions.java
@@ -2,6 +2,7 @@
 
 import static com.redis.spring.batch.reader.DataStructureGeneratorItemReader.*;
 
+import java.time.Instant;
 import java.util.Arrays;
 import java.util.List;
 import java.util.Optional;
@@ -43,6 +44,8 @@ public class DataStructureGeneratorOptions extends GeneratorOptions {
 	private IntRange stringSize = DEFAULT_STRING_SIZE;
 	@Option(names = "--ts-size", description = "Number of samples in timeseries (default: ${DEFAULT-VALUE}).", paramLabel = "<range>")
 	private IntRange timeseriesSize = DEFAULT_TIMESERIES_SIZE;
+	@Option(names = "--ts-time", description = "Start time for samples in timeseries, e.g. 2007-12-03T10:15:30.00Z (default: now).", paramLabel = "<epoch>")
+	private Optional<Instant> timeseriesStartTime = Optional.empty();
 	@Option(names = "--zset-size", description = "Number of elements in sorted sets (default: ${DEFAULT-VALUE}).", paramLabel = "<range>")
 	private IntRange zsetSize = DEFAULT_ZSET_SIZE;
 	@Option(names = "--zset-score", description = "Score of sorted sets (default: ${DEFAULT-VALUE}).", paramLabel = "<range>")
@@ -172,6 +175,14 @@ public void setTimeseriesSize(IntRange timeseriesSize) {
 		this.timeseriesSize = timeseriesSize;
 	}
 
+	public Optional<Instant> getTimeseriesStartTime() {
+		return timeseriesStartTime;
+	}
+
+	public void setTimeseriesStartTime(Optional<Instant> timeseriesStartTime) {
+		this.timeseriesStartTime = timeseriesStartTime;
+	}
+
 	public void setTypes(List<Type> types) {
 		this.types = types;
 	}