Skip to content

Commit 41291f4

Browse files
committed
Address feedbacks
1 parent 99928c1 commit 41291f4

File tree

4 files changed

+16
-10
lines changed

4 files changed

+16
-10
lines changed

extension/persistence/relational-jdbc/src/main/java/org/apache/polaris/extension/persistence/relational/jdbc/DatabaseType.java

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,8 @@
1818
*/
1919
package org.apache.polaris.extension.persistence.relational.jdbc;
2020

21+
import java.util.Locale;
22+
2123
public enum DatabaseType {
2224
POSTGRES("postgres"),
2325
H2("h2");
@@ -34,10 +36,14 @@ public String getDisplayName() {
3436
}
3537

3638
public static DatabaseType fromDisplayName(String displayName) {
37-
return switch (displayName) {
39+
return switch (displayName.toLowerCase(Locale.ROOT)) {
3840
case "h2" -> DatabaseType.H2;
3941
case "postgresql" -> DatabaseType.POSTGRES;
4042
default -> throw new IllegalStateException("Unsupported DatabaseType: '" + displayName + "'");
4143
};
4244
}
45+
46+
public String getInitScriptResource() {
47+
return String.format("%s/schema-v1.sql", this.getDisplayName());
48+
}
4349
}

extension/persistence/relational-jdbc/src/main/java/org/apache/polaris/extension/persistence/relational/jdbc/JdbcMetaStoreManagerFactory.java

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
*/
1919
package org.apache.polaris.extension.persistence.relational.jdbc;
2020

21+
import com.google.common.base.Preconditions;
2122
import io.smallrye.common.annotation.Identifier;
2223
import jakarta.annotation.Nullable;
2324
import jakarta.enterprise.context.ApplicationScoped;
@@ -26,7 +27,6 @@
2627
import java.sql.SQLException;
2728
import java.util.HashMap;
2829
import java.util.List;
29-
import java.util.Locale;
3030
import java.util.Map;
3131
import java.util.function.Supplier;
3232
import javax.sql.DataSource;
@@ -112,16 +112,13 @@ private void initializeForRealm(
112112

113113
private DatasourceOperations getDatasourceOperations(boolean isBootstrap) {
114114
// TODO: remove when multiple dataSources are supported.
115-
if (dataSources.size() > 1) {
116-
throw new IllegalStateException("More than one dataSources configured");
117-
}
115+
Preconditions.checkState(dataSources.size() == 1, "More than one dataSources configured");
118116
DatasourceOperations databaseOperations = new DatasourceOperations(dataSources.getFirst());
119117
if (isBootstrap) {
120118
try {
121119
DatabaseType databaseType;
122120
try (Connection connection = dataSources.getFirst().getConnection()) {
123-
String productName =
124-
connection.getMetaData().getDatabaseProductName().toLowerCase(Locale.ROOT);
121+
String productName = connection.getMetaData().getDatabaseProductName();
125122
databaseType = DatabaseType.fromDisplayName(productName);
126123
}
127124
databaseOperations.executeScript(

extension/persistence/relational-jdbc/src/main/java/org/apache/polaris/extension/persistence/relational/jdbc/models/ModelEntity.java

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -152,8 +152,11 @@ public ModelEntity fromResultSet(ResultSet r) throws SQLException {
152152
.purgeTimestamp(r.getObject("purge_timestamp", Long.class))
153153
.toPurgeTimestamp(r.getObject("to_purge_timestamp", Long.class))
154154
.lastUpdateTimestamp(r.getObject("last_update_timestamp", Long.class))
155-
.properties(r.getString("properties"))
156-
.internalProperties(r.getString("internal_properties"))
155+
.properties(
156+
r.getString("properties")) // required for extracting when the underlying type is JSONB
157+
.internalProperties(
158+
r.getString(
159+
"internal_properties")) // required for extracting when the underlying type is JSONB
157160
.grantRecordsVersion(r.getObject("grant_records_version", Integer.class))
158161
.build();
159162
}

integration-tests/src/main/java/org/apache/polaris/service/it/test/PolarisRestCatalogViewIntegrationBase.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,7 @@ static void close() throws Exception {
8484

8585
@BeforeEach
8686
public void before(TestInfo testInfo) {
87-
org.junit.jupiter.api.Assumptions.assumeFalse(shouldSkip());
87+
Assumptions.assumeThat(shouldSkip()).isFalse();
8888

8989
String principalName = client.newEntityName("snowman-rest");
9090
String principalRoleName = client.newEntityName("rest-admin");

0 commit comments

Comments
 (0)