jixuan1989
/
IoTDB
mirror of https://github.com/apache/iotdb
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

add session module and rename cli to client

This commit is contained in:
qiaojialin 2019-08-24 11:25:42 +08:00
parent c9d040ca4d
commit 5b1a318cbd
36 changed files with 736 additions and 657 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
README.md
View File

@ -183,9 +183,9 @@ The server can be stopped with ctrl-C or the following script:
> $IOTDB_HOME\sbin\stop-server.bat
```
### Start Cli
### Start Client
Now let's trying to read and write some data from IoTDB using our Cli. To start the client, you need to explicit the server's IP and PORT as well as the USER_NAME and PASSWORD.
Now let's trying to read and write some data from IoTDB using our Client. To start the client, you need to explicit the server's IP and PORT as well as the USER_NAME and PASSWORD.
```
# You can first build cli project

2
client/pom.xml
View File

@ -28,7 +28,7 @@
<relativePath>../pom.xml</relativePath>
</parent>
<artifactId>iotdb-client</artifactId>
<name>IoTDB Client and Cli</name>
<name>IoTDB Client</name>
<properties>
<cli.test.skip>false</cli.test.skip>
<cli.it.skip>${cli.test.skip}</cli.it.skip>

2
client/src/assembly/resources/sbin/start-cli.bat → client/src/assembly/resources/sbin/start-client.bat
View File

@ -28,7 +28,7 @@ pushd %~dp0..
if NOT DEFINED IOTDB_CLI_HOME set IOTDB_CLI_HOME=%CD%
popd
if NOT DEFINED MAIN_CLASS set MAIN_CLASS=org.apache.iotdb.cli.Cli
if NOT DEFINED MAIN_CLASS set MAIN_CLASS=org.apache.iotdb.client.Client
if NOT DEFINED JAVA_HOME goto :err
@REM -----------------------------------------------------------------------------

2
client/src/assembly/resources/sbin/start-cli.sh → client/src/assembly/resources/sbin/start-client.sh
View File

@ -24,7 +24,7 @@ if [ -z "${IOTDB_CLI_HOME}" ]; then
fi
MAIN_CLASS=org.apache.iotdb.cli.Cli
MAIN_CLASS=org.apache.iotdb.client.Client
CLASSPATH=""

187
client/src/main/java/org/apache/iotdb/cli/Cli.java
View File

@ -1,187 +0,0 @@
/**
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package org.apache.iotdb.cli;
import java.io.IOException;
import java.sql.DriverManager;
import java.sql.SQLException;
import jline.console.ConsoleReader;
import org.apache.commons.cli.CommandLine;
import org.apache.commons.cli.CommandLineParser;
import org.apache.commons.cli.DefaultParser;
import org.apache.commons.cli.HelpFormatter;
import org.apache.commons.cli.Options;
import org.apache.commons.cli.ParseException;
import org.apache.iotdb.exception.ArgsErrorException;
import org.apache.iotdb.jdbc.Config;
import org.apache.iotdb.jdbc.IoTDBConnection;
import org.apache.thrift.TException;
public class Cli extends AbstractCli {
private static CommandLine commandLine;
/**
* IoTDB CLI main function.
*
* @param args launch arguments
* @throws ClassNotFoundException ClassNotFoundException
*/
public static void main(String[] args) throws ClassNotFoundException {
Class.forName(Config.JDBC_DRIVER_NAME);
Options options = createOptions();
HelpFormatter hf = new HelpFormatter();
hf.setWidth(MAX_HELP_CONSOLE_WIDTH);
commandLine = null;
String[] newArgs;
String[] newArgs2;
if (args == null || args.length == 0) {
println(
"Require more params input, eg. ./start-cli.sh(start-cli.bat if Windows) "
+ "-h xxx.xxx.xxx.xxx -p xxxx -u xxx.");
println("For more information, please check the following hint.");
hf.printHelp(SCRIPT_HINT, options, true);
return;
}
init();
newArgs = removePasswordArgs(args);
newArgs2 = processExecuteArgs(newArgs);
boolean continues = parseCommandLine(options, newArgs2, hf);
if (!continues) {
return;
}
serve();
}
private static boolean parseCommandLine(Options options, String[] newArgs, HelpFormatter hf) {
try {
CommandLineParser parser = new DefaultParser();
commandLine = parser.parse(options, newArgs);
if (commandLine.hasOption(HELP_ARGS)) {
hf.printHelp(SCRIPT_HINT, options, true);
return false;
}
if (commandLine.hasOption(ISO8601_ARGS)) {
setTimeFormat("long");
}
if (commandLine.hasOption(MAX_PRINT_ROW_COUNT_ARGS)) {
setMaxDisplayNumber(commandLine.getOptionValue(MAX_PRINT_ROW_COUNT_ARGS));
}
} catch (ParseException e) {
println(
"Require more params input, eg. ./start-cli.sh(start-cli.bat if Windows) "
+ "-h xxx.xxx.xxx.xxx -p xxxx -u xxx.");
println("For more information, please check the following hint.");
hf.printHelp(IOTDB_CLI_PREFIX, options, true);
handleException(e);
return false;
} catch (NumberFormatException e) {
println(
IOTDB_CLI_PREFIX + "> error format of max print row count, it should be number");
handleException(e);
return false;
}
return true;
}
private static void serve() {
try (ConsoleReader reader = new ConsoleReader()) {
reader.setExpandEvents(false);
host = checkRequiredArg(HOST_ARGS, HOST_NAME, commandLine, false, host);
port = checkRequiredArg(PORT_ARGS, PORT_NAME, commandLine, false, port);
username = checkRequiredArg(USERNAME_ARGS, USERNAME_NAME, commandLine, true, null);
password = commandLine.getOptionValue(PASSWORD_ARGS);
if (password == null) {
password = reader.readLine("please input your password:", '\0');
}
if (hasExecuteSQL) {
try (IoTDBConnection connection = (IoTDBConnection) DriverManager
.getConnection(Config.IOTDB_URL_PREFIX + host + ":" + port + "/", username, password)) {
properties = connection.getServerProperties();
AGGREGRATE_TIME_LIST.addAll(properties.getSupportedTimeAggregationOperations());
processCmd(execute, connection);
return;
} catch (SQLException e) {
handleException(e);
}
}
receiveCommands(reader);
} catch (ArgsErrorException e) {
println(IOTDB_CLI_PREFIX + "> input params error because" + e.getMessage());
handleException(e);
} catch (Exception e) {
println(IOTDB_CLI_PREFIX + "> exit client with error " + e.getMessage());
handleException(e);
}
}
private static void receiveCommands(ConsoleReader reader) throws TException, IOException {
try (IoTDBConnection connection = (IoTDBConnection) DriverManager
.getConnection(Config.IOTDB_URL_PREFIX + host + ":" + port + "/", username, password)) {
String s;
properties = connection.getServerProperties();
AGGREGRATE_TIME_LIST.addAll(properties.getSupportedTimeAggregationOperations());
TIMESTAMP_PRECISION = properties.getTimestampPrecision();
echoStarting();
displayLogo(properties.getVersion());
println(IOTDB_CLI_PREFIX + "> login successfully");
while (true) {
s = reader.readLine(IOTDB_CLI_PREFIX + "> ", null);
boolean continues = processCmd(s, connection);
if (!continues) {
break;
}
}
} catch (SQLException e) {
println(String
.format("%s> %s Host is %s, port is %s.", IOTDB_CLI_PREFIX, e.getMessage(), host,
port));
handleException(e);
}
}
private static boolean processCmd(String s, IoTDBConnection connection) {
if (s == null) {
return true;
}
String[] cmds = s.trim().split(";");
for (int i = 0; i < cmds.length; i++) {
String cmd = cmds[i];
if (cmd != null && !"".equals(cmd.trim())) {
OperationResult result = handleInputCmd(cmd, connection);
switch (result) {
case STOP_OPER:
return false;
case CONTINUE_OPER:
continue;
default:
break;
}
}
}
return true;
}
}

10
client/src/main/java/org/apache/iotdb/cli/AbstractCli.java → client/src/main/java/org/apache/iotdb/client/AbstractClient.java
View File

@ -16,7 +16,7 @@
* specific language governing permissions and limitations
* under the License.
*/
package org.apache.iotdb.cli;
package org.apache.iotdb.client;
import java.io.PrintStream;
import java.sql.ResultSet;
@ -47,7 +47,7 @@ import org.apache.iotdb.jdbc.IoTDBQueryResultSet;
import org.apache.iotdb.service.rpc.thrift.ServerProperties;
import org.apache.thrift.TException;
public abstract class AbstractCli {
public abstract class AbstractClient {
static final String HOST_ARGS = "h";
static final String HOST_NAME = "host";
@ -79,7 +79,7 @@ public abstract class AbstractCli {
static final String SHOW_FETCH_SIZE = "show fetch_size";
private static final String HELP = "help";
static final String IOTDB_CLI_PREFIX = "IoTDB";
static final String SCRIPT_HINT = "./start-cli.sh(start-cli.bat if Windows)";
static final String SCRIPT_HINT = "./start-client.sh(start-client.bat if Windows)";
static final String QUIT_COMMAND = "quit";
static final String EXIT_COMMAND = "exit";
static final String SHOW_METADATA_COMMAND = "show timeseries";
@ -176,7 +176,7 @@ public abstract class AbstractCli {
}
/**
* CLI result output.
* Client result output.
*
* @param res result set
* @param printToConsole print to console
@ -585,7 +585,7 @@ public abstract class AbstractCli {
static void echoStarting(){
println("---------------------");
println("Starting IoTDB Cli");
println("Starting IoTDB Client");
println("---------------------");
}

308
client/src/main/java/org/apache/iotdb/client/Client.java
View File

@ -7,7 +7,7 @@
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
@ -18,172 +18,170 @@
*/
package org.apache.iotdb.client;
import java.time.ZoneId;
import org.apache.iotdb.exception.IoTDBSessionException;
import java.io.IOException;
import java.sql.DriverManager;
import java.sql.SQLException;
import jline.console.ConsoleReader;
import org.apache.commons.cli.CommandLine;
import org.apache.commons.cli.CommandLineParser;
import org.apache.commons.cli.DefaultParser;
import org.apache.commons.cli.HelpFormatter;
import org.apache.commons.cli.Options;
import org.apache.commons.cli.ParseException;
import org.apache.iotdb.exception.ArgsErrorException;
import org.apache.iotdb.jdbc.Config;
import org.apache.iotdb.jdbc.IoTDBConnection;
import org.apache.iotdb.jdbc.IoTDBSQLException;
import org.apache.iotdb.jdbc.Utils;
import org.apache.iotdb.service.rpc.thrift.TSBatchInsertionReq;
import org.apache.iotdb.service.rpc.thrift.TSCloseSessionReq;
import org.apache.iotdb.service.rpc.thrift.TSExecuteBatchStatementResp;
import org.apache.iotdb.service.rpc.thrift.TSGetTimeZoneResp;
import org.apache.iotdb.service.rpc.thrift.TSIService;
import org.apache.iotdb.service.rpc.thrift.TSOpenSessionReq;
import org.apache.iotdb.service.rpc.thrift.TSOpenSessionResp;
import org.apache.iotdb.service.rpc.thrift.TSProtocolVersion;
import org.apache.iotdb.service.rpc.thrift.TSSetTimeZoneReq;
import org.apache.iotdb.service.rpc.thrift.TSSetTimeZoneResp;
import org.apache.iotdb.service.rpc.thrift.TS_SessionHandle;
import org.apache.iotdb.tsfile.write.record.RowBatch;
import org.apache.iotdb.tsfile.write.schema.MeasurementSchema;
import org.apache.thrift.TException;
import org.apache.thrift.protocol.TBinaryProtocol;
import org.apache.thrift.protocol.TCompactProtocol;
import org.apache.thrift.transport.TSocket;
import org.apache.thrift.transport.TTransportException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
public class Client {
public class Client extends AbstractClient {
private static final Logger logger = LoggerFactory.getLogger(Client.class);
private String host;
private int port;
private String username;
private String password;
private final TSProtocolVersion protocolVersion = TSProtocolVersion.IOTDB_SERVICE_PROTOCOL_V1;
public TSIService.Iface client = null;
private TS_SessionHandle sessionHandle = null;
private TSocket transport;
private boolean isClosed = true;
private ZoneId zoneId;
private static CommandLine commandLine;
public Client(String host, int port) {
this(host, port, Config.DEFAULT_USER, Config.DEFALUT_PASSWORD);
}
/**
* IoTDB Client main function.
*
* @param args launch arguments
* @throws ClassNotFoundException ClassNotFoundException
*/
public static void main(String[] args) throws ClassNotFoundException {
Class.forName(Config.JDBC_DRIVER_NAME);
Options options = createOptions();
HelpFormatter hf = new HelpFormatter();
hf.setWidth(MAX_HELP_CONSOLE_WIDTH);
commandLine = null;
public Client(String host, String port, String username, String password) {
this(host, Integer.parseInt(port), username, password);
}
String[] newArgs;
String[] newArgs2;
public Client(String host, int port, String username, String password) {
this.host = host;
this.port = port;
this.username = username;
this.password = password;
}
public void open() {
open(false, 0);
}
public void open(boolean enableRPCCompression, int connectionTimeoutInMs) {
transport = new TSocket(host, port, connectionTimeoutInMs);
if (!transport.isOpen()) {
try {
transport.open();
} catch (TTransportException e) {
throw new IoTDBSessionException(e);
}
}
if(enableRPCCompression) {
client = new TSIService.Client(new TCompactProtocol(transport));
}
else {
client = new TSIService.Client(new TBinaryProtocol(transport));
}
TSOpenSessionReq openReq = new TSOpenSessionReq(TSProtocolVersion.IOTDB_SERVICE_PROTOCOL_V1);
openReq.setUsername(username);
openReq.setPassword(password);
try {
TSOpenSessionResp openResp = client.openSession(openReq);
// validate connection
try {
Utils.verifySuccess(openResp.getStatus());
} catch (IoTDBSQLException e) {
transport.close();
throw e;
}
if (protocolVersion.getValue() != openResp.getServerProtocolVersion().getValue()) {
throw new TException(String
.format("Protocol not supported, Client version is {}, but Server version is {}",
protocolVersion.getValue(), openResp.getServerProtocolVersion().getValue()));
}
sessionHandle = openResp.getSessionHandle();
if (zoneId != null) {
setTimeZone(zoneId.toString());
} else {
zoneId = ZoneId.of(getTimeZone());
}
} catch (TException | IoTDBSQLException e) {
throw new IoTDBSessionException(String.format("Can not open session to %s:%s with user: %s.",
host, port, username), e);
}
isClosed = false;
client = IoTDBConnection.newSynchronizedClient(client);
}
public void close() {
if (isClosed) {
if (args == null || args.length == 0) {
println(
"Require more params input, eg. ./start-client.sh(start-client.bat if Windows) "
+ "-h xxx.xxx.xxx.xxx -p xxxx -u xxx.");
println("For more information, please check the following hint.");
hf.printHelp(SCRIPT_HINT, options, true);
return;
}
TSCloseSessionReq req = new TSCloseSessionReq(sessionHandle);
init();
newArgs = removePasswordArgs(args);
newArgs2 = processExecuteArgs(newArgs);
boolean continues = parseCommandLine(options, newArgs2, hf);
if (!continues) {
return;
}
serve();
}
private static boolean parseCommandLine(Options options, String[] newArgs, HelpFormatter hf) {
try {
client.closeSession(req);
} catch (TException e) {
throw new IoTDBSessionException("Error occurs when closing session at server. Maybe server is down.", e);
} finally {
isClosed = true;
if (transport != null) {
transport.close();
CommandLineParser parser = new DefaultParser();
commandLine = parser.parse(options, newArgs);
if (commandLine.hasOption(HELP_ARGS)) {
hf.printHelp(SCRIPT_HINT, options, true);
return false;
}
if (commandLine.hasOption(ISO8601_ARGS)) {
setTimeFormat("long");
}
if (commandLine.hasOption(MAX_PRINT_ROW_COUNT_ARGS)) {
setMaxDisplayNumber(commandLine.getOptionValue(MAX_PRINT_ROW_COUNT_ARGS));
}
} catch (ParseException e) {
println(
"Require more params input, eg. ./start-client.sh(start-client.bat if Windows) "
+ "-h xxx.xxx.xxx.xxx -p xxxx -u xxx.");
println("For more information, please check the following hint.");
hf.printHelp(IOTDB_CLI_PREFIX, options, true);
handleException(e);
return false;
} catch (NumberFormatException e) {
println(
IOTDB_CLI_PREFIX + "> error format of max print row count, it should be number");
handleException(e);
return false;
}
return true;
}
private static void serve() {
try (ConsoleReader reader = new ConsoleReader()) {
reader.setExpandEvents(false);
host = checkRequiredArg(HOST_ARGS, HOST_NAME, commandLine, false, host);
port = checkRequiredArg(PORT_ARGS, PORT_NAME, commandLine, false, port);
username = checkRequiredArg(USERNAME_ARGS, USERNAME_NAME, commandLine, true, null);
password = commandLine.getOptionValue(PASSWORD_ARGS);
if (password == null) {
password = reader.readLine("please input your password:", '\0');
}
if (hasExecuteSQL) {
try (IoTDBConnection connection = (IoTDBConnection) DriverManager
.getConnection(Config.IOTDB_URL_PREFIX + host + ":" + port + "/", username, password)) {
properties = connection.getServerProperties();
AGGREGRATE_TIME_LIST.addAll(properties.getSupportedTimeAggregationOperations());
processCmd(execute, connection);
return;
} catch (SQLException e) {
handleException(e);
}
}
receiveCommands(reader);
} catch (ArgsErrorException e) {
println(IOTDB_CLI_PREFIX + "> input params error because" + e.getMessage());
handleException(e);
} catch (Exception e) {
println(IOTDB_CLI_PREFIX + "> exit client with error " + e.getMessage());
handleException(e);
}
}
private static void receiveCommands(ConsoleReader reader) throws TException, IOException {
try (IoTDBConnection connection = (IoTDBConnection) DriverManager
.getConnection(Config.IOTDB_URL_PREFIX + host + ":" + port + "/", username, password)) {
String s;
properties = connection.getServerProperties();
AGGREGRATE_TIME_LIST.addAll(properties.getSupportedTimeAggregationOperations());
TIMESTAMP_PRECISION = properties.getTimestampPrecision();
echoStarting();
displayLogo(properties.getVersion());
println(IOTDB_CLI_PREFIX + "> login successfully");
while (true) {
s = reader.readLine(IOTDB_CLI_PREFIX + "> ", null);
boolean continues = processCmd(s, connection);
if (!continues) {
break;
}
}
} catch (SQLException e) {
println(String
.format("%s> %s Host is %s, port is %s.", IOTDB_CLI_PREFIX, e.getMessage(), host,
port));
handleException(e);
}
}
private static boolean processCmd(String s, IoTDBConnection connection) {
if (s == null) {
return true;
}
String[] cmds = s.trim().split(";");
for (int i = 0; i < cmds.length; i++) {
String cmd = cmds[i];
if (cmd != null && !"".equals(cmd.trim())) {
OperationResult result = handleInputCmd(cmd, connection);
switch (result) {
case STOP_OPER:
return false;
case CONTINUE_OPER:
continue;
default:
break;
}
}
}
return true;
}
public TSExecuteBatchStatementResp insertBatch(RowBatch rowBatch) {
TSBatchInsertionReq request = new TSBatchInsertionReq();
request.deviceId = rowBatch.deviceId;
for (MeasurementSchema measurementSchema: rowBatch.measurements) {
request.addToMeasurements(measurementSchema.getMeasurementId());
request.addToTypes(measurementSchema.getType().ordinal());
}
request.setTimestamps(Utils.getTimeBuffer(rowBatch));
request.setValues(Utils.getValueBuffer(rowBatch));
request.setSize(rowBatch.batchSize);
try {
return client.insertBatch(request);
} catch (TException e) {
throw new IoTDBSessionException(e);
}
}
public String getTimeZone() throws TException, IoTDBSQLException {
if (zoneId != null) {
return zoneId.toString();
}
TSGetTimeZoneResp resp = client.getTimeZone();
Utils.verifySuccess(resp.getStatus());
return resp.getTimeZone();
}
public void setTimeZone(String zoneId) throws TException, IoTDBSQLException {
TSSetTimeZoneReq req = new TSSetTimeZoneReq(zoneId);
TSSetTimeZoneResp resp = client.setTimeZone(req);
Utils.verifySuccess(resp.getStatus());
this.zoneId = ZoneId.of(zoneId);
}
}

6
client/src/main/java/org/apache/iotdb/cli/WinCli.java → client/src/main/java/org/apache/iotdb/client/WinClient.java
View File

@ -16,7 +16,7 @@
* specific language governing permissions and limitations
* under the License.
*/
package org.apache.iotdb.cli;
package org.apache.iotdb.client;
import java.io.Console;
import java.sql.DriverManager;
@ -33,7 +33,7 @@ import org.apache.iotdb.jdbc.Config;
import org.apache.iotdb.jdbc.IoTDBConnection;
import org.apache.thrift.TException;
public class WinCli extends AbstractCli {
public class WinClient extends AbstractClient {
private static CommandLine commandLine;
@ -71,7 +71,7 @@ public class WinCli extends AbstractCli {
private static String readPassword() {
Console c = System.console();
if (c == null) { // IN ECLIPSE IDE
if (c == null) { // IN ECLIENTPSE IDE
print(IOTDB_CLI_PREFIX + "> please input password: ");
Scanner scanner = new Scanner(System.in);
return scanner.nextLine();

4
client/src/main/java/org/apache/iotdb/tool/ExportCsv.java
View File

@ -41,7 +41,7 @@ import org.apache.commons.cli.HelpFormatter;
import org.apache.commons.cli.Option;
import org.apache.commons.cli.Options;
import org.apache.commons.cli.ParseException;
import org.apache.iotdb.cli.AbstractCli;
import org.apache.iotdb.client.AbstractClient;
import org.apache.iotdb.exception.ArgsErrorException;
import org.apache.iotdb.jdbc.Config;
import org.apache.iotdb.jdbc.IoTDBConnection;
@ -322,7 +322,7 @@ public class ExportCsv extends AbstractCsvTool {
switch (timeFormat) {
case "default":
long timestamp = rs.getLong(1);
String str = AbstractCli
String str = AbstractClient
.parseLongToDateWithPrecision(DateTimeFormatter.ISO_OFFSET_DATE_TIME, timestamp, zoneId,
TIMESTAMP_PRECISION);
bw.write(str + ",");

223
client/src/test/java/org/apache/iotdb/cli/AbstractCliIT.java
View File

@ -1,223 +0,0 @@
/**
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package org.apache.iotdb.cli;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.fail;
import static org.mockito.Mockito.when;
import org.apache.commons.cli.CommandLine;
import org.apache.commons.cli.CommandLineParser;
import org.apache.commons.cli.DefaultParser;
import org.apache.commons.cli.Options;
import org.apache.commons.cli.ParseException;
import org.apache.iotdb.cli.AbstractCli.OperationResult;
import org.apache.iotdb.exception.ArgsErrorException;
import org.apache.iotdb.jdbc.IoTDBConnection;
import org.apache.iotdb.jdbc.IoTDBDatabaseMetadata;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
public class AbstractCliIT {
private static Logger logger = LoggerFactory.getLogger(AbstractCliIT.class);
@Mock
private IoTDBConnection connection;
@Mock
private IoTDBDatabaseMetadata databaseMetadata;
@Before
public void setUp() throws Exception {
MockitoAnnotations.initMocks(this);
when(connection.getMetaData()).thenReturn(databaseMetadata);
when(connection.getTimeZone()).thenReturn("Asia/Shanghai");
when(databaseMetadata.getMetadataInJson()).thenReturn("test metadata");
}
@After
public void tearDown() throws Exception {
}
@Test
public void testInit() {
AbstractCli.init();
String[] keywords = {AbstractCli.HOST_ARGS, AbstractCli.HELP_ARGS,
AbstractCli.PORT_ARGS,
AbstractCli.PASSWORD_ARGS, AbstractCli.USERNAME_ARGS, AbstractCli.ISO8601_ARGS,
AbstractCli.MAX_PRINT_ROW_COUNT_ARGS,};
for (String keyword : keywords) {
if (!AbstractCli.keywordSet.contains("-" + keyword)) {
logger.error(keyword);
fail();
}
}
}
@Test
public void testCheckRequiredArg() throws ParseException, ArgsErrorException {
Options options = AbstractCli.createOptions();
CommandLineParser parser = new DefaultParser();
String[] args = new String[]{"-u", "user1"};
CommandLine commandLine = parser.parse(options, args);
String str = AbstractCli
.checkRequiredArg(AbstractCli.USERNAME_ARGS, AbstractCli.USERNAME_NAME,
commandLine, true, "root");
assertEquals("user1", str);
args = new String[]{"-u", "root",};
commandLine = parser.parse(options, args);
str = AbstractCli
.checkRequiredArg(AbstractCli.HOST_ARGS, AbstractCli.HOST_NAME, commandLine, false,
"127.0.0.1");
assertEquals("127.0.0.1", str);
try {
str = AbstractCli
.checkRequiredArg(AbstractCli.HOST_ARGS, AbstractCli.HOST_NAME, commandLine, true,
"127.0.0.1");
} catch (ArgsErrorException e) {
assertEquals("IoTDB: Required values for option 'host' not provided", e.getMessage());
}
try {
str = AbstractCli
.checkRequiredArg(AbstractCli.HOST_ARGS, AbstractCli.HOST_NAME, commandLine,
false, null);
} catch (ArgsErrorException e) {
assertEquals("IoTDB: Required values for option 'host' is null.", e.getMessage());
}
}
@Test
public void testRemovePasswordArgs() {
AbstractCli.init();
String[] input = new String[]{"-h", "127.0.0.1", "-p", "6667", "-u", "root", "-pw", "root"};
String[] res = new String[]{"-h", "127.0.0.1", "-p", "6667", "-u", "root", "-pw", "root"};
isTwoStringArrayEqual(res, AbstractCli.removePasswordArgs(input));
input = new String[]{"-h", "127.0.0.1", "-p", "6667", "-pw", "root", "-u", "root"};
res = new String[]{"-h", "127.0.0.1", "-p", "6667", "-pw", "root", "-u", "root"};
isTwoStringArrayEqual(res, AbstractCli.removePasswordArgs(input));
input = new String[]{"-h", "127.0.0.1", "-p", "6667", "root", "-u", "root", "-pw"};
res = new String[]{"-h", "127.0.0.1", "-p", "6667", "root", "-u", "root"};
isTwoStringArrayEqual(res, AbstractCli.removePasswordArgs(input));
input = new String[]{"-h", "127.0.0.1", "-p", "6667", "-pw", "-u", "root"};
res = new String[]{"-h", "127.0.0.1", "-p", "6667", "-u", "root"};
isTwoStringArrayEqual(res, AbstractCli.removePasswordArgs(input));
input = new String[]{"-pw", "-h", "127.0.0.1", "-p", "6667", "root", "-u", "root"};
res = new String[]{"-h", "127.0.0.1", "-p", "6667", "root", "-u", "root"};
isTwoStringArrayEqual(res, AbstractCli.removePasswordArgs(input));
input = new String[]{};
res = new String[]{};
isTwoStringArrayEqual(res, AbstractCli.removePasswordArgs(input));
}
private void isTwoStringArrayEqual(String[] expected, String[] actual) {
for (int i = 0; i < expected.length; i++) {
assertEquals(expected[i], actual[i]);
}
}
@Test
public void testHandleInputInputCmd() {
assertEquals(OperationResult.STOP_OPER, AbstractCli
.handleInputCmd(AbstractCli.EXIT_COMMAND, connection));
assertEquals(OperationResult.STOP_OPER, AbstractCli
.handleInputCmd(AbstractCli.QUIT_COMMAND, connection));
assertEquals(OperationResult.CONTINUE_OPER, AbstractCli
.handleInputCmd(AbstractCli.SHOW_METADATA_COMMAND, connection));
assertEquals(OperationResult.CONTINUE_OPER, AbstractCli
.handleInputCmd(String.format("%s=", AbstractCli.SET_TIMESTAMP_DISPLAY), connection));
assertEquals(OperationResult.CONTINUE_OPER, AbstractCli
.handleInputCmd(String.format("%s=xxx", AbstractCli.SET_TIMESTAMP_DISPLAY), connection));
assertEquals(OperationResult.CONTINUE_OPER, AbstractCli
.handleInputCmd(String.format("%s=default", AbstractCli.SET_TIMESTAMP_DISPLAY), connection));
testSetTimeFormat();
assertEquals(OperationResult.CONTINUE_OPER, AbstractCli
.handleInputCmd(String.format("%s=", AbstractCli.SET_MAX_DISPLAY_NUM), connection));
assertEquals(OperationResult.CONTINUE_OPER, AbstractCli
.handleInputCmd(String.format("%s=xxx", AbstractCli.SET_MAX_DISPLAY_NUM),connection));
assertEquals(OperationResult.CONTINUE_OPER, AbstractCli
.handleInputCmd(String.format("%s=1", AbstractCli.SET_MAX_DISPLAY_NUM), connection));
testSetMaxDisplayNumber();
assertEquals(OperationResult.CONTINUE_OPER, AbstractCli
.handleInputCmd(AbstractCli.SHOW_TIMEZONE, connection));
assertEquals(OperationResult.CONTINUE_OPER, AbstractCli
.handleInputCmd(AbstractCli.SHOW_TIMESTAMP_DISPLAY, connection));
assertEquals(OperationResult.CONTINUE_OPER, AbstractCli
.handleInputCmd(AbstractCli.SHOW_FETCH_SIZE, connection));
assertEquals(OperationResult.CONTINUE_OPER, AbstractCli
.handleInputCmd(String.format("%s=", AbstractCli.SET_TIME_ZONE), connection));
assertEquals(OperationResult.CONTINUE_OPER, AbstractCli
.handleInputCmd(String.format("%s=+08:00", AbstractCli.SET_TIME_ZONE), connection));
assertEquals(OperationResult.CONTINUE_OPER, AbstractCli
.handleInputCmd(String.format("%s=", AbstractCli.SET_FETCH_SIZE), connection));
assertEquals(OperationResult.CONTINUE_OPER, AbstractCli
.handleInputCmd(String.format("%s=111", AbstractCli.SET_FETCH_SIZE), connection));
}
private void testSetTimeFormat() {
AbstractCli.setTimeFormat("long");
assertEquals(AbstractCli.maxTimeLength, AbstractCli.maxValueLength);
assertEquals(AbstractCli.formatTime, "%" + AbstractCli.maxTimeLength + "s|");
AbstractCli.setTimeFormat("number");
assertEquals(AbstractCli.maxTimeLength, AbstractCli.maxValueLength);
assertEquals(AbstractCli.formatTime, "%" + AbstractCli.maxTimeLength + "s|");
AbstractCli.setTimeFormat("default");
assertEquals(AbstractCli.ISO_DATETIME_LEN, AbstractCli.maxTimeLength);
assertEquals(AbstractCli.formatTime, "%" + AbstractCli.maxTimeLength + "s|");
AbstractCli.setTimeFormat("iso8601");
assertEquals(AbstractCli.ISO_DATETIME_LEN, AbstractCli.maxTimeLength);
assertEquals(AbstractCli.formatTime, "%" + AbstractCli.maxTimeLength + "s|");
AbstractCli.setTimeFormat("yyyy-MM-dd HH:mm:ssZZ");
assertEquals(AbstractCli.maxTimeLength, "yyyy-MM-dd HH:mm:ssZZ".length());
assertEquals(AbstractCli.formatTime, "%" + AbstractCli.maxTimeLength + "s|");
AbstractCli.setTimeFormat("dd");
assertEquals(AbstractCli.maxTimeLength, AbstractCli.TIMESTAMP_STR.length());
assertEquals(AbstractCli.formatTime, "%" + AbstractCli.maxTimeLength + "s|");
}
private void testSetMaxDisplayNumber() {
AbstractCli.setMaxDisplayNumber("10");
assertEquals(10, AbstractCli.maxPrintRowCount);
AbstractCli.setMaxDisplayNumber("111111111111111");
assertEquals(Integer.MAX_VALUE, AbstractCli.maxPrintRowCount);
AbstractCli.setMaxDisplayNumber("-10");
assertEquals(Integer.MAX_VALUE, AbstractCli.maxPrintRowCount);
}
}

223
client/src/test/java/org/apache/iotdb/client/AbstractClientIT.java Normal file
View File

@ -0,0 +1,223 @@
/**
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package org.apache.iotdb.client;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.fail;
import static org.mockito.Mockito.when;
import org.apache.commons.cli.CommandLine;
import org.apache.commons.cli.CommandLineParser;
import org.apache.commons.cli.DefaultParser;
import org.apache.commons.cli.Options;
import org.apache.commons.cli.ParseException;
import org.apache.iotdb.client.AbstractClient.OperationResult;
import org.apache.iotdb.exception.ArgsErrorException;
import org.apache.iotdb.jdbc.IoTDBConnection;
import org.apache.iotdb.jdbc.IoTDBDatabaseMetadata;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
public class AbstractClientIT {
private static Logger logger = LoggerFactory.getLogger(AbstractClientIT.class);
@Mock
private IoTDBConnection connection;
@Mock
private IoTDBDatabaseMetadata databaseMetadata;
@Before
public void setUp() throws Exception {
MockitoAnnotations.initMocks(this);
when(connection.getMetaData()).thenReturn(databaseMetadata);
when(connection.getTimeZone()).thenReturn("Asia/Shanghai");
when(databaseMetadata.getMetadataInJson()).thenReturn("test metadata");
}
@After
public void tearDown() throws Exception {
}
@Test
public void testInit() {
AbstractClient.init();
String[] keywords = {AbstractClient.HOST_ARGS, AbstractClient.HELP_ARGS,
AbstractClient.PORT_ARGS,
AbstractClient.PASSWORD_ARGS, AbstractClient.USERNAME_ARGS, AbstractClient.ISO8601_ARGS,
AbstractClient.MAX_PRINT_ROW_COUNT_ARGS,};
for (String keyword : keywords) {
if (!AbstractClient.keywordSet.contains("-" + keyword)) {
logger.error(keyword);
fail();
}
}
}
@Test
public void testCheckRequiredArg() throws ParseException, ArgsErrorException {
Options options = AbstractClient.createOptions();
CommandLineParser parser = new DefaultParser();
String[] args = new String[]{"-u", "user1"};
CommandLine commandLine = parser.parse(options, args);
String str = AbstractClient
.checkRequiredArg(AbstractClient.USERNAME_ARGS, AbstractClient.USERNAME_NAME,
commandLine, true, "root");
assertEquals("user1", str);
args = new String[]{"-u", "root",};
commandLine = parser.parse(options, args);
str = AbstractClient
.checkRequiredArg(AbstractClient.HOST_ARGS, AbstractClient.HOST_NAME, commandLine, false,
"127.0.0.1");
assertEquals("127.0.0.1", str);
try {
str = AbstractClient
.checkRequiredArg(AbstractClient.HOST_ARGS, AbstractClient.HOST_NAME, commandLine, true,
"127.0.0.1");
} catch (ArgsErrorException e) {
assertEquals("IoTDB: Required values for option 'host' not provided", e.getMessage());
}
try {
str = AbstractClient
.checkRequiredArg(AbstractClient.HOST_ARGS, AbstractClient.HOST_NAME, commandLine,
false, null);
} catch (ArgsErrorException e) {
assertEquals("IoTDB: Required values for option 'host' is null.", e.getMessage());
}
}
@Test
public void testRemovePasswordArgs() {
AbstractClient.init();
String[] input = new String[]{"-h", "127.0.0.1", "-p", "6667", "-u", "root", "-pw", "root"};
String[] res = new String[]{"-h", "127.0.0.1", "-p", "6667", "-u", "root", "-pw", "root"};
isTwoStringArrayEqual(res, AbstractClient.removePasswordArgs(input));
input = new String[]{"-h", "127.0.0.1", "-p", "6667", "-pw", "root", "-u", "root"};
res = new String[]{"-h", "127.0.0.1", "-p", "6667", "-pw", "root", "-u", "root"};
isTwoStringArrayEqual(res, AbstractClient.removePasswordArgs(input));
input = new String[]{"-h", "127.0.0.1", "-p", "6667", "root", "-u", "root", "-pw"};
res = new String[]{"-h", "127.0.0.1", "-p", "6667", "root", "-u", "root"};
isTwoStringArrayEqual(res, AbstractClient.removePasswordArgs(input));
input = new String[]{"-h", "127.0.0.1", "-p", "6667", "-pw", "-u", "root"};
res = new String[]{"-h", "127.0.0.1", "-p", "6667", "-u", "root"};
isTwoStringArrayEqual(res, AbstractClient.removePasswordArgs(input));
input = new String[]{"-pw", "-h", "127.0.0.1", "-p", "6667", "root", "-u", "root"};
res = new String[]{"-h", "127.0.0.1", "-p", "6667", "root", "-u", "root"};
isTwoStringArrayEqual(res, AbstractClient.removePasswordArgs(input));
input = new String[]{};
res = new String[]{};
isTwoStringArrayEqual(res, AbstractClient.removePasswordArgs(input));
}
private void isTwoStringArrayEqual(String[] expected, String[] actual) {
for (int i = 0; i < expected.length; i++) {
assertEquals(expected[i], actual[i]);
}
}
@Test
public void testHandleInputInputCmd() {
assertEquals(OperationResult.STOP_OPER, AbstractClient
.handleInputCmd(AbstractClient.EXIT_COMMAND, connection));
assertEquals(OperationResult.STOP_OPER, AbstractClient
.handleInputCmd(AbstractClient.QUIT_COMMAND, connection));
assertEquals(OperationResult.CONTINUE_OPER, AbstractClient
.handleInputCmd(AbstractClient.SHOW_METADATA_COMMAND, connection));
assertEquals(OperationResult.CONTINUE_OPER, AbstractClient
.handleInputCmd(String.format("%s=", AbstractClient.SET_TIMESTAMP_DISPLAY), connection));
assertEquals(OperationResult.CONTINUE_OPER, AbstractClient
.handleInputCmd(String.format("%s=xxx", AbstractClient.SET_TIMESTAMP_DISPLAY), connection));
assertEquals(OperationResult.CONTINUE_OPER, AbstractClient
.handleInputCmd(String.format("%s=default", AbstractClient.SET_TIMESTAMP_DISPLAY), connection));
testSetTimeFormat();
assertEquals(OperationResult.CONTINUE_OPER, AbstractClient
.handleInputCmd(String.format("%s=", AbstractClient.SET_MAX_DISPLAY_NUM), connection));
assertEquals(OperationResult.CONTINUE_OPER, AbstractClient
.handleInputCmd(String.format("%s=xxx", AbstractClient.SET_MAX_DISPLAY_NUM),connection));
assertEquals(OperationResult.CONTINUE_OPER, AbstractClient
.handleInputCmd(String.format("%s=1", AbstractClient.SET_MAX_DISPLAY_NUM), connection));
testSetMaxDisplayNumber();
assertEquals(OperationResult.CONTINUE_OPER, AbstractClient
.handleInputCmd(AbstractClient.SHOW_TIMEZONE, connection));
assertEquals(OperationResult.CONTINUE_OPER, AbstractClient
.handleInputCmd(AbstractClient.SHOW_TIMESTAMP_DISPLAY, connection));
assertEquals(OperationResult.CONTINUE_OPER, AbstractClient
.handleInputCmd(AbstractClient.SHOW_FETCH_SIZE, connection));
assertEquals(OperationResult.CONTINUE_OPER, AbstractClient
.handleInputCmd(String.format("%s=", AbstractClient.SET_TIME_ZONE), connection));
assertEquals(OperationResult.CONTINUE_OPER, AbstractClient
.handleInputCmd(String.format("%s=+08:00", AbstractClient.SET_TIME_ZONE), connection));
assertEquals(OperationResult.CONTINUE_OPER, AbstractClient
.handleInputCmd(String.format("%s=", AbstractClient.SET_FETCH_SIZE), connection));
assertEquals(OperationResult.CONTINUE_OPER, AbstractClient
.handleInputCmd(String.format("%s=111", AbstractClient.SET_FETCH_SIZE), connection));
}
private void testSetTimeFormat() {
AbstractClient.setTimeFormat("long");
assertEquals(AbstractClient.maxTimeLength, AbstractClient.maxValueLength);
assertEquals(AbstractClient.formatTime, "%" + AbstractClient.maxTimeLength + "s|");
AbstractClient.setTimeFormat("number");
assertEquals(AbstractClient.maxTimeLength, AbstractClient.maxValueLength);
assertEquals(AbstractClient.formatTime, "%" + AbstractClient.maxTimeLength + "s|");
AbstractClient.setTimeFormat("default");
assertEquals(AbstractClient.ISO_DATETIME_LEN, AbstractClient.maxTimeLength);
assertEquals(AbstractClient.formatTime, "%" + AbstractClient.maxTimeLength + "s|");
AbstractClient.setTimeFormat("iso8601");
assertEquals(AbstractClient.ISO_DATETIME_LEN, AbstractClient.maxTimeLength);
assertEquals(AbstractClient.formatTime, "%" + AbstractClient.maxTimeLength + "s|");
AbstractClient.setTimeFormat("yyyy-MM-dd HH:mm:ssZZ");
assertEquals(AbstractClient.maxTimeLength, "yyyy-MM-dd HH:mm:ssZZ".length());
assertEquals(AbstractClient.formatTime, "%" + AbstractClient.maxTimeLength + "s|");
AbstractClient.setTimeFormat("dd");
assertEquals(AbstractClient.maxTimeLength, AbstractClient.TIMESTAMP_STR.length());
assertEquals(AbstractClient.formatTime, "%" + AbstractClient.maxTimeLength + "s|");
}
private void testSetMaxDisplayNumber() {
AbstractClient.setMaxDisplayNumber("10");
assertEquals(10, AbstractClient.maxPrintRowCount);
AbstractClient.setMaxDisplayNumber("111111111111111");
assertEquals(Integer.MAX_VALUE, AbstractClient.maxPrintRowCount);
AbstractClient.setMaxDisplayNumber("-10");
assertEquals(Integer.MAX_VALUE, AbstractClient.maxPrintRowCount);
}
}

2
client/src/test/java/org/apache/iotdb/cli/AbstractScript.java → client/src/test/java/org/apache/iotdb/client/AbstractScript.java
View File

@ -16,7 +16,7 @@
* specific language governing permissions and limitations
* under the License.
*/
package org.apache.iotdb.cli;
package org.apache.iotdb.client;
import static org.junit.Assert.assertEquals;

10
client/src/test/java/org/apache/iotdb/cli/StartCliScriptIT.java → client/src/test/java/org/apache/iotdb/client/StartClientScriptIT.java
View File

@ -16,7 +16,7 @@
* specific language governing permissions and limitations
* under the License.
*/
package org.apache.iotdb.cli;
package org.apache.iotdb.client;
import java.io.File;
import java.io.IOException;
@ -25,7 +25,7 @@ import org.junit.After;
import org.junit.Before;
import org.junit.Test;
public class StartCliScriptIT extends AbstractScript {
public class StartClientScriptIT extends AbstractScript {
@Before
public void setUp() throws Exception {
@ -47,12 +47,12 @@ public class StartCliScriptIT extends AbstractScript {
@Override
protected void testOnWindows() throws IOException {
final String[] output = {"````````````````````````", "Starting IoTDB Cli",
final String[] output = {"````````````````````````", "Starting IoTDB Client",
"````````````````````````",
"IoTDB> Connection Error, please check whether the network is available or the server has started. Host is 127.0.0.1, port is 6668."};
String dir = getCliPath();
ProcessBuilder builder = new ProcessBuilder("cmd.exe", "/c",
dir + File.separator + "sbin" + File.separator + "start-cli.bat",
dir + File.separator + "sbin" + File.separator + "start-client.bat",
"-h",
"127.0.0.1", "-p", "6668", "-u", "root", "-pw", "root");
testOutput(builder, output);
@ -64,7 +64,7 @@ public class StartCliScriptIT extends AbstractScript {
"IoTDB> Connection Error, please check whether the network is available or the server has started. Host is 127.0.0.1, port is 6668."};
String dir = getCliPath();
ProcessBuilder builder = new ProcessBuilder("sh",
dir + File.separator + "sbin" + File.separator + "start-cli.sh",
dir + File.separator + "sbin" + File.separator + "start-client.sh",
"-h",
"127.0.0.1", "-p", "6668", "-u", "root", "-pw", "root");
testOutput(builder, output);

2
client/src/test/java/org/apache/iotdb/tool/ExportCsvTestIT.java
View File

@ -20,7 +20,7 @@ package org.apache.iotdb.tool;
import java.io.File;
import java.io.IOException;
import org.apache.iotdb.cli.AbstractScript;
import org.apache.iotdb.client.AbstractScript;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

2
client/src/test/java/org/apache/iotdb/tool/ImportCsvTestIT.java
View File

@ -20,7 +20,7 @@ package org.apache.iotdb.tool;
import java.io.File;
import java.io.IOException;
import org.apache.iotdb.cli.AbstractScript;
import org.apache.iotdb.client.AbstractScript;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

4
docs/Development.md
View File

@ -74,8 +74,8 @@ you can just choose menu "import" -> "Maven" -> "Existing Maven Projects".
## Debugging IoTDB
The main class of IoTDB server is `org.apache.iotdb.db.service.IoTDB`.
The main class of IoTDB cli is `org.apache.iotdb.cli.Cli`
(or `org.apache.iotdb.cli.WinCli` on Win OS).
The main class of IoTDB cli is `org.apache.iotdb.client.Client`
(or `org.apache.iotdb.client.WinClient` on Win OS).
You can run/debug IoTDB by using the two classes as the entrance.

4
docs/Documentation-CHN/UserGuide/1-Overview/3-Scenario.md
View File

@ -57,7 +57,7 @@
为了保证机械手的监控数据能够及时监控和分析，公司需要收集这些机械手传感器信息，将其发送至可以连接外部网络的服务器上，而后将原始数据信息上传到数据中心进行复杂的计算和分析。
此时，可以采用IoTDB套件中的IoTDB、IoTDB-CLI工具、TsFileSync工具和Hadoop/Spark集成组件等。将IoTDB服务器安装在工厂连接外网的服务器上，用户接收机械手传输的数据并将数据上传到数据中心。将IoTDB-CLI工具安装在每一个连接工厂内网的机械手上，用于将传感器产生的实时数据上传到工厂内部服务器。再使用TsFileSync工具将原始数据上传到数据中心。此外还需要部署Hadoop/Spark集群用于数据中心端的数据存储和分析。如图1.6中间场景所示。
此时，可以采用IoTDB套件中的IoTDB、IoTDB-Client工具、TsFileSync工具和Hadoop/Spark集成组件等。将IoTDB服务器安装在工厂连接外网的服务器上，用户接收机械手传输的数据并将数据上传到数据中心。将IoTDB-Client工具安装在每一个连接工厂内网的机械手上，用于将传感器产生的实时数据上传到工厂内部服务器。再使用TsFileSync工具将原始数据上传到数据中心。此外还需要部署Hadoop/Spark集群用于数据中心端的数据存储和分析。如图1.6中间场景所示。
<img style="width:100%; max-width:800px; max-height:600px; margin-left:auto; margin-right:auto; display:block;" src="https://user-images.githubusercontent.com/13203019/51579080-96aba780-1efa-11e9-87ac-940c45b19dd7.jpg">
@ -71,7 +71,7 @@
为了能够实时接收汽车传感器所采集的物联网数据，公司需要在车辆行驶的过程中将传感器数据通过窄带物联网实时发送至数据中心，而后在数据中心的服务器上进行复杂的计算和分析。
此时，可以采用IoTDB套件中的IoTDB、IoTDB-CLI和Hadoop/Spark集成组件等。将IoTDB-CLI工具安装在每一辆车联网内的车辆上，使用IoTDB-JDBC工具将数据直接传回数据中心的服务器。
此时，可以采用IoTDB套件中的IoTDB、IoTDB-Client和Hadoop/Spark集成组件等。将IoTDB-Client工具安装在每一辆车联网内的车辆上，使用IoTDB-JDBC工具将数据直接传回数据中心的服务器。
此外还需要部署Hadoop/Spark集群用于数据中心端的数据存储和分析。如图1.8所示。

44
docs/Documentation-CHN/UserGuide/7-Client API/1-Client API.md → docs/Documentation-CHN/UserGuide/7-Session API/1-Session API.md
View File

@ -19,7 +19,7 @@
-->
# 第7章: JDBC API
# 第7章: Session API
# 使用方式
@ -28,33 +28,30 @@
* JDK >= 1.8
* Maven >= 3.0
## 只打包 Client 模块
In root directory:
> mvn clean package -pl client -am -Dmaven.test.skip=true
## 安装到本地 maven 库
In root directory:
> mvn clean install -pl client -am -Dmaven.test.skip=true
> mvn clean install -pl session -am -Dmaven.test.skip=true
## 在 maven 中使用 Client 接口
## 在 maven 中使用 session 接口
```
<dependencies>
<dependency>
<groupId>org.apache.iotdb</groupId>
<artifactId>iotdb-client</artifactId>
<artifactId>iotdb-session</artifactId>
<version>0.9.0-SNAPSHOT</version>
</dependency>
</dependencies>
```
## Client 示例
## Session 接口使用示例
```Java
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.Statement;
import org.apache.iotdb.session.Session;
import org.apache.iotdb.tsfile.file.metadata.enums.TSDataType;
import org.apache.iotdb.tsfile.file.metadata.enums.TSEncoding;
@ -62,9 +59,20 @@ import org.apache.iotdb.tsfile.write.record.RowBatch;
import org.apache.iotdb.tsfile.write.schema.MeasurementSchema;
import org.apache.iotdb.tsfile.write.schema.Schema;
public static void main(String[] args) {
Client client = new Client("127.0.0.1", 6667, "root", "root");
client.open();
public static void main(String[] args) throws ClassNotFoundException {
Class.forName("org.apache.iotdb.jdbc.IoTDBDriver");
try (Connection connection = DriverManager.getConnection("jdbc:iotdb://127.0.0.1:6667/", "root", "root");
Statement statement = connection.createStatement()) {
statement.execute("SET STORAGE GROUP TO root.sg1");
statement.execute("CREATE TIMESERIES root.sg1.d1.s1 WITH DATATYPE=INT64, ENCODING=RLE");
statement.execute("CREATE TIMESERIES root.sg1.d1.s2 WITH DATATYPE=INT64, ENCODING=RLE");
statement.execute("CREATE TIMESERIES root.sg1.d1.s3 WITH DATATYPE=INT64, ENCODING=RLE");
} catch (Exception e) {
System.out.println(e.getMessage());
}
Session session = new Session("127.0.0.1", 6667, "root", "root");
session.open();
Schema schema = new Schema();
schema.registerMeasurement(new MeasurementSchema("s1", TSDataType.INT64, TSEncoding.RLE));
@ -76,7 +84,7 @@ import org.apache.iotdb.tsfile.write.schema.Schema;
long[] timestamps = rowBatch.timestamps;
Object[] values = rowBatch.values;
for (long time = 0; time < 1000; time++) {
for (long time = 0; time < 30000; time++) {
int row = rowBatch.batchSize++;
timestamps[row] = time;
for (int i = 0; i < 3; i++) {
@ -88,14 +96,14 @@ import org.apache.iotdb.tsfile.write.schema.Schema;
rowBatch.reset();
}
}
if (rowBatch.batchSize != 0) {
session.insertBatch(rowBatch);
rowBatch.reset();
}
client.close();
session.close();
}
```
> The code is in example/client/src/main/java/org/apache/iotdb/client/ClientExample.java
> The code is in example/session/src/main/java/org/apache/iotdb/session/SessionExample.java

6
docs/Documentation-CHN/UserGuide/9-Tools-Cli.md
View File

@ -29,7 +29,7 @@
<!-- /TOC -->
# Cli / Shell工具
IOTDB为用户提供CLI/Shell工具用于启动客户端和服务端程序。下面介绍每个CLI/Shell工具的运行方式和相关参数。
IOTDB为用户提供Client/Shell工具用于启动客户端和服务端程序。下面介绍每个Client/Shell工具的运行方式和相关参数。
> \$IOTDB\_HOME表示IoTDB的安装目录所在路径。
## Cli / Shell运行方式
@ -91,7 +91,7 @@ Windows系统启动命令如下：
为了避免SQL语句和其他参数混淆，现在只支持-e参数作为最后的参数使用。
针对CLI/Shell工具的-e参数用法如下：
针对Client/Shell工具的-e参数用法如下：
```
Shell > ./sbin/start-cli.sh -h {host} -p {port} -u {user} -pw {password} -e {sql for iotdb}
@ -109,7 +109,7 @@ Windows系统启动命令如下：
4.查询验证数据是否插入成功
那么通过使用CLI/Shell工具的-e参数，可以采用如下的脚本：
那么通过使用Client/Shell工具的-e参数，可以采用如下的脚本：
```
# !/bin/bash

6
docs/Documentation/Frequently asked questions.md
View File

@ -31,7 +31,7 @@
- Can I use Hadoop and Spark to read TsFile in IoTDB?
- How does IoTDB handle duplicate points?
- How can I tell what type of the specific timeseries?
- How can I change IoTDB's CLI time display format?
- How can I change IoTDB's Client time display format?
<!-- /TOC -->
# Frequently Asked Questions
@ -137,9 +137,9 @@ Otherwise, you can also use wildcard in timeseries path:
IoTDB> show timeseries root.fit.d1.*
```
## How can I change IoTDB's CLI time display format?
## How can I change IoTDB's Client time display format?
The default IoTDB's CLI time display format is human readable (e.g. ```1970-01-01T08:00:00.001```), if you want to display time in timestamp type or other readable format, add parameter ```-disableIS08601``` in start command:
The default IoTDB's Client time display format is human readable (e.g. ```1970-01-01T08:00:00.001```), if you want to display time in timestamp type or other readable format, add parameter ```-disableIS08601``` in start command:
```
> $IOTDB_CLI_HOME/sbin/start-cli.sh -h 127.0.0.1 -p 6667 -u root -pw root -disableIS08601

4
docs/Documentation/UserGuide/1-Overview/3-Scenario.md
View File

@ -61,7 +61,7 @@ In order to ensure that the data of the robot can be monitored and analyzed in t
<img style="width:100%; max-width:800px; max-height:600px; margin-left:auto; margin-right:auto; display:block;" src="https://user-images.githubusercontent.com/13203019/51579080-96aba780-1efa-11e9-87ac-940c45b19dd7.jpg">
At this point, IoTDB, IoTDB-CLI tools, TsFileSync tools, and Hadoop/Spark integration components in the IoTDB suite can be used. IoTDB-CLI tool is installed on the robot and each of them is connected to the LAN of the factory. When sensors generate real-time data, the data will be uploaded to the server in the factory. The IoTDB server and TsFileSync is installed on the server connected to the external network. Once triggered, the data on the server will be upload to the data center. In addition, Hadoop/Spark clusters need to be deployed for data storage and analysis on the data center side. As shown in Figure 1.6. Figure 1.7 shows the architecture at this time.
At this point, IoTDB, IoTDB-Client tools, TsFileSync tools, and Hadoop/Spark integration components in the IoTDB suite can be used. IoTDB-Client tool is installed on the robot and each of them is connected to the LAN of the factory. When sensors generate real-time data, the data will be uploaded to the server in the factory. The IoTDB server and TsFileSync is installed on the server connected to the external network. Once triggered, the data on the server will be upload to the data center. In addition, Hadoop/Spark clusters need to be deployed for data storage and analysis on the data center side. As shown in Figure 1.6. Figure 1.7 shows the architecture at this time.
<img style="width:100%; max-width:800px; max-height:600px; margin-left:auto; margin-right:auto; display:block;" src="https://user-images.githubusercontent.com/13203019/51579085-9dd2b580-1efa-11e9-97b9-f56bc8d342b0.jpg">
@ -71,7 +71,7 @@ A car company installed sensors on its cars to collect monitoring information su
In order to receive the IoT data collected by the car sensor in real time, the company needs to send the sensor data to the data center in real time through the narrowband IoT while the vehicle is running. Thus, they can perform complex calculations and analysis on the server in the data center.
At this point, IoTDB, IoTDB-CLI, and Hadoop/Spark integration components in the IoTDB suite can be used. IoTDB-CLI tool is installed on each car and use IoTDB-JDBC tool to send data directly back to the server in the data center.
At this point, IoTDB, IoTDB-Client, and Hadoop/Spark integration components in the IoTDB suite can be used. IoTDB-Client tool is installed on each car and use IoTDB-JDBC tool to send data directly back to the server in the data center.
In addition, Hadoop/Spark clusters need to be deployed for data storage and analysis on the data center side. As shown in Figure 1.8.

2
docs/Documentation/UserGuide/3-Operation Manual/3-Data Import.md
View File

@ -28,7 +28,7 @@ This feature is not supported in version 0.7.0.
### Import Real-time Data
IoTDB provides users with a variety of ways to insert real-time data, such as directly inputting [INSERT SQL statement](/#/Documents/0.8.0/chap5/sec1) in [Cli/Shell tools](/#/Tools/Cli), or using [Java JDBC](/#/Documents/0.8.0/chap6/sec1) to perform single or batch execution of [INSERT SQL statement](/#/Documents/0.8.0/chap5/sec1).
IoTDB provides users with a variety of ways to insert real-time data, such as directly inputting [INSERT SQL statement](/#/Documents/0.8.0/chap5/sec1) in [Client/Shell tools](/#/Tools/Client), or using [Java JDBC](/#/Documents/0.8.0/chap6/sec1) to perform single or batch execution of [INSERT SQL statement](/#/Documents/0.8.0/chap5/sec1).
This section mainly introduces the use of [INSERT SQL statement](/#/Documents/0.8.0/chap5/sec1) for real-time data import in the scenario. See Section 5.1 for a detailed syntax of [INSERT SQL statement](/#/Documents/0.8.0/chap5/sec1).

6
docs/Documentation/UserGuide/5-IoTDB SQL Documentation/1-IoTDB Query Statement.md
View File

@ -74,7 +74,7 @@ Eg: IoTDB > DELETE TIMESERIES root.ln.wf01.wt01.*
```
SHOW TIMESERIES
Eg: IoTDB > SHOW TIMESERIES
Note: This statement can only be used in IoTDB Cli. If you need to show all timeseries in JDBC, please use `DataBaseMetadata` interface.
Note: This statement can only be used in IoTDB Client. If you need to show all timeseries in JDBC, please use `DataBaseMetadata` interface.
```
* Show Specific Timeseries Statement
@ -86,7 +86,7 @@ Eg: IoTDB > SHOW TIMESERIES root.ln
Eg: IoTDB > SHOW TIMESERIES root.ln.*.*.status
Eg: IoTDB > SHOW TIMESERIES root.ln.wf01.wt01.status
Note: The path can be prefix path, star path or timeseries path
Note: This statement can be used in IoTDB Cli and JDBC.
Note: This statement can be used in IoTDB Client and JDBC.
```
* Show Storage Group Statement
@ -94,7 +94,7 @@ Note: This statement can be used in IoTDB Cli and JDBC.
```
SHOW STORAGE GROUP
Eg: IoTDB > SHOW STORAGE GROUP
Note: This statement can be used in IoTDB Cli and JDBC.
Note: This statement can be used in IoTDB Client and JDBC.
```
### Data Management Statement

4
docs/Documentation/UserGuide/6-JDBC API/1-JDBC API.md
View File

@ -19,7 +19,7 @@
-->
# Chaper6: JDBC API
# Chapter6: JDBC API
# Usage
@ -57,7 +57,7 @@ This chapter provides an example of how to open a database connection, execute a
Requires that you include the packages containing the JDBC classes needed for database programming.
**NOTE: For faster insertion, the insertBatch() in Client is recommended.**
**NOTE: For faster insertion, the insertBatch() in Session is recommended.**
```Java
import java.sql.*;

36
docs/Documentation/UserGuide/7-Client API/1-Client API.md → docs/Documentation/UserGuide/7-Session API/1-Session API.md
View File

@ -19,7 +19,7 @@
-->
# Chaper7: JDBC API
# Chapter7: Session API
# Usage
@ -38,7 +38,7 @@ In root directory:
In root directory:
> mvn clean install -pl client -am -Dmaven.test.skip=true
## Using IoTDB Client with Maven
## Using IoTDB Session with Maven
```
<dependencies>
@ -51,23 +51,37 @@ In root directory:
```
## Examples with Client
## Examples with Session
This chapter provides an example of how to open an IoTDB session, execute a batch insertion.
Requires that you include the packages containing the Client classes needed for database programming.
```Java
import org.apache.iotdb.client.Client;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.Statement;
import org.apache.iotdb.session.Session;
import org.apache.iotdb.tsfile.file.metadata.enums.TSDataType;
import org.apache.iotdb.tsfile.file.metadata.enums.TSEncoding;
import org.apache.iotdb.tsfile.write.record.RowBatch;
import org.apache.iotdb.tsfile.write.schema.MeasurementSchema;
import org.apache.iotdb.tsfile.write.schema.Schema;
public static void main(String[] args) {
Client client = new Client("127.0.0.1", 6667, "root", "root");
client.open();
public static void main(String[] args) throws ClassNotFoundException {
Class.forName("org.apache.iotdb.jdbc.IoTDBDriver");
try (Connection connection = DriverManager.getConnection("jdbc:iotdb://127.0.0.1:6667/", "root", "root");
Statement statement = connection.createStatement()) {
statement.execute("SET STORAGE GROUP TO root.sg1");
statement.execute("CREATE TIMESERIES root.sg1.d1.s1 WITH DATATYPE=INT64, ENCODING=RLE");
statement.execute("CREATE TIMESERIES root.sg1.d1.s2 WITH DATATYPE=INT64, ENCODING=RLE");
statement.execute("CREATE TIMESERIES root.sg1.d1.s3 WITH DATATYPE=INT64, ENCODING=RLE");
} catch (Exception e) {
System.out.println(e.getMessage());
}
Session session = new Session("127.0.0.1", 6667, "root", "root");
session.open();
Schema schema = new Schema();
schema.registerMeasurement(new MeasurementSchema("s1", TSDataType.INT64, TSEncoding.RLE));
@ -79,7 +93,7 @@ import org.apache.iotdb.tsfile.write.schema.Schema;
long[] timestamps = rowBatch.timestamps;
Object[] values = rowBatch.values;
for (long time = 0; time < 1000; time++) {
for (long time = 0; time < 30000; time++) {
int row = rowBatch.batchSize++;
timestamps[row] = time;
for (int i = 0; i < 3; i++) {
@ -91,14 +105,14 @@ import org.apache.iotdb.tsfile.write.schema.Schema;
rowBatch.reset();
}
}
if (rowBatch.batchSize != 0) {
session.insertBatch(rowBatch);
rowBatch.reset();
}
client.close();
session.close();
}
```
> The code is in example/client/src/main/java/org/apache/iotdb/client/ClientExample.java
> The code is in example/session/src/main/java/org/apache/iotdb/session/SessionExample.java

2
example/pom.xml
View File

@ -36,7 +36,7 @@
<modules>
<module>kafka</module>
<module>rocketmq</module>
<module>client</module>
<module>session</module>
<module>tsfile</module>
</modules>
<build>

2
example/client/pom.xml → example/session/pom.xml
View File

@ -36,7 +36,7 @@
<dependencies>
<dependency>
<groupId>org.apache.iotdb</groupId>
<artifactId>iotdb-client</artifactId>
<artifactId>iotdb-session</artifactId>
<version>0.9.0-SNAPSHOT</version>
</dependency>
</dependencies>

11
example/client/src/main/java/org/apache/iotdb/client/JDBCExample.java → example/session/src/main/java/org/apache/iotdb/JDBCExample.java
View File

@ -16,7 +16,7 @@
* specific language governing permissions and limitations
* under the License.
*/
package org.apache.iotdb.client;
package org.apache.iotdb;
import java.sql.Connection;
import java.sql.DriverManager;
@ -29,10 +29,8 @@ public class JDBCExample {
public static void main(String[] args) throws ClassNotFoundException, SQLException {
Class.forName("org.apache.iotdb.jdbc.IoTDBDriver");
Connection connection = null;
try {
connection = DriverManager.getConnection("jdbc:iotdb://127.0.0.1:6667/", "root", "root");
Statement statement = connection.createStatement();
try (Connection connection = DriverManager.getConnection("jdbc:iotdb://127.0.0.1:6667/", "root", "root");
Statement statement = connection.createStatement()) {
statement.execute("SET STORAGE GROUP TO root.sg1");
statement.execute("CREATE TIMESERIES root.sg1.d1.s1 WITH DATATYPE=INT64, ENCODING=RLE");
statement.execute("CREATE TIMESERIES root.sg1.d1.s2 WITH DATATYPE=INT64, ENCODING=RLE");
@ -54,9 +52,6 @@ public class JDBCExample {
}
System.out.println(builder);
}
statement.close();
} finally {
connection.close();
}
}
}

15
example/client/src/main/java/org/apache/iotdb/client/ClientExample.java → example/session/src/main/java/org/apache/iotdb/SessionExample.java
View File

@ -16,11 +16,12 @@
* specific language governing permissions and limitations
* under the License.
*/
package org.apache.iotdb.client;
package org.apache.iotdb;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.Statement;
import org.apache.iotdb.session.Session;
import org.apache.iotdb.tsfile.file.metadata.enums.TSDataType;
import org.apache.iotdb.tsfile.file.metadata.enums.TSEncoding;
import org.apache.iotdb.tsfile.write.record.RowBatch;
@ -37,23 +38,21 @@ import org.apache.iotdb.tsfile.write.schema.Schema;
* CREATE TIMESERIES root.sg1.d1.s2 WITH DATATYPE=FLOAT, ENCODING=RLE
* CREATE TIMESERIES root.sg1.d1.s3 WITH DATATYPE=FLOAT, ENCODING=RLE
*/
public class ClientExample {
public class SessionExample {
public static void main(String[] args) throws ClassNotFoundException {
Class.forName("org.apache.iotdb.jdbc.IoTDBDriver");
Connection connection = null;
try {
connection = DriverManager.getConnection("jdbc:iotdb://127.0.0.1:6667/", "root", "root");
Statement statement = connection.createStatement();
try (Connection connection = DriverManager.getConnection("jdbc:iotdb://127.0.0.1:6667/", "root", "root");
Statement statement = connection.createStatement()) {
statement.execute("SET STORAGE GROUP TO root.sg1");
statement.execute("CREATE TIMESERIES root.sg1.d1.s1 WITH DATATYPE=INT64, ENCODING=RLE");
statement.execute("CREATE TIMESERIES root.sg1.d1.s2 WITH DATATYPE=INT64, ENCODING=RLE");
statement.execute("CREATE TIMESERIES root.sg1.d1.s3 WITH DATATYPE=INT64, ENCODING=RLE");
} catch (Exception e) {
System.out.println(e.getMessage());
}
Client session = new Client("127.0.0.1", 6667, "root", "root");
Session session = new Session("127.0.0.1", 6667, "root", "root");
session.open();
Schema schema = new Schema();

1
pom.xml
View File

@ -388,6 +388,7 @@
<module>spark-tsfile</module>
<!-- <module>hadoop</module> -->
<module>distribution</module>
<module>session</module>
</modules>
<build>
<pluginManagement>

2
server/src/main/antlr3/org/apache/iotdb/db/sql/parse/TSParser.g
View File

@ -119,7 +119,7 @@ ArrayList<ParseError> errors = new ArrayList<ParseError>();
private static HashMap<String, String> xlateMap;
static {
//this is used to support auto completion in CLI
//this is used to support auto completion in Client
xlateMap = new HashMap<String, String>();
// Keywords

4
server/src/main/java/org/apache/iotdb/db/utils/MemUtils.java
View File

@ -60,7 +60,7 @@ public class MemUtils {
case BOOLEAN:
memSize += 8L + 1L; break;
case TEXT:
memSize += 8L + insertPlan.getValues()[i].length() * 2; break;
memSize += 8L + insertPlan.getValues()[i].length(); break;
default:
memSize += 8L + 8L;
}
@ -85,7 +85,7 @@ public class MemUtils {
case TEXT:
memSize += batchInsertPlan.getRowCount() * 8L;
for (int j = 0; j < batchInsertPlan.getRowCount(); j++) {
memSize += ((Binary[]) batchInsertPlan.getColumns()[i])[j].getLength() * 2;
memSize += ((Binary[]) batchInsertPlan.getColumns()[i])[j].getLength();
}
break;
default:

2
service-rpc/src/main/thrift/rpc.thrift
View File

@ -37,7 +37,7 @@ struct TS_Status {
2: optional list<string> infoMessages
// If status is ERROR, then the following fields may be set
3: optional string sqlState // as defined in the ISO/IEF CLI specification
3: optional string sqlState // as defined in the ISO/IEF CLIENT specification
4: optional i32 errorCode // internal error code
5: optional string errorMessage
}

62
session/pom.xml Normal file
View File

@ -0,0 +1,62 @@
<?xml version="1.0" encoding="UTF-8"?>
<!--
Licensed to the Apache Software Foundation (ASF) under one
or more contributor license agreements. See the NOTICE file
distributed with this work for additional information
regarding copyright ownership. The ASF licenses this file
to you under the Apache License, Version 2.0 (the
"License"); you may not use this file except in compliance
with the License. You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing,
software distributed under the License is distributed on an
"AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
KIND, either express or implied. See the License for the
specific language governing permissions and limitations
under the License.
-->
<project xmlns="http://maven.apache.org/POM/4.0.0"
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
<parent>
<artifactId>iotdb-parent</artifactId>
<groupId>org.apache.iotdb</groupId>
<version>0.9.0-SNAPSHOT</version>
</parent>
<modelVersion>4.0.0</modelVersion>
<artifactId>iotdb-session</artifactId>
<name>IoTDB Session</name>
<build>
<plugins>
<plugin>
<groupId>org.apache.maven.plugins</groupId>
<artifactId>maven-compiler-plugin</artifactId>
<configuration>
<source>8</source>
<target>8</target>
</configuration>
</plugin>
</plugins>
</build>
<properties>
<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
<maven.compiler.source>1.7</maven.compiler.source>
<maven.compiler.target>1.7</maven.compiler.target>
</properties>
<dependencies>
<dependency>
<groupId>org.apache.iotdb</groupId>
<artifactId>iotdb-jdbc</artifactId>
<version>0.9.0-SNAPSHOT</version>
</dependency>
</dependencies>
</project>

0
client/src/main/java/org/apache/iotdb/exception/IoTDBSessionException.java → session/src/main/java/org/apache/iotdb/exception/IoTDBSessionException.java
View File

189
session/src/main/java/org/apache/iotdb/session/Session.java Normal file
View File

@ -0,0 +1,189 @@
/**
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package org.apache.iotdb.session;
import java.time.ZoneId;
import org.apache.iotdb.exception.IoTDBSessionException;
import org.apache.iotdb.jdbc.Config;
import org.apache.iotdb.jdbc.IoTDBConnection;
import org.apache.iotdb.jdbc.IoTDBSQLException;
import org.apache.iotdb.jdbc.Utils;
import org.apache.iotdb.service.rpc.thrift.TSBatchInsertionReq;
import org.apache.iotdb.service.rpc.thrift.TSCloseSessionReq;
import org.apache.iotdb.service.rpc.thrift.TSExecuteBatchStatementResp;
import org.apache.iotdb.service.rpc.thrift.TSGetTimeZoneResp;
import org.apache.iotdb.service.rpc.thrift.TSIService;
import org.apache.iotdb.service.rpc.thrift.TSOpenSessionReq;
import org.apache.iotdb.service.rpc.thrift.TSOpenSessionResp;
import org.apache.iotdb.service.rpc.thrift.TSProtocolVersion;
import org.apache.iotdb.service.rpc.thrift.TSSetTimeZoneReq;
import org.apache.iotdb.service.rpc.thrift.TSSetTimeZoneResp;
import org.apache.iotdb.service.rpc.thrift.TS_SessionHandle;
import org.apache.iotdb.tsfile.write.record.RowBatch;
import org.apache.iotdb.tsfile.write.schema.MeasurementSchema;
import org.apache.thrift.TException;
import org.apache.thrift.protocol.TBinaryProtocol;
import org.apache.thrift.protocol.TCompactProtocol;
import org.apache.thrift.transport.TSocket;
import org.apache.thrift.transport.TTransportException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
public class Session {
private static final Logger logger = LoggerFactory.getLogger(Session.class);
private String host;
private int port;
private String username;
private String password;
private final TSProtocolVersion protocolVersion = TSProtocolVersion.IOTDB_SERVICE_PROTOCOL_V1;
public TSIService.Iface client = null;
private TS_SessionHandle sessionHandle = null;
private TSocket transport;
private boolean isClosed = true;
private ZoneId zoneId;
public Session(String host, int port) {
this(host, port, Config.DEFAULT_USER, Config.DEFALUT_PASSWORD);
}
public Session(String host, String port, String username, String password) {
this(host, Integer.parseInt(port), username, password);
}
public Session(String host, int port, String username, String password) {
this.host = host;
this.port = port;
this.username = username;
this.password = password;
}
public void open() {
open(false, 0);
}
public void open(boolean enableRPCCompression, int connectionTimeoutInMs) {
transport = new TSocket(host, port, connectionTimeoutInMs);
if (!transport.isOpen()) {
try {
transport.open();
} catch (TTransportException e) {
throw new IoTDBSessionException(e);
}
}
if(enableRPCCompression) {
client = new TSIService.Client(new TCompactProtocol(transport));
}
else {
client = new TSIService.Client(new TBinaryProtocol(transport));
}
TSOpenSessionReq openReq = new TSOpenSessionReq(TSProtocolVersion.IOTDB_SERVICE_PROTOCOL_V1);
openReq.setUsername(username);
openReq.setPassword(password);
try {
TSOpenSessionResp openResp = client.openSession(openReq);
// validate connection
try {
Utils.verifySuccess(openResp.getStatus());
} catch (IoTDBSQLException e) {
transport.close();
throw e;
}
if (protocolVersion.getValue() != openResp.getServerProtocolVersion().getValue()) {
throw new TException(String
.format("Protocol not supported, Client version is {}, but Server version is {}",
protocolVersion.getValue(), openResp.getServerProtocolVersion().getValue()));
}
sessionHandle = openResp.getSessionHandle();
if (zoneId != null) {
setTimeZone(zoneId.toString());
} else {
zoneId = ZoneId.of(getTimeZone());
}
} catch (TException | IoTDBSQLException e) {
throw new IoTDBSessionException(String.format("Can not open session to %s:%s with user: %s.",
host, port, username), e);
}
isClosed = false;
client = IoTDBConnection.newSynchronizedClient(client);
}
public void close() {
if (isClosed) {
return;
}
TSCloseSessionReq req = new TSCloseSessionReq(sessionHandle);
try {
client.closeSession(req);
} catch (TException e) {
throw new IoTDBSessionException("Error occurs when closing session at server. Maybe server is down.", e);
} finally {
isClosed = true;
if (transport != null) {
transport.close();
}
}
}
public TSExecuteBatchStatementResp insertBatch(RowBatch rowBatch) {
TSBatchInsertionReq request = new TSBatchInsertionReq();
request.deviceId = rowBatch.deviceId;
for (MeasurementSchema measurementSchema: rowBatch.measurements) {
request.addToMeasurements(measurementSchema.getMeasurementId());
request.addToTypes(measurementSchema.getType().ordinal());
}
request.setTimestamps(Utils.getTimeBuffer(rowBatch));
request.setValues(Utils.getValueBuffer(rowBatch));
request.setSize(rowBatch.batchSize);
try {
return client.insertBatch(request);
} catch (TException e) {
throw new IoTDBSessionException(e);
}
}
public String getTimeZone() throws TException, IoTDBSQLException {
if (zoneId != null) {
return zoneId.toString();
}
TSGetTimeZoneResp resp = client.getTimeZone();
Utils.verifySuccess(resp.getStatus());
return resp.getTimeZone();
}
public void setTimeZone(String zoneId) throws TException, IoTDBSQLException {
TSSetTimeZoneReq req = new TSSetTimeZoneReq(zoneId);
TSSetTimeZoneResp resp = client.setTimeZone(req);
Utils.verifySuccess(resp.getStatus());
this.zoneId = ZoneId.of(zoneId);
}
}