package com.vci.server.base.persistence.order;

import java.sql.Types;

import org.hibernate.Criteria;
import org.hibernate.HibernateException;
import org.hibernate.criterion.CriteriaQuery;
import org.hibernate.criterion.Order;
import org.hibernate.engine.SessionFactoryImplementor;
import org.hibernate.type.Type;
  
public class GBKOrder extends Order {   
    private boolean ascending;   
    private boolean ignoreCase;   
    private String propertyName;   
  
    @Override  
    public String toString() {   
//        return "CONVERT( " + propertyName + " USING " + encoding + " ) " + (ascending ? "asc" : "desc");
        return "nlssort(" + propertyName + ",'NLS_SORT=SCHINESE_PINYIN_M') " + (ascending ? "asc" : "desc");
    }   
  
    @Override  
    public Order ignoreCase() {   
        ignoreCase = true;   
        return this;   
    }   
  
    /**  
     * Constructor for Order.  
     */  
    protected GBKOrder(String propertyName, boolean ascending) {   
        super(propertyName, ascending);   
        this.propertyName = propertyName;   
        this.ascending = ascending;   
    }   
  
    /**  
     * Constructor for Order.  
     */  
    protected GBKOrder(String propertyName, String dir) {   
        super(propertyName, dir.equalsIgnoreCase("ASC") ? true : false);   
        ascending = dir.equalsIgnoreCase("ASC") ? true : false;   
        this.propertyName = propertyName;   
        this.ascending = ascending;   
    }   
  
    /**  
     * Render the SQL fragment  
     *   
     */  
    @Override  
    public String toSqlString(Criteria criteria, CriteriaQuery criteriaQuery) throws HibernateException {   
        String[] columns = criteriaQuery.getColumnsUsingProjection(criteria, propertyName);   
        Type type = criteriaQuery.getTypeUsingProjection(criteria, propertyName);   
        StringBuffer fragment = new StringBuffer();   
        for (int i = 0; i < columns.length; i++) {   
            SessionFactoryImplementor factory = criteriaQuery.getFactory();   
            boolean lower = ignoreCase && type.sqlTypes(factory)[i] == Types.VARCHAR;   
            if (lower) {   
                fragment.append(factory.getDialect().getLowercaseFunction()).append('(');   
            }   
//            fragment.append("CONVERT( " + columns[i] + " USING " + encoding + " )"); 
            fragment.append("nlssort(" + columns[i] + ",'NLS_SORT=SCHINESE_PINYIN_M')");
            if (lower)   
                fragment.append(')');   
            fragment.append(ascending ? " asc" : " desc");   
            if (i < columns.length - 1)   
                fragment.append(", ");   
        }   
        return fragment.toString();   
    }   
  
    /**  
     * Ascending order  
     *   
     * @param propertyName  
     * @return Order  
     */  
    public static Order asc(String propertyName) {   
        return new GBKOrder(propertyName, true);   
    }   
  
    /**  
     * Descending order  
     *   
     * @param propertyName  
     * @return Order  
     */  
    public static Order desc(String propertyName) {   
        return new GBKOrder(propertyName, false);   
    }   
}