Skip to content

Commit 5337577

Browse files
author
jruaux
committed
feat: Added convenience methods
1 parent 9409b44 commit 5337577

File tree

2 files changed

+15
-1
lines changed

2 files changed

+15
-1
lines changed

core/redis-enterprise-admin/src/main/java/com/redis/enterprise/Admin.java

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -242,10 +242,15 @@ public Database createDatabase(Database database) throws IOException, GeneralSec
242242
.until(() -> executeCommand(uid, new Command("PING")).getResponse().asBoolean());
243243
return response;
244244
}
245-
245+
246246
public List<Database> getDatabases() throws IOException, GeneralSecurityException {
247247
return get(v1(BDBS), objectMapper.getTypeFactory().constructCollectionType(List.class, Database.class));
248248
}
249+
250+
public void deleteAllDatabases() throws IOException, GeneralSecurityException {
251+
getDatabases().stream().map(Database::getUid).forEach(this::deleteDatabase);
252+
Awaitility.await().until(() -> getDatabases().isEmpty());
253+
}
249254

250255
public void deleteDatabase(long uid) {
251256
Awaitility.await().pollInterval(Duration.ofSeconds(1)).until(() -> {

core/redis-enterprise-admin/src/main/java/com/redis/enterprise/Database.java

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
import java.util.List;
66
import java.util.Objects;
77
import java.util.stream.Collectors;
8+
import java.util.stream.Stream;
89

910
import org.apache.hc.core5.util.Asserts;
1011
import org.springframework.util.unit.DataSize;
@@ -155,6 +156,10 @@ public List<ShardKeyRegex> getShardKeyRegex() {
155156
return shardKeyRegex;
156157
}
157158

159+
public void setShardKeyRegex(String... regexes) {
160+
setShardKeyRegex(Stream.of(regexes).map(ShardKeyRegex::new).collect(Collectors.toList()));
161+
}
162+
158163
public void setShardKeyRegex(List<ShardKeyRegex> shardKeyRegex) {
159164
this.shardKeyRegex = shardKeyRegex;
160165
}
@@ -181,6 +186,10 @@ public void setShardPlacement(ShardPlacement shardPlacement) {
181186
public List<ModuleConfig> getModules() {
182187
return modules;
183188
}
189+
190+
public void setModules(RedisModule... modules) {
191+
setModules(Stream.of(modules).map(RedisModule::getModuleName).map(ModuleConfig::new).collect(Collectors.toList()));
192+
}
184193

185194
public void setModules(List<ModuleConfig> modules) {
186195
this.modules = modules;

0 commit comments

Comments
 (0)