pt/jrunner
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 5 Wiki Activity

Compare commits

base: pt:feature/query-mode
Branches Tags
pt:master
pt:feature/query-mode
pt:rename-app-to-jrunner
pt:testread
pt:v1.0
pt:v0.41
pt:v0.40
pt:v0.38
pt:v0.36
pt:v0.35
pt:v0.34
pt:v0.33
pt:v0.32
pt:v0.31
pt:v0.30
pt:v0.29
pt:v0.28
..
compare: pt:master
Branches Tags
pt:feature/query-mode
pt:master
pt:rename-app-to-jrunner
pt:testread
pt:v1.0
pt:v0.41
pt:v0.40
pt:v0.38
pt:v0.36
pt:v0.35
pt:v0.34
pt:v0.33
pt:v0.32
pt:v0.31
pt:v0.30
pt:v0.29
pt:v0.28

No commits in common. "feature/query-mode" and "master" have entirely different histories.
feature/qu ... master

1 changed files with 32 additions and 148 deletions
Show Stats Expand all files Collapse all files

118
jrunner/src/main/java/jrunner/jrunner.java
View File

@ -27,8 +27,6 @@ public class jrunner {
String nr = ""; String nr = "";
String nc = ""; String nc = "";
String nl = "\n"; String nl = "\n";
Boolean queryMode = false;
String outputFormat = "csv";
String msg = ""; String msg = "";
Connection scon = null; Connection scon = null;
Connection dcon = null; Connection dcon = null;
@ -52,7 +50,6 @@ public class jrunner {
msg = msg + nl + "-dt fully qualified name of destination table"; msg = msg + nl + "-dt fully qualified name of destination table";
msg = msg + nl + "-t trim text"; msg = msg + nl + "-t trim text";
msg = msg + nl + "-c clear target table"; msg = msg + nl + "-c clear target table";
msg = msg + nl + "-f output format (csv, tsv, table, json) - default: csv";
msg = msg + nl + "--help info"; msg = msg + nl + "--help info";
//---------------------------------------parse args into variables------------------------------------------------- //---------------------------------------parse args into variables-------------------------------------------------
@ -107,9 +104,6 @@ public class jrunner {
case "-c": case "-c":
clear = true; clear = true;
break; break;
case "-f":
outputFormat = args[i+1].toLowerCase();
break;
case "-v": case "-v":
System.out.println(msg); System.out.println(msg);
return; return;
@ -133,15 +127,6 @@ public class jrunner {
} }
} }
// Detect query mode when destination flags are not provided
queryMode = dcu.isEmpty() && dcn.isEmpty() && dcp.isEmpty() && dt.isEmpty();
if (queryMode) {
System.err.println("------------query mode------------------------------------");
System.err.println("source db uri: " + scu);
System.err.println("source db username: " + scn);
System.err.println("output format: " + outputFormat);
} else {
System.out.println("------------db info---------------------------------------"); System.out.println("------------db info---------------------------------------");
System.out.println("source db uri: " + scu); System.out.println("source db uri: " + scu);
System.out.println("source db username: " + scn); System.out.println("source db username: " + scn);
@ -151,7 +136,6 @@ public class jrunner {
System.out.println(sq); System.out.println(sq);
System.out.println("------------destination table-----------------------------"); System.out.println("------------destination table-----------------------------");
System.out.println(dt); System.out.println(dt);
}
//return; //return;
@ -165,11 +149,7 @@ public class jrunner {
//-------------------------------------------establish connections------------------------------------------------- //-------------------------------------------establish connections-------------------------------------------------
//source database //source database
if (queryMode) {
System.err.println("------------open database connection---------------------");
} else {
System.out.println("------------open database connections---------------------"); System.out.println("------------open database connections---------------------");
}
try { try {
scon = DriverManager.getConnection(scu, scn, scp); scon = DriverManager.getConnection(scu, scn, scp);
} catch (SQLException e) { } catch (SQLException e) {
@ -179,7 +159,6 @@ public class jrunner {
} }
System.out.println(" ✅ source database"); System.out.println(" ✅ source database");
//destination database //destination database
if (!queryMode) {
try { try {
dcon = DriverManager.getConnection(dcu, dcn, dcp); dcon = DriverManager.getConnection(dcu, dcn, dcp);
} catch (SQLException e) { } catch (SQLException e) {
@ -188,7 +167,6 @@ public class jrunner {
System.exit(0); System.exit(0);
} }
System.out.println(" ✅ destination database"); System.out.println(" ✅ destination database");
}
//----------------------------------------open resultset------------------------------------------------------------ //----------------------------------------open resultset------------------------------------------------------------
try { try {
stmt = scon.createStatement(); stmt = scon.createStatement();
@ -226,7 +204,7 @@ public class jrunner {
System.exit(0); System.exit(0);
} }
//-------------------------clear the target table if requeted---------------------------------------------------- //-------------------------clear the target table if requeted----------------------------------------------------
if (!queryMode && clear) { if (clear) {
System.out.println("------------clear target table----------------------------"); System.out.println("------------clear target table----------------------------");
sql = "DELETE FROM " + dt; sql = "DELETE FROM " + dt;
try { try {
@ -239,15 +217,6 @@ public class jrunner {
System.exit(0); System.exit(0);
} }
} }
if (queryMode) {
//-------------------------------output query results--------------------------------------------------------
try {
outputQueryResults(rs, cols, dtn, outputFormat);
} catch (SQLException e) {
e.printStackTrace();
System.exit(0);
}
} else {
System.out.println("------------row count-------------------------------------"); System.out.println("------------row count-------------------------------------");
//-------------------------------build & execute sql------------------------------------------------------------- //-------------------------------build & execute sql-------------------------------------------------------------
try { try {
@ -385,107 +354,22 @@ public class jrunner {
e.printStackTrace(); e.printStackTrace();
System.exit(0); System.exit(0);
} }
}
//System.out.println(sql); //System.out.println(sql);
//---------------------------------------close connections-------------------------------------------------------- //---------------------------------------close connections--------------------------------------------------------
try { try {
scon.close(); scon.close();
if (!queryMode && dcon != null) {
dcon.close(); dcon.close();
}
} catch (SQLException e) { } catch (SQLException e) {
System.out.println("issue closing connections"); System.out.println("issue closing connections");
e.printStackTrace(); e.printStackTrace();
} }
if (!queryMode) {
System.out.println(" rows written"); System.out.println(" rows written");
tsEnd = Timestamp.from(Instant.now()); tsEnd = Timestamp.from(Instant.now());
System.out.println(tsStart); System.out.println(tsStart);
System.out.println(tsEnd); System.out.println(tsEnd);
}
//long time = Duration.between(tsStart, tsEnd).toMillis(); //long time = Duration.between(tsStart, tsEnd).toMillis();
//System.out.println("time elapsed: " + time); //System.out.println("time elapsed: " + time);
System.out.println(); System.out.println();
} }
private static void outputQueryResults(ResultSet rs, int cols, String[] dtn, String format) throws SQLException {
switch (format) {
case "csv":
outputCSV(rs, cols);
break;
case "tsv":
outputTSV(rs, cols);
break;
case "table":
System.err.println("Table format not yet implemented, defaulting to CSV");
outputCSV(rs, cols);
break;
case "json":
System.err.println("JSON format not yet implemented, defaulting to CSV");
outputCSV(rs, cols);
break;
default:
System.err.println("Unknown format: " + format + ", defaulting to CSV");
outputCSV(rs, cols);
break;
}
}
private static void outputCSV(ResultSet rs, int cols) throws SQLException {
// Print header row
for (int i = 1; i <= cols; i++) {
if (i > 1) System.out.print(",");
System.out.print(escapeCSV(rs.getMetaData().getColumnName(i)));
}
System.out.println();
// Print data rows
while (rs.next()) {
for (int i = 1; i <= cols; i++) {
if (i > 1) System.out.print(",");
String value = rs.getString(i);
if (rs.wasNull() || value == null) {
// Empty field for NULL
} else {
System.out.print(escapeCSV(value));
}
}
System.out.println();
}
}
private static void outputTSV(ResultSet rs, int cols) throws SQLException {
// Print header row
for (int i = 1; i <= cols; i++) {
if (i > 1) System.out.print("\t");
System.out.print(escapeTSV(rs.getMetaData().getColumnName(i)));
}
System.out.println();
// Print data rows
while (rs.next()) {
for (int i = 1; i <= cols; i++) {
if (i > 1) System.out.print("\t");
String value = rs.getString(i);
if (rs.wasNull() || value == null) {
// Empty field for NULL
} else {
System.out.print(escapeTSV(value));
}
}
System.out.println();
}
}
private static String escapeCSV(String value) {
if (value.contains(",") || value.contains("\"") || value.contains("\n") || value.contains("\r")) {
return "\"" + value.replaceAll("\"", "\"\"") + "\"";
}
return value;
}
private static String escapeTSV(String value) {
return value.replaceAll("\t", " ").replaceAll("\n", " ").replaceAll("\r", " ");
}
} }