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 15:12:48 +08:00
parent 3ef5dea4a4
commit 99ef14af98
1 changed files with 21 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
tests/examples/JDBC/connectionPools/src/main/java/com/taosdata/example/ConnectionPoolDemo.java
View File

@ -20,13 +20,14 @@ public class ConnectionPoolDemo {
private static Logger logger = Logger.getLogger(DruidPoolBuilder.class); private static Logger logger = Logger.getLogger(DruidPoolBuilder.class);
private static final String dbName = "pool_test"; private static final String dbName = "pool_test";
private static long totalSize = 1_000_000l;
private static int batchSize = 10;
private static int sleep = 1000;
private static int poolSize = 50;
private static int tableSize = 1000;
private static int threadCount = 50;
private static String poolType = "hikari"; private static String poolType = "hikari";
private static long totalSize = 1_000_000l;
private static long tableSize = 1;
private static long batchSize = 1;
private static int poolSize = 50;
private static int threadCount = 50;
private static int sleep = 1000;
public static void main(String[] args) { public static void main(String[] args) {
@ -35,16 +36,26 @@ public class ConnectionPoolDemo {
if ("-host".equalsIgnoreCase(args[i]) && i < args.length - 1) { if ("-host".equalsIgnoreCase(args[i]) && i < args.length - 1) {
host = args[++i]; host = args[++i];
} }
if ("-poolType".equalsIgnoreCase(args[i]) && i < args.length - 1) {
poolType = args[++i];
}
if ("-recordNumber".equalsIgnoreCase(args[i]) && i < args.length - 1) { if ("-recordNumber".equalsIgnoreCase(args[i]) && i < args.length - 1) {
totalSize = Long.parseLong(args[++i]); totalSize = Long.parseLong(args[++i]);
} }
if ("-poolType".equalsIgnoreCase(args[i]) && i < args.length - 1) { if ("-tableNumber".equalsIgnoreCase(args[i]) && i < args.length - 1) {
poolType = args[++i]; tableSize = Long.parseLong(args[++i]);
}
if ("-batchNumber".equalsIgnoreCase(args[i]) && i < args.length - 1) {
batchSize = Long.parseLong(args[++i]);
} }
} }
if (host == null) { if (host == null) {
System.out.println("Usage: java -jar XXX.jar -host <hostname> -recordNumber <totalNumber> -poolType <c3p0| dbcp| druid| hikari>"); System.out.println("Usage: java -jar XXX.jar -host <hostname> " +
"-recordNumber <number> " +
"-tableNumber <number> " +
"-batchNumber <number> " +
"-poolType <c3p0| dbcp| druid| hikari>");
return; return;
} }
@ -83,7 +94,7 @@ public class ConnectionPoolDemo {
// } // }
ExecutorService executor = Executors.newFixedThreadPool(threadCount); ExecutorService executor = Executors.newFixedThreadPool(threadCount);
for (long i = 0; i < totalSize / batchSize / tableSize; i++) { for (long i = 0; i < totalSize / tableSize / batchSize; i++) {
executor.execute(new InsertTask(dataSource, dbName, tableSize, batchSize)); executor.execute(new InsertTask(dataSource, dbName, tableSize, batchSize));
// sleep few seconds // sleep few seconds
try { try {