模仿grails NamingStrategy

MyImprovedNamingStrategy.java

package com.ys.scs.persistence;

import org.hibernate.AssertionFailure;
import org.hibernate.cfg.ImprovedNamingStrategy;
import org.hibernate.cfg.NamingStrategy;
import org.hibernate.internal.util.StringHelper;

/**
 * User: Frank Tang <br/>
 * Date: 15/11/26<br/>
 * Time: 下午6:52<br/>
 * Email: lovefree103@gmail.com<br/>
 */
public class MyImprovedNamingStrategy extends ImprovedNamingStrategy {

    public static final NamingStrategy INSTANCE = new MyImprovedNamingStrategy();

    @Override
    public String columnName(String columnName) {
        String colName = addUnderscores(columnName);
        return colName;
    }

    @Override
    public String propertyToColumnName(String propertyName) {
        propertyName = propertyName.replace("id.","").replace('.', '_');
        return addUnderscores(propertyName);
    }

    @Override
    public String foreignKeyColumnName(String propertyName, String propertyEntityName, String propertyTableName, String referencedColumnName) {
        String header = propertyName != null ? propertyName : propertyTableName;
        if (header == null) throw new AssertionFailure("NamingStrategy not properly filled");
        header = header.replace("id.","");
        header = addUnderscores(header);
        return columnName(header + "_id");
    }

    @Override
    public String logicalColumnName(String columnName, String propertyName) {
        return StringHelper.isNotEmpty(columnName) ? columnName : propertyToColumnName(propertyName);
    }
}