<!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.0 Transitional//EN">
<html>
<body link="#355491" alink="#4262a1" vlink="#355491" style="background: #e2e2e2; margin: 0; padding: 20px;">
<div>
<table cellpadding="0" bgcolor="#FFFFFF" border="0" cellspacing="0" style="border: 1px solid #dadada; margin-bottom: 30px; width: 100%; -moz-border-radius: 6px; -webkit-border-radius: 6px;">
<tbody>
<tr>
<td>
<table border="0" cellpadding="0" cellspacing="0" bgcolor="#FFFFFF" style="border: solid 2px #ccc; background: #dadada; width: 100%; -moz-border-radius: 6px; -webkit-border-radius: 6px;">
<tbody>
<tr>
<td bgcolor="#000000" valign="middle" height="58px" style="border-bottom: 1px solid #ccc; padding: 20px; -moz-border-radius-topleft: 3px; -moz-border-radius-topright: 3px; -webkit-border-top-right-radius: 5px; -webkit-border-top-left-radius: 5px;">
<h1 style="color: #333333; font: bold 22px Arial, Helvetica, sans-serif; margin: 0; display: block !important;">
<!-- To have a header image/logo replace the name below with your img tag -->
<!-- Email clients will render the images when the message is read so any image -->
<!-- must be made available on a public server, so that all recipients can load the image. -->
<a href="https://community.jboss.org/index.jspa" style="text-decoration: none; color: #E1E1E1">JBoss Community</a></h1>
</td>
</tr>
<tr>
<td bgcolor="#FFFFFF" style="font: normal 12px Arial, Helvetica, sans-serif; color:#333333; padding: 20px; -moz-border-radius-bottomleft: 4px; -moz-border-radius-bottomright: 4px; -webkit-border-bottom-right-radius: 5px; -webkit-border-bottom-left-radius: 5px;"><h3 style="margin: 10px 0 5px; font-size: 17px; font-weight: normal;">
Re: Hibernate Tools question
</h3>
<span style="margin-bottom: 10px;">
created by <a href="https://community.jboss.org/people/alexpote">Alex Popa-Tesileanu</a> in <i>JBoss Tools</i> - <a href="https://community.jboss.org/message/761353#761353">View the full discussion</a>
</span>
<hr style="margin: 20px 0; border: none; background-color: #dadada; height: 1px;">
<div class="jive-rendered-content"><p>Hi,</p><p>The problem was how to get RuntimeInfo in order to get the connection.</p><p>The obvious response is to override configure method:</p><blockquote class="jive-quote"><p>@Override</p><p>          public void configure(ReverseEngineeringRuntimeInfo runtimeInfo) {</p><p>                    // TODO Auto-generated method stub</p><p>                    super.configure(runtimeInfo);</p><p>                    setRuntimeInfo(runtimeInfo);</p><p>                    try {</p><p>                              setConn(getRuntimeInfo().getConnectionProvider().getConnection());</p><p>                    } catch (SQLException e) {</p><p>                              // TODO Auto-generated catch block</p><p>                              e.printStackTrace();</p><p>                    }</p><p>          }</p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p></blockquote><p>Problem solved. It work.</p><p>Bellow is all code, maybe useful for someone.</p><p>It is hardcoded for MySQL and doesn't include all data types. </p><p>Can be improved if will possible to get column defaults and hibernate datatype in <span style="font-family: 'Lucida Grande', Arial, Helvetica, sans-serif; background-color: #ffffff;">columnToMetaAttributes</span> function.</p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p><blockquote class="jive-quote"><p>package ro.alex.reveng;</p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p><p>import java.sql.Connection;</p><p>import java.sql.ResultSet;</p><p>import java.sql.SQLException;</p><p>import java.sql.Statement;</p><p>import java.util.HashMap;</p><p>import java.util.Map;</p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p><p>import org.hibernate.cfg.reveng.DelegatingReverseEngineeringStrategy;</p><p>import org.hibernate.cfg.reveng.ReverseEngineeringRuntimeInfo;</p><p>import org.hibernate.cfg.reveng.ReverseEngineeringStrategy;</p><p>import org.hibernate.cfg.reveng.TableIdentifier;</p><p>import org.hibernate.mapping.MetaAttribute;</p><p>import org.apache.log4j.Logger;</p><p>import org.apache.log4j.BasicConfigurator;</p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p><p>public class CustReveng extends DelegatingReverseEngineeringStrategy {</p><p>          private ReverseEngineeringRuntimeInfo RuntimeInfo;</p><p>          private Connection conn;</p><p>          static Logger logger = Logger.getLogger(CustReveng.class);</p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p><p>          public CustReveng(ReverseEngineeringStrategy strategy) {</p><p>                    super(strategy);</p><p>          }</p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p><p>          public Connection getConn() {</p><p>                    return conn;</p><p>          }</p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p><p>          public void setConn(Connection conn) {</p><p>                    this.conn = conn;</p><p>          }</p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p><p>          public void setRuntimeInfo(ReverseEngineeringRuntimeInfo runtimeInfo) {</p><p>                    RuntimeInfo = runtimeInfo;</p><p>          }</p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p><p>          public ReverseEngineeringRuntimeInfo getRuntimeInfo() {</p><p>                    return RuntimeInfo;</p><p>          }</p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p><p>          @Override</p><p>          public void configure(ReverseEngineeringRuntimeInfo runtimeInfo) {</p><p>                    // TODO Auto-generated method stub</p><p>                    super.configure(runtimeInfo);</p><p>                    setRuntimeInfo(runtimeInfo);</p><p>                    try {</p><p>                              setConn(getRuntimeInfo().getConnectionProvider().getConnection());</p><p>                    } catch (SQLException e) {</p><p>                              // TODO Auto-generated catch block</p><p>                              e.printStackTrace();</p><p>                    }</p><p>          }</p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p><p>          public Map<String, MetaAttribute> columnToMetaAttributes(</p><p>                              TableIdentifier identifier, String column) {</p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p><p>                    Map<String, MetaAttribute> result = new HashMap<String, MetaAttribute>();</p><p>                    MetaAttribute ma = new MetaAttribute("default-value");</p><p>                    String defaultValue = getTableColumnDefault(identifier, column);</p><p>                    if (defaultValue != null && !defaultValue.isEmpty()) {</p><p>                              ma.addValue(defaultValue);</p><p>                              result.put("default-value", ma);</p><p>                    }</p><p>                    return result;</p><p>          }</p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p><p>          public String tableToClassName(TableIdentifier tableIdentifier) {</p><p>                    tableIdentifier = removePrefix(tableIdentifier);</p><p>                    return super.tableToClassName(tableIdentifier);</p><p>          }</p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p><p>          private TableIdentifier removePrefix(TableIdentifier tableIdentifier) {</p><p>                    String name = tableIdentifier.getName();</p><p>                    if (name.toLowerCase().startsWith("aaa_")) {</p><p>                              tableIdentifier = new TableIdentifier(tableIdentifier.getSchema(),</p><p>                                                  tableIdentifier.getCatalog(), name.substring(4));</p><p>                    }</p><p>                    return tableIdentifier;</p><p>          }</p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p><p>          public String getTableColumnDefault(TableIdentifier identifier,</p><p>                              String column) {</p><p>                    String xdefault = "";</p><p>                    String xType = "";</p><p>                    String query = "SHOW COLUMNS FROM " + identifier.getName() + " LIKE '"</p><p>                                        + column + "'";</p><p>                    try {</p><p>                              Connection con = getConn();</p><p>                              Statement stmt = con.createStatement();</p><p>                              ResultSet rs = stmt.executeQuery(query);</p><p>                              while (rs.next()) {</p><p>                                        xdefault = rs.getString(5);</p><p>                                        xType = rs.getString(2);</p><p>                              } // end while</p><p>                    } // end try</p><p>                    catch (SQLException e) {</p><p>                              e.printStackTrace();</p><p>                              logger.info("SQL Error: " + e.getMessage());</p><p>                    }</p><p>                    // Code to calculate default value</p><p>                    if (xType.substring(0, Math.min(xType.length(), 4)).equals("char")) {</p><p>                              if (xType.trim().equals("char(1)")) {</p><p>                                        if (xdefault.isEmpty()) {</p><p>                                                  xdefault = "\' \'";</p><p>                                        } else {</p><p>                                                  xdefault = "\'" + xdefault + "\'";</p><p>                                        }</p><p>                              } else {</p><p>                                        xdefault = "\"" + xdefault + "\"";</p><p>                              }</p><p>                    } else if (xType.substring(0, Math.min(xType.length(), 7)).equals(</p><p>                                        "varchar")) {</p><p>                              if (xType.trim().equals("varchar(1)")) {</p><p>                                        xdefault = "\'" + xdefault + "\'";</p><p>                              } else {</p><p>                                        xdefault = "\"" + xdefault + "\"";</p><p>                              }</p><p>                    } else if (xType.trim() == "text") {</p><p>                              xdefault = "\"" + xdefault + "\"";</p><p>                    } else if (xType.trim().equals("timestamp")) {</p><p>                              xdefault = "";</p><p>                    } else if (xdefault != null</p><p>                                        && !xdefault.isEmpty()</p><p>                                        && (xType.trim().equals("date") || xType.trim().equals(</p><p>                                                            "datetime"))) {</p><p>                              xdefault = "new Date(0)";</p><p>                    } else if (xType.substring(0, Math.min(xType.length(), 7)).equals(</p><p>                                        "decimal")) {</p><p>                              if (xdefault != null && xdefault.isEmpty()) {</p><p>                                        xdefault = "new BigDecimal(0)";</p><p>                              } else {</p><p>                                        xdefault = "new BigDecimal(" + xdefault + ")";</p><p>                              }</p><p>                    } else if (xType.substring(0, Math.min(xType.length(), 3))</p><p>                                        .equals("int")</p><p>                                        || xType.substring(0, Math.min(xType.length(), 8)).equals(</p><p>                                                            "smallint")</p><p>                                        || xType.substring(0, Math.min(xType.length(), 7)).equals(</p><p>                                                            "tinyint")) {</p><p>                              xdefault = "0";</p><p>                    } else {</p><p>                              xdefault = "\"ccc\""; //Dummy value for control</p><p>                    }</p><p>  </p><p>                    ///For compositeId's don't need to initialize variables </p><p>                    boolean isPk = false;</p><p>                    try {</p><p>                              ResultSet PKColumns = conn.getMetaData().getPrimaryKeys(</p><p>                                                  conn.getCatalog(), "conta", identifier.getName());</p><p>                              while (PKColumns.next()) {</p><p>                                        if (PKColumns.getString(4).equals(column)) {</p><p>                                                  isPk = true;</p><p>                                        }</p><p>                              }</p><p>                    } catch (SQLException e) {</p><p>                              // TODO Auto-generated catch block</p><p>                              e.printStackTrace();</p><p>                    }</p><p>                    if (isPk) {</p><p>                              xdefault = "";</p><p>                    }</p><p>                    // }</p><p>                    return xdefault;</p><p>          }</p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p><p>          public void finalize() {</p><p>                    try {</p><p>                              getConn().close();</p><p>                    } catch (SQLException e) {</p><p>                              // TODO Auto-generated catch block</p><p>                              e.printStackTrace();</p><p>                    }</p><p>                    try {</p><p>                              super.finalize();</p><p>                    } catch (Throwable e) {</p><p>                              // TODO Auto-generated catch block</p><p>                              e.printStackTrace();</p><p>                    }</p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p><p>          }</p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p><p>}</p><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p></blockquote><p style="min-height: 8pt; height: 8pt; padding: 0px;"> </p><p><strong>Thank you very much for help,</strong></p><p>Alex</p></div>
<div style="background-color: #f4f4f4; padding: 10px; margin-top: 20px;">
<p style="margin: 0;">Reply to this message by <a href="https://community.jboss.org/message/761353#761353">going to Community</a></p>
<p style="margin: 0;">Start a new discussion in JBoss Tools at <a href="https://community.jboss.org/choose-container!input.jspa?contentType=1&containerType=14&container=2128">Community</a></p>
</div></td>
</tr>
</tbody>
</table>
</td>
</tr>
</tbody>
</table>
</div>
</body>
</html>