Sha256: d18d4630e28d58ff0ea63b5d5744430b2003484de3e4887393115cc5ce47b7f6
Contents?: true
Size: 1.52 KB
Versions: 10
Compression:
Stored size: 1.52 KB
Contents
package org.embulk.input.redshift;
import com.google.common.base.Throwables;
import com.google.common.io.ByteStreams;
import org.embulk.config.ConfigSource;
import org.embulk.test.EmbulkTests;
import java.io.IOException;
import static java.util.Locale.ENGLISH;
public class RedshiftTests
{
public static ConfigSource baseConfig()
{
return EmbulkTests.config("EMBULK_INPUT_REDSHIFT_TEST_CONFIG");
}
public static void execute(String sql)
{
ConfigSource config = baseConfig();
ProcessBuilder pb = new ProcessBuilder("psql", "-w", "--set", "ON_ERROR_STOP=1", "-c", sql);
pb.environment().put("PGUSER", config.get(String.class, "user"));
pb.environment().put("PGPASSWORD", config.get(String.class, "password"));
pb.environment().put("PGDATABASE", config.get(String.class, "database"));
pb.environment().put("PGHOST", config.get(String.class, "host", "localhost"));
pb.environment().put("PGPORT", config.get(String.class, "port", "5439"));
pb.redirectErrorStream(true);
int code;
try {
Process process = pb.start();
ByteStreams.copy(process.getInputStream(), System.out);
code = process.waitFor();
} catch (IOException | InterruptedException ex) {
throw Throwables.propagate(ex);
}
if (code != 0) {
throw new RuntimeException(String.format(ENGLISH,
"Command finished with non-zero exit code. Exit code is %d.", code));
}
}
}
Version data entries
10 entries across 10 versions & 1 rubygems