如何定制myeclipse逆向生成ORM映射实体

最新推荐文章于 2022-07-20 20:26:38 发布

liweisnake

最新推荐文章于 2022-07-20 20:26:38 发布

阅读量1.3k

点赞数

分类专栏： develop 文章标签： ORM 定制 MyEclipse

本文链接：https://blog.csdn.net/liweisnake/article/details/9290129

版权

develop 专栏收录该内容

24 篇文章 0 订阅

订阅专栏

ORM比较方便的一点就是能够自动生成实体类和DAO类，可以说当entity和dao生成好了，基本上已经完成了一半的工作，大大提高了开发效率。

平时我常用的就是myeclipse，生成的过程简单快捷，代码也简洁不繁琐；当然，还有些框架如rapid-framework可以生成比较复杂的实体，而且配置起来也很麻烦。

但是当我们需要定制生成的实体类时，比如我的数据库的表中没有ID，逆向之后就会将所有字段合成ID，从而生成另一个实体XXXID，如果是自己定义的框架不支持额外的这个实体，应该咋办呢？

从界面上看，能够让我们定制的地方有这么几个：但是目测第一个template是实体类的一个模板，第二个rev-setting是你用于反向工程的配置（每次进来都会自动填充上次的配置），而第三custom rev-eng strategy则是定制逆向工程过程中的设置。

为了解决这个问题，我需要定制逆向策略。过程比较简单，参考如下两个链接：

http://www.myeclipseide.com/documentation/quickstarts/hibernate/#5-6

http://www.javaplex.com/blog/myeclipse-customizing-hibernate-reverse-engineering/

以下是我定制的策略，主要就是要去修改主键，将第一个字段当做主键，“骗”过逆向代码，让它认为该实体有主键，从而不会为我们自动合成复合主键。

public class ReverseStrategy extends DelegatingReverseEngineeringStrategy {

	Connection connection;

	public ReverseStrategy(ReverseEngineeringStrategy delegate) {
		super(delegate);
		ConnectionConfiguration cc = new ConnectionConfiguration();
		try {
			Class.forName(cc.getDriverClass());
			connection = DriverManager.getConnection(cc.getConnectionUrl(), cc
					.getUsername(), cc.getPassword());
		} catch (ClassNotFoundException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		} catch (SQLException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
	}

	@Override
	public List getPrimaryKeyColumnNames(TableIdentifier identifier) {
		List list = new ArrayList();
		DatabaseMetaData d;
		try {
			d = connection.getMetaData();
			ResultSet rs = d.getColumns(null, "%", identifier.getName(), "%");
			if(rs.next())
			{
			    String strFieldName = rs.getString(4);
			    list.add(strFieldName);
			}
		} catch (SQLException e) {
			e.printStackTrace();
		}
		return list;
	}

	@Override
	public void close() {
		super.close();
		try {
			connection.close();
//			RemoveId.removeIds();
		} catch (SQLException e) {
			e.printStackTrace();
		} catch (Exception e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
			}

}

完成之后需要删掉源码之中生成的@Id，也需要写个程序处理一下

public class RemoveId {

	private final static String sourcePath = "C:\\xxxx\\xxxx\\xxx\\xx";

	public static void removeIds() throws IOException {
		File dirs = new File(sourcePath);
		if (dirs.isDirectory()) {
			File[] fs = dirs.listFiles();
			for (File file : fs) {
				System.out.println(file.getName());
				String str = FileUtils.readFileToString(file);
				FileUtils
						.writeStringToFile(file, str.replaceAll("@Id", ""));
			}
		}
	}

	public static void main(String[] args) throws IOException {
		removeIds();
	}
}

hibernate tool：http://docs.jboss.org/tools/4.0.0.Final/en/hibernatetools/html_single/index.html#download_base_project