cherry.foundation.sql.SqlExecutorImpl.java Source code

Introduction

Here is the source code for cherry.foundation.sql.SqlExecutorImpl.java

Source

/*
 * Copyright 2012,2015 agwlvssainokuni
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package cherry.foundation.sql;

import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.Reader;
import java.nio.charset.Charset;
import java.util.Map;

import javax.sql.DataSource;

import org.springframework.core.io.Resource;
import org.springframework.dao.DataAccessException;
import org.springframework.jdbc.core.namedparam.NamedParameterJdbcTemplate;

import cherry.goods.sql.SimpleSqlParser;

/**
 * SQL
 */
public class SqlExecutorImpl implements SqlExecutor {

    private Charset charset;

    public void setCharset(Charset charset) {
        this.charset = charset;
    }

    /**
     * SQL?
     * 
     * @param dataSource 
     * @param resource SQL??
     * @param paramMap SQL???
     * @param continueOnError SQL?????
     * @throws IOException SQL???
     */
    @Override
    public void execute(DataSource dataSource, Resource resource, Map<String, ?> paramMap, boolean continueOnError)
            throws IOException {
        try (InputStream in = resource.getInputStream()) {
            execute(dataSource, in, paramMap, continueOnError);
        }
    }

    /**
     * SQL?
     * 
     * @param dataSource 
     * @param in SQL??
     * @param paramMap SQL???
     * @param continueOnError SQL?????
     * @throws IOException SQL???
     */
    @Override
    public void execute(DataSource dataSource, InputStream in, Map<String, ?> paramMap, boolean continueOnError)
            throws IOException {
        try (Reader reader = new InputStreamReader(in, charset)) {
            execute(dataSource, reader, paramMap, continueOnError);
        }
    }

    /**
     * SQL?
     * 
     * @param dataSource 
     * @param reader SQL??
     * @param paramMap SQL???
     * @param continueOnError SQL?????
     * @throws IOException SQL???
     */
    @Override
    public void execute(DataSource dataSource, Reader reader, Map<String, ?> paramMap, boolean continueOnError)
            throws IOException {

        NamedParameterJdbcTemplate template = new NamedParameterJdbcTemplate(dataSource);

        String sql;
        while ((sql = SimpleSqlParser.nextStatement(reader)) != null) {

            sql = sql.trim();
            if (sql.isEmpty()) {
                continue;
            }

            try {
                template.update(sql, paramMap);
            } catch (DataAccessException ex) {
                if (!continueOnError) {
                    throw ex;
                }
            }
        }
    }

}