diff --git a/src/main/java/io/ebean/test/containers/YugabyteContainer.java b/src/main/java/io/ebean/test/containers/YugabyteContainer.java index df4eaca..316e607 100644 --- a/src/main/java/io/ebean/test/containers/YugabyteContainer.java +++ b/src/main/java/io/ebean/test/containers/YugabyteContainer.java @@ -36,6 +36,7 @@ public static class Builder extends BaseDbBuilder { int port9042 = 9042; int port9000 = 9000; int port7000 = 7000; + String tserverFlags = "yb_enable_read_committed_isolation=true"; private Builder(String version) { super("yugabyte", 6433, 5433, version); @@ -58,6 +59,14 @@ public YugabyteContainer.Builder port9042(int port9042) { return self(); } + /** + * Set the tserverFlags to be used. Defaults to yb_enable_read_committed_isolation=true. + */ + public YugabyteContainer.Builder tserverFlags(String tserverFlags) { + this.tserverFlags = tserverFlags; + return self(); + } + @Override protected String buildJdbcUrl() { return "jdbc:postgresql://" + getHost() + ":" + getPort() + "/" + getDbName(); @@ -101,18 +110,13 @@ protected ProcessBuilder runProcess() { args.add("-p"); args.add(builder.port9042 +":9042"); -// if (dbConfig.isInMemory() && dbConfig.getTmpfs() != null) { -// args.add("--tmpfs"); -// args.add(dbConfig.getTmpfs()); -// } -// if (!dbConfig.adminPassword.isEmpty()) { -// args.add("-e"); -// args.add("POSTGRES_PASSWORD=" + dbConfig.getAdminPassword()); -// } args.add(config.getImage()); args.add("bin/yugabyted"); args.add("start"); // args.add("--base_dir=/home/yugabyte/yb_data"); + if (builder.tserverFlags != null) { + args.add("--tserver_flags=" + builder.tserverFlags); + } args.add("--daemon=false"); return createProcessBuilder(args); }