Implementd maven-plugin "Hibernate 4"
authorKai Moritz <kai@coolibri.de>
Sun, 20 Jun 2010 18:17:07 +0000 (20:17 +0200)
committerKai Moritz <kai@coolibri.de>
Tue, 11 Dec 2012 20:53:56 +0000 (21:53 +0100)
pom.xml [new file with mode: 0644]
src/main/java/de/juplo/plugins/hibernate4/Hbm2DdlMojo.java [new file with mode: 0644]

diff --git a/pom.xml b/pom.xml
new file mode 100644 (file)
index 0000000..3b8c444
--- /dev/null
+++ b/pom.xml
@@ -0,0 +1,134 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/maven-v4_0_0.xsd">
+
+  <modelVersion>4.0.0</modelVersion>
+
+  <groupId>de.juplo</groupId>
+  <artifactId>hibernate4-maven-plugin</artifactId>
+  <name>Hibernate 4 Maven Plugin</name>
+  <description>Plugin for generating a database-schema from Hibernate-4-Mapping-Annotations</description>
+  <version>1.0-SNAPSHOT</version>
+  <packaging>maven-plugin</packaging>
+  <url>http://juplo.de/hibernate4-maven-plugin</url>
+
+  <prerequisites>
+    <maven>2.0.6</maven>
+  </prerequisites>
+
+  <developers>
+    <developer>
+      <id>kai</id>
+      <name>Kai Moritz</name>
+      <email>kai@juplo.de</email>
+    </developer>
+  </developers>
+
+  <properties>
+    <!-- Zeichensatz -->
+    <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
+    <!-- Verwendete Versionen -->
+    <hibernate-core.version>4.1.5.SP1</hibernate-core.version>
+    <maven.version>3.0.4</maven.version>
+    <maven-plugin-log4j.version>1.0.1</maven-plugin-log4j.version>
+    <scannotation.version>1.0.2</scannotation.version>
+  </properties>
+
+  <dependencies>
+    <dependency>
+      <groupId>org.apache.maven</groupId>
+      <artifactId>maven-core</artifactId>
+      <version>${maven.version}</version>
+      <exclusions>
+        <exclusion>
+          <groupId>org.codehaus.plexus</groupId>
+          <artifactId>plexus-utils</artifactId>
+        </exclusion>
+      </exclusions>
+    </dependency>
+    <dependency>
+      <groupId>org.apache.maven</groupId>
+      <artifactId>maven-plugin-api</artifactId>
+      <version>${maven.version}</version>
+    </dependency>
+    <dependency>
+      <groupId>org.hibernate</groupId>
+      <artifactId>hibernate-core</artifactId>
+      <version>${hibernate-core.version}</version>
+    </dependency>
+    <dependency>
+      <groupId>net.sf.scannotation</groupId>
+      <artifactId>scannotation</artifactId>
+      <version>${scannotation.version}</version>
+    </dependency>
+    <dependency>
+      <groupId>com.pyx4j</groupId>
+      <artifactId>maven-plugin-log4j</artifactId>
+      <version>${maven-plugin-log4j.version}</version>
+      <exclusions>
+        <exclusion>
+          <groupId>org.apache.maven</groupId>
+          <artifactId>maven-artifact</artifactId>
+        </exclusion>
+        <exclusion>
+          <groupId>org.apache.maven</groupId>
+          <artifactId>maven-plugin-api</artifactId>
+        </exclusion>
+      </exclusions>
+    </dependency>
+  </dependencies>
+
+  <distributionManagement>
+    <repository>
+      <id>juplo.internal</id>
+      <name>Internal Release Repository</name>
+      <url>http://juplo.de/archiva/repository/internal/</url>
+    </repository>
+    <snapshotRepository>
+      <id>juplo.snapshots</id>
+      <name>Internal Snapshot Repository</name>
+      <url>http://juplo.de/archiva/repository/snapshots/</url>
+    </snapshotRepository>
+  </distributionManagement>
+
+  <build>
+    <plugins>
+      <plugin>
+        <groupId>org.apache.maven.plugins</groupId>
+        <artifactId>maven-compiler-plugin</artifactId>
+        <configuration>
+          <source>1.6</source>
+          <target>1.6</target>
+          <encoding>utf8</encoding>
+          <showWarnings>true</showWarnings>
+        </configuration>
+      </plugin>
+      <plugin>
+        <groupId>org.apache.maven.plugins</groupId>
+        <artifactId>maven-dependency-plugin</artifactId>
+        <executions>
+          <execution>
+            <id>install</id>
+            <phase>install</phase>
+            <goals>
+              <goal>sources</goal>
+            </goals>
+          </execution>
+        </executions>
+      </plugin>
+      <plugin>
+        <groupId>org.apache.maven.plugins</groupId>
+        <artifactId>maven-source-plugin</artifactId>
+        <executions>
+          <execution>
+            <id>attach-sources</id>
+            <phase>verify</phase>
+            <goals>
+              <goal>jar</goal>
+            </goals>
+          </execution>
+        </executions>
+      </plugin>
+    </plugins>
+  </build>
+
+</project>
diff --git a/src/main/java/de/juplo/plugins/hibernate4/Hbm2DdlMojo.java b/src/main/java/de/juplo/plugins/hibernate4/Hbm2DdlMojo.java
new file mode 100644 (file)
index 0000000..f2fd232
--- /dev/null
@@ -0,0 +1,433 @@
+package de.juplo.plugins.hibernate4;
+
+/*
+ * Copyright 2001-2005 The Apache Software Foundation.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import com.pyx4j.log4j.MavenLogAppender;
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.IOException;
+import java.net.URL;
+import java.net.URLClassLoader;
+import java.sql.Connection;
+import java.sql.Driver;
+import java.sql.DriverManager;
+import java.sql.DriverPropertyInfo;
+import java.sql.SQLException;
+import java.util.Enumeration;
+import java.util.HashSet;
+import java.util.List;
+import java.util.Map.Entry;
+import java.util.Properties;
+import java.util.Set;
+import javax.persistence.Entity;
+import javax.persistence.Embeddable;
+import javax.persistence.MappedSuperclass;
+import org.apache.maven.plugin.AbstractMojo;
+import org.apache.maven.plugin.MojoExecutionException;
+import org.apache.maven.plugin.MojoFailureException;
+import org.apache.maven.project.MavenProject;
+import org.hibernate.cfg.Configuration;
+import org.hibernate.tool.hbm2ddl.SchemaExport;
+import org.hibernate.tool.hbm2ddl.SchemaExport.Type;
+import org.hibernate.tool.hbm2ddl.Target;
+import org.scannotation.AnnotationDB;
+
+
+/**
+ * Goal which extracts the hibernate-mapping-configuration and
+ * exports an according SQL-database-schema.
+ *
+ * @goal export
+ * @phase process-classes
+ * @threadSafe
+ * @requiresDependencyResolution runtime
+ */
+public class Hbm2DdlMojo extends AbstractMojo
+{
+  /**
+   * The project whose project files to create.
+   *
+   * @parameter expression="${project}"
+   * @required
+   * @readonly
+   */
+  private MavenProject project;
+
+  /**
+   * Directories to scan.
+   *
+   * @parameter expression="${project.build.outputDirectory}"
+   */
+  private String outputDirectory;
+
+  /**
+   * Skip execution
+   *
+   * @parameter default-value="false"
+   */
+  private boolean skip;
+
+  /**
+   * SQL-Driver name.
+   *
+   * @parameter expression="${hibernate.connection.driver_class}
+   */
+  private String driverClassName;
+
+  /**
+   * Database URL.
+   *
+   * @parameter expression="${hibernate.connection.url}"
+   */
+  private String url;
+
+  /**
+   * Database username
+   *
+   * @parameter expression="${hibernate.connection.username}"
+   */
+  private String username;
+
+  /**
+   * Database password
+   *
+   * @parameter expression="${hibernate.connection.password}"
+   */
+  private String password;
+
+  /**
+   * Hibernate dialect.
+   *
+   * @parameter expression="${hibernate.dialect}"
+   */
+  private String hibernateDialect;
+
+  /**
+   * Hibernate configuration file.
+   *
+   * @parameter default-value="${project.build.outputDirectory}/hibernate.properties"
+   */
+  private String hibernateProperties;
+
+  /**
+   * Target of execution:
+   * <ul>
+   *   <li><strong>NONE</strong> do nothing - just validate the configuration</li>
+   *   <li><strong>EXPORT</strong> create database <strong>(DEFAULT!)</strong></li>
+   *   <li><strong>SCRIPT</strong> export schema to SQL-script</li>
+   *   <li><strong>BOTH</strong></li>
+   * </ul>
+   * @parameter default-value="EXPORT"
+   */
+  private String target;
+
+  /**
+   * Type of export.
+   * <ul>
+   *   <li><strong>NONE</strong> do nothing - just validate the configuration</li>
+   *   <li><strong>CREATE</strong> create database-schema</li>
+   *   <li><strong>DROP</strong> drop database-schema</li>
+   *   <li><strong>BOTH</strong> <strong>(DEFAULT!)</strong></li>
+   * </ul>
+   * @parameter default-value="BOTH"
+   */
+  private String type;
+
+  /**
+   * Output file.
+   *
+   * @parameter default-value="${project.build.outputDirectory}/schema.sql"
+   */
+  private String outputFile;
+
+  /**
+   * Delimiter in output-file.
+   *
+   * @parameter default-value=";"
+   */
+  private String delimiter;
+
+  /**
+   * Format output-file.
+   *
+   * @parameter default-value="true"
+   */
+  private boolean format;
+
+
+  @Override
+  public void execute()
+      throws
+        MojoFailureException,
+        MojoExecutionException
+  {
+    if (skip)
+      return;
+
+    File dir = new File(outputDirectory);
+    if (!dir.exists())
+      throw new MojoExecutionException("Cannot scan for annotated classes in " + outputDirectory + ": directory does not exist!");
+
+
+    Set<String> classes = new HashSet<String>();
+    URL dirUrl = null;
+    try {
+      AnnotationDB db = new AnnotationDB();
+      getLog().info("Scanning directory " + outputDirectory + " for annotated classes...");
+      dirUrl = dir.toURI().toURL();
+      db.scanArchives(dirUrl);
+      if (db.getAnnotationIndex().containsKey(Entity.class.getName()))
+        classes.addAll(db.getAnnotationIndex().get(Entity.class.getName()));
+      if (db.getAnnotationIndex().containsKey(MappedSuperclass.class.getName()))
+        classes.addAll(db.getAnnotationIndex().get(MappedSuperclass.class.getName()));
+      if (db.getAnnotationIndex().containsKey(Embeddable.class.getName()))
+        classes.addAll(db.getAnnotationIndex().get(Embeddable.class.getName()));
+    }
+    catch (IOException e) {
+      getLog().error("Error while scanning!", e);
+      throw new MojoFailureException(e.getMessage());
+    }
+    if (classes.isEmpty())
+      throw new MojoFailureException("No annotated classes found in directory " + outputDirectory);
+
+    Properties properties = new Properties();
+
+    /** Try to read configuration from properties-file */
+    try {
+      File file = new File(hibernateProperties);
+      if (file.exists()) {
+        getLog().info("Reading properties from file " + hibernateProperties + "...");
+        properties.load(new FileInputStream(file));
+      }
+      else
+        getLog().info("Ignoring nonexistent properties-file " + hibernateProperties + "!");
+    }
+    catch (IOException e) {
+      getLog().error("Error while reading properties!", e);
+      throw new MojoExecutionException(e.getMessage());
+    }
+
+    /** Overwrite values from propertie-file or set  if given */
+    if (driverClassName != null)
+      properties.setProperty("hibernate.connection.driver_class", driverClassName);
+    if (url != null)
+      properties.setProperty("hibernate.connection.url", url);
+    if (username != null)
+      properties.setProperty("hibernate.connection.username", username);
+    if (password != null)
+      properties.setProperty("hibernate.connection.password", password);
+    if (hibernateDialect != null)
+      properties.setProperty("hibernate.dialect", hibernateDialect);
+
+    if (properties.isEmpty())
+      getLog().warn("No properties set!");
+    for (Entry<Object,Object> entry : properties.entrySet())
+      getLog().debug(entry.getKey() + " = " + entry.getValue());
+
+    ClassLoader classLoader = null;
+    try {
+      getLog().debug("Creating ClassLoader for project-dependencies...");
+      List<String> classpathFiles = project.getCompileClasspathElements();
+      URL[] urls = new URL[classpathFiles.size()];
+      for (int i = 0; i < classpathFiles.size(); ++i) {
+        getLog().debug("Dependency: " + classpathFiles.get(i));
+        urls[i] = new File(classpathFiles.get(i)).toURI().toURL();
+      }
+      classLoader = new URLClassLoader(urls, getClass().getClassLoader());
+    }
+    catch (Exception e) {
+      getLog().error("Error while creating ClassLoader!", e);
+      throw new MojoExecutionException(e.getMessage());
+    }
+
+    Configuration config = new Configuration();
+    config.setProperties(properties);
+    try {
+      getLog().debug("Adding annotated classes to hibernate-mapping-configuration...");
+      for (String annotatedClass : classes) {
+        getLog().debug("Class " + annotatedClass);
+        config.addAnnotatedClass(classLoader.loadClass(annotatedClass));
+      }
+    }
+    catch (ClassNotFoundException e) {
+      getLog().error("Error while adding annotated classes!", e);
+      throw new MojoExecutionException(e.getMessage());
+    }
+
+    Target target = null;
+    try {
+      target = Target.valueOf(this.target);
+    }
+    catch (IllegalArgumentException e) {
+      getLog().error("Invalid value for configuration-option \"target\": " + this.target);
+      getLog().error("Valid values are: NONE, SCRIPT, EXPORT, BOTH");
+      throw new MojoExecutionException("Invalid value for configuration-option \"target\"");
+    }
+    Type type = null;
+    try {
+      type = Type.valueOf(this.type);
+    }
+    catch (IllegalArgumentException e) {
+      getLog().error("Invalid value for configuration-option \"type\": " + this.type);
+      getLog().error("Valid values are: NONE, CREATE, DROP, BOTH");
+      throw new MojoExecutionException("Invalid value for configuration-option \"type\"");
+    }
+
+    Connection connection = null;
+    try {
+      /**
+       * The connection must be established outside of hibernate, because
+       * hibernate does not use the context-classloader of the current
+       * thread and, hence, would not be able to resolve the driver-class!
+       */
+      switch (target) {
+        case EXPORT:
+        case BOTH:
+          switch (type) {
+            case CREATE:
+            case DROP:
+            case BOTH:
+              Class driverClass = classLoader.loadClass(driverClassName);
+              getLog().debug("Registering JDBC-driver " + driverClass.getName());
+              DriverManager.registerDriver(new DriverProxy((Driver)driverClass.newInstance()));
+              getLog().debug("Opening JDBC-connection to " + url + " as " + username + " with password " + password);
+              connection = DriverManager.getConnection(url, username, password);
+          }
+      }
+    }
+    catch (ClassNotFoundException e) {
+      getLog().error("Dependency for driver-class " + driverClassName + " is missing!");
+      throw new MojoExecutionException(e.getMessage());
+    }
+    catch (Exception e) {
+      getLog().error("Cannot establish connection to database!");
+      Enumeration<Driver> drivers = DriverManager.getDrivers();
+      if (!drivers.hasMoreElements())
+        getLog().error("No drivers registered!");
+      while (drivers.hasMoreElements())
+        getLog().debug("Driver: " + drivers.nextElement());
+      throw new MojoExecutionException(e.getMessage());
+    }
+
+    ClassLoader contextClassLoader = Thread.currentThread().getContextClassLoader();
+    MavenLogAppender.startPluginLog(this);
+    try {
+      /**
+       * Change class-loader of current thread, so that hibernate can
+       * see all dependencies!
+       */
+      Thread.currentThread().setContextClassLoader(classLoader);
+
+      SchemaExport export = new SchemaExport(config, connection);
+      export.setOutputFile(outputFile);
+      export.setDelimiter(delimiter);
+      export.setFormat(format);
+      export.execute(target, type);
+
+      for (Object exception : export.getExceptions())
+        getLog().debug(exception.toString());
+    }
+    finally {
+      /** Stop Log-Capturing */
+      MavenLogAppender.endPluginLog(this);
+
+      /** Restore the old class-loader (TODO: is this really necessary?) */
+      Thread.currentThread().setContextClassLoader(contextClassLoader);
+
+      /** Close the connection */
+      try {
+        connection.close();
+      }
+      catch (SQLException e) {
+        getLog().error("Error while closing connection: " + e.getMessage());
+      }
+    }
+  }
+
+  /**
+   * Needed, because DriverManager won't pick up drivers, that were not
+   * loaded by the system-classloader!
+   * See:
+   * http://stackoverflow.com/questions/288828/how-to-use-a-jdbc-driver-from-an-arbitrary-location
+   */
+  static final class DriverProxy implements Driver {
+
+    private final Driver target;
+
+    DriverProxy(Driver target) {
+      if (target == null) {
+        throw new NullPointerException();
+      }
+      this.target = target;
+    }
+
+    public java.sql.Driver getTarget() {
+      return target;
+    }
+
+    @Override
+    public boolean acceptsURL(String url) throws SQLException {
+      return target.acceptsURL(url);
+    }
+
+    @Override
+    public java.sql.Connection connect(
+        String url, java.util.Properties info) throws SQLException {
+      return target.connect(url, info);
+    }
+
+    @Override
+    public int getMajorVersion() {
+      return target.getMajorVersion();
+    }
+
+    @Override
+    public int getMinorVersion() {
+      return target.getMinorVersion();
+    }
+
+    @Override
+    public DriverPropertyInfo[] getPropertyInfo(
+        String url, Properties info) throws SQLException {
+      return target.getPropertyInfo(url, info);
+    }
+
+    @Override
+    public boolean jdbcCompliant() {
+      return target.jdbcCompliant();
+    }
+
+    @Override
+    public String toString() {
+      return "Proxy: " + target;
+    }
+
+    @Override
+    public int hashCode() {
+      return target.hashCode();
+    }
+
+    @Override
+    public boolean equals(Object obj) {
+      if (!(obj instanceof DriverProxy)) {
+        return false;
+      }
+      DriverProxy other = (DriverProxy) obj;
+      return this.target.equals(other.target);
+    }
+}
+}