cn.mybatisboost.util.SqlUtils Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of mybatis-boost-core Show documentation
The newest version!
package cn.mybatisboost.util;

import cn.mybatisboost.util.tuple.BinaryTuple;
import org.apache.commons.lang3.StringUtils;
import org.apache.ibatis.session.RowBounds;

import java.util.Arrays;
import java.util.List;
import java.util.Objects;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import java.util.stream.Collectors;
import java.util.stream.Stream;

public abstract class SqlUtils {

    public static final Pattern PATTERN_PLACEHOLDER = Pattern.compile("(? placeholderCountCache = new ConcurrentHashMap<>();

    public static StringBuilder appendWhere(StringBuilder sqlBuilder, Stream stream) {
        sqlBuilder.append(" WHERE ");
        stream.forEach(c -> sqlBuilder.append(c).append(" = ? AND "));
        sqlBuilder.setLength(sqlBuilder.length() - 5);
        return sqlBuilder;
    }

    public static String appendLimit(String sql, RowBounds rowBounds) {
        if (rowBounds.getOffset() != RowBounds.NO_ROW_OFFSET || rowBounds.getLimit() != RowBounds.NO_ROW_LIMIT) {
            sql += " LIMIT " + rowBounds.getOffset() + ", " + rowBounds.getLimit();
        }
        return sql;
    }

    public static String appendLimitOffset(String sql, RowBounds rowBounds) {
        if (rowBounds.getOffset() != RowBounds.NO_ROW_OFFSET || rowBounds.getLimit() != RowBounds.NO_ROW_LIMIT) {
            sql += " LIMIT " + rowBounds.getLimit() + " OFFSET " + rowBounds.getOffset();
        }
        return sql;
    }

    public static int countPlaceholders(String sql) {
        return placeholderCountCache.computeIfAbsent(sql, k -> {
            int count = 0;
            Matcher matcher = PATTERN_PLACEHOLDER.matcher(sql);
            while (matcher.find()) count++;
            return count;
        });
    }

    public static String normalizeColumn(String column, boolean mapUnderscoreToCamelCase) {
        Stream stream = Arrays.stream(StringUtils.splitByCharacterTypeCamelCase(column))
                .filter(it -> !Objects.equals(it, "_"));
        if (mapUnderscoreToCamelCase) {
            return stream.map(StringUtils::uncapitalize).collect(Collectors.joining("_"));
        } else {
            return stream.map(StringUtils::capitalize).collect(Collectors.joining());
        }
    }

    public static BinaryTuple, List> getPropertiesAndColumnsFromLiteralColumns
            (String literalColumns, Class entityType, boolean mapUnderscoreToCamelCase) {
        List properties, columns;
        if (Objects.equals(literalColumns, "*") || literalColumns.toUpperCase().startsWith("NOT ")) {
            properties = EntityUtils.getProperties(entityType);
            if (literalColumns.toUpperCase().startsWith("NOT ")) {
                properties.removeAll
                        (Arrays.stream(literalColumns.substring(4).split(","))
                                .map(String::trim).map(PropertyUtils::normalizeProperty)
                                .collect(Collectors.toList()));
            }
            columns = properties.stream()
                    .map(it -> SqlUtils.normalizeColumn(it, mapUnderscoreToCamelCase)).collect(Collectors.toList());
        } else {
            columns = Arrays.stream(literalColumns.split(",")).map(String::trim)
                    .map(it -> SqlUtils.normalizeColumn(it, mapUnderscoreToCamelCase)).collect(Collectors.toList());
            properties = columns.stream().map(PropertyUtils::normalizeProperty).collect(Collectors.toList());
        }
        return new BinaryTuple<>(properties, columns);
    }
}