文档章节

编写自己的代码生成工具四:数据库信息查询实现

Yemon
 Yemon
发布于 2016/05/17 10:33
字数 1126
阅读 5
收藏 0

数据库信息的查询,无非就是查询表拥有的列,列的备注以及表的主键,实现起来十分简单。

这里主要考虑的就是,如何能有一个好的扩展,在更换数据库的时候可以方便的切换而不用修改源代码。

我们先来定义一个接口,这个接口只有一个方法,就是查询表的列信息:


  
  1. /**
  2. * 数据库操作接口
  3. *
  4. * User: liyd
  5. * Date: 13-12-6
  6. * Time: 上午10:30
  7. */
  8. public interface DatabaseProvider {
  9. /**
  10. * 获取数据库表的信息
  11. *
  12. * @param tableName 表名
  13. * @return meta data
  14. */
  15. public List<Column> getTableMetaData(String tableName);
  16. }

Column类代码如下:


  
  1. /**
  2. * 列信息
  3. *
  4. * User: liyd
  5. * Date: 13-11-28
  6. * Time: 下午5:19
  7. */
  8. public class Column {
  9. /** 列名 */
  10. private String name;
  11. /** 是否主键 */
  12. private boolean isPrimaryKey;
  13. /** 列备注 */
  14. private String comment;
  15. /** 数据库类型 */
  16. private String dbType;
  17. /** jdbc类型 */
  18. private String jdbcType;
  19. /** java类型 例:String */
  20. private String javaType;
  21. /** java类型class名称例:java.lang.String */
  22. private String javaClass;
  23. getter and setter...
  24. }

定义了接口,接下来就是要实现它了,但是如果我们直接进行实现,那么数据库连接的获取关闭等操作都要在这个方法里面完成,而且每实现一个数据查询类都需要这些,将会比较麻烦。因此,我们先来一个抽象的实现:


  
  1. /**
  2. * 数据库操作抽象类
  3. *
  4. * User: liyd
  5. * Date: 13-12-6
  6. * Time: 上午11:13
  7. */
  8. public abstract class AbstractDatabaseProvider implements DatabaseProvider {
  9. /**
  10. * 获取数据库表的信息
  11. *
  12. * @param tableName 表名
  13. * @return meta data
  14. */
  15. @Override
  16. public List<Column> getTableMetaData(String tableName) {
  17. Connection connection = DBUtils.getDefaultConnection();
  18. return getMetaData(tableName, connection);
  19. }
  20. /**
  21. * 获取数据库表元信息
  22. *
  23. * @param tableName the table name
  24. * @param connection the connection
  25. * @return meta data
  26. */
  27. public abstract List<Column> getMetaData(String tableName, Connection connection);
  28. }

在这个抽象类中,我们统一对数据库连接进行了获取,然后再调用getMetaData(String tableName, Connection connection)方法将表名和连接对象作为参数传入,由子类来实现。

这里获取的连接即是前面在代码生成组织者GenerationOrganizer中创建打开的连接,这里只进行获取,由外围负责打开和关闭。这样做的好处是一次性生成多张表的时候,连接只打开和关闭一次。还有就是具体的实现者不用关心这个连接是哪里来的,不会迷惑于连接到底是自己创建还是从哪里获取,清晰明了。

下面贴上具体的mysql实现类:


  
  1. /**
  2. * mysql操作类
  3. *
  4. * User: liyd
  5. * Date: 14-1-13
  6. * Time: 上午11:40
  7. */
  8. public class MysqlProvider extends AbstractDatabaseProvider {
  9. /**
  10. * 获取数据库表元信息
  11. *
  12. * @param tableName the table name
  13. * @param connection the connection
  14. * @return meta data
  15. */
  16. @Override
  17. public List<Column> getMetaData(String tableName, Connection connection) {
  18. List<Column> columnList = new ArrayList<Column>();
  19. PreparedStatement pst = null;
  20. ResultSet rs = null;
  21. ResultSetMetaData rsd = null;
  22. try {
  23. //查询时没有数据,只返回表头信息
  24. pst = connection.prepareStatement("select * from " + tableName + " where 1=2");
  25. rsd = pst.executeQuery().getMetaData();
  26. //查询主键
  27. String primaryKey = null;
  28. pst = connection
  29. .prepareStatement("SELECT COLUMN_NAME FROM INFORMATION_SCHEMA.KEY_COLUMN_USAGE WHERE CONSTRAINT_NAME='PRIMARY' and TABLE_NAME = ?");
  30. pst.setString(1, tableName.toUpperCase());
  31. rs = pst.executeQuery();
  32. if (rs.next()) {
  33. primaryKey = rs.getString(1);
  34. }
  35. //查询列备注
  36. pst = connection
  37. .prepareStatement("select column_name, column_comment from information_schema.columns where table_name = ?");
  38. pst.setString(1, tableName.toUpperCase());
  39. rs = pst.executeQuery();
  40. //先将注释放入到map再获取,防止有些列没有注释获取不对应的问题
  41. Map<String, String> commentMap = new HashMap<String, String>();
  42. while (rs.next()) {
  43. commentMap.put(rs.getString("COLUMN_NAME"), rs.getString("column_comment"));
  44. }
  45. for (int i = 1; i <= rsd.getColumnCount(); i++) {
  46. String name = rsd.getColumnName(i);
  47. String dbType = rsd.getColumnTypeName(i);
  48. String javaClass = rsd.getColumnClassName(i);
  49. String comment = commentMap.get(name);
  50. boolean b = StringUtils.equalsIgnoreCase(primaryKey, name) ? true : false;
  51. Column column = new Column();
  52. column.setName(name);
  53. column.setDbType(dbType);
  54. column.setJdbcType(dbType);
  55. column.setJavaClass(javaClass);
  56. column.setComment(comment);
  57. column.setIsPrimaryKey(b);
  58. columnList.add(column);
  59. }
  60. } catch (SQLException e) {
  61. throw new EasyCodeException(e);
  62. } finally {
  63. try {
  64. if (pst != null) {
  65. pst.close();
  66. }
  67. if (rs != null) {
  68. rs.close();
  69. }
  70. } catch (SQLException e) {
  71. //ignore
  72. }
  73. }
  74. return columnList;
  75. }
  76. }

oracle的实现类:


  
  1. /**
  2. * Oracle 操作类
  3. *
  4. * User: liyd
  5. * Date: 13-12-6
  6. * Time: 上午11:11
  7. */
  8. public class OracleProvider extends AbstractDatabaseProvider {
  9. /**
  10. * 获取数据库表元信息
  11. *
  12. * @param tableName the table name
  13. * @param connection the connection
  14. * @return meta data
  15. */
  16. @Override
  17. public List<Column> getMetaData(String tableName, Connection connection) {
  18. List<Column> columnList = new ArrayList<Column>();
  19. PreparedStatement pst = null;
  20. ResultSet rs = null;
  21. ResultSetMetaData rsd = null;
  22. try {
  23. //查询时没有数据,只返回表头信息
  24. pst = connection.prepareStatement("select * from " + tableName + " where 1=2");
  25. rsd = pst.executeQuery().getMetaData();
  26. //查询主键
  27. String primaryKey = null;
  28. pst = connection
  29. .prepareStatement("select col.column_name from user_constraints con, user_cons_columns col where con.constraint_name = col.constraint_name and con.constraint_type = 'P' and col.table_name = ?");
  30. pst.setString(1, tableName.toUpperCase());
  31. rs = pst.executeQuery();
  32. if (rs.next()) {
  33. primaryKey = rs.getString(1);
  34. }
  35. //查询列备注
  36. pst = connection
  37. .prepareStatement("SELECT * FROM USER_COL_COMMENTS WHERE TABLE_NAME = ?");
  38. pst.setString(1, tableName.toUpperCase());
  39. rs = pst.executeQuery();
  40. //先将注释放入到map再获取,防止有些列没有注释获取不对应的问题
  41. Map<String, String> commentMap = new HashMap<String, String>();
  42. while (rs.next()) {
  43. commentMap.put(rs.getString("COLUMN_NAME"), rs.getString("COMMENTS"));
  44. }
  45. for (int i = 1; i <= rsd.getColumnCount(); i++) {
  46. String name = rsd.getColumnName(i);
  47. String dbType = rsd.getColumnTypeName(i);
  48. String javaClass = rsd.getColumnClassName(i);
  49. String comment = commentMap.get(name);
  50. boolean b = StringUtils.equalsIgnoreCase(primaryKey, name) ? true : false;
  51. Column column = new Column();
  52. column.setName(name);
  53. column.setDbType(dbType);
  54. column.setJdbcType(dbType);
  55. column.setJavaClass(javaClass);
  56. column.setComment(comment);
  57. column.setIsPrimaryKey(b);
  58. columnList.add(column);
  59. }
  60. } catch (SQLException e) {
  61. throw new EasyCodeException(e);
  62. } finally {
  63. try {
  64. if (pst != null) {
  65. pst.close();
  66. }
  67. if (rs != null) {
  68. rs.close();
  69. }
  70. } catch (SQLException e) {
  71. }
  72. }
  73. return columnList;
  74. }

本文转载自:http://blog.csdn.net/hj7jay/article/details/51130540

共有 人打赏支持
Yemon
粉丝 10
博文 320
码字总数 22799
作品 0
广州
架构师
基于 JavaFX8 的图形界面 Spring 代码生成器 - Spring-generator

Spring-generator 是基于 javafx8 开发的图形界面 Spring 代码生成器,使用 Apache FreeMarker 作为代码文件的模板,用户可以一键将数据库中的表生成为任意风格的 .java 代码文件(比如经典的...

Mirren
06/19
0
0
敢不敢更懒,自己编写的mysql C++自动代码生成工具

我一直有一个梦想,就是那天能够让计算机去帮我编写代码,而我只需要将我自己的规划告诉它,就可以自动生成我需要的代码。 梦想永远是梦想,一定要自己一步步的去实现,否则和咸鱼有什么区别...

自由的眼
2016/12/13
80
0
Spring-generator 1.0.0 版正式发布,Spring 代码生成器

Spring-generator是基于javafx8开发的图形界面Spring代码生成器,使用 Apache FreeMarker作为代码文件的模板,用户可以一键将数据库中的表生成为任意风格的.java代码文件(比如经典的三层模型)...

Mirren
06/19
0
0
ShenzhenMirren/Spring-generator

基本介绍 官方QQ交流群号:519714660; Spring-generator是基于javafx8开发的图形界面Spring代码生成器,使用 Apache FreeMarker 作为代码文件的模板,用户可以一键将数据库中的表生成为任意风格...

ShenzhenMirren
06/19
0
0
OpenJWeb(v1.0) Java Web快速开发平台功能清单

一、OpenJWeb功能清单 OpenJWeb作为一款Java快速Web应用开发平台产品,目前提供了以下功能: 序 功能点 功能说明 1 系统基础数据—数据字典类型定义 定义数据字典类型,即编码种类 2 系统基础...

迷途d书童
2012/03/09
196
0

没有更多内容

加载失败,请刷新页面

加载更多

读书(附电子书)|小狗钱钱之白色的拉布拉多

关注公众号,在公众号中回复“小狗钱钱”可免费获得电子书。 一、背景 之前写了一篇文章 《小狗钱钱》 理财小白应该读的一本书,那时候我才看那本书,现在看了一大半了,发现这本书确实不错,...

tiankonguse
29分钟前
0
0
Permissions 0777 for ‘***’ are too open

异常显示: @@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@ @ WARNING: UNPROTECTED PRIVATE KEY FILE! @ @@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@ ......

李玉长
31分钟前
0
0
区块链10年了,还未落地,它失败了吗?

导读 几乎每个人,甚至是对通证持怀疑态度的人,都对区块链的技术有积极的看法,因为它有可能改变世界。然而,区块链技术问世已经10年了,我们仍然没有真正的用上区块链技术。 几乎每个人,甚...

问题终结者
59分钟前
2
0
20180921 su与sudo命令、限制root用户通过ssh远程登录

su 命令 用户切换。 su # 切换到root用户su username # 切换到username用户# su 后面加-时,会初始化当前用户的各种环境su - username # 指定用户执行某些命令 su - -c "touch /tm...

野雪球
今天
2
0
Windows 下双 Python 开发环境配置

Windows 下双 Python 开发环境配置作者:老农民(刘启华)QQ: 46715422Email: 46715422@qq.com微信: 46715422 本人曾经在 Windows 下被两个版本环境折腾够呛,现在总结两个 Python...

新疆老农民
昨天
4
0

没有更多内容

加载失败,请刷新页面

加载更多

返回顶部
顶部