HibernateEnversDDL.java

/**
 * Copyright (C) 2014 Premium Minds.
 *
 * This file is part of pm-persistence-utils.
 *
 * pm-persistence-utils is free software: you can redistribute it and/or modify it under
 * the terms of the GNU Lesser General Public License as published by the Free
 * Software Foundation, either version 3 of the License, or (at your option) any
 * later version.
 *
 * pm-persistence-utils is distributed in the hope that it will be useful, but WITHOUT ANY
 * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
 * A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
 * details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with pm-persistence-utils. If not, see <http://www.gnu.org/licenses/>.
 */
package com.premiumminds.persistence.utils;

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.util.List;

import org.hibernate.cfg.Configuration;
import org.hibernate.dialect.Dialect;
import org.hibernate.envers.configuration.spi.AuditConfiguration;
import org.hibernate.envers.tools.hbm2ddl.EnversSchemaGenerator;
import org.hibernate.tool.hbm2ddl.DatabaseMetadata;
import org.hibernate.tool.hbm2ddl.SchemaUpdateScript;


public class HibernateEnversDDL  {

	/**
	 * --create unitName [filename]
	 * --create-drop unitName [filename]
	 * --update unitName jdbcUrl jdbcUsername jdbcPassword [filename]
	 * 
	 * @param args usage arguments
	 */
	public static void main(String[] args) {
		if(args.length==0){
			System.out.println("Usage: ");
			System.out.println("\t--create unitName filename - Create table commands");
			System.out.println("\t--create-drop unitName filename - Create table and drop commands");
			System.out.println("\t--update unitName jdbcUrl jdbcUsername jdbcPassword filename - Alter table commands based on your database");
			System.out.println("\n\tfilename is the name of the file where to write");
		} else {
			if("--create".equals(args[0].toLowerCase())) createCommand(args);
			if("--create-drop".equals(args[0].toLowerCase())) createDropCommand(args);
			if("--update".equals(args[0].toLowerCase())) updateCommand(args);

		}
	}

	private static void createCommand(String[] args) {
		if(args.length<3){
			System.out.println("Expected unitName and filename");
		} else {
			String unitName = args[1];
			String filename = args[2];

			EnversSchemaGenerator esg = new EnversSchemaGenerator(HibernateDDL.getConfiguration(unitName));
			org.hibernate.tool.hbm2ddl.SchemaExport se = esg.export();
			se.setOutputFile(filename);
			se.setFormat(true);
			se.setDelimiter(";");
			se.execute(false, false, false, true);
		}
	}

	private static void createDropCommand(String[] args) {
		if(args.length<3){
			System.out.println("Expected unitName and filename");
		} else {
			String unitName = args[1];
			String filename = args[2];

			EnversSchemaGenerator esg = new EnversSchemaGenerator(HibernateDDL.getConfiguration(unitName));
			org.hibernate.tool.hbm2ddl.SchemaExport se = esg.export();
			se.setOutputFile(filename);
			se.setFormat(true);
			se.setDelimiter(";");
			se.execute(false, false, false, false);
		}
	}

	private static void updateCommand(String[] args) {
		String unitName, filename=null, url, username, password;
		if(args.length<5) System.out.println("Expected unitName jdbcUrl jdbcUsername jdbcPassword [filename]");
		else {
			unitName = args[1];
			url = args[2];
			username = args[3];
			password = args[4];
			if(args.length>5) filename = args[5];
			
			Configuration configuration = HibernateDDL.getConfiguration(unitName);
			configuration.buildMappings();
			AuditConfiguration.getFor(configuration);
			Dialect dialect = Dialect.getDialect(configuration.getProperties());
	
			Connection conn = null;
			DatabaseMetadata meta = null;
			try {
				conn = DriverManager.getConnection(url, username, password);
				meta = new DatabaseMetadata(conn, dialect, configuration, true);

				List<SchemaUpdateScript> updateScriptList = configuration.generateSchemaUpdateScriptList(dialect, meta);
				String[] updateSQL = SchemaUpdateScript.toStringArray(updateScriptList);
				
				HibernateDDL.stringToStream(updateSQL, filename);
				
			} catch (SQLException e) {
				e.printStackTrace();
			}
		}
	}

}