io.github.jinghui70.rainbow.dbaccess.NamedSql Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of rainbow-dbaccess Show documentation
Rainbow DBAccess
The newest version!
package io.github.jinghui70.rainbow.dbaccess;

import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.map.MapUtil;
import io.github.jinghui70.rainbow.dbaccess.cnd.Cnd;
import org.springframework.dao.DataAccessException;
import org.springframework.dao.EmptyResultDataAccessException;
import org.springframework.jdbc.core.RowCallbackHandler;
import org.springframework.jdbc.core.RowMapper;
import org.springframework.jdbc.core.namedparam.MapSqlParameterSource;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

import static io.github.jinghui70.rainbow.dbaccess.DbaUtil.enumCheck;

/**
 * 封装了一个Sql的内容对象
 *
 * @author lijinghui
 */
public class NamedSql extends SqlWrapper {

    private final Map params = new HashMap<>();

    public NamedSql() {
        super();
    }

    public NamedSql(String str) {
        super(str);
    }

    public NamedSql(Dba dao) {
        super(dao);
    }

    public NamedSql append(NamedSql sql) {
        append(sql.getSql());
        this.params.putAll(sql.getParams());
        return this;
    }

    @Override
    public NamedSql append(Cnd cnd) {
        cnd.toNamedSql(this);
        return this;
    }

    @Override
    public NamedSql set(String key) {
        if (key.contains("=:")) return super.set(key);
        set();
        append(key).append("=:").append(key);
        return this;
    }

    @Override
    public NamedSql set(String key, Object value) {
        set(key);
        return setParam(key, value);
    }

    public Map getParams() {
        return params;
    }

    public NamedSql setParam(String key, Object value) {
        this.params.put(key, enumCheck(value));
        return this;
    }

    public NamedSql clearParam() {
        this.params.clear();
        return this;
    }

    public NamedSql setParam(Map params) {
        if (MapUtil.isNotEmpty(params)) {
            for (Map.Entry entry : params.entrySet()) {
                this.params.put(entry.getKey(), enumCheck(entry.getValue()));
            }
        }
        return this;
    }

    public NamedSql resetParams(Map params) {
        return clearParam().setParam(params);
    }

    /**
     * 执行当前sql
     *
     * @return 影响的行数
     */
    @Override
    public int execute() {
        return dba.getNamedParameterJdbcTemplate().update(getSql(), params);
    }

    @Override
    public void query(RowCallbackHandler rch) {
        dba.getNamedParameterJdbcTemplate().query(getSql(), params, rch);
    }

    @Override
    public  T queryForObject(String sql, RowMapper mapper) throws DataAccessException {
        try {
            return dba.getNamedParameterJdbcTemplate().queryForObject(sql, params, mapper);
        } catch (EmptyResultDataAccessException e) {
            return null;
        }
    }

    @Override
    protected  List queryForList(String sql, RowMapper rowMapper) throws DataAccessException {
        return dba.getNamedParameterJdbcTemplate().query(sql, params, rowMapper);
    }

    public int[] batchUpdate(List> data) {
        MapSqlParameterSource[] sources = data.stream().map(MapSqlParameterSource::new)
                .toArray(MapSqlParameterSource[]::new);
        if (CollUtil.isNotEmpty(getParams())) {
            for (MapSqlParameterSource s : sources)
                s.addValues(getParams());
        }
        return dba.getNamedParameterJdbcTemplate().batchUpdate(getSql(), sources);
    }

}