修正使用HikariDataSource时CacheDB(2016)生成文档报错的问题

yanchie · Sep 14, 2020 · e890b82 · e890b82
2 parents d6882c7 + 6e5409b
commit e890b82
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/screw-core/src/main/java/cn/smallbun/screw/core/query/AbstractDatabaseQuery.java b/screw-core/src/main/java/cn/smallbun/screw/core/query/AbstractDatabaseQuery.java
@@ -140,7 +140,13 @@ protected String getSchema() throws QueryException {
      * @return Schema
      */
     private String verifySchema(DataSource dataSource) throws SQLException {
-        String schema = dataSource.getConnection().getSchema();
+        String schema;
+        if (dataSource instanceof HikariDataSource) {
+            schema = ((HikariDataSource) dataSource).getSchema();
+        } else {
+            schema = dataSource.getConnection().getSchema();
+        }
+
         //验证是否有此Schema
         ResultSet resultSet = this.getConnection().getMetaData().getSchemas();
         while (resultSet.next()) {