jsqlparser(四):实现MySQL 函数DATE_FORMAT到Phoenix函数TO_CHAR的替换

10km

已于 2024-01-05 17:01:00 修改

阅读量635

点赞数

分类专栏： java 文章标签： mysql phoenix jsqlparser DATE_FORMAT TO_CHAR

于 2022-12-03 14:53:56 首次发布

本文链接：https://blog.csdn.net/10km/article/details/128161556

版权

java 专栏收录该内容

229 篇文章 4 订阅

订阅专栏

上一篇博客《jsqlparser:基于语法分析实现SQL中的CAST函数替换》中我用jsqlparser这个神器实现了 TO_DATE(TO_CHAR("local_time"), 'yyyy-MM-dd')替换CAST(local_time AS DATE)。
现在又遇到新问题了，
MySQL中日期格式化函数为DATE_FORMAT,对应Phoenix中的函数为TO_CHAR,如果应用层使用DATE_FORMAT函数，也要替换为TO_CHAR才能在HBase数据库中正常执行。有了上次经验，这次就简单多了，
不同的是DATE_FORMAT的日期格式定义与Phoenix不同，所以DATE_FORMAT的format参数也需要自动替换，但是只能不完全替换。

以下为实现代码

import net.sf.jsqlparser.expression.Expression;
import net.sf.jsqlparser.expression.Function;
import net.sf.jsqlparser.expression.StringValue;
import net.sf.jsqlparser.util.TablesNamesFinder;

import java.util.List;
/**
 * 基于SQL语法对象实现对SQL的修改<br>
 * 将MySQL函数DATE_FORMAT替换为Phoenix的函数TO_CHAR
 * @author guyadong
 *
 */
public class PhoenixNormalizer  extends TablesNamesFinder{
    public PhoenixNormalizer() {
	}
	
    /**
     * 将MySQL的 DATE_FORMAT格式转为Java 日期格式<br>     
     * 参见 <a href="https://dev.mysql.com/doc/refman/5.7/en/date-and-time-functions.html#function_date-format">function_date-format</a>,
     * 因为不能实现完全转换，所以当有不支持的格式时返回原字符串 
     * @param format
     * @see java.text.SimpleDateFormat
     */
    String castFormat(String format){
        if(null != format){
            String fmt=format.replace("%Y", "yyyy").replace("%y", "yy").replace("%m", "MM").replaceAll("%[de]", "dd")
            .replaceAll("%[Hk]", "HH").replace("%i", "mm").replaceAll("%[sS]", "ss")
            .replace("%W", "E")
            .replace("%j", "D")
            .replace("%T", "hh:mm:ss")
            .replace("%r", "hh:mm:ss a")
            .replaceAll("%[Uu]", "w")
            .replace("%%", "%");
            // 当有不支持的格式时返回原字符串 
            if(fmt.indexOf('%') < 0){
                return fmt;
            }
        }
        return format;
    }
    
    @Override
    public void visit(Function function){
        if("DATE_FORMAT".equalsIgnoreCase(function.getName())){
            /** DATE_FORMAT 改为 TO_CHAR */
            List<Expression> expressions = function.getParameters().getExpressions();
            if(null != expressions && 2 == expressions.size() ){
            	/** 修改函数名 */
                function.setName("TO_CHAR");
                Expression formatExp = expressions.get(1);
                if(formatExp instanceof StringValue){
                	/** 修改第二个参数(format) */
                    StringValue stringValue = (StringValue)formatExp;
                    stringValue.setValue(castFormat(stringValue.getValue()));
                }
            }
            
        }
        super.visit(function);
    }
}