jdbc

Author: wendell-dev

generator/src/main/java/lazy/fast/code/generator/Config.java

@@ -1,7 +1,5 @@
 package lazy.fast.code.generator;
 
-import lazy.fast.code.generator.util.JdbcUtils;
-
 import java.util.HashMap;
 import java.util.Map;
 
@@ -57,6 +55,11 @@ public class Config {
      */
     public static String classDescription;
 
+    public static String jdbcClassName;
+    public static String jdbcUrl;
+    public static String jdbcUser;
+    public static String jdbcPassword;
+
     /**
      * 模板填充数据值
      */
@@ -74,10 +77,10 @@ public static void initTemplateData() {
         DATA.put("className", className);
         DATA.put("classDescription", classDescription);
 
-        DATA.put("jdbcClassName", JdbcUtils.className);
-        DATA.put("jdbcUrl", JdbcUtils.url);
-        DATA.put("jdbcUser", JdbcUtils.user);
-        DATA.put("jdbcPassword", JdbcUtils.password);
+        DATA.put("jdbcClassName", jdbcClassName);
+        DATA.put("jdbcUrl", jdbcUrl);
+        DATA.put("jdbcUser", jdbcUser);
+        DATA.put("jdbcPassword", jdbcPassword);
     }
 
 }

generator/src/main/java/lazy/fast/code/generator/QuickStart.java

@@ -48,6 +48,11 @@ public static void main(String[] args) {
         // 项目骨架模板文件根路径
         Config.layoutTemplateRootPath = System.getProperty("template.layout", null);
 
+        Config.jdbcClassName = System.getProperty("jdbc.class", null);
+        Config.jdbcUrl = System.getProperty("jdbc.url", null);
+        Config.jdbcUser = System.getProperty("jdbc.user", null);
+        Config.jdbcPassword = System.getProperty("jdbc.password", null);
+
         // 生成项目骨架
         Generator.generateLayout();
     }

generator/src/main/java/lazy/fast/code/generator/util/JdbcUtils.java

@@ -1,11 +1,10 @@
 package lazy.fast.code.generator.util;
 
-import java.io.IOException;
-import java.io.InputStream;
+import lazy.fast.code.generator.Config;
+
 import java.sql.Connection;
 import java.sql.DriverManager;
 import java.sql.SQLException;
-import java.util.Properties;
 
 /**
  * 数据库工具操作类
@@ -14,38 +13,19 @@
  */
 public class JdbcUtils {
 
-    public static String className;
-    public static String url;
-    public static String user;
-    public static String password;
-
-    static {
-        InputStream is = JdbcUtils.class.getClassLoader().getResourceAsStream("jdbc.properties");
-        Properties props = new Properties();
-        try {
-            props.load(is);
-            className = props.getProperty("className");
-            url = props.getProperty("url");
-            user = props.getProperty("user");
-            password = props.getProperty("password");
-        } catch (IOException e) {
-            throw new RuntimeException();
-        }
-    }
-
     private Connection con = null;
 
     public JdbcUtils() {
         try {
-            Class.forName(className);
+            Class.forName(Config.jdbcClassName);
         } catch (ClassNotFoundException e) {
             throw new RuntimeException();
         }
     }
 
     public Connection getConnection() {
         try {
-            con = DriverManager.getConnection(url, user, password);
+            con = DriverManager.getConnection(Config.jdbcUrl, Config.jdbcUser, Config.jdbcPassword);
         } catch (SQLException e) {
             throw new RuntimeException();
         }

generator/src/main/resources/jdbc.properties

@@ -16,6 +16,7 @@ public class GeneratorTest {
         // 模板文件根路径, 为空的时候就默认使用内置模板，如果有自定义模板就写模板绝对路径
         Config.moduleTemplateRootPath = null;
         Config.layoutTemplateRootPath = null;
+
         // 作者签名, 如果不填写则按System.getProperty("user.name")取值
         Config.author = "${author}";
 
@@ -38,6 +39,11 @@ public class GeneratorTest {
         // 【必填项】类描述
         Config.classDescription = "用户信息";
 
+        Config.jdbcClassName = "${jdbcClassName!'com.mysql.cj.jdbc.Driver'}";
+        Config.jdbcUrl = "${jdbcUrl!'jdbc:mysql://localhost:3306/demo'}";
+        Config.jdbcUser = "${jdbcUser!'root'}";
+        Config.jdbcPassword = "${jdbcPassword!'123456'}";
+
         // 生成所有模板类
         Generator.generateAll();
     }

generator/src/main/resources/template/layout/generator.ftl

@@ -2,10 +2,10 @@ spring:
   application:
     name: ${projectName}
   datasource:
-    url: ${jdbcUrl}
-    username: ${jdbcUser}
-    password: ${jdbcPassword}
-    driver-class-name: ${jdbcClassName}
+    url: ${jdbcUrl!"jdbc:mysql://localhost:3306/demo?useUnicode=true&characterEncoding=utf-8&useSSL=false&autoReconnect=true&serverTimezone=Asia/Shanghai"}
+    username: ${jdbcUser!"root"}
+    password: ${jdbcPassword!"123456"}
+    driver-class-name: ${jdbcClassName!"com.mysql.cj.jdbc.Driver"}
     hikari:
       minimum-idle: 5
       maximum-pool-size: 15

generator/src/main/resources/template/layout/resource.ftl

@@ -11,6 +11,7 @@ public static void main(String[] args) {
         // 模板文件根路径, 为空的时候就默认使用内置模板，如果有自定义模板就写模板绝对路径
         Config.moduleTemplateRootPath = null;
         Config.layoutTemplateRootPath = null;
+
         // 作者签名, 如果不填写则按System.getProperty("user.name")取值
         Config.author = "wendell";
 
@@ -33,6 +34,11 @@ public static void main(String[] args) {
         // 【必填项】类描述
         Config.classDescription = "用户信息";
 
+        Config.jdbcClassName = "com.mysql.cj.jdbc.Driver";
+        Config.jdbcUrl = "jdbc:mysql://localhost:3306/demo";
+        Config.jdbcUser = "root";
+        Config.jdbcPassword = "123456";
+
         // 生成所有模板类
         Generator.generateAll();
     }