maj sqlcsv

This commit is contained in:
Jephté Clain 2017-09-07 15:32:23 +04:00
parent f05382cdac
commit 1aee0cb4e6
1 changed files with 51 additions and 9 deletions

60
sqlcsv
View File

@ -49,6 +49,7 @@ import java.util.Properties;
import java.util.logging.ConsoleHandler;
import java.util.logging.Level;
import java.util.logging.Logger;
import java.util.prefs.Preferences;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
@ -572,6 +573,10 @@ public class sqlcsv {
"org.hsqldb.jdbcDriver",
"org.postgresql.Driver"};
static final boolean isOracle(String jdbcUrl) {
return jdbcUrl.startsWith("jdbc:oracle:");
}
static final class Exit extends RuntimeException {
private static final long serialVersionUID = 1L;
@ -789,14 +794,25 @@ public class sqlcsv {
}
}
// requêtes à ignorer, en l'occurence lignes spéciques à sqlplus pour une connexion à Oracle
static final Pattern RE_ORACLE_IGNORE_STMT = Pattern.compile("(?i)set\\b");
static final Pattern RE_CREATE_STMT = Pattern
.compile("(?i)create\\s+(?:or\\s+replace)?(?:procedure|trigger|function|package)");
static final Pattern RE_SEPARATOR = Pattern.compile(";$");
static String nextStatement(BufferedReader inf) throws IOException {
String line = nextLine(inf);
if (line == null) return null;
static String nextStatement(BufferedReader inf, String jdbcUrl) throws IOException {
boolean ignore;
String line;
do {
ignore = false;
line = nextLine(inf);
if (line == null) return null;
if (isOracle(jdbcUrl) && RE_ORACLE_IGNORE_STMT.matcher(line).lookingAt()) {
ignore = true;
}
} while (ignore);
StringBuilder sb = new StringBuilder();
boolean first = true;
@ -889,7 +905,7 @@ public class sqlcsv {
static final Pattern RE_NAME_EXT = Pattern.compile("(.+?)(?:(\\d+))?(?:(\\.[^.]+))?");
void executeQueries(Connection conn, Object[] args, ArrayList<BufferedReader> infs,
ArrayList<String> outputs, Options o) throws IOException, SQLException {
ArrayList<String> outputs, Options o, String jdbcUrl) throws IOException, SQLException {
int stcount = 0;
Iterator<String> outputIt = outputs.iterator();
String output = null, prevOutput = null, actualOutput = null;
@ -898,7 +914,7 @@ public class sqlcsv {
for (BufferedReader inf : infs) {
try {
while (true) {
String statement = nextStatement(inf);
String statement = nextStatement(inf, jdbcUrl);
if (statement == null) break;
stcount++;
@ -1052,6 +1068,11 @@ public class sqlcsv {
+ "\n obligatoire. Si cette option n'est pas fournie, il faut spécifier un"
+ "\n fichier de configuration avec l'option -C dans lequel *une seule*"
+ "\n propriété 'CONN.url' est définie."
+ "\n Si -C n'est pas spécifié et que -l est spécifié, alors l'identifiant de"
+ "\n connexion est enregistré dans les préférences utilisateur. Si ni -C"
+ "\n ni -l ne sont spécifiés, alors le dernier identifiant est utilisé s'il"
+ "\n existe. L'option -z permet de démarrer comme si aucun identifiant n'avait"
+ "\n jamais été enregistré."
+ "\n -u, --user USER"
+ "\n -p, --password PASSWORD"
+ "\n Spécifier un nom de connexion et un mot de passe si l'url ne le fournit"
@ -1106,13 +1127,18 @@ public class sqlcsv {
+ "\n l'application avec le niveau de log ALL ce qui permet de voir les logs"
+ "\n concernant le chargement des jar. Un fichier SQL_DEBUG permet d'activer"
+ "\n la trace de DriverManager. Exemple:"
+ "\n mkdir -p ~/.sqlcsv && touch ~/.sqlcsv/{DEBUG,SQL_DEBUG}");
+ "\n mkdir -p ~/.sqlcsv && touch ~/.sqlcsv/{DEBUG,SQL_DEBUG}"
+ "\n -v, --verbose"
+ "\n Equivalent à --loglevel FINER pour afficher le maximum de messages"
+ "\n -z, --reset-prefs"
+ "\n Effacer les préférences avant de faire le calcul de l'identifiant par"
+ "\n défaut");
return;
}
String config = null, connid = null, user = null, password = null;
Options o = new Options();
boolean autocommit = false, autoNa = false;
boolean autocommit = false, autoNa = false, resetPrefs = false;
String loglevel = null;
ArrayList<String> inputs = new ArrayList<String>();
ArrayList<String> outputs = new ArrayList<String>();
@ -1156,6 +1182,10 @@ public class sqlcsv {
o.outputUc = true;
} else if (arg.equals("--loglevel")) {
loglevel = getArg(args, ++i, "Vous devez spécifier le niveau de logs");
} else if (arg.equals("-v") || arg.equals("--verbose")) {
loglevel = "FINER";
} else if (arg.equals("-z") || arg.equals("--reset-prefs")) {
resetPrefs = true;
} else if (arg.equals("--")) {
i++;
break;
@ -1170,6 +1200,14 @@ public class sqlcsv {
log.setLevel(Level.parse(loglevel.toUpperCase()));
}
Preferences prefs = null;
boolean usePrefs = config == null;
if (usePrefs || resetPrefs) {
prefs = Preferences.userNodeForPackage(sqlcsv.class);
if (resetPrefs) prefs.clear();
if (usePrefs && connid == null) connid = prefs.get("lastConnid", null);
}
// Charger les propriétés...
// essayer depuis le répertoire courant et les répertoires parents jusqu'à $HOME
if (config == null) {
@ -1252,7 +1290,11 @@ public class sqlcsv {
jdbcUrl = connid;
}
}
if (jdbcUrl == null) throw new Exit("Vous devez spécifier l'url de connexion");
if (jdbcUrl == null) {
if (resetPrefs) throw new Exit();
else throw new Exit("Vous devez spécifier l'url de connexion");
}
if (usePrefs) prefs.put("lastConnid", connid);
// Ouvrir les fichiers en entrée
ArrayList<BufferedReader> infs = new ArrayList<BufferedReader>();
@ -1298,7 +1340,7 @@ public class sqlcsv {
// Exécuter les requêtes
try {
executeQueries(conn, null, infs, outputs, o);
executeQueries(conn, null, infs, outputs, o, jdbcUrl);
} finally {
close(conn);
for (BufferedReader inf : infs) {