xxq250
/
homework-jianmu
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

change

This commit is contained in:
zyyang 2021-01-27 17:10:08 +08:00
parent a763cb48fe
commit b346f51889
1 changed files with 27 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

64
tests/examples/JDBC/JDBCDemo/src/main/java/com/taosdata/example/SubscribeDemo.java
View File

@ -7,52 +7,45 @@ import com.taosdata.jdbc.TSDBSubscribe;
import java.sql.DriverManager; import java.sql.DriverManager;
import java.util.Properties; import java.util.Properties;
import java.util.concurrent.TimeUnit;
public class SubscribeDemo { public class SubscribeDemo {
private static final String usage = "java -jar SubscribeDemo.jar -host <hostname> -database <database name> -topic <topic> -sql <sql>";
public static TSDBConnection getConnection(String host, String database) throws Exception { public static void main(String[] args) {
Class.forName("com.taosdata.jdbc.TSDBDriver"); // parse args from command line
Properties properties = new Properties(); String host = "", database = "", topic = "", sql = "";
properties.setProperty(TSDBDriver.PROPERTY_KEY_CHARSET, "UTF-8");
properties.setProperty(TSDBDriver.PROPERTY_KEY_LOCALE, "en_US.UTF-8");
properties.setProperty(TSDBDriver.PROPERTY_KEY_TIME_ZONE, "UTC-8");
String cs = String.format("jdbc:TAOS://%s:0/%s", host, database);
return (TSDBConnection) DriverManager.getConnection(cs, properties);
}
public static void main(String[] args) throws Exception {
String usage = "java -Djava.ext.dirs=../ TestTSDBSubscribe [-host host] <-db database> <-topic topic> <-sql sql>";
if (args.length < 2) {
System.err.println(usage);
return;
}
String host = "localhost", database = "", topic = "", sql = "";
for (int i = 0; i < args.length; i++) { for (int i = 0; i < args.length; i++) {
if ("-db".equalsIgnoreCase(args[i]) && i < args.length - 1) { if ("-host".equalsIgnoreCase(args[i]) && i < args.length - 1) {
host = args[++i];
}
if ("-database".equalsIgnoreCase(args[i]) && i < args.length - 1) {
database = args[++i]; database = args[++i];
} }
if ("-topic".equalsIgnoreCase(args[i]) && i < args.length - 1) { if ("-topic".equalsIgnoreCase(args[i]) && i < args.length - 1) {
topic = args[++i]; topic = args[++i];
} }
if ("-host".equalsIgnoreCase(args[i]) && i < args.length - 1) {
host = args[++i];
}
if ("-sql".equalsIgnoreCase(args[i]) && i < args.length - 1) { if ("-sql".equalsIgnoreCase(args[i]) && i < args.length - 1) {
sql = args[++i]; sql = args[++i];
} }
} }
if (database.isEmpty() || topic.isEmpty() || sql.isEmpty()) { if (host.isEmpty() || database.isEmpty() || topic.isEmpty() || sql.isEmpty()) {
System.err.println(usage); System.out.println(usage);
return; return;
} }
/*********************************************************************************************/
TSDBConnection connection = null;
TSDBSubscribe sub = null;
try { try {
connection = getConnection(host, database); Class.forName("com.taosdata.jdbc.TSDBDriver");
sub = ((TSDBConnection) connection).subscribe(topic, sql, false); Properties properties = new Properties();
properties.setProperty(TSDBDriver.PROPERTY_KEY_CHARSET, "UTF-8");
properties.setProperty(TSDBDriver.PROPERTY_KEY_LOCALE, "en_US.UTF-8");
properties.setProperty(TSDBDriver.PROPERTY_KEY_TIME_ZONE, "UTC-8");
final String url = "jdbc:TAOS://" + host + ":6030/" + database + "?user=root&password=taosdata";
// get TSDBConnection
TSDBConnection connection = DriverManager.getConnection(url, properties).unwrap(TSDBConnection.class);
// create TSDBSubscribe
TSDBSubscribe sub = connection.subscribe(topic, sql, false);
int total = 0; int total = 0;
while (true) { while (true) {
@ -63,17 +56,14 @@ public class SubscribeDemo {
} }
total += count; total += count;
System.out.printf("%d rows consumed, total %d\n", count, total); System.out.printf("%d rows consumed, total %d\n", count, total);
Thread.sleep(900); if (total >= 10)
break;
TimeUnit.SECONDS.sleep(1);
} }
sub.close(false);
connection.close();
} catch (Exception e) { } catch (Exception e) {
e.printStackTrace(); e.printStackTrace();
} finally {
if (null != sub) {
sub.close(true);
}
if (null != connection) {
connection.close();
}
} }
} }
} }