com.dt.jdbc.parser.UpdateOrInsertParser Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of spring-dt-jdbc Show documentation

A project for Spring JDBC

The newest version!

package com.dt.jdbc.parser;

import java.util.Map;

/**
 * 更新插入数据解析器
 *
 * @author 白超
 * @version 1.0
 * @since 2018/7/10
 */
public class UpdateOrInsertParser {

    public String updateOrInsert(String tableName, Map columnAliasMap, int recordSize) {
        StringBuilder sql = new StringBuilder(64);
        sql.append("insert into ")
                .append(tableName)
                .append(" (");
        int i = 0;
        StringBuilder on = new StringBuilder(64);
        for (Map.Entry entry : columnAliasMap.entrySet()) {
            if (i++ != 0) {
                sql.append(",");
                on.append(",");
            }
            sql.append("`").append(entry.getKey()).append("`");
            on.append("`").append(entry.getKey()).append("` = values(`").append(entry.getKey()).append("`)");
        }
        sql.append(") values ");
        StringBuilder values = new StringBuilder(32).append("(");
        for (; i > 0; i--) {
            if (i == 1) {
                values.append("?)");
            } else {
                values.append("?,");
            }
        }
        for (; recordSize > 0; recordSize--) {
            if (recordSize == 1) {
                sql.append(values.toString());
            } else {
                sql.append(values.toString()).append(",");
            }
        }
        return sql.append(" on duplicate key update ").append(on).toString();
    }

}