Java读取数据库表（二）

Java读取数据库表（二）


application.properties

1. db.driver.name=com.mysql.cj.jdbc.Driver
2. db.url=jdbc:mysql://localhost:3306/easycrud?useUnicode=true&characterEncoding=utf8&serverTimezone=Asia/Shanghai&useSSL=false
3. db.username=root
4. db.password=xpx24167830
6. #是否忽略表前缀
7. ignore.table.prefix=true
9. #参数bean后缀
10. suffix.bean.param=Query

复制代码

辅助阅读

配置文件中部分信息被读取到之前文档说到的Constants.java中以常量的形式存储，BuildTable.java中会用到，常量命名和上面类似。
StringUtils.java

1. package com.easycrud.utils;
3. /**
4. * @BelongsProject: EasyCrud
5. * @BelongsPackage: com.easycrud.utils
6. * @Author: xpx
7. * @Email: 2436846019@qq.com
8. * @CreateTime: 2023-05-03  13:30
9. * @Description: 字符串大小写转换工具类
10. * @Version: 1.0
11. */
13. public class StringUtils {
14.     /**
15.      * 首字母转大写
16.      * @param field
17.      * @return
18.      */
19.     public static String uperCaseFirstLetter(String field) {
20.         if (org.apache.commons.lang3.StringUtils.isEmpty(field)) {
21.             return field;
22.         }
23.         return field.substring(0, 1).toUpperCase() + field.substring(1);
24.     }
26.     /**
27.      * 首字母转小写
28.      * @param field
29.      * @return
30.      */
31.     public static String lowerCaseFirstLetter(String field) {
32.         if (org.apache.commons.lang3.StringUtils.isEmpty(field)) {
33.             return field;
34.         }
35.         return field.substring(0, 1).toLowerCase() + field.substring(1);
36.     }
38.     /**
39.      * 测试
40.      * @param args
41.      */
42.     public static void main(String[] args) {
43.         System.out.println(lowerCaseFirstLetter("Abcdef"));
44.         System.out.println(uperCaseFirstLetter("abcdef"));
45.     }
46. }

复制代码

辅助阅读

org.apache.commons.lang3.StringUtils.isEmpty()

只能判断String类型是否为空（org.springframework.util包下的Empty可判断其他类型），源码如下

1. public static boolean isEmpty(final CharSequence cs) {
2.         return cs == null || cs.length() == 0;
3. }

复制代码

xx.toUpperCase()

字母转大写

xx.toLowerCase()

字母转小写

xx.substring()

返回字符串的子字符串

1. 索引从0开始
2. public String substring(int beginIndex)        //起始索引，闭
3. public String substring(int beginIndex, int endIndex)        //起始索引到结束索引，左闭右开

复制代码

BuildTable.java完整代码

1. package com.easycrud.builder;
3. import com.easycrud.bean.Constants;
4. import com.easycrud.bean.FieldInfo;
5. import com.easycrud.bean.TableInfo;
6. import com.easycrud.utils.JsonUtils;
7. import com.easycrud.utils.PropertiesUtils;
8. import com.easycrud.utils.StringUtils;
9. import org.apache.commons.lang3.ArrayUtils;
10. import org.slf4j.Logger;
11. import org.slf4j.LoggerFactory;
12. import java.sql.*;
13. import java.util.ArrayList;
14. import java.util.HashMap;
15. import java.util.List;
16. import java.util.Map;
18. /**
19. * @BelongsProject: EasyCrud
20. * @BelongsPackage: com.easycrud.builder
21. * @Author: xpx
22. * @Email: 2436846019@qq.com
23. * @CreateTime: 2023-05-02  18:02
24. * @Description: 读Table
25. * @Version: 1.0
26. */
28. public class BuildTable {
30.     private static final Logger logger = LoggerFactory.getLogger(BuildTable.class);
31.     private static Connection conn = null;
33.     /**
34.      * 查表信息，表名，表注释等
35.      */
36.     private static String SQL_SHOW_TABLE_STATUS = "show table status";
38.     /**
39.      * 将表结构当作表读出字段的信息，如字段名(field),类型(type),自增(extra)...
40.      */
41.     private static String SQL_SHOW_TABLE_FIELDS = "show full fields from %s";
43.     /**
44.      * 检索索引
45.      */
46.     private static String SQL_SHOW_TABLE_INDEX = "show index from %s";
48.     /**
49.      * 读配置,连接数据库
50.      */
51.     static {
52.         String driverName = PropertiesUtils.getString("db.driver.name");
53.         String url = PropertiesUtils.getString("db.url");
54.         String user = PropertiesUtils.getString("db.username");
55.         String password = PropertiesUtils.getString("db.password");
57.         try {
58.             Class.forName(driverName);
59.             conn = DriverManager.getConnection(url,user,password);
60.         } catch (Exception e) {
61.             logger.error("数据库连接失败",e);
62.         }
63.     }
65.     /**
66.      * 读取表
67.      */
68.     public static List<TableInfo> getTables() {
69.         PreparedStatement ps = null;
70.         ResultSet tableResult = null;
72.         List<TableInfo> tableInfoList = new ArrayList();
73.         try{
74.             ps = conn.prepareStatement(SQL_SHOW_TABLE_STATUS);
75.             tableResult = ps.executeQuery();
76.             while(tableResult.next()) {
77.                 String tableName = tableResult.getString("name");
78.                 String comment = tableResult.getString("comment");
79.                 //logger.info("tableName:{},comment:{}",tableName,comment);
81.                 String beanName = tableName;
82.                 /**
83.                  * 去xx_前缀
84.                  */
85.                 if (Constants.IGNORE_TABLE_PREFIX) {
86.                     beanName = tableName.substring(beanName.indexOf("_")+1);
87.                 }
88.                 beanName = processFiled(beanName,true);
90. //                logger.info("bean:{}",beanName);
92.                 TableInfo tableInfo = new TableInfo();
93.                 tableInfo.setTableName(tableName);
94.                 tableInfo.setBeanName(beanName);
95.                 tableInfo.setComment(comment);
96.                 tableInfo.setBeanParamName(beanName + Constants.SUFFIX_BEAN_PARAM);
98.                 /**
99.                  * 读字段信息
100.                  */
101.                 readFieldInfo(tableInfo);
103.                 /**
104.                  * 读索引
105.                  */
106.                 getKeyIndexInfo(tableInfo);
108. //                logger.info("tableInfo:{}",JsonUtils.convertObj2Json(tableInfo));
110.                 tableInfoList.add(tableInfo);
112. //                logger.info("表名：{},备注：{}，JavaBean:{},JavaParamBean:{}",tableInfo.getTableName(),tableInfo.getComment(),tableInfo.getBeanName(),tableInfo.getBeanParamName());
113.             }
114.             logger.info("tableInfoList:{}",JsonUtils.convertObj2Json(tableInfoList));
115.         }catch (Exception e){
116.             logger.error("读取表失败",e);
117.         }finally {
118.             if (tableResult != null) {
119.                 try {
120.                     tableResult.close();
121.                 } catch (SQLException e) {
122.                     e.printStackTrace();
123.                 }
124.             }
125.             if (ps != null) {
126.                 try {
127.                     ps.close();
128.                 } catch (SQLException e) {
129.                     e.printStackTrace();
130.                 }
131.             }
132.             if (conn != null) {
133.                 try {
134.                     conn.close();
135.                 } catch (SQLException e) {
136.                     e.printStackTrace();
137.                 }
138.             }
139.         }
140.         return tableInfoList;
141.     }
143.     /**
144.      * 将表结构当作表读出字段的信息，如字段名(field),类型(type),自增(extra)...
145.      * @param tableInfo
146.      * @return
147.      */
148.     private static void readFieldInfo(TableInfo tableInfo) {
149.         PreparedStatement ps = null;
150.         ResultSet fieldResult = null;
152.         List<FieldInfo> fieldInfoList = new ArrayList();
153.         try{
154.             ps = conn.prepareStatement(String.format(SQL_SHOW_TABLE_FIELDS,tableInfo.getTableName()));
155.             fieldResult = ps.executeQuery();
156.             while(fieldResult.next()) {
157.                 String field = fieldResult.getString("field");
158.                 String type = fieldResult.getString("type");
159.                 String extra = fieldResult.getString("extra");
160.                 String comment = fieldResult.getString("comment");
162.                 /**
163.                  * 类型例如varchar(50)我们只需要得到varchar
164.                  */
165.                 if (type.indexOf("(") > 0) {
166.                     type = type.substring(0, type.indexOf("("));
167.                 }
168.                 /**
169.                  * 将aa_bb变为aaBb
170.                  */
171.                 String propertyName = processFiled(field, false);
173. //                logger.info("f:{},p:{},t:{},e:{},c:{},",field,propertyName,type,extra,comment);
175.                 FieldInfo fieldInfo = new FieldInfo();
176.                 fieldInfoList.add(fieldInfo);
178.                 fieldInfo.setFieldName(field);
179.                 fieldInfo.setComment(comment);
180.                 fieldInfo.setSqlType(type);
181.                 fieldInfo.setAutoIncrement("auto_increment".equals(extra) ? true : false);
182.                 fieldInfo.setPropertyName(propertyName);
183.                 fieldInfo.setJavaType(processJavaType(type));
185. //                logger.info("JavaType:{}",fieldInfo.getJavaType());
187.                 if (ArrayUtils.contains(Constants.SQL_DATE_TIME_TYPES, type)) {
188.                     tableInfo.setHaveDataTime(true);
189.                 }else {
190.                     tableInfo.setHaveDataTime(false);
191.                 }
192.                 if (ArrayUtils.contains(Constants.SQL_DATE_TYPES, type)) {
193.                     tableInfo.setHaveData(true);
194.                 }else {
195.                     tableInfo.setHaveData(false);
196.                 }
197.                 if (ArrayUtils.contains(Constants.SQL_DECIMAL_TYPE, type)) {
198.                     tableInfo.setHaveBigDecimal(true);
199.                 }else {
200.                     tableInfo.setHaveBigDecimal(false);
201.                 }
202.             }
203.             tableInfo.setFieldList(fieldInfoList);
204.         }catch (Exception e){
205.             logger.error("读取表失败",e);
206.         }finally {
207.             if (fieldResult != null) {
208.                 try {
209.                     fieldResult.close();
210.                 } catch (SQLException e) {
211.                     e.printStackTrace();
212.                 }
213.             }
214.             if (ps != null) {
215.                 try {
216.                     ps.close();
217.                 } catch (SQLException e) {
218.                     e.printStackTrace();
219.                 }
220.             }
221.         }
222.     }
224.     /**
225.      * 检索唯一索引
226.      * @param tableInfo
227.      * @return
228.      */
229.     private static List<FieldInfo> getKeyIndexInfo(TableInfo tableInfo) {
230.         PreparedStatement ps = null;
231.         ResultSet fieldResult = null;
233.         List<FieldInfo> fieldInfoList = new ArrayList();
234.         try{
235.             /**
236.              * 缓存Map
237.              */
238.             Map<String,FieldInfo> tempMap = new HashMap();
239.             /**
240.              * 遍历表中字段
241.              */
242.             for (FieldInfo fieldInfo : tableInfo.getFieldList()) {
243.                 tempMap.put(fieldInfo.getFieldName(),fieldInfo);
244.             }
246.             ps = conn.prepareStatement(String.format(SQL_SHOW_TABLE_INDEX,tableInfo.getTableName()));
247.             fieldResult = ps.executeQuery();
248.             while(fieldResult.next()) {
249.                 String keyName = fieldResult.getString("key_name");
250.                 Integer nonUnique = fieldResult.getInt("non_unique");
251.                 String columnName = fieldResult.getString("column_name");
253.                 /**
254.                  * 0是唯一索引，1不唯一
255.                  */
256.                 if (nonUnique == 1) {
257.                     continue;
258.                 }
260.                 List<FieldInfo> keyFieldList = tableInfo.getKeyIndexMap().get(keyName);
262.                 if (null == keyFieldList) {
263.                     keyFieldList = new ArrayList();
264.                     tableInfo.getKeyIndexMap().put(keyName,keyFieldList);
265.                 }
267.                 keyFieldList.add(tempMap.get(columnName));
268.             }
269.         }catch (Exception e){
270.             logger.error("读取索引失败",e);
271.         }finally {
272.             if (fieldResult != null) {
273.                 try {
274.                     fieldResult.close();
275.                 } catch (SQLException e) {
276.                     e.printStackTrace();
277.                 }
278.             }
279.             if (ps != null) {
280.                 try {
281.                     ps.close();
282.                 } catch (SQLException e) {
283.                     e.printStackTrace();
284.                 }
285.             }
286.         }
287.         return fieldInfoList;
288.     }
290.     /**
291.      * aa_bb__cc==>AaBbCc || aa_bb_cc==>aaBbCc
292.      * @param field
293.      * @param uperCaseFirstLetter，首字母是否大写
294.      * @return
295.      */
296.     private static String processFiled(String field,Boolean uperCaseFirstLetter) {
297.         StringBuffer sb = new StringBuffer();
298.         String[] fields=field.split("_");
299.         sb.append(uperCaseFirstLetter ? StringUtils.uperCaseFirstLetter(fields[0]):fields[0]);
300.         for (int i = 1,len = fields.length; i < len; i++){
301.             sb.append(StringUtils.uperCaseFirstLetter(fields[i]));
302.         }
303.         return sb.toString();
304.     }
306.     /**
307.      * 为数据库字段类型匹配对应Java属性类型
308.      * @param type
309.      * @return
310.      */
311.     private static String processJavaType(String type) {
312.         if (ArrayUtils.contains(Constants.SQL_INTEGER_TYPE,type)) {
313.             return "Integer";
314.         }else if (ArrayUtils.contains(Constants.SQL_LONG_TYPE,type)) {
315.             return "Long";
316.         }else if (ArrayUtils.contains(Constants.SQL_STRING_TYPE,type)) {
317.             return "String";
318.         }else if (ArrayUtils.contains(Constants.SQL_DATE_TIME_TYPES,type) || ArrayUtils.contains(Constants.SQL_DATE_TYPES,type)) {
319.             return "Date";
320.         }else if (ArrayUtils.contains(Constants.SQL_DECIMAL_TYPE,type)) {
321.             return "BigDecimal";
322.         }else {
323.             throw new RuntimeException("无法识别的类型:"+type);
324.         }
325.     }
326. }

复制代码

辅助阅读

去表名前缀，如tb_test-->test

1. beanName = tableName.substring(beanName.indexOf("_")+1);

复制代码

indexOf("_")定位第一次出现下划线的索引位置，substring截取后面的字符串。

processFiled(String,Boolean)

自定义方法，用于将表名或字段名转换为Java中的类名或属性名，如aa_bb__cc-->AaBbCc || aa_bb_cc-->aaBbCc

processFiled(String,Boolean)中的String[] fields=field.split("_")

xx.split("_")是将xx字符串按照下划线进行分割。

processFiled(String,Boolean)中的append()

StringBuffer类包含append()方法，相当于“+”，将指定的字符串追加到此字符序列。

processJavaType(String)

自定义方法，用于做数据库字段类型与Java属性类型之间的匹配。

processJavaType(String)中的ArrayUtils.contains(A,B)

判断B是否在A中出现过。

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！