前言
想要自己实现一个mybatis-generator类似的轮子，目前只实现MySQL部分的方法。利用下班时间，写了一个小项目，实现了这个功能。我准备分成三篇博客来写这个东西。
基类
/**
*基类
*通用方法和属性
*getMapper 获取mapper 例如：cc.vvxtoys.mapper.StudentMapper
*getPojo 获取实体 例如：cc.vvxtoys.pojo.Student
*getFileName 生成地址
*@author vvxtoys
*/
package cc.vvxtoys.bean;import cc.vvxtoys.config.Engine;public class BaseBean {protected String clazz ="";//实体类名
protected String context = "";//bean内容public String getClazz() {
return clazz==null?"":clazz;
}
public void setClazz(String clazz) {
this.clazz = clazz;
}
public String getContext() {
return context==null?"":context;
}
public void setContext(String context) {
this.context = context;
}
public String getMapper(){
return Engine.targetMapper+"."+clazz+"Mapper";
}
public String getPojo(){
return Engine.targetPackage+"."+clazz;
}
public String getFileName() {return null;
}
}
Bean
/**
*ClassBean
*dependency 生成实体是否存在Date类型 存在的话导入java.util.Date
*@author vvxtoys
*/
package cc.vvxtoys.bean;import java.io.File;import cc.vvxtoys.config.Engine;public class ClassBean extends BaseBean{private String attributes = "";//属性
private String methods= "";//方法
private String constructors = "";//构造
private String packages= Engine.targetPackage;
private boolean dependency= false;//依赖public ClassBean(String clazz) {
this.clazz = clazz;
}public String getAttributes() {
return attributes;
}
public void setAttributes(String attributes) {
this.attributes = attributes;
}
public String getMethods() {
return methods;
}
public void setMethods(String methods) {
this.methods = methods;
}
public String getConstructors() {
return constructors;
}
public void setConstructors(String constructors) {
this.constructors = constructors;
}
public String getPackages() {
return packages;
}
public boolean isDependency() {
return dependency;
}
public void setDependency(boolean dependency) {
this.dependency = dependency;
}public String getFileName(){
return Engine.targetProject+File.separator+Engine.targetPackage.replace(".", File.separator)+File.separator+clazz+".java";
}}
Mapper
/**
*MapperBean
*primary 主键 一般来说建表是有的，默认为true
*@author vvxtoys
*/
package cc.vvxtoys.bean;import java.io.File;import cc.vvxtoys.config.Engine;public class MapperBean extends BaseBean{private String packages = Engine.targetMapper;
private boolean primary = true;public MapperBean(String clazz) {
this.clazz = clazz;
}public boolean isPrimary() {
return primary;
}public String getPackages() {
return packages;
}
public void setPrimary(boolean primary) {
this.primary = primary;
}
public String getFileName() {
return Engine.targetProject + File.separator + Engine.targetMapper.replace(".", File.separator)
+ File.separator + clazz + "Mapper.java";
}
}
XML
/**
*XmlBean
*这个类其实我是不想写的，原定计划使用dom4j来实现，但是最终使用dom4j换行没有达到我想要的
*效果，所以我就自己写了一个简单的xml方法，可以满足我现在的需求了。
*方法就不写注释了，感觉应该能看懂
*@author vvxtoys
*/
package cc.vvxtoys.bean;import java.io.File;
import java.util.ArrayList;
import java.util.List;import cc.vvxtoys.utils.Vconst;
import cc.vvxtoys.config.Engine;
import cc.vvxtoys.utils.Common;public class XMLBean extends BaseBean {private String name;//节点名
private List<XMLBean> childrens = new ArrayList<XMLBean>();//子节点
private int cnt;//用来判断空格
private boolean isEnd = true;//是否关闭节点
private String tname;//表名public XMLBean(String name, String tname, String clazz) {
this.name = name;
this.tname = tname;
this.clazz = clazz;if (name.equals(Vconst.ROOTELEMENT[0])) {
cnt = 0;
}
}public XMLBean(String name) {
this(name, null, null);
}public void init() {
for (int i = 0; i < cnt; i++) {
this.context += Vconst.SPACE;
}
context += "<" + name + ">";
}public void addElement(XMLBean item) {
item.setCnt(this.cnt + 2);
childrens.add(item);
isEnd = false;
}public void addText(String text) {
StringBuffer buff = new StringBuffer(context);
buff.append(Vconst.LINE);
for (int i = 0; i < cnt + 1; i++) {
buff.append(Vconst.SPACE);
}
buff.append(text);
context = buff.toString();
isEnd = false;
}public void addEndText(String text) {
if (childrens != null && childrens.size() > 0) {
context += Vconst.LINE;
for (XMLBean i : childrens) {
context += i.toString();
}
}
childrens.clear();
StringBuffer buff = new StringBuffer(context);
for (int i = 0; i < cnt + 1; i++) {
buff.append(Vconst.SPACE);
}
buff.append(text);
context = buff.toString();
isEnd = false;
}public void addAttribute(String Qname, String value) {
context = context.substring(0, getSize() - 1);
StringBuffer buff = new StringBuffer(context);
buff.append(Common.ConcatAttr(Qname, value));
buff.append(" >");
context = buff.toString();
}public int getSize() {
return context.length();
}public void setCnt(int cnt) {
this.cnt = cnt;
}public String getTname() {
return tname;
}public void setTname(String tname) {
this.tname = tname;
}public String getFileName() {
return Engine.targetProject + File.separator + Engine.targetXML.replace(".", File.separator) + File.separator
+ clazz + "Mapper.xml";
}public String toString() {
StringBuffer buffer = new StringBuffer(context);
if (isEnd) {
return context.substring(0, getSize() - 1) + "/>\n";
}
buffer.append(Vconst.LINE);
if (childrens != null && childrens.size() > 0) {
for (XMLBean i : childrens) {
buffer.append(i.toString());
}
}
for (int i = 0; i < cnt; i++) {
buffer.append(Vconst.SPACE);
}
buffer.append("</").append(name).append(">");
buffer.append(Vconst.LINE);
return buffer.toString();
}
}
配置工具类
/**
*读取配置文件
*初始化,如果没有自定义配置文件，默认全文检索默认配置文件
*author vvxtoys
*/
package cc.vvxtoys.config;import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.InputStreamReader;
import java.io.UnsupportedEncodingException;
import java.util.Properties;import cc.vvxtoys.utils.Common;public class Engine extends Properties{/**
*
*/
private static final long serialVersionUID = 1L;private static boolean isDefault = true;
private static String DEFAULT_CONFIG = "vgenerator.properties";public static String targetPackage;//实体类包
public static String targetProject;//项目路径
public static String targetMapper;//mapper包
public static String targetXML;//xml包
public static String [] objectName;//实体名
public static String [] tableName;//表名public static String user;
public static String pass;
public static String name;
public static String url;public static void setConfigFile(String path){
DEFAULT_CONFIG = path;
isDefault = false;
}private Engine() {
// TODO Auto-generated constructor stub
}public static Engine getInstance(){
Engine engine = null;
String path = isDefault?Common.searchAbsolutePath(DEFAULT_CONFIG):DEFAULT_CONFIG;
try {
engine = new Engine();
InputStreamReader is = new InputStreamReader(new FileInputStream(path), "UTF-8");
engine.load(is);targetPackage = (String)engine.getProperty("targetPackage");
targetProject = (String)engine.getProperty("targetProject");
targetMapper = (String)engine.getProperty("targetMapper");
targetXML = (String)engine.getProperty("targetXML");
tableName = ((String)engine.getProperty("tableName")).split(",");
objectName = ((String)engine.getProperty("objectName")).split(",");user = (String)engine.getProperty("user");
pass = (String)engine.getProperty("pass");
name = (String)engine.getProperty("name");
url = (String)engine.getProperty("url");} catch (UnsupportedEncodingException e) {
e.printStackTrace();
} catch (FileNotFoundException e) {
e.printStackTrace();
}catch (Exception e) {
e.printStackTrace();
}
return engine;
}}
