feat: Added flush option to demo

redis-field-engineering · Jul 21, 2022 · b61c4e8 · b61c4e8
1 parent d6bb21b
commit b61c4e8
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/demos/redis-sidecar-spring-demo/src/main/java/com/redis/sidecar/springdemo/Config.java b/demos/redis-sidecar-spring-demo/src/main/java/com/redis/sidecar/springdemo/Config.java
@@ -9,6 +9,15 @@ public class Config {
 
 	private Loader loader = new Loader();
 	private Query query = new Query();
+	private boolean flush;
+
+	public boolean isFlush() {
+		return flush;
+	}
+
+	public void setFlush(boolean flush) {
+		this.flush = flush;
+	}
 
 	public Query getQuery() {
 		return query;

diff --git a/...sidecar-spring-demo/src/main/java/com/redis/sidecar/springdemo/PostgresQueryExecutor.java b/...sidecar-spring-demo/src/main/java/com/redis/sidecar/springdemo/PostgresQueryExecutor.java
@@ -21,6 +21,7 @@
 import com.zaxxer.hikari.HikariConfig;
 import com.zaxxer.hikari.HikariDataSource;
 
+import io.lettuce.core.RedisClient;
 import io.lettuce.core.RedisURI;
 
 @Component
@@ -50,6 +51,9 @@ public PostgresQueryExecutor(RedisURI redisURI, DataSourceProperties dataSourceP
 
 	@PostConstruct
 	public void executeQueries() {
+		if (config.isFlush()) {
+			RedisClient.create(redisURI).connect().sync().flushall();
+		}
 		HikariConfig hikariConfig = new HikariConfig();
 		hikariConfig.setJdbcUrl("jdbc:" + redisURI.toString());
 		hikariConfig.setDriverClassName(SidecarDriver.class.getName());