Mercurial > hg > monetdb-java
annotate src/main/java/org/monetdb/client/JdbcClient.java @ 833:e890195256ac
Update copyright for the new year, move to MonetDB Foundation, add SPDX.
author | Sjoerd Mullender <sjoerd@acm.org> |
---|---|
date | Fri, 29 Dec 2023 14:37:42 +0100 (15 months ago) |
parents | 1074ad1a4c63 |
children | d9a45743536d |
rev | line source |
---|---|
675
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1 /* |
833
e890195256ac
Update copyright for the new year, move to MonetDB Foundation, add SPDX.
Sjoerd Mullender <sjoerd@acm.org>
parents:
826
diff
changeset
|
2 * SPDX-License-Identifier: MPL-2.0 |
e890195256ac
Update copyright for the new year, move to MonetDB Foundation, add SPDX.
Sjoerd Mullender <sjoerd@acm.org>
parents:
826
diff
changeset
|
3 * |
675
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
4 * This Source Code Form is subject to the terms of the Mozilla Public |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
5 * License, v. 2.0. If a copy of the MPL was not distributed with this |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
6 * file, You can obtain one at http://mozilla.org/MPL/2.0/. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
7 * |
833
e890195256ac
Update copyright for the new year, move to MonetDB Foundation, add SPDX.
Sjoerd Mullender <sjoerd@acm.org>
parents:
826
diff
changeset
|
8 * Copyright 2024 MonetDB Foundation; |
e890195256ac
Update copyright for the new year, move to MonetDB Foundation, add SPDX.
Sjoerd Mullender <sjoerd@acm.org>
parents:
826
diff
changeset
|
9 * Copyright August 2008 - 2023 MonetDB B.V.; |
e890195256ac
Update copyright for the new year, move to MonetDB Foundation, add SPDX.
Sjoerd Mullender <sjoerd@acm.org>
parents:
826
diff
changeset
|
10 * Copyright 1997 - July 2008 CWI. |
675
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
11 */ |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
12 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
13 package org.monetdb.client; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
14 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
15 import org.monetdb.jdbc.MonetDriver; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
16 import org.monetdb.jdbc.MonetConnection; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
17 import org.monetdb.util.CmdLineOpts; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
18 import org.monetdb.util.Exporter; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
19 import org.monetdb.util.FileTransferHandler; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
20 import org.monetdb.util.MDBvalidator; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
21 import org.monetdb.util.OptionsException; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
22 import org.monetdb.util.SQLExporter; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
23 import org.monetdb.util.XMLExporter; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
24 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
25 import java.io.BufferedReader; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
26 import java.io.BufferedWriter; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
27 import java.io.Console; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
28 import java.io.IOException; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
29 import java.io.InputStreamReader; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
30 import java.io.File; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
31 import java.io.PrintWriter; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
32 import java.net.HttpURLConnection; |
770
1b9fe515f574
Fixed javac version 20 compiler warning.
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
768
diff
changeset
|
33 import java.net.URI; |
675
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
34 import java.net.URL; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
35 import java.nio.charset.Charset; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
36 import java.sql.Connection; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
37 import java.sql.DatabaseMetaData; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
38 import java.sql.DriverManager; // this import is required as it will trigger loading the org.monetdb.jdbc.MonetDriver class |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
39 import java.sql.ResultSet; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
40 import java.sql.Statement; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
41 import java.sql.SQLException; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
42 import java.sql.SQLWarning; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
43 import java.util.ArrayList; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
44 import java.util.LinkedList; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
45 import java.util.List; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
46 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
47 /** |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
48 * This program acts like an extended client program for MonetDB. Its |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
49 * look and feel is very much like PostgreSQL's interactive terminal |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
50 * program. Although it looks like this client is designed for MonetDB, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
51 * it demonstrates the power of the JDBC interface since it built on top |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
52 * of JDBC only. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
53 * |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
54 * @author Fabian Groffen |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
55 * @author Martin van Dinther |
768
a80c21fe7bb2
Removed deprecated nl.cwi.monetdb.*.* classes and package.
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
723
diff
changeset
|
56 * @version 1.8 |
675
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
57 */ |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
58 |
768
a80c21fe7bb2
Removed deprecated nl.cwi.monetdb.*.* classes and package.
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
723
diff
changeset
|
59 public final class JdbcClient { |
675
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
60 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
61 private static Connection con; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
62 private static DatabaseMetaData dbmd; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
63 private static Statement stmt; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
64 private static BufferedReader in; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
65 private static PrintWriter out; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
66 private static Exporter exporter; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
67 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
68 /** |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
69 * JdbcClient is a command line query tool for MonetDB, similar to mclient. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
70 * It uses the JDBC API and the MonetDB JDBC driver to communicate with a |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
71 * MonetDB server. The MonetDB JDBC driver is included in the jdbcclient.jre8.jar |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
72 * for ease of use, so only 1 jar file is needed to use it. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
73 * |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
74 * <pre>Usage java -jar jdbcclient.jre8.jar |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
75 * [-h host[:port]] [-p port] [-f file] [-u user] |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
76 * [-l language] [-d database] [-e] [-D [table]] |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
77 * [--csvdir /path/to/csvfiles] [-X<opt>] |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
78 * | [--help] | [--version] |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
79 * or using long option equivalents --host --port --file --user --language |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
80 * --dump --echo --database. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
81 * Arguments may be written directly after the option like -p50000. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
82 * |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
83 * If no host and port are given, localhost and 50000 are assumed. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
84 * An .monetdb file may exist in the user's home directory. This file can contain |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
85 * preferences to use each time JdbcClient is started. Options given on the |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
86 * command line override the preferences file. The .monetdb file syntax is |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
87 * <option>=<value> where option is one of the options host, port, file, mode |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
88 * debug, or password. Note that the last one is perilous and therefore not |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
89 * available as command line option. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
90 * If no input file is given using the -f flag, an interactive session is |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
91 * started on the terminal. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
92 * |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
93 * OPTIONS |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
94 * -h --host The hostname of the host that runs the MonetDB database. A port |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
95 * number can be supplied by use of a colon, i.e. -h somehost:12345. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
96 * -p --port The port number to connect to. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
97 * -f --file A file name to use either for reading or writing. The file will |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
98 * be used for writing when dump mode is used (-D --dump). In read |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
99 * mode, the file can also be an URL pointing to a plain text file |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
100 * that is optionally gzip compressed. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
101 * -u --user The username to use when connecting to the database. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
102 * -d --database Try to connect to the given database (only makes sense if |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
103 * connecting to monetdbd). |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
104 * -l --language Use the given language, defaults to 'sql'. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
105 * --help This help screen. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
106 * --version Display driver version and exit. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
107 * -e --echo Also outputs the contents of the input file, if any. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
108 * -q --quiet Suppress printing the welcome header. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
109 * -D --dump Dumps the given table(s), or the complete database if none given. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
110 * --csvdir The directory path where csv data files wil be read from or |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
111 * written to when COPY ... ON CLIENT commands are executed. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
112 * -Xoutput The output mode when dumping. Default is sql, xml may be used for |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
113 * an experimental XML output. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
114 * -Xhash Use the given hash algorithm during challenge response. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
115 * Supported algorithm names: SHA512, SHA384, SHA256 and SHA1. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
116 * -Xdebug Writes a transmission log to disk for debugging purposes. If a |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
117 * file name is given, it is used, otherwise a file called |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
118 * monet<timestamp>.log is created. A given file never be |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
119 * overwritten; instead a unique variation of the file is used. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
120 * -Xbatching Indicates that a batch should be used instead of direct |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
121 * communication with the server for each statement. If a number is |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
122 * given, it is used as batch size. i.e. 8000 would execute the |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
123 * contents on the batch after each 8000 statements read. Batching |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
124 * can greatly speedup the process of restoring a database dump.</pre> |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
125 * |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
126 * @param args optional list of startup arguments |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
127 * @throws Exception if uncaught exception is thrown |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
128 */ |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
129 public final static void main(String[] args) throws Exception { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
130 final CmdLineOpts copts = new CmdLineOpts(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
131 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
132 // arguments which take exactly one argument |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
133 copts.addOption("h", "host", CmdLineOpts.CAR_ONE, "localhost", |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
134 "The hostname of the host that runs the MonetDB database. " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
135 "A port number can be supplied by use of a colon, i.e. " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
136 "-h somehost:12345."); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
137 copts.addOption("p", "port", CmdLineOpts.CAR_ONE, "50000", |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
138 "The port number to connect to."); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
139 // todo make it CAR_ONE_MANY |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
140 copts.addOption("f", "file", CmdLineOpts.CAR_ONE, null, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
141 "A file name to use either for reading or writing. The " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
142 "file will be used for writing when dump mode is used " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
143 "(-D --dump). In read mode, the file can also be an URL " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
144 "pointing to a plain text file that is optionally gzip " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
145 "compressed."); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
146 copts.addOption("u", "user", CmdLineOpts.CAR_ONE, System.getProperty("user.name"), |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
147 "The username to use when connecting to the database."); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
148 // this one is only here for the .monetdb file parsing, it is |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
149 // removed before the command line arguments are parsed |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
150 copts.addOption(null, "password", CmdLineOpts.CAR_ONE, null, null); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
151 copts.addOption("d", "database", CmdLineOpts.CAR_ONE, "", |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
152 "Try to connect to the given database (only makes sense " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
153 "if connecting to monetdbd)."); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
154 copts.addOption("l", "language", CmdLineOpts.CAR_ONE, "sql", |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
155 "Use the given language, defaults to 'sql'."); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
156 copts.addOption(null, "csvdir", CmdLineOpts.CAR_ONE, null, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
157 "The directory path where csv data files are read or " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
158 "written when using ON CLIENT clause of COPY command."); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
159 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
160 // arguments which have no argument(s) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
161 copts.addOption(null, "help", CmdLineOpts.CAR_ZERO, null, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
162 "This help screen."); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
163 copts.addOption(null, "version", CmdLineOpts.CAR_ZERO, null, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
164 "Display driver version and exit."); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
165 copts.addOption("e", "echo", CmdLineOpts.CAR_ZERO, null, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
166 "Also outputs the contents of the input file, if any."); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
167 copts.addOption("q", "quiet", CmdLineOpts.CAR_ZERO, null, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
168 "Suppress printing the welcome header."); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
169 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
170 // arguments which have zero to many arguments |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
171 copts.addOption("D", "dump", CmdLineOpts.CAR_ZERO_MANY, null, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
172 "Dumps the given table(s), or the complete database if " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
173 "none given."); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
174 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
175 // extended options |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
176 copts.addOption(null, "Xoutput", CmdLineOpts.CAR_ONE, null, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
177 "The output mode when dumping. Default is sql, xml may " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
178 "be used for an experimental XML output."); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
179 copts.addOption(null, "Xhash", CmdLineOpts.CAR_ONE, null, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
180 "Use the given hash algorithm during challenge response. " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
181 "Supported algorithm names: SHA512, SHA384, SHA256 and SHA1."); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
182 // arguments which can have zero or one argument(s) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
183 copts.addOption(null, "Xdebug", CmdLineOpts.CAR_ZERO_ONE, null, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
184 "Writes a transmission log to disk for debugging purposes. " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
185 "If a file name is given, it is used, otherwise a file " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
186 "called monet<timestamp>.log is created. A given file " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
187 "never be overwritten; instead a unique variation of the " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
188 "file is used."); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
189 copts.addOption(null, "Xbatching", CmdLineOpts.CAR_ZERO_ONE, null, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
190 "Indicates that a batch should be used instead of direct " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
191 "communication with the server for each statement. If a " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
192 "number is given, it is used as batch size. i.e. 8000 " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
193 "would execute the contents on the batch after each 8000 " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
194 "statements read. Batching can greatly speedup the " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
195 "process of restoring a database dump."); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
196 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
197 // we store user and password in separate variables in order to |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
198 // be able to properly act on them like forgetting the password |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
199 // from the user's file if the user supplies a username on the |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
200 // command line arguments |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
201 String pass = null; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
202 String user = null; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
203 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
204 // look for a file called .monetdb in the current dir or in the |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
205 // user's homedir and read its preferences |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
206 File pref = new File(".monetdb"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
207 if (!pref.exists()) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
208 pref = new File(System.getProperty("user.home"), ".monetdb"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
209 if (pref.exists()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
210 try { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
211 copts.processFile(pref); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
212 } catch (OptionsException e) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
213 System.err.println("Error in " + pref.getAbsolutePath() + ": " + e.getMessage()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
214 System.exit(1); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
215 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
216 user = copts.getOption("user").getArgument(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
217 pass = copts.getOption("password").getArgument(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
218 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
219 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
220 // process the command line arguments, remove password option |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
221 // first, and save the user we had at this point |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
222 copts.removeOption("password"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
223 try { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
224 copts.processArgs(args); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
225 } catch (OptionsException e) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
226 System.err.println("Error: " + e.getMessage()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
227 System.exit(1); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
228 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
229 // we can actually compare pointers (objects) here |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
230 if (user != copts.getOption("user").getArgument()) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
231 pass = null; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
232 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
233 if (copts.getOption("help").isPresent()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
234 System.out.print( |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
235 "Usage java -jar jdbcclient.jre8.jar\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
236 "\t\t[-h host[:port]] [-p port] [-f file] [-u user]\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
237 "\t\t[-l language] [-d database] [-e] [-D [table]]\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
238 "\t\t[--csvdir /path/to/csvfiles]] [-X<opt>]\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
239 "\t\t| [--help] | [--version]\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
240 "or using long option equivalents --host --port --file --user --language\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
241 "--dump --echo --database.\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
242 "Arguments may be written directly after the option like -p50000.\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
243 "\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
244 "If no host and port are given, localhost and 50000 are assumed.\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
245 "An .monetdb file may exist in the user's home directory. This file can contain\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
246 "preferences to use each time JdbcClient is started. Options given on the\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
247 "command line override the preferences file. The .monetdb file syntax is\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
248 "<option>=<value> where option is one of the options host, port, file, mode\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
249 "debug, or password. Note that the last one is perilous and therefore not\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
250 "available as command line option.\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
251 "If no input file is given using the -f flag, an interactive session is\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
252 "started on the terminal.\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
253 "\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
254 "OPTIONS\n" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
255 copts.produceHelpMessage() |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
256 ); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
257 System.exit(0); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
258 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
259 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
260 if (copts.getOption("version").isPresent()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
261 // We cannot use the DatabaseMetaData here, because we |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
262 // cannot get a Connection. So instead, we just get the |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
263 // values we want out of the Driver directly. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
264 System.out.println("JDBC Driver: v" + MonetDriver.getDriverVersion()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
265 System.exit(0); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
266 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
267 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
268 // whether the semi-colon at the end of a String terminates the |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
269 // query or not (default = yes => SQL) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
270 final boolean scolonterm = true; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
271 final boolean xmlMode = "xml".equals(copts.getOption("Xoutput").getArgument()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
272 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
273 // we need the password from the user, fetch it with a pseudo |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
274 // password protector |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
275 if (pass == null) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
276 final Console syscon = System.console(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
277 char[] tmp = null; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
278 if (syscon != null) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
279 tmp = syscon.readPassword("password: "); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
280 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
281 if (tmp == null) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
282 System.err.println("Invalid password!"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
283 System.exit(1); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
284 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
285 pass = String.valueOf(tmp); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
286 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
287 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
288 user = copts.getOption("user").getArgument(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
289 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
290 // build the hostname |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
291 String host = copts.getOption("host").getArgument(); |
698
6beecac8aa1b
Use indexOf(char) instead of indexOf(String) where possible.
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
675
diff
changeset
|
292 if (host.indexOf(':') == -1) { |
675
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
293 host = host + ":" + copts.getOption("port").getArgument(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
294 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
295 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
296 // build the extra arguments of the JDBC connect string |
723
7200f7942b16
For JdbcClient enlarge the fetchsize at connection moment.
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
716
diff
changeset
|
297 // increase the fetchsize from the default 250 to 10000 |
7200f7942b16
For JdbcClient enlarge the fetchsize at connection moment.
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
716
diff
changeset
|
298 String attr = "?fetchsize=10000&"; |
675
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
299 CmdLineOpts.OptionContainer oc = copts.getOption("language"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
300 final String lang = oc.getArgument(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
301 if (oc.isPresent()) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
302 attr += "language=" + lang + "&"; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
303 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
304 /* Xquery is no longer functional or supported |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
305 // set some behaviour based on the language XQuery |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
306 if (lang.equals("xquery")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
307 scolonterm = false; // no ; to end a statement |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
308 if (!copts.getOption("Xoutput").isPresent()) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
309 xmlMode = true; // the user will like xml results, most probably |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
310 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
311 */ |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
312 oc = copts.getOption("Xdebug"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
313 if (oc.isPresent()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
314 attr += "debug=true&"; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
315 if (oc.getArgumentCount() == 1) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
316 attr += "logfile=" + oc.getArgument() + "&"; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
317 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
318 oc = copts.getOption("Xhash"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
319 if (oc.isPresent()) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
320 attr += "hash=" + oc.getArgument() + "&"; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
321 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
322 // request a connection suitable for MonetDB from the driver |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
323 // manager note that the database specifier is only used when |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
324 // connecting to a proxy-like service, since MonetDB itself |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
325 // can't access multiple databases. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
326 con = null; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
327 final String database = copts.getOption("database").getArgument(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
328 try { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
329 // make sure the driver class is loaded (and thus register itself with the DriverManager) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
330 Class.forName("org.monetdb.jdbc.MonetDriver"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
331 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
332 con = DriverManager.getConnection( |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
333 "jdbc:monetdb://" + host + "/" + database + attr, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
334 user, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
335 pass |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
336 ); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
337 SQLWarning warn = con.getWarnings(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
338 while (warn != null) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
339 System.err.println("Connection warning: " + warn.getMessage()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
340 warn = warn.getNextWarning(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
341 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
342 con.clearWarnings(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
343 } catch (SQLException e) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
344 System.err.println("Database connect failed: " + e.getMessage()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
345 System.exit(1); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
346 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
347 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
348 try { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
349 dbmd = con.getMetaData(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
350 } catch (SQLException e) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
351 // we ignore this because it's probably because we don't use |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
352 // SQL language |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
353 dbmd = null; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
354 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
355 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
356 oc = copts.getOption("csvdir"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
357 if (oc.isPresent()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
358 final String csvdir = oc.getArgument(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
359 if (csvdir != null) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
360 // check if provided csvdir is an existing dir |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
361 // else a download of data into file will terminate the JDBC connection!! |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
362 if (java.nio.file.Files.isDirectory(java.nio.file.Paths.get(csvdir))) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
363 final FileTransferHandler FThandler = new FileTransferHandler(csvdir, Charset.defaultCharset()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
364 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
365 // register file data uploadHandler to allow support |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
366 // for: COPY INTO mytable FROM 'data.csv' ON CLIENT; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
367 ((MonetConnection) con).setUploadHandler(FThandler); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
368 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
369 // register file data downloadHandler to allow support |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
370 // for: COPY select_query INTO 'data.csv' ON CLIENT; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
371 ((MonetConnection) con).setDownloadHandler(FThandler); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
372 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
373 System.err.println("Warning: provided csvdir \"" + csvdir + "\" does not exist. Ignoring csvdir setting."); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
374 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
375 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
376 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
377 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
378 stmt = con.createStatement(); // is used by processInteractive(), processBatch(), doDump() |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
379 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
380 in = new BufferedReader(new InputStreamReader(System.in)); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
381 out = new PrintWriter(new BufferedWriter(new java.io.OutputStreamWriter(System.out))); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
382 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
383 // see if we will have to perform a database dump (only in SQL mode) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
384 if ("sql".equals(lang) && copts.getOption("dump").isPresent() && dbmd != null) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
385 final int argcount = copts.getOption("dump").getArgumentCount(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
386 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
387 // use the given file for writing |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
388 oc = copts.getOption("file"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
389 if (oc.isPresent()) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
390 out = new PrintWriter(new BufferedWriter(new java.io.FileWriter(oc.getArgument()))); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
391 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
392 // we only want user tables and views to be dumped (DDL and optional data), unless a specific table is requested |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
393 final String[] types = {"TABLE","VIEW","MERGE TABLE","REMOTE TABLE","REPLICA TABLE","STREAM TABLE"}; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
394 // Future: fetch all type names using dbmd.getTableTypes() and construct String[] with all |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
395 // table type names excluding the SYSTEM ... ones and LOCAL TEMPORARY TABLE ones. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
396 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
397 // request the list of tables/views available in the current schema in the database |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
398 ResultSet tbl = dbmd.getTables(null, con.getSchema(), null, (argcount == 0) ? types : null); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
399 // fetch all tables and store them in a LinkedList of Table objects |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
400 final LinkedList<Table> tables = new LinkedList<Table>(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
401 while (tbl.next()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
402 tables.add(new Table( |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
403 tbl.getString(2), // 2 = "TABLE_SCHEM" |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
404 tbl.getString(3), // 3 = "TABLE_NAME" |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
405 tbl.getString(4))); // 4 = "TABLE_TYPE" |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
406 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
407 tbl.close(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
408 tbl = null; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
409 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
410 if (xmlMode) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
411 exporter = new XMLExporter(out); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
412 exporter.setProperty(XMLExporter.TYPE_NIL, XMLExporter.VALUE_XSI); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
413 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
414 exporter = new SQLExporter(out); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
415 // stick with SQL INSERT INTO commands for now |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
416 // in the future we might do COPY INTO's here using VALUE_COPY |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
417 exporter.setProperty(SQLExporter.TYPE_OUTPUT, SQLExporter.VALUE_INSERT); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
418 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
419 exporter.useSchemas(true); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
420 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
421 // start SQL output |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
422 if (!xmlMode) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
423 out.println("START TRANSACTION;\n"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
424 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
425 // dump specific table(s) or not? |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
426 if (argcount > 0) { // yes we do |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
427 final String[] dumpers = copts.getOption("dump").getArguments(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
428 for (int i = 0; i < tables.size(); i++) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
429 Table ttmp = tables.get(i); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
430 for (int j = 0; j < dumpers.length; j++) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
431 String dumptblnm = dumpers[j].toString(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
432 if (ttmp.getName().equalsIgnoreCase(dumptblnm) || |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
433 ttmp.getFqname().equalsIgnoreCase(dumptblnm)) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
434 { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
435 // dump the table |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
436 doDump(out, ttmp); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
437 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
438 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
439 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
440 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
441 /* this returns everything, so including SYSTEM TABLE constraints */ |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
442 tbl = dbmd.getImportedKeys(null, null, null); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
443 while (tbl.next()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
444 // find FK table object 6 = "FKTABLE_SCHEM", 7 = "FKTABLE_NAME" |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
445 Table fk = Table.findTable(tbl.getString(6), tbl.getString(7), tables); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
446 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
447 // find PK table object 2 = "PKTABLE_SCHEM", 3 = "PKTABLE_NAME" |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
448 Table pk = Table.findTable(tbl.getString(2), tbl.getString(3), tables); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
449 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
450 // this happens when a system table has referential constraints |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
451 if (fk == null || pk == null) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
452 continue; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
453 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
454 // add PK table dependency to FK table |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
455 fk.addDependency(pk); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
456 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
457 tbl.close(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
458 tbl = null; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
459 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
460 // search for cycles of type a -> (x ->)+ b probably not |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
461 // the most optimal way, but it works by just scanning |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
462 // every table for loops in a recursive manor |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
463 for (Table t : tables) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
464 Table.checkForLoop(t, new ArrayList<Table>()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
465 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
466 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
467 // find the graph, at this point we know there are no |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
468 // cycles, thus a solution exists |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
469 for (int i = 0; i < tables.size(); i++) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
470 final List<Table> needs = tables.get(i).requires(tables.subList(0, i + 1)); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
471 if (needs.size() > 0) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
472 tables.removeAll(needs); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
473 tables.addAll(i, needs); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
474 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
475 // re-evaluate this position, for there is a new |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
476 // table now |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
477 i--; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
478 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
479 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
480 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
481 // we now have the right order to dump tables |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
482 for (Table t : tables) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
483 // dump the table |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
484 doDump(out, t); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
485 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
486 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
487 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
488 if (!xmlMode) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
489 out.println("COMMIT;"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
490 out.flush(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
491 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
492 // free resources, close the statement |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
493 stmt.close(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
494 // close the connection with the database |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
495 con.close(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
496 // completed database dump |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
497 System.exit(0); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
498 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
499 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
500 if (xmlMode) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
501 exporter = new XMLExporter(out); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
502 exporter.setProperty(XMLExporter.TYPE_NIL, XMLExporter.VALUE_XSI); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
503 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
504 exporter = new SQLExporter(out); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
505 // we want nice table formatted output |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
506 exporter.setProperty(SQLExporter.TYPE_OUTPUT, SQLExporter.VALUE_TABLE); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
507 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
508 exporter.useSchemas(false); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
509 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
510 try { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
511 // use the given file for reading |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
512 final boolean hasFile = copts.getOption("file").isPresent(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
513 final boolean doEcho = hasFile && copts.getOption("echo").isPresent(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
514 if (hasFile) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
515 final String tmp = copts.getOption("file").getArgument(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
516 try { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
517 in = getReader(tmp); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
518 } catch (Exception e) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
519 System.err.println("Error: " + e.getMessage()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
520 System.exit(1); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
521 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
522 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
523 // check for batch mode |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
524 int batchSize = 0; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
525 oc = copts.getOption("Xbatching"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
526 if (oc.isPresent()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
527 if (oc.getArgumentCount() == 1) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
528 // parse the number |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
529 try { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
530 batchSize = Integer.parseInt(oc.getArgument()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
531 } catch (NumberFormatException ex) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
532 // complain to the user |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
533 throw new IllegalArgumentException("Illegal argument for Xbatching: " + oc.getArgument() + " is not a parseable number!"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
534 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
535 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
536 processBatch(batchSize); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
537 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
538 processInteractive(true, doEcho, scolonterm, user); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
539 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
540 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
541 if (!copts.getOption("quiet").isPresent()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
542 // print welcome message |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
543 out.println("Welcome to the MonetDB interactive JDBC terminal!"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
544 if (dbmd != null) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
545 out.println("JDBC Driver: " + dbmd.getDriverName() + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
546 " v" + dbmd.getDriverVersion()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
547 out.println("Database Server: " + dbmd.getDatabaseProductName() + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
548 " v" + dbmd.getDatabaseProductVersion()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
549 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
550 out.println("Current Schema: " + con.getSchema()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
551 out.println("Type \\q to quit (you can also use: quit or exit), \\? or \\h for a list of available commands"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
552 out.flush(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
553 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
554 processInteractive(false, doEcho, scolonterm, user); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
555 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
556 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
557 // free resources, close the statement |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
558 stmt.close(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
559 // close the connection with the database |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
560 con.close(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
561 // close the file (if we used a file) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
562 in.close(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
563 } catch (Exception e) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
564 System.err.println("A fatal exception occurred: " + e.toString()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
565 e.printStackTrace(System.err); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
566 // at least try to close the connection properly, since it will |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
567 // close all statements associated with it |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
568 try { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
569 con.close(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
570 } catch (SQLException ex) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
571 // ok... nice try |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
572 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
573 System.exit(1); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
574 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
575 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
576 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
577 /** |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
578 * Tries to interpret the given String as URL or file. Returns the |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
579 * assigned BufferedReader, or throws an Exception if the given |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
580 * string couldn't be identified as a valid URL or file. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
581 * |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
582 * @param uri URL or filename as String |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
583 * @return a BufferedReader for the uri |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
584 * @throws Exception if uri cannot be identified as a valid URL or file |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
585 */ |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
586 static BufferedReader getReader(final String uri) throws Exception { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
587 BufferedReader ret = null; |
772
9493dacdd186
Fix problem where passed file name contained a : character but was not an URL,
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
770
diff
changeset
|
588 URI u = null; |
675
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
589 |
772
9493dacdd186
Fix problem where passed file name contained a : character but was not an URL,
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
770
diff
changeset
|
590 // Try and parse as URL |
9493dacdd186
Fix problem where passed file name contained a : character but was not an URL,
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
770
diff
changeset
|
591 try { |
9493dacdd186
Fix problem where passed file name contained a : character but was not an URL,
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
770
diff
changeset
|
592 // Note: as of Java version 20 java.net.URL(String) constructor is deprecated. |
9493dacdd186
Fix problem where passed file name contained a : character but was not an URL,
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
770
diff
changeset
|
593 // https://docs.oracle.com/en/java/javase/20/docs/api/java.base/java/net/URL.html#%3Cinit%3E(java.lang.String) |
9493dacdd186
Fix problem where passed file name contained a : character but was not an URL,
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
770
diff
changeset
|
594 u = new java.net.URI(uri); |
9493dacdd186
Fix problem where passed file name contained a : character but was not an URL,
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
770
diff
changeset
|
595 // the URL must start with a scheme such as: "http://" or "https://" else u.toURL() errors |
9493dacdd186
Fix problem where passed file name contained a : character but was not an URL,
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
770
diff
changeset
|
596 if (u != null && u.isAbsolute()) { |
9493dacdd186
Fix problem where passed file name contained a : character but was not an URL,
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
770
diff
changeset
|
597 final URL url = u.toURL(); |
9493dacdd186
Fix problem where passed file name contained a : character but was not an URL,
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
770
diff
changeset
|
598 HttpURLConnection.setFollowRedirects(true); |
9493dacdd186
Fix problem where passed file name contained a : character but was not an URL,
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
770
diff
changeset
|
599 final HttpURLConnection con = (HttpURLConnection)url.openConnection(); |
9493dacdd186
Fix problem where passed file name contained a : character but was not an URL,
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
770
diff
changeset
|
600 con.setRequestMethod("GET"); |
9493dacdd186
Fix problem where passed file name contained a : character but was not an URL,
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
770
diff
changeset
|
601 final String ct = con.getContentType(); |
9493dacdd186
Fix problem where passed file name contained a : character but was not an URL,
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
770
diff
changeset
|
602 if ("application/x-gzip".equals(ct)) { |
9493dacdd186
Fix problem where passed file name contained a : character but was not an URL,
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
770
diff
changeset
|
603 // open gzip stream |
9493dacdd186
Fix problem where passed file name contained a : character but was not an URL,
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
770
diff
changeset
|
604 ret = new BufferedReader(new InputStreamReader( |
9493dacdd186
Fix problem where passed file name contained a : character but was not an URL,
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
770
diff
changeset
|
605 new java.util.zip.GZIPInputStream(con.getInputStream()))); |
9493dacdd186
Fix problem where passed file name contained a : character but was not an URL,
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
770
diff
changeset
|
606 } else { |
9493dacdd186
Fix problem where passed file name contained a : character but was not an URL,
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
770
diff
changeset
|
607 // text/plain otherwise just attempt to read as is |
9493dacdd186
Fix problem where passed file name contained a : character but was not an URL,
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
770
diff
changeset
|
608 ret = new BufferedReader(new InputStreamReader(con.getInputStream())); |
770
1b9fe515f574
Fixed javac version 20 compiler warning.
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
768
diff
changeset
|
609 } |
675
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
610 } |
772
9493dacdd186
Fix problem where passed file name contained a : character but was not an URL,
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
770
diff
changeset
|
611 } catch (IOException e) { |
9493dacdd186
Fix problem where passed file name contained a : character but was not an URL,
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
770
diff
changeset
|
612 // failed to open/read the url |
9493dacdd186
Fix problem where passed file name contained a : character but was not an URL,
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
770
diff
changeset
|
613 throw new Exception("Failed to open/read http URL: " + e.getMessage()); |
9493dacdd186
Fix problem where passed file name contained a : character but was not an URL,
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
770
diff
changeset
|
614 } catch (Exception e) { |
9493dacdd186
Fix problem where passed file name contained a : character but was not an URL,
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
770
diff
changeset
|
615 if (u != null) |
9493dacdd186
Fix problem where passed file name contained a : character but was not an URL,
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
770
diff
changeset
|
616 throw new Exception("Invalid http URL: " + uri + "\n" + e.getMessage()); |
675
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
617 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
618 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
619 if (ret == null) { |
772
9493dacdd186
Fix problem where passed file name contained a : character but was not an URL,
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
770
diff
changeset
|
620 // uri is not a valid URI/URL, so probably a file name |
675
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
621 try { |
770
1b9fe515f574
Fixed javac version 20 compiler warning.
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
768
diff
changeset
|
622 ret = new BufferedReader(new java.io.FileReader(uri)); |
1b9fe515f574
Fixed javac version 20 compiler warning.
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
768
diff
changeset
|
623 } catch (java.io.FileNotFoundException fnfe) { |
1b9fe515f574
Fixed javac version 20 compiler warning.
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
768
diff
changeset
|
624 // the message is descriptive enough, |
1b9fe515f574
Fixed javac version 20 compiler warning.
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
768
diff
changeset
|
625 // adds "(No such file or directory)" itself. |
1b9fe515f574
Fixed javac version 20 compiler warning.
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
768
diff
changeset
|
626 throw new Exception(fnfe.getMessage()); |
675
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
627 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
628 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
629 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
630 return ret; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
631 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
632 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
633 /** |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
634 * Starts an interactive processing loop, where output is adjusted to an |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
635 * user session. This processing loop is not suitable for bulk processing |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
636 * as in executing the contents of a file, since processing on the given |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
637 * input is done after each row that has been entered. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
638 * |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
639 * @param hasFile a boolean indicating whether a file is used as input |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
640 * @param doEcho a boolean indicating whether to echo the given input |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
641 * @param scolonterm whether a ';' makes this query part complete |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
642 * @param user a String representing the username of the current user |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
643 * @throws IOException if an IO exception occurs |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
644 * @throws SQLException if a database related error occurs |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
645 */ |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
646 private static void processInteractive( |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
647 final boolean hasFile, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
648 final boolean doEcho, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
649 final boolean scolonterm, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
650 final String user) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
651 throws IOException, SQLException |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
652 { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
653 // an SQL stack keeps track of ( " and ' |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
654 final SQLStack stack = new SQLStack(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
655 boolean lastac = false; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
656 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
657 if (!hasFile) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
658 lastac = con.getAutoCommit(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
659 out.println("auto commit mode: " + (lastac ? "on" : "off")); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
660 out.print(getPrompt(stack, true)); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
661 out.flush(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
662 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
663 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
664 String curLine; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
665 String query = ""; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
666 boolean doProcess; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
667 boolean wasComplete = true; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
668 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
669 // the main (interactive) process loop |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
670 for (long i = 1; true; i++) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
671 // Manually read a line, because we want to detect an EOF |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
672 // (ctrl-D). Doing so allows to have a terminator for query |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
673 // which is not based on a special character, as is the |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
674 // problem for XQuery |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
675 curLine = in.readLine(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
676 if (curLine == null) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
677 out.println(""); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
678 if (!query.isEmpty()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
679 try { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
680 executeQuery(query, stmt, out, !hasFile); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
681 } catch (SQLException e) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
682 out.flush(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
683 do { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
684 if (hasFile) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
685 System.err.println("Error on line " + i + ": [" + e.getSQLState() + "] " + e.getMessage()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
686 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
687 System.err.println("Error [" + e.getSQLState() + "]: " + e.getMessage()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
688 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
689 // print all error messages in the chain (if any) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
690 } while ((e = e.getNextException()) != null); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
691 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
692 query = ""; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
693 wasComplete = true; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
694 if (!hasFile) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
695 final boolean ac = con.getAutoCommit(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
696 if (ac != lastac) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
697 out.println("auto commit mode: " + (ac ? "on" : "off")); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
698 lastac = ac; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
699 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
700 out.print(getPrompt(stack, wasComplete)); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
701 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
702 out.flush(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
703 // try to read again |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
704 continue; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
705 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
706 // user did ctrl-D without something in the buffer, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
707 // so terminate |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
708 break; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
709 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
710 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
711 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
712 if (doEcho) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
713 out.println(curLine); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
714 out.flush(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
715 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
716 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
717 // a query part is a line of an SQL query |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
718 QueryPart qp = scanQuery(curLine, stack, scolonterm); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
719 if (!qp.isEmpty()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
720 final String command = qp.getQuery(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
721 doProcess = true; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
722 if (wasComplete) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
723 doProcess = false; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
724 // check for commands only when the previous row was complete |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
725 if (command.equals("\\q") || command.equals("quit") || command.equals("exit")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
726 break; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
727 } else if (dbmd != null && command.startsWith("\\d")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
728 processDescribeCmd(command, scolonterm); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
729 } else if (command.startsWith("\\v")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
730 if (command.equals("\\vsci")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
731 MDBvalidator.validateSqlCatalogIntegrity(con, true); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
732 } else if (command.equals("\\vsci_noheader")) { // used only for internal automated testing |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
733 MDBvalidator.validateSqlCatalogIntegrity(con, false); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
734 } else if (command.equals("\\vsni")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
735 MDBvalidator.validateSqlNetcdfTablesIntegrity(con, true); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
736 } else if (command.equals("\\vsni_noheader")) { // used only for internal automated testing |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
737 MDBvalidator.validateSqlNetcdfTablesIntegrity(con, false); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
738 } else if (command.equals("\\vsgi")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
739 MDBvalidator.validateSqlGeomTablesIntegrity(con, true); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
740 } else if (command.equals("\\vsgi_noheader")) { // used only for internal automated testing |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
741 MDBvalidator.validateSqlGeomTablesIntegrity(con, false); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
742 } else if (command.startsWith("\\vsi ")) { |
826
1074ad1a4c63
Enhance \vsi command by removing a trailing ; terminator from the provided schema name.
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
777
diff
changeset
|
743 String schema_nm = command.substring(5).trim(); |
1074ad1a4c63
Enhance \vsi command by removing a trailing ; terminator from the provided schema name.
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
777
diff
changeset
|
744 if (schema_nm.endsWith(";")) |
1074ad1a4c63
Enhance \vsi command by removing a trailing ; terminator from the provided schema name.
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
777
diff
changeset
|
745 schema_nm = schema_nm.substring(0, schema_nm.length() - 1); |
675
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
746 MDBvalidator.validateSchemaIntegrity(con, schema_nm, true); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
747 } else if (command.startsWith("\\vsi_noheader ")) { // used only for internal automated testing |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
748 String schema_nm = command.substring(14); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
749 MDBvalidator.validateSchemaIntegrity(con, schema_nm, false); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
750 } else if (command.equals("\\vdbi")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
751 MDBvalidator.validateDBIntegrity(con, true); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
752 } else if (command.equals("\\vdbi_noheader")) { // used only for internal automated testing |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
753 MDBvalidator.validateDBIntegrity(con, false); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
754 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
755 showCommands(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
756 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
757 } else if (command.startsWith("\\l") || command.startsWith("\\i")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
758 String object = command.substring(2).trim(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
759 if (scolonterm && object.endsWith(";")) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
760 object = object.substring(0, object.length() - 1); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
761 if (object.isEmpty()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
762 System.err.println("Usage: '" + command.substring(0, 2) + "<uri>' where <uri> is a file or URL"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
763 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
764 // temporarily redirect input from in |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
765 final BufferedReader console = in; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
766 try { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
767 in = getReader(object); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
768 if (command.startsWith("\\l")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
769 processInteractive(true, doEcho, scolonterm, user); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
770 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
771 processBatch(0); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
772 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
773 } catch (Exception e) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
774 out.flush(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
775 System.err.println("Error: " + e.getMessage()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
776 } finally { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
777 // put back in redirection |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
778 in = console; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
779 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
780 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
781 } else if (command.equals("\\?") || command.equals("\\h") || command.startsWith("\\")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
782 showCommands(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
783 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
784 doProcess = true; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
785 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
786 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
787 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
788 if (doProcess) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
789 query += command + (qp.hasOpenQuote() ? "\\n" : " "); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
790 if (qp.isComplete()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
791 // strip off trailing ';' |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
792 query = query.substring(0, query.length() - 2); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
793 // execute query |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
794 try { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
795 executeQuery(query, stmt, out, !hasFile); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
796 } catch (SQLException e) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
797 out.flush(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
798 final String startmsg = (hasFile ? ("Error on line " + i + ": [") : "Error ["); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
799 do { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
800 System.err.println(startmsg + e.getSQLState() + "] " + e.getMessage()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
801 // print all error messages in the chain (if any) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
802 } while ((e = e.getNextException()) != null); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
803 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
804 query = ""; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
805 wasComplete = true; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
806 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
807 wasComplete = false; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
808 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
809 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
810 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
811 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
812 if (!hasFile) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
813 final boolean ac = con.getAutoCommit(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
814 if (ac != lastac) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
815 out.println("auto commit mode: " + (ac ? "on" : "off")); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
816 lastac = ac; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
817 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
818 out.print(getPrompt(stack, wasComplete)); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
819 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
820 out.flush(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
821 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
822 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
823 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
824 private static void showCommands() |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
825 { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
826 out.println("Available commands:"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
827 out.println("\\q quits this program (you can also use: quit or exit)"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
828 if (dbmd != null) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
829 out.println("\\d list available user tables and views in current schema"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
830 out.println("\\dt list available user tables in current schema"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
831 out.println("\\dv list available user views in current schema"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
832 out.println("\\df list available user functions in current schema"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
833 out.println("\\dp list available user procedures in current schema"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
834 out.println("\\ds list available user sequences in current schema"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
835 out.println("\\dn list available user schemas"); |
777
5788507c01b9
Adapt JdbcClient to show information_schema views when using commands: \dS or \dSv or \dvS and current schema is information_schema.
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
772
diff
changeset
|
836 out.println("\\dS list available system tables and views in sys/tmp/information_schema schema"); |
675
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
837 out.println("\\dSt list available system tables in sys/tmp schema"); |
777
5788507c01b9
Adapt JdbcClient to show information_schema views when using commands: \dS or \dSv or \dvS and current schema is information_schema.
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
772
diff
changeset
|
838 out.println("\\dSv list available system views in sys/information_schema/logging schema"); |
675
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
839 out.println("\\dSf list available system functions in current schema"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
840 out.println("\\dSp list available system procedures in current schema"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
841 out.println("\\dSn list available system schemas"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
842 out.println("\\d <obj> describes the given table or view"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
843 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
844 out.println("\\l<uri> executes the contents of the given file or URL"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
845 out.println("\\i<uri> batch executes the inserts from the given file or URL"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
846 out.println("\\vsci validate sql system catalog integrity"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
847 // out.println("\\vsni validate sql system netcdf tables integrity"); // do not print as it depends on availability of netcdf library on server |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
848 // out.println("\\vsgi validate sql system geom tables integrity"); // do not print as it depends on availability of geom library on server |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
849 out.println("\\vsi <schema> validate integrity of data in the given schema"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
850 out.println("\\vdbi validate integrity of data in all user schemas in the database"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
851 out.println("\\? or \\h this help screen"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
852 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
853 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
854 /** |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
855 * Process the commands that start with \d |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
856 * These commands list the available user or system tables, views, functions, procedures, schemas |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
857 * or user sequences |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
858 * or describe a specific table or view by printing the CREATE DDL statement |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
859 * See also showCommands() |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
860 * These commands are almost the same as in mclient program |
709
bdeabbd46ec6
Resolve javac and javadoc warnings when compiled with JDK19.
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
698
diff
changeset
|
861 * |
bdeabbd46ec6
Resolve javac and javadoc warnings when compiled with JDK19.
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
698
diff
changeset
|
862 * @param dcommand the command starting with \d |
bdeabbd46ec6
Resolve javac and javadoc warnings when compiled with JDK19.
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
698
diff
changeset
|
863 * @param scolonterm whether a ';' makes this query part complete |
675
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
864 */ |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
865 private static void processDescribeCmd(final String dcommand, final boolean scolonterm) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
866 { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
867 final String command = dcommand.substring(1); // remove the leading \ character first |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
868 ResultSet rs = null; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
869 Statement st = null; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
870 try { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
871 if (command.equals("d") || command.equals("dt") || command.equals("dv")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
872 // list available user tables and/or views in current schema (maybe tmp schema to show temporary tables) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
873 boolean tablesOnly = command.equals("dt"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
874 boolean viewsOnly = command.equals("dv"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
875 rs = dbmd.getTables(null, con.getSchema(), null, null); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
876 while (rs.next()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
877 final String tableType = rs.getString(4); // 4 = "TABLE_TYPE" |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
878 boolean include = (tableType != null && !tableType.startsWith("SYSTEM ")); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
879 if (include && tablesOnly && !tableType.contains("TABLE")) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
880 include = false; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
881 if (include && viewsOnly && !tableType.contains("VIEW")) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
882 include = false; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
883 if (include) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
884 out.println(tableType + "\t" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
885 optAddDqs(rs.getString(2)) + "." + // 2 = "TABLE_SCHEM" |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
886 optAddDqs(rs.getString(3)) ); // 3 = "TABLE_NAME" |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
887 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
888 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
889 } else |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
890 if (command.equals("dS") || command.equals("dSt") || command.equals("dSv") || command.equals("dtS") || command.equals("dvS")) { |
777
5788507c01b9
Adapt JdbcClient to show information_schema views when using commands: \dS or \dSv or \dvS and current schema is information_schema.
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
772
diff
changeset
|
891 // list available system tables and/or views in sys/tmp/information_schema/logging schema |
675
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
892 boolean tablesOnly = command.contains("t"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
893 boolean viewsOnly = command.contains("v"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
894 String curSchema = con.getSchema(); |
777
5788507c01b9
Adapt JdbcClient to show information_schema views when using commands: \dS or \dSv or \dvS and current schema is information_schema.
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
772
diff
changeset
|
895 // only schemas: sys, tmp, information_schema and logging contain system tables and/or system views. |
5788507c01b9
Adapt JdbcClient to show information_schema views when using commands: \dS or \dSv or \dvS and current schema is information_schema.
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
772
diff
changeset
|
896 if (!("sys".equals(curSchema) || "tmp".equals(curSchema) || "information_schema".equals(curSchema) || "logging".equals(curSchema))) |
5788507c01b9
Adapt JdbcClient to show information_schema views when using commands: \dS or \dSv or \dvS and current schema is information_schema.
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
772
diff
changeset
|
897 // when currently in another schema, default to the sys schema else nothing will be listed |
675
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
898 curSchema = "sys"; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
899 rs = dbmd.getTables(null, curSchema, null, null); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
900 while (rs.next()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
901 final String tableType = rs.getString(4); // 4 = "TABLE_TYPE" |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
902 boolean include = (tableType != null && tableType.startsWith("SYSTEM ")); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
903 if (include && tablesOnly && !tableType.contains("TABLE")) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
904 include = false; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
905 if (include && viewsOnly && !tableType.contains("VIEW")) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
906 include = false; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
907 if (include) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
908 out.println(tableType + "\t" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
909 optAddDqs(rs.getString(2)) + "." + // 2 = "TABLE_SCHEM" |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
910 optAddDqs(rs.getString(3)) ); // 3 = "TABLE_NAME" |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
911 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
912 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
913 } else |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
914 if (command.equals("df") || command.equals("dp") || command.equals("dSf") || command.equals("dSp") || command.equals("dfS") || command.equals("dpS")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
915 // list available user/system functions or procedures in current schema (maybe tmp schema) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
916 boolean proceduresOnly = command.contains("p"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
917 boolean systemOnly = command.contains("S"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
918 // Note we cannot use: dbmd.getFunctions(null, con.getSchema(), null); or dbmd.getProcedures(null, con.getSchema(), null); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
919 // as they do not return the full function type name, nor provide information whether it is a system function or system procedure. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
920 // Instead we query the MonetDB system catog tables directly using: |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
921 String qy = "SELECT DISTINCT " + (systemOnly ? "'SYSTEM ' || " : "") + "\"function_type_keyword\" as func_type, " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
922 "\"schemas\".\"name\" as schem_name, \"functions\".\"name\" as func_name" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
923 " FROM sys.\"functions\"" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
924 " JOIN sys.\"function_types\" ON \"functions\".\"type\" = \"function_types\".\"function_type_id\"" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
925 " JOIN sys.\"schemas\" ON \"functions\".\"schema_id\" = \"schemas\".\"id\"" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
926 " WHERE \"functions\".\"system\" = " + (systemOnly ? "true" : "false") + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
927 " AND \"functions\".\"type\" " + (proceduresOnly ? "= 2" : "<> 2") + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
928 " AND \"schemas\".\"name\" = current_schema" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
929 " ORDER BY 2, 3"; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
930 st = con.createStatement(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
931 rs = st.executeQuery(qy); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
932 while (rs.next()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
933 out.println(rs.getString(1) + " \t" + // 1 = func_type |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
934 optAddDqs(rs.getString(2)) + "." + // 2 = schem_name |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
935 optAddDqs(rs.getString(3)) ); // 3 = func_name |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
936 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
937 } else |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
938 if (command.equals("ds")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
939 // list available (user) sequences in current schema |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
940 String qy = "SELECT \"schemas\".\"name\", \"sequences\".\"name\"" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
941 " FROM sys.\"sequences\"" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
942 " JOIN sys.\"schemas\" ON \"sequences\".\"schema_id\" = \"schemas\".\"id\"" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
943 " WHERE \"schemas\".\"name\" = current_schema" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
944 " ORDER BY 1, 2"; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
945 st = con.createStatement(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
946 rs = st.executeQuery(qy); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
947 while (rs.next()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
948 out.println("SEQUENCE " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
949 optAddDqs(rs.getString(1)) + "." + // 1 = schemas.name |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
950 optAddDqs(rs.getString(2)) ); // 2 = sequences.name |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
951 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
952 } else |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
953 if (command.equals("dn") || command.equals("dSn") || command.equals("dnS")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
954 // list available user/system schemas in the database |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
955 boolean systemOnly = command.contains("S"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
956 String qy = "SELECT \"name\"" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
957 " FROM sys.\"schemas\"" + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
958 " WHERE \"system\" = " + (systemOnly ? "true" : "false") + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
959 " ORDER BY 1"; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
960 String schema_type = (systemOnly ? "SYSTEM SCHEMA\t" : "SCHEMA\t"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
961 st = con.createStatement(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
962 rs = st.executeQuery(qy); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
963 while (rs.next()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
964 out.println(schema_type + optAddDqs(rs.getString(1)) ); // 1 = schemas.name |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
965 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
966 } else |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
967 if (command.startsWith("d ")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
968 // describes the given table or view name. It may be a fully qualified name such as: sys."keys" |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
969 String object = command.substring(2).trim(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
970 if (scolonterm && object.endsWith(";")) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
971 object = object.substring(0, object.length() - 1); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
972 if (object.isEmpty()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
973 System.err.println("Missing name of object to describe"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
974 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
975 // extract the name of the schema first. If none found use current schema |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
976 String schema; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
977 String obj_nm = object; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
978 int len; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
979 boolean found = false; |
698
6beecac8aa1b
Use indexOf(char) instead of indexOf(String) where possible.
Martin van Dinther <martin.van.dinther@monetdbsolutions.com>
parents:
675
diff
changeset
|
980 final int dot = object.indexOf('.'); |
675
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
981 if (dot > 0) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
982 // use specified schema |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
983 schema = object.substring(0, dot); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
984 obj_nm = object.substring(dot + 1); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
985 // remove potential surrounding double quotes around schema name |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
986 len = schema.length(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
987 if (len > 2 && schema.charAt(0) == '"' && schema.charAt(len -1) == '"') |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
988 schema = schema.substring(1, len -1); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
989 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
990 // use current schema |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
991 schema = con.getSchema(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
992 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
993 // remove potential surrounding double quotes around table or view name |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
994 len = obj_nm.length(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
995 if (len > 2 && obj_nm.charAt(0) == '"' && obj_nm.charAt(len -1) == '"') |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
996 obj_nm = obj_nm.substring(1, len -1); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
997 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
998 // System.err.println("calling dbmd.getTables(" + schema + ", " + obj_nm + ")"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
999 rs = dbmd.getTables(null, schema, obj_nm, null); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1000 while (rs.next() && !found) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1001 final String schemaName = rs.getString(2); // 2 = "TABLE_SCHEM" |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1002 final String tableName = rs.getString(3); // 3 = "TABLE_NAME" |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1003 if (obj_nm.equals(tableName) && schema.equals(schemaName)) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1004 String tableType = rs.getString(4); // 4 = "TABLE_TYPE" |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1005 if (tableType.startsWith("SYSTEM ")) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1006 tableType = tableType.substring(7); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1007 // we found it, describe it |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1008 exporter.dumpSchema(dbmd, tableType, schemaName, tableName); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1009 found = true; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1010 break; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1011 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1012 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1013 if (!found) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1014 System.err.println("table or view: " + schema + "." + obj_nm + " does not exist"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1015 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1016 } else |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1017 if (command.startsWith("d")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1018 System.err.println("unknown sub-command for \\d: " + command.substring(1)); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1019 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1020 showCommands(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1021 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1022 } catch (SQLException e) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1023 out.flush(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1024 do { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1025 System.err.println("Error [" + e.getSQLState() + "]: " + e.getMessage()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1026 // print all error messages in the chain (if any) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1027 } while ((e = e.getNextException()) != null); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1028 } finally { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1029 if (rs != null) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1030 try { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1031 rs.close(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1032 } catch (SQLException e) { /* ignore */ } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1033 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1034 if (st != null) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1035 try { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1036 st.close(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1037 } catch (SQLException e) { /* ignore */ } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1038 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1039 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1040 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1041 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1042 private static String optAddDqs(final String name) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1043 { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1044 if (name.contains(" ") || name.contains("\t") || name.contains("\n")) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1045 return Exporter.dq(name); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1046 return name; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1047 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1048 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1049 /** |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1050 * Executes the given query and prints the result tabularised to the |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1051 * given PrintWriter stream. The result of this method is the |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1052 * default output of a query: tabular data. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1053 * |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1054 * @param query the query to execute |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1055 * @param stmt the Statement to execute the query on |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1056 * @param out the PrintWriter to write to |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1057 * @param showTiming flag to specify if timing information nees to be printed |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1058 * @throws SQLException if a database related error occurs |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1059 */ |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1060 private static void executeQuery(final String query, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1061 final Statement stmt, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1062 final PrintWriter out, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1063 final boolean showTiming) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1064 throws SQLException |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1065 { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1066 // warnings generated during querying |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1067 SQLWarning warn; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1068 long startTime = (showTiming ? System.currentTimeMillis() : 0); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1069 long finishTime = 0; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1070 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1071 // execute the query, let the driver decide what type it is |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1072 int aff = -1; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1073 boolean nextRslt = stmt.execute(query, Statement.RETURN_GENERATED_KEYS); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1074 if (!nextRslt) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1075 aff = stmt.getUpdateCount(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1076 do { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1077 if (nextRslt) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1078 // we have a ResultSet, print it |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1079 final ResultSet rs = stmt.getResultSet(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1080 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1081 exporter.dumpResultSet(rs); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1082 if (showTiming) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1083 finishTime = System.currentTimeMillis(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1084 out.println("Elapsed Time: " + (finishTime - startTime) + " ms"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1085 startTime = finishTime; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1086 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1087 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1088 // if there were warnings for this result, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1089 // show them! |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1090 warn = rs.getWarnings(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1091 if (warn != null) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1092 // force stdout to be written so the |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1093 // warning appears below it |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1094 out.flush(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1095 do { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1096 System.err.println("ResultSet warning: " + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1097 warn.getMessage()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1098 warn = warn.getNextWarning(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1099 } while (warn != null); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1100 rs.clearWarnings(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1101 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1102 rs.close(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1103 } else if (aff != -1) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1104 String timingoutput = ""; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1105 if (showTiming) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1106 finishTime = System.currentTimeMillis(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1107 timingoutput = ". Elapsed Time: " + (finishTime - startTime) + " ms"; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1108 startTime = finishTime; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1109 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1110 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1111 if (aff == Statement.SUCCESS_NO_INFO) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1112 out.println("Operation successful" + timingoutput); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1113 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1114 // we have an update count |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1115 // see if a key was generated |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1116 final ResultSet rs = stmt.getGeneratedKeys(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1117 final boolean hasGeneratedKeyData = rs.next(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1118 out.println(aff + " affected row" + (aff != 1 ? "s" : "") + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1119 (hasGeneratedKeyData ? ", last generated key: " + rs.getString(1) : "") + |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1120 timingoutput); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1121 rs.close(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1122 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1123 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1124 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1125 out.flush(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1126 } while ((nextRslt = stmt.getMoreResults()) || |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1127 (aff = stmt.getUpdateCount()) != -1); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1128 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1129 // if there were warnings for this statement show them! |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1130 warn = stmt.getWarnings(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1131 while (warn != null) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1132 System.err.println("Statement warning: " + warn.getMessage()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1133 warn = warn.getNextWarning(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1134 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1135 stmt.clearWarnings(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1136 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1137 // if there were warnings for this connection show them! |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1138 warn = con.getWarnings(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1139 while (warn != null) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1140 // suppress warning when issueing a "set schema xyz;" command |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1141 // if ( !(warn.getMessage()).equals("Server enabled auto commit mode while local state already was auto commit.") ) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1142 System.err.println("Connection warning: " + warn.getMessage()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1143 warn = warn.getNextWarning(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1144 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1145 con.clearWarnings(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1146 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1147 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1148 /** |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1149 * Starts a processing loop optimized for processing (large) chunks of |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1150 * continuous data, such as input from a file. Unlike in the interactive |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1151 * loop above, queries are sent only to the database if a certain batch |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1152 * amount is reached. No client side query checks are made, but everything |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1153 * is sent to the server as-is. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1154 * |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1155 * @param batchSize the number of items to store in the batch before |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1156 * sending them to the database for execution. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1157 * @throws IOException if an IO exception occurs. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1158 */ |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1159 private static void processBatch(final int batchSize) throws IOException { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1160 final StringBuilder query = new StringBuilder(2048); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1161 int i = 0; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1162 try { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1163 String curLine; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1164 // the main loop |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1165 for (i = 1; (curLine = in.readLine()) != null; i++) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1166 query.append(curLine); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1167 if (curLine.endsWith(";")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1168 // lousy check for end of statement, but in batch mode it |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1169 // is not very important to catch all end of statements... |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1170 stmt.addBatch(query.toString()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1171 query.setLength(0); // clear the buffer |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1172 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1173 query.append('\n'); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1174 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1175 if (batchSize > 0 && i % batchSize == 0) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1176 stmt.executeBatch(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1177 // stmt.clearBatch(); // this is no longer needed after call executeBatch(), see https://www.monetdb.org/bugzilla/show_bug.cgi?id=6953 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1178 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1179 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1180 stmt.addBatch(query.toString()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1181 stmt.executeBatch(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1182 // stmt.clearBatch(); // this is no longer needed after call executeBatch(), see https://www.monetdb.org/bugzilla/show_bug.cgi?id=6953 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1183 } catch (SQLException e) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1184 do { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1185 System.err.println("Error at line " + i + ": [" + e.getSQLState() + "] " + e.getMessage()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1186 // print all error messages in the chain (if any) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1187 } while ((e = e.getNextException()) != null); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1188 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1189 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1190 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1191 /** |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1192 * Wrapper method that decides to dump SQL or XML. In the latter case, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1193 * this method does the XML data generation. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1194 * |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1195 * @param out a Writer to write the data to |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1196 * @param table the table to dump |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1197 * @throws SQLException if a database related error occurs |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1198 */ |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1199 private static void doDump(final PrintWriter out, final Table table) throws SQLException { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1200 final String tableType = table.getType(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1201 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1202 // dump CREATE definition of this table/view |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1203 exporter.dumpSchema(dbmd, tableType, table.getSchem(), table.getName()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1204 out.println(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1205 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1206 // only dump data from real tables, not from VIEWs / MERGE / REMOTE / REPLICA / STREAM tables |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1207 if (tableType.contains("TABLE") |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1208 && !tableType.equals("MERGE TABLE") |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1209 && !tableType.equals("REMOTE TABLE") |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1210 && !tableType.equals("REPLICA TABLE") |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1211 && !tableType.equals("STREAM TABLE")) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1212 final ResultSet rs = stmt.executeQuery("SELECT * FROM " + table.getFqnameQ()); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1213 if (rs != null) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1214 exporter.dumpResultSet(rs); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1215 rs.close(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1216 out.println(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1217 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1218 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1219 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1220 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1221 /** |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1222 * Simple helper method that generates a prompt. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1223 * |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1224 * @param stack the current SQLStack |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1225 * @param compl whether the statement is complete |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1226 * @return a prompt which consist of "sql" plus the top of the stack |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1227 */ |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1228 private static String getPrompt(final SQLStack stack, final boolean compl) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1229 return (compl ? "sql" : "more") + (stack.empty() ? ">" : stack.peek()) + " "; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1230 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1231 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1232 /** |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1233 * Scans the given string and tries to discover if it is a complete query |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1234 * or that there needs something to be added. If a string doesn't end with |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1235 * a ; it is considered not to be complete. SQL string quotation using ' and |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1236 * SQL identifier quotation using " is taken into account when scanning a |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1237 * string this way. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1238 * Additionally, this method removes comments from the SQL statements, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1239 * identified by -- and removes white space where appropriate. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1240 * |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1241 * @param query the query to parse |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1242 * @param stack query stack to work with |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1243 * @param scolonterm whether a ';' makes this query part complete |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1244 * @return a QueryPart object containing the results of this parse |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1245 */ |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1246 private static QueryPart scanQuery( |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1247 final String query, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1248 final SQLStack stack, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1249 final boolean scolonterm) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1250 { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1251 // examine string, char for char |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1252 final boolean wasInString = (stack.peek() == '\''); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1253 final boolean wasInIdentifier = (stack.peek() == '"'); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1254 boolean escaped = false; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1255 int len = query.length(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1256 for (int i = 0; i < len; i++) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1257 switch(query.charAt(i)) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1258 case '\\': |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1259 escaped = !escaped; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1260 break; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1261 default: |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1262 escaped = false; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1263 break; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1264 case '\'': |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1265 /** |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1266 * We can not be in a string if we are in an identifier. So |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1267 * If we find a ' and are not in an identifier, and not in |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1268 * a string we can safely assume we will be now in a string. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1269 * If we are in a string already, we should stop being in a |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1270 * string if we find a quote which is not prefixed by a \, |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1271 * for that would be an escaped quote. However, a nasty |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1272 * situation can occur where the string is like 'test \\'. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1273 * As obvious, a test for a \ in front of a ' doesn't hold |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1274 * here. Because 'test \\\'' can exist as well, we need to |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1275 * know if a quote is prefixed by an escaping slash or not. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1276 */ |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1277 if (!escaped && stack.peek() != '"') { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1278 if (stack.peek() != '\'') { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1279 // although it makes no sense to escape a quote |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1280 // outside a string, it is escaped, thus not meant |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1281 // as quote for us, apparently |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1282 stack.push('\''); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1283 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1284 stack.pop(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1285 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1286 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1287 // reset escaped flag |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1288 escaped = false; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1289 break; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1290 case '"': |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1291 if (!escaped && stack.peek() != '\'') { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1292 if (stack.peek() != '"') { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1293 stack.push('"'); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1294 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1295 stack.pop(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1296 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1297 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1298 // reset escaped flag |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1299 escaped = false; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1300 break; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1301 case '-': |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1302 if (!escaped && stack.peek() != '\'' && stack.peek() != '"' && i + 1 < len && query.charAt(i + 1) == '-') { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1303 len = i; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1304 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1305 escaped = false; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1306 break; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1307 case '(': |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1308 if (!escaped && stack.peek() != '\'' && stack.peek() != '"') { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1309 stack.push('('); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1310 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1311 escaped = false; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1312 break; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1313 case ')': |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1314 if (!escaped && stack.peek() == '(') { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1315 stack.pop(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1316 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1317 escaped = false; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1318 break; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1319 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1320 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1321 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1322 int start = 0; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1323 if (!wasInString && !wasInIdentifier && len > 0) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1324 // trim spaces at the start of the string |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1325 for (; start < len && Character.isWhitespace(query.charAt(start)); start++); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1326 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1327 int stop = len - 1; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1328 if (stack.peek() != '\'' && !wasInIdentifier && stop > start) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1329 // trim spaces at the end of the string |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1330 for (; stop >= start && Character.isWhitespace(query.charAt(stop)); stop--); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1331 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1332 stop++; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1333 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1334 if (start == stop) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1335 // we have an empty string |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1336 return new QueryPart(false, null, stack.peek() == '\'' || stack.peek() == '"'); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1337 } else if (stack.peek() == '\'' || stack.peek() == '"') { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1338 // we have an open quote |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1339 return new QueryPart(false, query.substring(start, stop), true); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1340 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1341 // see if the string is complete |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1342 if (scolonterm && query.charAt(stop - 1) == ';') { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1343 return new QueryPart(true, query.substring(start, stop), false); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1344 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1345 return new QueryPart(false, query.substring(start, stop), false); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1346 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1347 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1348 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1349 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1350 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1351 /** |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1352 * A QueryPart is (a part of) a SQL query. In the QueryPart object information |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1353 * like the actual SQL query string, whether it has an open quote and the like |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1354 * is stored. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1355 */ |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1356 final class QueryPart { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1357 private final boolean complete; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1358 private final String query; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1359 private final boolean open; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1360 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1361 QueryPart(final boolean complete, final String query, final boolean open) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1362 this.complete = complete; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1363 this.query = query; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1364 this.open = open; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1365 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1366 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1367 boolean isEmpty() { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1368 return query == null; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1369 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1370 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1371 boolean isComplete() { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1372 return complete; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1373 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1374 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1375 String getQuery() { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1376 return query; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1377 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1378 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1379 boolean hasOpenQuote() { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1380 return open; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1381 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1382 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1383 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1384 /** |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1385 * An SQLStack is a simple stack that keeps track of open brackets and |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1386 * (single and double) quotes in an SQL query. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1387 */ |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1388 final class SQLStack { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1389 final StringBuilder stack = new StringBuilder(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1390 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1391 char peek() { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1392 if (empty()) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1393 return '\0'; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1394 } else { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1395 return stack.charAt(stack.length() - 1); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1396 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1397 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1398 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1399 char pop() { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1400 final char tmp = peek(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1401 if (tmp != '\0') { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1402 stack.setLength(stack.length() - 1); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1403 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1404 return tmp; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1405 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1406 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1407 char push(char item) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1408 stack.append(item); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1409 return item; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1410 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1411 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1412 boolean empty() { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1413 return stack.length() == 0; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1414 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1415 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1416 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1417 /** |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1418 * A Table represents an SQL table. All data required to |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1419 * generate a fully qualified name is stored, as well as dependency |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1420 * data. |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1421 */ |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1422 final class Table { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1423 final String schem; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1424 final String name; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1425 final String type; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1426 final String fqname; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1427 final ArrayList<Table> needs = new ArrayList<Table>(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1428 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1429 Table(final String schem, final String name, final String type) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1430 this.schem = schem; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1431 this.name = name; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1432 this.type = type; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1433 this.fqname = schem + "." + name; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1434 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1435 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1436 void addDependency(final Table dependsOn) throws Exception { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1437 if (this.fqname.equals(dependsOn.fqname)) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1438 throw new Exception("Cyclic dependency graphs are not supported (foreign key relation references self)"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1439 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1440 if (dependsOn.needs.contains(this)) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1441 throw new Exception("Cyclic dependency graphs are not supported (foreign key relation a->b and b->a)"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1442 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1443 if (!needs.contains(dependsOn)) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1444 needs.add(dependsOn); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1445 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1446 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1447 List<Table> requires(final List<Table> existingTables) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1448 if (existingTables == null || existingTables.isEmpty()) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1449 return new ArrayList<Table>(needs); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1450 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1451 final ArrayList<Table> req = new ArrayList<Table>(); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1452 for (Table n : needs) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1453 if (!existingTables.contains(n)) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1454 req.add(n); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1455 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1456 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1457 return req; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1458 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1459 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1460 final String getSchem() { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1461 return schem; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1462 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1463 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1464 final String getName() { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1465 return name; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1466 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1467 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1468 final String getType() { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1469 return type; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1470 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1471 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1472 final String getFqname() { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1473 return fqname; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1474 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1475 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1476 final String getFqnameQ() { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1477 return Exporter.dq(schem) + "." + Exporter.dq(name); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1478 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1479 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1480 public final String toString() { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1481 return fqname; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1482 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1483 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1484 static final Table findTable(final String schname, final String tblname, final List<Table> list) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1485 for (Table t : list) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1486 if (t.schem.equals(schname) && t.name.equals(tblname)) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1487 return t; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1488 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1489 // not found |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1490 return null; |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1491 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1492 |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1493 static final void checkForLoop(final Table table, final List<Table> parents) throws Exception { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1494 parents.add(table); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1495 for (int i = 0; i < table.needs.size(); i++) { |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1496 Table child = table.needs.get(i); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1497 if (parents.contains(child)) |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1498 throw new Exception("Cyclic dependency graphs are not supported (cycle detected for " + child.fqname + ")"); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1499 checkForLoop(child, parents); |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1500 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1501 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1502 } |
844139b33cdd
Remove carriage returns (\r) from files.
Sjoerd Mullender <sjoerd@acm.org>
parents:
628
diff
changeset
|
1503 |