add DynamicTableGeneratorQueryTest

Author: xlorne

springboot-starter-data-fast/src/main/java/com/codingapi/springboot/fast/generator/DynamicTableGenerator.java

@@ -5,7 +5,6 @@
 import org.hibernate.boot.registry.StandardServiceRegistry;
 import org.hibernate.boot.registry.StandardServiceRegistryBuilder;
 import org.hibernate.cfg.AvailableSettings;
-import org.hibernate.dialect.Dialect;
 import org.hibernate.tool.hbm2ddl.ImportSqlCommandExtractor;
 import org.hibernate.tool.schema.SourceType;
 import org.hibernate.tool.schema.TargetType;
@@ -23,7 +22,7 @@ public class DynamicTableGenerator {
     /**
      * 数据库方言
      */
-    private final Dialect dialect;
+    private final Class<?> dialect;
     private final StandardServiceRegistry serviceRegistry;
     private final SchemaManagementTool managementTool;
 
@@ -32,14 +31,9 @@ public DynamicTableGenerator(Class<?> dialectClass, String jdbcUrl) {
     }
 
     public DynamicTableGenerator(Class<?> dialectClass, String jdbcUrl, String username, String password) {
-        try {
-            this.dialect = (Dialect) dialectClass.getDeclaredConstructor().newInstance();
-        } catch (Exception e) {
-            throw new RuntimeException("Failed to instantiate dialect", e);
-        }
-
+        this.dialect = dialectClass;
         StandardServiceRegistryBuilder builder = new StandardServiceRegistryBuilder()
-                .applySetting(AvailableSettings.DIALECT, dialect.getClass().getName())
+                .applySetting(AvailableSettings.DIALECT, dialect.getName())
                 .applySetting("hibernate.connection.url", jdbcUrl);
         if (StringUtils.hasText(username)) {
             builder.applySetting("hibernate.connection.username", username);
@@ -56,7 +50,7 @@ private class ExecutionOptionsImpl implements ExecutionOptions {
         @Override
         public Map<String, Object> getConfigurationValues() {
             Map<String, Object> config = new HashMap<>();
-            config.put(AvailableSettings.DIALECT, dialect.getClass().getName());
+            config.put(AvailableSettings.DIALECT, dialect.getName());
             return config;
         }
 

springboot-starter-data-fast/src/test/java/com/codingapi/springboot/fast/generator/DynamicTableGeneratorQueryTest.java

@@ -0,0 +1,71 @@
+package com.codingapi.springboot.fast.generator;
+
+import com.codingapi.springboot.fast.jdbc.JdbcQuery;
+import com.codingapi.springboot.fast.jdbc.JdbcQueryContext;
+import com.codingapi.springboot.fast.jpa.JPAQuery;
+import com.codingapi.springboot.fast.jpa.JpaQueryContext;
+import com.codingapi.springboot.fast.metadata.TableEntityMetadata;
+import javax.persistence.EntityManagerFactory;
+import javax.persistence.GenerationType;
+import org.hibernate.cfg.AvailableSettings;
+import org.junit.jupiter.api.Test;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.core.env.Environment;
+import org.springframework.jdbc.core.JdbcTemplate;
+
+import java.util.List;
+import java.util.Map;
+
+import static org.junit.jupiter.api.Assertions.assertFalse;
+import static org.junit.jupiter.api.Assertions.assertThrows;
+
+@SpringBootTest
+public class DynamicTableGeneratorQueryTest {
+
+    @Autowired
+    private EntityManagerFactory entityManagerFactory;
+
+    @Autowired
+    private Environment environment;
+
+    @Autowired
+    private JdbcTemplate jdbcTemplate;
+
+
+    @Test
+    void generateThenUpdateQuery() throws ClassNotFoundException {
+        String dialect = entityManagerFactory.getProperties().get(AvailableSettings.DIALECT).toString();
+        String jdbcUrl = environment.getProperty("spring.datasource.url");
+        String username = environment.getProperty("spring.datasource.username");
+        String password = environment.getProperty("spring.datasource.password");
+
+        DynamicTableGenerator dynamicTableGenerator = new DynamicTableGenerator(Class.forName(dialect), jdbcUrl, username, password);
+
+        TableEntityMetadata tableEntityMetadata = new TableEntityMetadata("com.codingapi.springboot.fast.entity.Test");
+        tableEntityMetadata.setTable("test");
+        tableEntityMetadata.addPrimaryKeyColumn(Long.class, "id", GenerationType.IDENTITY, "主键");
+        tableEntityMetadata.addColumn(String.class, "name", "姓名");
+
+        Class<?> entityClass = tableEntityMetadata.buildClass();
+
+        // 创建表
+        dynamicTableGenerator.generateMigratorTableDDL(entityClass, true);
+
+        // 通过DynamicTableGenerator 动态创建的entity无法通过jpa的 EntityManager进行数据查询处理, 因为Entity不在JPA扫描支持的范围内。
+        JPAQuery jpaQuery = JpaQueryContext.getInstance().getJPAQuery();
+        assertThrows(Exception.class,()->{
+            jpaQuery.listQuery(entityClass,"select t from com.codingapi.springboot.fast.entity.Test");
+        });
+
+        // 插入数据
+        jdbcTemplate.update("insert into test(name) values(?)", "小明");
+
+        // 查询数据
+        JdbcQuery jdbcQuery = JdbcQueryContext.getInstance().getJdbcQuery();
+        List<Map<String, Object>> list = jdbcQuery.queryForMapList("select * from test");
+        System.out.println(list);
+        assertFalse(list.isEmpty());
+
+    }
+}