`

根据表信息生成JavaVO

 
阅读更多
package com.ustcsoft.util;

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.Date;

/**
 * 类:DBToVO
 * 功能:根据数据库表生成VO--Value Object
 * @author sam
 * @e-mail:sanlai_lee@lisanlai.cn
 */
public class DBToJavaVO {
	private static final String TARGET_DIR="C:/tempVO/";							//类文件存放的路径
	private static final String DIVER_NAME="org.gjt.mm.mysql.Driver";
	private static final String URL="jdbc:mysql://localhost:3306/bms"; 
	private static final String USERNAME="root";
	private static final String PASSWORD="root";
	private static final String DATABASE_NAME="BMS";								//数据库名称
	private static final String AUTHOR="sam";										//作者
	private static final String E_MAIL="sanlai_lee@lisanlai.cn";					//作者邮箱
	private static final String PACKAGE_NAME="com.ustcsoft.vo";						//包名
	private Connection conn;
	private Statement stmt;
	private String sql;
	private ResultSet rs;
	private String[] fields;	//属性
	private String[] dataTypes;	//数据类型
	private String[] comments;	//属性的注释
	/**
	 * 方法:根据数据库表生成VO--Value Object
	 * @param tableName
	 */
	private void tableToVo(){
		
		String tableName;
		String className;
		try {
			InputStreamReader reader = new InputStreamReader(System.in);
			BufferedReader bf = new BufferedReader(reader);
			System.out.println("输入要转换成VO的表名称:");
			tableName = bf.readLine().trim();
			System.out.println("输入生成的类名,不输入的时候默认跟表名相同:");
			className = bf.readLine().trim();
			Class.forName(DIVER_NAME);
			conn = DriverManager.getConnection(URL,USERNAME,PASSWORD);
			stmt = conn.createStatement(ResultSet.TYPE_SCROLL_SENSITIVE,ResultSet.CONCUR_READ_ONLY);	//创建可滚动的,只读的结果集
			sql = "SELECT COLUMN_NAME, DATA_TYPE, COLUMN_COMMENT"+
				  " FROM INFORMATION_SCHEMA.COLUMNS"+
				  " WHERE table_name = '"+tableName.toUpperCase()+"'"+		//表名
				  " AND table_schema = '"+DATABASE_NAME+"'";				//数据库名
			System.out.println(sql);
			rs = stmt.executeQuery(sql);
			rs.last();										//把指针指向结果集的最后
			int fieldNum = rs.getRow();						//取得最后一条结果的行号,作为类的属性个数
			int n = fieldNum;
			if(n>0){									//判断数据表中是否存在字段
				fields = new String[n];
				dataTypes = new String[n];
				comments = new String[n];
				fields[--n] = rs.getString(1);
				dataTypes[n] = rs.getString(2);
				comments[n] = rs.getString(3);
				while(rs.previous()){
					fields[--n]=rs.getString(1);			//取得结果集的第一列数据,对应的列名:Field
					dataTypes[n] = rs.getString(2);
					comments[n] = rs.getString(3);
				}
				//打印相关信息
				System.out.println("你要转换的表是:"+tableName);
				System.out.println("该表中共有"+fieldNum+"个字段,信息如下:");
				for(int i=0,j=fields.length;i<j;i++){
					System.out.println("----------------------------------------");
					String field = fields[i];
					System.out.println("字段名称:"+field);
					//把字段名称格式化成java命名规则形式
					field=formatField(field);
					fields[i]=field;						//把格式化后的字段放入属性数组中
					System.out.println("数据类型:"+dataTypes[i]);
					//把数据库字段类型转换成java数据类型
					String dataType = dataTypes[i].toLowerCase();
					dataType = formatDataType(dataType);
					dataTypes[i] = dataType;
					
					System.out.println("字段注释:"+comments[i]);
					if("".equals(comments[i])||comments[i]==null){
						comments[i]=fields[i];
					}
					System.out.println("----------------------------------------");
				}
				//格式化类名称
				foramtClassName(className,tableName);
				//生成类文件,写入到磁盘中
				writeObjectToFile(className);
			}else{
				System.out.println("该表不存在或者表中没有字段");
			}	
		} catch (ClassNotFoundException e) {
			System.out.println("未找到数据库驱动");
			System.out.println(e.getMessage());
		} catch (SQLException e) {
			System.out.println(e.getMessage());
			//e.printStackTrace();
		} catch (IOException e) {
			System.out.println(e.getMessage());
			//e.printStackTrace();
		}finally{
			try {
				if(stmt!=null){
					if(!stmt.isClosed()){
						stmt.close();
						stmt=null;
						System.gc();
					}
				}
				if(conn!=null){
					if(!conn.isClosed()){
						conn.close();
						conn=null;
						System.gc();
					}
				}
			} catch (SQLException e) {
				System.out.println(e.getMessage());
			}
		}
	}
	
	/**
	 * 把数据库字段格式成java变量名
	 * @param field
	 * @return
	 */
	private String formatField(String field){
		String[] strs = field.split("_");
		field="";
		for(int m = 0,length=strs.length;m<length;m++){
			if(m>0){
				String tempStr = strs[m].toLowerCase();
				tempStr = tempStr.substring(0, 1).toUpperCase()+tempStr.substring(1, tempStr.length());
				field += tempStr;
			}else{
				field += strs[m].toLowerCase();
			}
		}
		return field;
	}
	
	private String formatDataType(String dataType){
		if(dataType.contains("char")){
			dataType="java.lang.String";
		}else if(dataType.contains("int")){
			dataType="java.lang.Integer";
		}else if(dataType.contains("float")){
			dataType="java.lang.Float";
		}else if(dataType.contains("double")){
			dataType="java.lang.Double";
		}else if(dataType.contains("number")||dataType.contains("decimal")){
			dataType="java.math.BigDecimal";
		}else if(dataType.contains("date")){
			dataType="java.sql.Date";
		}else if(dataType.contains("time")){
			dataType="java.sql.Timestamp";
		}else if(dataType.contains("clob")){
			dataType="java.sql.Clob";
		}else{
			dataType="java.lang.Object";
		}
		return dataType;
	}
	/**
	 * 格式化类名
	 * @param className
	 * @param tableName
	 * @return
	 */
	private String foramtClassName(String className,String tableName){
		//如果类名不是自己定义的,那么根据表名格式化类名
		if("".equals(className)||className==null){
			className="";
			String[] tempArr = tableName.split("_");
			for(int m=0,length=tempArr.length;m<length;m++){		
				className += tempArr[m].substring(0,1).toUpperCase()+tempArr[m].substring(1,tempArr[m].length()).toLowerCase();
			}
		}else{
			//如果类名已经输入,那么不管三七二十一就把类名的第一个字母大写
			className = className.substring(0,1).toUpperCase()+className.substring(1,className.length());
		}
		return className;
	}
	
	/**
	 * 生成类并写到文件
	 * @param className
	 */
	@SuppressWarnings("deprecation")
	private void writeObjectToFile(String className){
		PrintWriter writer = null;
		try {
			 File dir = new File(TARGET_DIR);
			 if(!dir.exists()){
				 dir.mkdirs();
			 }
			 File file = new File(TARGET_DIR+className+".java");
			 if(!file.exists()){
				 file.createNewFile();
			 }
			 writer = new PrintWriter(new BufferedWriter(new FileWriter(file)));
			 
			 //拼文件内容
			 StringBuffer content = new StringBuffer();
			 content.append("package "+PACKAGE_NAME+";\n\n");
			 content.append("/**\n");
			 content.append(" *@类:"+className+"\n");
			 content.append(" *@作者:"+AUTHOR+"\n");
			 content.append(" *@E-mail:"+E_MAIL+"\n");
			 content.append(" *@日期:"+new Date().toLocaleString().substring(0,10)+"\n");
			 content.append(" */\n\n");
			 content.append("public class "+className+"{\n\n");
			 for(int i=0,j=fields.length;i<j;i++){
				 content.append("	/**"+comments[i]+"*/\n");
				 content.append("	private "+dataTypes[i]+" "+fields[i]+";\n\n");
			 }
			 //get set 方法
			 for(int i=0,j=fields.length;i<j;i++){
				 content.append("	/**\n");
				 content.append("	 *方法: 取得"+fields[i]+"\n");
				 content.append("	 *@return: "+dataTypes[i]+"  "+fields[i]+"\n");
				 content.append("	 */\n");
				 content.append("	public "+dataTypes[i]+" get"+fields[i].substring(0,1).toUpperCase()+fields[i].substring(1,fields[i].length())+"(){\n");
				 content.append("		return this."+fields[i]+";\n");
				 content.append("	}\n\n");
				 content.append("	/**\n");
				 content.append("	 *方法: 设置"+fields[i]+"\n");
				 content.append("	 *@param: "+dataTypes[i]+"  "+fields[i]+"\n");
				 content.append("	 */\n");
				 content.append("	public void set"+fields[i].substring(0,1).toUpperCase()+fields[i].substring(1,fields[i].length())+"("+dataTypes[i]+" "+fields[i]+"){\n");
				 content.append("		this."+fields[i]+" = "+fields[i]+";\n");
				 content.append("	}\n\n");
			 }
			 content.append("}");
			 writer.write(content.toString());
			 writer.flush();
			 System.out.println("类生成成功,存放路径:"+TARGET_DIR);
		} catch (IOException e) {
			e.printStackTrace();
		}finally{
			if(writer!=null){
				writer.close();
			}
		}
	}
	
	public static void main(String[] args) {
		try {
			new DBToJavaVO().tableToVo();
		} catch (Exception e) {
			System.out.println(e.getMessage());
		}
		
	}
}

 http://lisanlai.iteye.com/blog/800389

分享到:
评论

相关推荐

    java根据数据库自动生成vo 类

    NULL 博文链接:https://joynet007.iteye.com/blog/972596

    自动生成vo类(mysql数据库为例)

    根据数据库中表的字段自动生成java中的vo类,方便调用

    扩展MyBatisPlus代码生成器实现自定义源码生成,可生成前端页面、vo对象、dto对象等代码

    原代码生成组件,默认会在最终生成文件名前,默认附加实体名前缀,例如实体名是User,VO.java.ftl会自动生成UserVO.java,这样没问题。但是对于前端页面,如list.vue,edit.vue,因为是分目录放实体,希望最终文件名...

    Java一键生成代码到controller层【源码】

    3、可大大提高开发效率,数据表创建完以后,自动生成entity,mapper.xml,dao,service,controller,vo,dto相关代码。 4、本项目集成了spring,aop,mybatis plus,swagger2,异常处理,分页,freemarker等多种技术。 5、操作...

    java 自动生成代码 entity,dao,service

    对于经常写重复性代码的entity,dao,service,本例子中写了一个简单的自动生成小程序,有需要者可以下载

    根据数据表自动生成基础CRUD代码

    "CRUD代码生成器(Java)" 是一个强大的工具,旨在帮助开发者根据数据库中的数据表自动生成基础CRUD代码。这包括实体类(Entity)、数据访问对象(DAO)、数据传输对象(DTO)、值对象(VO)、查询对象(QO)、转换器...

    java web代码生成器

    java web开发后台代码自动生成,支持SSH和SSM2大框架,数据库支持MySQL和sqlserver,根据数据库自动生成entity,vo,service,impl,action或者controller,dao,mapper等类,包含常用的增删改查方法,集成了pageheper分页...

    AutoCode代码生成器(JAVA版)

    您是不是因为java的开发和维护繁琐而烦恼,目前政府机关、企业大都要求用java来开发项目,本软件将彻底解决您的后顾之忧,通过本软件的自动生成功能,生成的java代码规范、全部开源,不存在用隐含的内容,全部代码...

    mybatis自动生成mapping文件、vo和dao接口

    使用说明 1.打开命令窗口 ...java -jar mybatis-generator-core-1.3.2.jar -configfile generatorConfig.xml -overwrite 回车 5.MyBatis Generator finished successfully. 表示成功生成mapping、model等文件

    快速代码生成器

    根据oracle数据库中的表,生成对应的vo/dao,vo_中提供了对应的get、set方法,dao中提供了增、删、改、查,各种通用的方法。

    java代码生成器

    该生成器连接的是oracle数据库,通过选择数据库中的表和视图,根据配置的路径自动在该路径下生成对应的po,vo,jsp,dao和service文件夹和下面的java文件 是初学java者的便捷小工具

    基于数据库的自动化生成工具,自动生成JavaBean、自动生成数据库文档等(v4.1.2版)

    3、新增根据字段快速生成JavaBean的功能,在VO中增量添加字段非常方便。 4、新增精确匹配(排除)和模糊匹配(排除),排除掉不需要生成的表。 5、新增数据源备注,该备注用于自动生成数据库设计文档的文件名。 6、...

    MkCodeTool代码生成工具

    1.您将的生成Java代码根据的的数据库或者创建数据库表的SQL 2.MkCodeTool将您要写的代码严格分开,使您业务逻辑改变或者表字段更变的时候不会影响您写的任何代码。也是MkCodeTool的特色之一 3.您将非常容易的扩展您...

    SSH代码生成工具 SSH代码生成器

    VO --&gt; bean.java及 bean.hbm.xml配置文件 Service --&gt; Service接口定义类 DAO --&gt; DAO接口定义类 DAOImpl --&gt; DAO接口Implements实现类 程序配置 --&gt; web.xml 验证框架--&gt; bean-validation.xml (实体验证)、...

    AutoCode代码生成器(Struts1.2版)

    ☆ 支持多表的多主键处理。 ☆ 支持数据事务的操作。 ☆ 生成内容可以依据客户的需要来选择性的生成。 ☆ 支持自定义查询接口的生成,用户可以定义查询条件。 ☆ 生成的jsp文件及Action完全处理好了精度字段的...

    DTOGenerator:从类生成 DTO 的 IntelliJIDEA 插件

    该插件将创建当前在编辑器中处于活动状态的 Java 类的 DTO。 可以使用CTRL + ALT + D或通过打开生成器弹出ALT + INSERT来调用插件 更多的信息 目前,该插件将读取您现有类的所有字段,并将它们复制到一个目标类,...

    ssh代码生成器轻松、快捷

    VO --&gt; bean.java及 bean.hbm.xml配置文件 Service --&gt; Service接口定义类 DAO --&gt; DAO接口定义类 DAOImpl --&gt; DAO接口Implements实现类 程序配置 --&gt; web.xml 验证框架--&gt; bean-validation.xml (实体验证)、Save...

    mybatisJavaCodeTarena (mybatis 项目代码生成器)

    这是我自己写的代码生成器,只支持mysql数据库,可以生成PO,VO,FORM,DAO,mapper.xml,IService,serviceImpl,IBiz,bizImpl,IWeb,webImpl,Controller等文件代码基本上一个项目所需要的文件都生成了。生成的代码,需要...

    NC65主子VO字段

    1. 建立UAP业务组件项目(模块名),新建UAP组件(组件名) 2. 切换MDP透视图,新建实体组件(实体组件名) ...8. 发布元数据,生成Java源代码(NC65生成的代码有问题,需要补全),生成建库SQL脚本并执行

Global site tag (gtag.js) - Google Analytics