src/com/xx/supermarket/pagehelper/SqlParser.java

这里只显示每个文件的开头 60 行。登录后可以解锁完整代码。

微信扫码关注公众号，代码整理好后第一时间通知你。

package com.xx.supermarket.pagehelper;

import net.sf.jsqlparser.JSQLParserException;
import net.sf.jsqlparser.expression.Alias;
import net.sf.jsqlparser.parser.CCJSqlParserUtil;
import net.sf.jsqlparser.schema.Column;
import net.sf.jsqlparser.statement.Statement;
import net.sf.jsqlparser.statement.select.*;
import org.apache.ibatis.mapping.BoundSql;
import org.apache.ibatis.mapping.MappedStatement;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

/**
 * sql解析类，使用该类时，必须引入jsqlparser-x.x.x.jar
 *
 * @author liuzh
 */
@SuppressWarnings("rawtypes")
public class SqlParser implements SqlUtil.Parser {
    private static final List<SelectItem> COUNT_ITEM;
    private static final Alias TABLE_ALIAS;
    private SqlUtil.Parser simpleParser;

    static {
        COUNT_ITEM = new ArrayList<SelectItem>();
        COUNT_ITEM.add(new SelectExpressionItem(new Column("count(*)")));

        TABLE_ALIAS = new Alias("table_count");
        TABLE_ALIAS.setUseAs(false);
    }

    //缓存已经修改过的sql
    private Map<String, String> CACHE = new ConcurrentHashMap<String, String>();

    public SqlParser(SqlUtil.Dialect dialect) {
        simpleParser = SqlUtil.SimpleParser.newParser(dialect);
    }

    public void isSupportedSql(String sql) {
        simpleParser.isSupportedSql(sql);
    }

    public String getCountSql(String sql) {
        //校验是否支持该sql
        isSupportedSql(sql);
        return parse(sql);
    }

    public String getPageSql(String sql) {
        return simpleParser.getPageSql(sql);
    }

    public Map setPageParameter(MappedStatement ms, Object parameterObject, BoundSql boundSql, Page page) {
        return simpleParser.setPageParameter(ms, parameterObject, boundSql, page);
    }