Add support for Create DB with Locale

yaojiejia · yaojiejia · commit afbdfe1b1846 · 2025-05-31T21:54:39.000-04:00
diff --git a/src/sqlancer/postgres/PostgresProvider.java b/src/sqlancer/postgres/PostgresProvider.java
@@ -340,9 +340,13 @@ private String getCreateDatabaseCommand(PostgresGlobalState state) {
                     sb.append(Randomly.fromOptions("utf8"));
                     sb.append("' ");
                 }
-                for (String lc : Arrays.asList("LC_COLLATE", "LC_CTYPE")) {
-                    if (!state.getCollates().isEmpty() && Randomly.getBoolean()) {
-                        sb.append(String.format(" %s = '%s'", lc, Randomly.fromList(state.getCollates())));
+                if (Randomly.getBoolean() && !state.getCollates().isEmpty()) {
+                    sb.append(String.format(" LOCALE = '%s' ", Randomly.fromList(state.getCollates())));
+                } else {
+                    for (String lc : Arrays.asList("LC_COLLATE", "LC_CTYPE")) {
+                        if (!state.getCollates().isEmpty() && Randomly.getBoolean()) {
+                            sb.append(String.format(" %s = '%s'", lc, Randomly.fromList(state.getCollates())));
+                        }
                     }
                 }
                 sb.append(" TEMPLATE template0");

Original file line number	Diff line number	Diff line change
`@@ -340,9 +340,13 @@ private String getCreateDatabaseCommand(PostgresGlobalState state) {`
`340`	`340`	`sb.append(Randomly.fromOptions("utf8"));`
`341`	`341`	`sb.append("' ");`
`342`	`342`	`}`
`343`		`- for (String lc : Arrays.asList("LC_COLLATE", "LC_CTYPE")) {`
`344`		`- if (!state.getCollates().isEmpty() && Randomly.getBoolean()) {`
`345`		`- sb.append(String.format(" %s = '%s'", lc, Randomly.fromList(state.getCollates())));`
	`343`	`+ if (Randomly.getBoolean() && !state.getCollates().isEmpty()) {`
	`344`	`+ sb.append(String.format(" LOCALE = '%s' ", Randomly.fromList(state.getCollates())));`
	`345`	`+ } else {`
	`346`	`+ for (String lc : Arrays.asList("LC_COLLATE", "LC_CTYPE")) {`
	`347`	`+ if (!state.getCollates().isEmpty() && Randomly.getBoolean()) {`
	`348`	`+ sb.append(String.format(" %s = '%s'", lc, Randomly.fromList(state.getCollates())));`
	`349`	`+ }`
`346`	`350`	`}`
`347`	`351`	`}`
`348`	`352`	`sb.append(" TEMPLATE template0");`