Skip to content

Commit

Permalink
能够显示sql shell的用法
Browse files Browse the repository at this point in the history
  • Loading branch information
@codefollower
codefollower committed Feb 29, 2016
1 parent 251affb commit d23836d
Show file tree
Hide file tree
Showing 4 changed files with 42 additions and 179 deletions.
190 changes: 25 additions & 165 deletions lealone-main/src/main/java/org/lealone/main/Shell.java
View file
Expand Up @@ -6,7 +6,6 @@
package org.lealone.main; package org.lealone.main;


import java.io.BufferedReader; import java.io.BufferedReader;
import java.io.ByteArrayInputStream;
import java.io.IOException; import java.io.IOException;
import java.io.InputStream; import java.io.InputStream;
import java.io.InputStreamReader; import java.io.InputStreamReader;
Expand All @@ -19,135 +18,66 @@
import java.sql.SQLException; import java.sql.SQLException;
import java.sql.Statement; import java.sql.Statement;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.Properties;


import org.lealone.api.ErrorCode;
import org.lealone.common.exceptions.DbException;
import org.lealone.common.util.JdbcUtils; import org.lealone.common.util.JdbcUtils;
import org.lealone.common.util.New; import org.lealone.common.util.New;
import org.lealone.common.util.StringUtils; import org.lealone.common.util.StringUtils;
import org.lealone.common.util.Utils;
import org.lealone.db.Constants; import org.lealone.db.Constants;
import org.lealone.db.util.ScriptReader; import org.lealone.db.util.ScriptReader;


/** /**
* Interactive command line tool to access a database using JDBC. * Interactive command line tool to access a database using JDBC.
* @h2.resource
* *
* @author H2 Group * @author H2 Group
* @author zhh * @author zhh
*/ */
public class Shell implements Runnable { public class Shell {


private static final int MAX_ROW_BUFFER = 5000; private static final int MAX_ROW_BUFFER = 5000;
private static final int HISTORY_COUNT = 20; private static final int HISTORY_COUNT = 20;
// Windows: '\u00b3'; // Windows: '\u00b3';
private static final char BOX_VERTICAL = '|'; private static final char BOX_VERTICAL = '|';


private PrintStream err = System.err; private final PrintStream err = System.err;
private InputStream in = System.in; private final InputStream in = System.in;
private final PrintStream out = System.out;
private BufferedReader reader; private BufferedReader reader;
private Connection conn; private Connection conn;
private Statement stat; private Statement stat;
private boolean listMode; private boolean listMode;
private int maxColumnSize = 100; private int maxColumnSize = 100;
private final ArrayList<String> history = New.arrayList(); private final ArrayList<String> history = New.arrayList();
private boolean stopHide;


private Properties resources;

/**
* The output stream where this tool writes to.
*/
protected PrintStream out = System.out;

// private String serverPropertiesDir = Constants.SERVER_PROPERTIES_DIR;

/**
* Options are case sensitive. Supported options are:
* <table>
* <tr><td>[-help] or [-?]</td>
* <td>Print the list of options</td></tr>
* <tr><td>[-url "&lt;url&gt;"]</td>
* <td>The database URL (jdbc:lealone:...)</td></tr>
* <tr><td>[-user &lt;user&gt;]</td>
* <td>The user name</td></tr>
* <tr><td>[-password &lt;pwd&gt;]</td>
* <td>The password</td></tr>
* <tr><td>[-sql "&lt;statements&gt;"]</td>
* <td>Execute the SQL statements and exit</td></tr>
* <tr><td>[-properties "&lt;dir&gt;"]</td>
* <td>Load the server properties from this directory</td></tr>
* </table>
* If special characters don't work as expected, you may need to use
* -Dfile.encoding=UTF-8 (Mac OS X) or CP850 (Windows).
* @h2.resource
*
* @param args the command line arguments
*/
public static void main(String... args) throws SQLException { public static void main(String... args) throws SQLException {
new Shell().runTool(args); new Shell().run(args);
}

/**
* Sets the standard error stream.
*
* @param err the new standard error stream
*/
public void setErr(PrintStream err) {
this.err = err;
}

/**
* Redirects the standard input. By default, System.in is used.
*
* @param in the input stream to use
*/
public void setIn(InputStream in) {
this.in = in;
} }


/** private void run(String... args) throws SQLException {
* Redirects the standard input. By default, System.in is used.
*
* @param reader the input stream reader to use
*/
public void setInReader(BufferedReader reader) {
this.reader = reader;
}

/**
* Run the shell tool with the given command line settings.
*
* @param args the command line settings
*/
public void runTool(String... args) throws SQLException {
String url = null; String url = null;
String user = ""; String user = "";
String password = ""; String password = "";
String sql = null; String sql = null;
for (int i = 0; args != null && i < args.length; i++) { for (int i = 0; args != null && i < args.length; i++) {
String arg = args[i]; String arg = args[i];
arg = arg.trim();
if (arg.isEmpty())
continue;
if (arg.equals("-url")) { if (arg.equals("-url")) {
url = args[++i]; url = args[++i];
} else if (arg.equals("-user")) { } else if (arg.equals("-user")) {
user = args[++i]; user = args[++i];
} else if (arg.equals("-password")) { } else if (arg.equals("-password")) {
password = args[++i]; password = args[++i];
// } else if (arg.equals("-driver")) {
// String driver = args[++i];
// Utils.loadUserClass(driver);
} else if (arg.equals("-sql")) { } else if (arg.equals("-sql")) {
sql = args[++i]; sql = args[++i];
} else if (arg.equals("-properties")) {
// serverPropertiesDir = args[++i];
} else if (arg.equals("-help") || arg.equals("-?")) { } else if (arg.equals("-help") || arg.equals("-?")) {
showUsage(); showUsage();
return; return;
} else if (arg.equals("-list")) { } else if (arg.equals("-list")) {
listMode = true; listMode = true;
} else { } else {
showUsageAndThrowUnsupportedOption(arg); showUsage();
return;
} }
} }
if (url != null) { if (url != null) {
Expand All @@ -171,22 +101,17 @@ public void runTool(String... args) throws SQLException {
} }
} }


/** private void showUsage() {
* Run the shell tool with the given connection and command line settings. println("Options are case sensitive. Supported options are:");
* The connection will be closed when the shell exits. println("[-help] or [-?] Print the list of options");
* This is primary used to integrate the Shell into another application. println("[-url \"<url>\"] The database URL (jdbc:lealone:...)");
* <p> println("[-user <user>] The user name");
* Note: using the "-url" option in {@code args} doesn't make much sense println("[-password <pwd>] The password");
* since it will override the {@code conn} parameter. println("[-sql \"<statements>\"] Execute the SQL statements and exit");
* </p> println("");
* println("If special characters don't work as expected, ");
* @param conn the connection println("you may need to use -Dfile.encoding=UTF-8 (Mac OS X) or CP850 (Windows).");
* @param args the command line settings println("");
*/
public void runTool(Connection conn, String... args) throws SQLException {
this.conn = conn;
this.stat = conn.createStatement();
runTool(args);
} }


private void showHelp() { private void showHelp() {
Expand All @@ -202,7 +127,7 @@ private void showHelp() {


private void promptLoop() { private void promptLoop() {
println(""); println("");
println("Welcome to Lealone Shell " + Constants.getFullVersion()); println("Welcome to Lealone Shell " + Constants.getVersion());
println("Exit with Ctrl+C"); println("Exit with Ctrl+C");
if (conn != null) { if (conn != null) {
showHelp(); showHelp();
Expand All @@ -227,7 +152,7 @@ private void promptLoop() {
break; break;
} }
String trimmed = line.trim(); String trimmed = line.trim();
if (trimmed.length() == 0) { if (trimmed.isEmpty()) {
continue; continue;
} }
boolean end = trimmed.endsWith(";"); boolean end = trimmed.endsWith(";");
Expand Down Expand Up @@ -346,12 +271,7 @@ private void connect() throws IOException, SQLException {
println("Connected"); println("Connected");
} }


/** private void print(String s) {
* Print the string without newline, and flush.
*
* @param s the string to print
*/
protected void print(String s) {
out.print(s); out.print(s);
out.flush(); out.flush();
} }
Expand All @@ -372,23 +292,6 @@ private String readPassword() throws IOException {
} }
} }


/**
* INTERNAL.
* Hides the password by repeatedly printing
* backspace, backspace, &gt;, &lt;.
*/
@Override
public void run() {
while (!stopHide) {
print("\b\b><");
try {
Thread.sleep(10);
} catch (InterruptedException e) {
// ignore
}
}
}

private String readLine(String defaultValue) throws IOException { private String readLine(String defaultValue) throws IOException {
String s = readLine(); String s = readLine();
return s.length() == 0 ? defaultValue : s; return s.length() == 0 ? defaultValue : s;
Expand Down Expand Up @@ -563,47 +466,4 @@ private int printResultAsList(ResultSet rs) throws SQLException {
return rowCount; return rowCount;
} }


/**
* Print the usage of the tool. This method reads the description from the
* resource file.
*/
protected void showUsage() {
if (resources == null) {
resources = new Properties();
String resourceName = "/org/lealone/res/javadoc.properties";
try {
byte[] buff = Utils.getResource(resourceName);
if (buff != null) {
resources.load(new ByteArrayInputStream(buff));
}
} catch (IOException e) {
out.println("Cannot load " + resourceName);
}
}
String className = getClass().getName();
out.println(resources.get(className));
out.println("Usage: java " + getClass().getName() + " <options>");
out.println(resources.get(className + ".main"));
}

/**
* Throw a SQLException saying this command line option is not supported.
*
* @param option the unsupported option
* @return this method never returns normally
*/
protected SQLException showUsageAndThrowUnsupportedOption(String option) throws SQLException {
showUsage();
throw throwUnsupportedOption(option);
}

/**
* Throw a SQLException saying this command line option is not supported.
*
* @param option the unsupported option
* @return this method never returns normally
*/
protected SQLException throwUnsupportedOption(String option) throws SQLException {
throw DbException.get(ErrorCode.FEATURE_NOT_SUPPORTED_1, option).getSQLException();
}
} }
4 changes: 3 additions & 1 deletion lealone/bin/lealone.bat
View file
Expand Up @@ -17,12 +17,14 @@
@echo off @echo off
if "%OS%" == "Windows_NT" setlocal if "%OS%" == "Windows_NT" setlocal


if NOT DEFINED JAVA_HOME goto :err

pushd %~dp0.. pushd %~dp0..
if NOT DEFINED LEALONE_HOME set LEALONE_HOME=%CD% if NOT DEFINED LEALONE_HOME set LEALONE_HOME=%CD%
popd popd


if NOT DEFINED LEALONE_MAIN set LEALONE_MAIN=org.lealone.main.Lealone if NOT DEFINED LEALONE_MAIN set LEALONE_MAIN=org.lealone.main.Lealone
if NOT DEFINED JAVA_HOME goto :err


REM ***** JAVA options ***** REM ***** JAVA options *****
REM set JAVA_OPTS=-ea^ REM set JAVA_OPTS=-ea^
Expand Down
14 changes: 6 additions & 8 deletions lealone/bin/sqlshell.bat
View file
Expand Up @@ -17,15 +17,13 @@
@echo off @echo off
if "%OS%" == "Windows_NT" setlocal if "%OS%" == "Windows_NT" setlocal


if NOT DEFINED JAVA_HOME goto :err

pushd %~dp0.. pushd %~dp0..
if NOT DEFINED LEALONE_HOME set LEALONE_HOME=%CD% if NOT DEFINED LEALONE_HOME set LEALONE_HOME=%CD%
popd popd


if NOT DEFINED LEALONE_MAIN set LEALONE_MAIN=org.lealone.main.Shell if NOT DEFINED LEALONE_MAIN set LEALONE_MAIN=org.lealone.main.Shell
if NOT DEFINED JAVA_HOME goto :err

REM ***** JAVA options *****
set JAVA_OPTS=-Dlogback.configurationFile=logback.xml


REM ***** CLASSPATH library setting ***** REM ***** CLASSPATH library setting *****


Expand All @@ -42,12 +40,12 @@ goto :eof


:okClasspath :okClasspath
set LEALONE_CLASSPATH=%CLASSPATH%; set LEALONE_CLASSPATH=%CLASSPATH%;
set LEALONE_PARAMS=-Dlealone.logdir="%LEALONE_HOME%\logs" set LEALONE_PARAMS="%1 %2"
REM set LEALONE_PARAMS=%LEALONE_PARAMS% -agentlib:jdwp=transport=dt_socket,address=8000,server=y,suspend=y REM set LEALONE_PARAMS=%LEALONE_PARAMS% -agentlib:jdwp=transport=dt_socket,address=8000,server=y,suspend=y
goto runDaemon goto runShell


:runDaemon :runShell
"%JAVA_HOME%\bin\java" %JAVA_OPTS% %LEALONE_PARAMS% -cp %LEALONE_CLASSPATH% "%LEALONE_MAIN%" "%JAVA_HOME%\bin\java" -cp %LEALONE_CLASSPATH% "%LEALONE_MAIN%" %LEALONE_PARAMS%
goto finally goto finally


:err :err
Expand Down
13 changes: 8 additions & 5 deletions lealone/bin/sqlshell.sh
View file
Expand Up @@ -14,6 +14,7 @@
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.

if [ "x$LEALONE_HOME" = "x" ]; then if [ "x$LEALONE_HOME" = "x" ]; then
LEALONE_HOME="`dirname "$0"`/.." LEALONE_HOME="`dirname "$0"`/.."
fi fi
Expand All @@ -25,17 +26,19 @@ fi


LEALONE_MAIN=org.lealone.main.Shell LEALONE_MAIN=org.lealone.main.Shell


JAVA_OPTS=-Dlogback.configurationFile=logback.xml

CLASSPATH=$LEALONE_HOME/conf CLASSPATH=$LEALONE_HOME/conf


for jar in "$LEALONE_HOME"/lib/*.jar; do for jar in "$LEALONE_HOME"/lib/*.jar; do
CLASSPATH="$CLASSPATH:$jar" CLASSPATH="$CLASSPATH:$jar"
done done


LEALONE_CLASSPATH="$CLASSPATH" LEALONE_CLASSPATH="$CLASSPATH"
LEALONE_PARAMS="-Dlealone.logdir=$LEALONE_HOME/logs" if [ "x$1" = "x" ]; then
LEALONE_PARAMS="$1"
fi
if [ "x$2" = "x" ]; then
LEALONE_PARAMS="$LEALONE_PARAMS $2"
fi
# LEALONE_PARAMS="$LEALONE_PARAMS -agentlib:jdwp=transport=dt_socket,address=8000,server=y,suspend=y" # LEALONE_PARAMS="$LEALONE_PARAMS -agentlib:jdwp=transport=dt_socket,address=8000,server=y,suspend=y"



"$JAVA_HOME/bin/java" -cp $LEALONE_CLASSPATH $LEALONE_MAIN $LEALONE_PARAMS
"$JAVA_HOME/bin/java" $JAVA_OPTS $LEALONE_PARAMS -cp $LEALONE_CLASSPATH $LEALONE_MAIN

0 comments on commit d23836d

Please sign in to comment.