scriptella.driver.text.TextConnection Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of scriptella-drivers Show documentation
The newest version!
/*
 * Copyright 2006-2012 The Scriptella Project Team.
 *
 * 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 scriptella.driver.text;

import scriptella.expression.PropertiesSubstitutor;
import scriptella.spi.ConnectionParameters;
import scriptella.spi.ParametersCallback;
import scriptella.spi.ProviderException;
import scriptella.spi.QueryCallback;
import scriptella.spi.Resource;
import scriptella.util.IOUtils;

import java.io.IOException;
import java.io.Reader;

/**
 * Represents a connection to a Text file.
 * For configuration details and examples see overview page.
 *
 * @author Fyodor Kupolov
 * @version 1.0
 */
public class TextConnection extends AbstractTextConnection {
    private TextScriptExecutor out;

    /**
     * For testing purposes only.
     */
    protected TextConnection() {
    }

    public TextConnection(ConnectionParameters parameters) {
        super(Driver.DIALECT, parameters);
    }

    public void executeScript(final Resource scriptContent, final ParametersCallback parametersCallback) throws ProviderException {
        initOut();
        Reader reader = null;
        try {
            reader = scriptContent.open();
            out.execute(reader, parametersCallback, counter);
            if (getConnectionParameters().isFlush()) {
                out.flush();
            }
        } catch (IOException e) {
            throw new TextProviderException("Failed to produce a text file", e);
        } finally {
            IOUtils.closeSilently(reader);
        }
    }

    /**
     * Lazily initializes script writer.
     */
    protected void initOut() {
        if (out == null) {
            try {
                this.out = new TextScriptExecutor(newOutputWriter(), getConnectionParameters());
            } catch (IOException e) {
                throw new TextProviderException("Unable to open file " + getConnectionParameters().getUrl() + " for writing", e);
            }
        }
    }

    public void executeQuery(Resource queryContent, ParametersCallback parametersCallback, QueryCallback queryCallback) throws ProviderException {
        if (out != null) {
            throw new TextProviderException("Cannot query and update a Text file simultaneously");
        }

        Reader in;
        try {
            in = newInputReader();
        } catch (IOException e) {
            throw new TextProviderException("Cannot open a text file for reading", e);
        }
        Reader q;
        try {
            q = queryContent.open();
        } catch (IOException e) {
            throw new TextProviderException("Cannot read a text query", e);
        }

        try {
            new TextQueryExecutor(q, new PropertiesSubstitutor(parametersCallback), getConnectionParameters()).
                    execute(in, queryCallback, counter);
        } finally {
            IOUtils.closeSilently(q);
            IOUtils.closeSilently(in);
        }
    }

    public void close() throws ProviderException {
        IOUtils.closeSilently(out);
        out = null;
    }
}