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

format the code

This commit is contained in:
xieyinglin 2019-10-24 12:19:49 +08:00
parent 76f82481f5
commit 2add95e9fe
3 changed files with 388 additions and 376 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

96
src/connector/jdbc/src/main/java/com/taosdata/jdbc/SavedPreparedStatement.java
View File

@ -33,9 +33,9 @@ public class SavedPreparedStatement {
/** /**
* insert or import sql template pattern, the template are the following: * insert or import sql template pattern, the template are the following:
* * <p>
* insert/import into tableName [(field1, field2, ...)] [using stables tags(?, ?, ...) ] values(?, ?, ...) (?, ?, ...) * insert/import into tableName [(field1, field2, ...)] [using stables tags(?, ?, ...) ] values(?, ?, ...) (?, ?, ...)
* * <p>
* we split it to three part: * we split it to three part:
* 1. prefix, insert/import * 1. prefix, insert/import
* 2. middle, tableName [(field1, field2, ...)] [using stables tags(?, ?, ...) ] * 2. middle, tableName [(field1, field2, ...)] [using stables tags(?, ?, ...) ]
@ -70,9 +70,9 @@ public class SavedPreparedStatement {
/** /**
* default param value * default param value
*/ */
private static final String DEFAULT_VALUE= "NULL"; private static final String DEFAULT_VALUE = "NULL";
private static final String PLACEHOLDER= "?"; private static final String PLACEHOLDER = "?";
private String tableName; private String tableName;
@ -81,7 +81,7 @@ public class SavedPreparedStatement {
*/ */
private boolean isAddBatch; private boolean isAddBatch;
public SavedPreparedStatement(String sql, TSDBPreparedStatement tsdbPreparedStatement)throws SQLException { public SavedPreparedStatement(String sql, TSDBPreparedStatement tsdbPreparedStatement) throws SQLException {
this.sql = sql; this.sql = sql;
this.tsdbPreparedStatement = tsdbPreparedStatement; this.tsdbPreparedStatement = tsdbPreparedStatement;
this.sqlParamList = new ArrayList<>(); this.sqlParamList = new ArrayList<>();
@ -91,17 +91,18 @@ public class SavedPreparedStatement {
/** /**
* parse the init param according the sql param * parse the init param according the sql param
*
* @param sql * @param sql
*/ */
private void parsePreparedParam(String sql)throws SQLException{ private void parsePreparedParam(String sql) throws SQLException {
Matcher matcher = sqlPattern.matcher(sql); Matcher matcher = sqlPattern.matcher(sql);
if(matcher.find()){ if (matcher.find()) {
tableName = matcher.group("tablename"); tableName = matcher.group("tablename");
if (tableName != null && PLACEHOLDER.equals(tableName)){ if (tableName != null && PLACEHOLDER.equals(tableName)) {
// the table name is dynamic // the table name is dynamic
this.isTableNameDynamic = true; this.isTableNameDynamic = true;
} }
@ -110,24 +111,24 @@ public class SavedPreparedStatement {
middle = matcher.group(2); middle = matcher.group(2);
valueList = matcher.group("valueList"); valueList = matcher.group("valueList");
if(middle != null && !"".equals(middle)){ if (middle != null && !"".equals(middle)) {
middleParamSize = parsePlaceholder(middle); middleParamSize = parsePlaceholder(middle);
} }
if(valueList != null && !"".equals(valueList)){ if (valueList != null && !"".equals(valueList)) {
valueListSize = parsePlaceholder(valueList); valueListSize = parsePlaceholder(valueList);
} }
initPreparedParam = initDefaultParam(tableName, middleParamSize, valueListSize); initPreparedParam = initDefaultParam(tableName, middleParamSize, valueListSize);
}else{ } else {
// not match // not match
throw new SQLException(TSDBConstants.WrapErrMsg("the sql is not complete!")); throw new SQLException(TSDBConstants.WrapErrMsg("the sql is not complete!"));
} }
} }
private TSDBPreparedParam initDefaultParam(String tableName, int middleParamSize, int valueListSize){ private TSDBPreparedParam initDefaultParam(String tableName, int middleParamSize, int valueListSize) {
TSDBPreparedParam tsdbPreparedParam = new TSDBPreparedParam(tableName); TSDBPreparedParam tsdbPreparedParam = new TSDBPreparedParam(tableName);
@ -140,14 +141,15 @@ public class SavedPreparedStatement {
/** /**
* generate the default param value list * generate the default param value list
*
* @param paramSize * @param paramSize
* @return * @return
*/ */
private List<Object> getDefaultParamList(int paramSize){ private List<Object> getDefaultParamList(int paramSize) {
List<Object> paramList = new ArrayList<>(paramSize); List<Object> paramList = new ArrayList<>(paramSize);
if (paramSize > 0){ if (paramSize > 0) {
for (int i = 0; i < paramSize; i++){ for (int i = 0; i < paramSize; i++) {
paramList.add(i, DEFAULT_VALUE); paramList.add(i, DEFAULT_VALUE);
} }
} }
@ -157,17 +159,18 @@ public class SavedPreparedStatement {
/** /**
* calculate the placeholder num * calculate the placeholder num
*
* @param value * @param value
* @return * @return
*/ */
private int parsePlaceholder(String value){ private int parsePlaceholder(String value) {
Pattern pattern = Pattern.compile("[?]"); Pattern pattern = Pattern.compile("[?]");
Matcher matcher = pattern.matcher(value); Matcher matcher = pattern.matcher(value);
int result = 0; int result = 0;
while (matcher.find()){ while (matcher.find()) {
result++; result++;
} }
return result; return result;
@ -175,34 +178,35 @@ public class SavedPreparedStatement {
/** /**
* set current row params * set current row params
*
* @param parameterIndex the first parameter is 1, the second is 2, ... * @param parameterIndex the first parameter is 1, the second is 2, ...
* @param x the parameter value * @param x the parameter value
*/ */
public void setParam(int parameterIndex, Object x) throws SQLException{ public void setParam(int parameterIndex, Object x) throws SQLException {
int paramSize = this.middleParamSize + this.valueListSize; int paramSize = this.middleParamSize + this.valueListSize;
String errorMsg = String.format("the parameterIndex %s out of the range [1, %s]", parameterIndex, this.middleParamSize + this.valueListSize); String errorMsg = String.format("the parameterIndex %s out of the range [1, %s]", parameterIndex, this.middleParamSize + this.valueListSize);
if (parameterIndex < 1 || parameterIndex > paramSize){ if (parameterIndex < 1 || parameterIndex > paramSize) {
throw new SQLException(TSDBConstants.WrapErrMsg(errorMsg)); throw new SQLException(TSDBConstants.WrapErrMsg(errorMsg));
} }
this.isAddBatch = false; //set isAddBatch to false this.isAddBatch = false; //set isAddBatch to false
if (x == null){ if (x == null) {
x = DEFAULT_VALUE; // set default null string x = DEFAULT_VALUE; // set default null string
} }
parameterIndex = parameterIndex -1; // start from 0 in param list parameterIndex = parameterIndex - 1; // start from 0 in param list
if (this.middleParamSize != 0 && parameterIndex >= 0 && parameterIndex < this.middleParamSize){ if (this.middleParamSize != 0 && parameterIndex >= 0 && parameterIndex < this.middleParamSize) {
this.initPreparedParam.setMiddleParam(parameterIndex, x); this.initPreparedParam.setMiddleParam(parameterIndex, x);
return; return;
} }
if (this.valueListSize != 0 && parameterIndex >= this.middleParamSize && parameterIndex < (this.middleParamSize + this.valueListSize)){ if (this.valueListSize != 0 && parameterIndex >= this.middleParamSize && parameterIndex < paramSize) {
this.initPreparedParam.setValueParam(parameterIndex - this.middleParamSize, x); this.initPreparedParam.setValueParam(parameterIndex - this.middleParamSize, x);
return; return;
@ -220,9 +224,9 @@ public class SavedPreparedStatement {
/** /**
* add current param to batch list * add current param to batch list
*/ */
private void addCurrentRowParamToList(){ private void addCurrentRowParamToList() {
if (initPreparedParam != null && (this.middleParamSize > 0 || this.valueListSize > 0)){ if (initPreparedParam != null && (this.middleParamSize > 0 || this.valueListSize > 0)) {
this.sqlParamList.add(initPreparedParam); // add current param to batch list this.sqlParamList.add(initPreparedParam); // add current param to batch list
} }
this.isAddBatch = true; this.isAddBatch = true;
@ -231,6 +235,7 @@ public class SavedPreparedStatement {
/** /**
* execute the sql with batch sql * execute the sql with batch sql
*
* @return * @return
* @throws SQLException * @throws SQLException
*/ */
@ -242,9 +247,9 @@ public class SavedPreparedStatement {
} }
public int executeBatchInternal() throws SQLException{ public int executeBatchInternal() throws SQLException {
if (!isAddBatch){ if (!isAddBatch) {
addCurrentRowParamToList(); // add current param to batch list addCurrentRowParamToList(); // add current param to batch list
} }
@ -262,26 +267,27 @@ public class SavedPreparedStatement {
/** /**
* generate the batch sql * generate the batch sql
*
* @return * @return
*/ */
private String generateExecuteSql(){ private String generateExecuteSql() {
StringBuilder stringBuilder = new StringBuilder(); StringBuilder stringBuilder = new StringBuilder();
stringBuilder.append(prefix); stringBuilder.append(prefix);
stringBuilder.append(" into "); stringBuilder.append(" into ");
if (!isTableNameDynamic){ if (!isTableNameDynamic) {
// tablename will not need to be replaced // tablename will not need to be replaced
stringBuilder.append(middle); stringBuilder.append(middle);
stringBuilder.append(" values"); stringBuilder.append(" values");
stringBuilder.append(replaceValueListParam(valueList, sqlParamList)); stringBuilder.append(replaceValueListParam(valueList, sqlParamList));
} else{ } else {
// need to replace tablename // need to replace tablename
if (sqlParamList.size() > 0 ){ if (sqlParamList.size() > 0) {
TSDBPreparedParam firstPreparedParam = sqlParamList.get(0); TSDBPreparedParam firstPreparedParam = sqlParamList.get(0);
@ -292,12 +298,12 @@ public class SavedPreparedStatement {
//the first param in the middleParamList is the tableName //the first param in the middleParamList is the tableName
String lastTableName = firstPreparedParam.getMiddleParamList().get(0).toString(); String lastTableName = firstPreparedParam.getMiddleParamList().get(0).toString();
if (sqlParamList.size() > 1){ if (sqlParamList.size() > 1) {
for (int i = 1; i < sqlParamList.size(); i++){ for (int i = 1; i < sqlParamList.size(); i++) {
TSDBPreparedParam currentParam = sqlParamList.get(i); TSDBPreparedParam currentParam = sqlParamList.get(i);
String currentTableName = currentParam.getMiddleParamList().get(0).toString(); String currentTableName = currentParam.getMiddleParamList().get(0).toString();
if (lastTableName.equalsIgnoreCase(currentTableName)){ if (lastTableName.equalsIgnoreCase(currentTableName)) {
// tablename is same with the last row ,so only need to append the part of value // tablename is same with the last row ,so only need to append the part of value
String values = replaceTemplateParam(valueList, currentParam.getValueList()); String values = replaceTemplateParam(valueList, currentParam.getValueList());
@ -313,7 +319,7 @@ public class SavedPreparedStatement {
} }
} }
}else{ } else {
stringBuilder.append(middle); stringBuilder.append(middle);
stringBuilder.append(" values"); stringBuilder.append(" values");
@ -327,10 +333,11 @@ public class SavedPreparedStatement {
/** /**
* replace the middle and value part * replace the middle and value part
*
* @param tsdbPreparedParam * @param tsdbPreparedParam
* @return * @return
*/ */
private String replaceMiddleAndValuePart(TSDBPreparedParam tsdbPreparedParam){ private String replaceMiddleAndValuePart(TSDBPreparedParam tsdbPreparedParam) {
StringBuilder stringBuilder = new StringBuilder(" "); StringBuilder stringBuilder = new StringBuilder(" ");
@ -348,17 +355,18 @@ public class SavedPreparedStatement {
/** /**
* replace the placeholder of the template with TSDBPreparedParam list * replace the placeholder of the template with TSDBPreparedParam list
*
* @param template * @param template
* @param sqlParamList * @param sqlParamList
* @return * @return
*/ */
private String replaceValueListParam(String template, List<TSDBPreparedParam> sqlParamList){ private String replaceValueListParam(String template, List<TSDBPreparedParam> sqlParamList) {
StringBuilder stringBuilder = new StringBuilder(); StringBuilder stringBuilder = new StringBuilder();
if (sqlParamList.size() > 0 ){ if (sqlParamList.size() > 0) {
for (TSDBPreparedParam tsdbPreparedParam : sqlParamList){ for (TSDBPreparedParam tsdbPreparedParam : sqlParamList) {
String tmp = replaceTemplateParam(template, tsdbPreparedParam.getValueList()); String tmp = replaceTemplateParam(template, tsdbPreparedParam.getValueList());
@ -374,13 +382,14 @@ public class SavedPreparedStatement {
/** /**
* replace the placeholder of the template with paramList * replace the placeholder of the template with paramList
*
* @param template * @param template
* @param paramList * @param paramList
* @return * @return
*/ */
private String replaceTemplateParam(String template, List<Object> paramList){ private String replaceTemplateParam(String template, List<Object> paramList) {
if (paramList.size() > 0){ if (paramList.size() > 0) {
String tmp = template; String tmp = template;
@ -402,10 +411,11 @@ public class SavedPreparedStatement {
/** /**
* get the string of param object * get the string of param object
*
* @param paramObj * @param paramObj
* @return * @return
*/ */
private String getParamString(Object paramObj){ private String getParamString(Object paramObj) {
String paraStr = paramObj.toString(); String paraStr = paramObj.toString();
if (paramObj instanceof Timestamp || (paramObj instanceof String && !DEFAULT_VALUE.equalsIgnoreCase(paraStr))) { if (paramObj instanceof Timestamp || (paramObj instanceof String && !DEFAULT_VALUE.equalsIgnoreCase(paraStr))) {
@ -415,7 +425,7 @@ public class SavedPreparedStatement {
} }
private int executeSql(String sql)throws SQLException{ private int executeSql(String sql) throws SQLException {
return tsdbPreparedStatement.executeUpdate(sql); return tsdbPreparedStatement.executeUpdate(sql);
} }

28
src/connector/jdbc/src/main/java/com/taosdata/jdbc/TSDBPreparedStatement.java
View File

@ -47,16 +47,16 @@ public class TSDBPreparedStatement extends TSDBStatement implements PreparedStat
init(sql); init(sql);
} }
private void init(String sql){ private void init(String sql) {
this.rawSql = sql; this.rawSql = sql;
preprocessSql(); preprocessSql();
this.isSaved = isSavedSql(this.rawSql); this.isSaved = isSavedSql(this.rawSql);
if (this.isSaved){ if (this.isSaved) {
try { try {
this.savedPreparedStatement = new SavedPreparedStatement(this.rawSql, this); this.savedPreparedStatement = new SavedPreparedStatement(this.rawSql, this);
} catch (SQLException e){ } catch (SQLException e) {
e.printStackTrace(); e.printStackTrace();
} }
} }
@ -64,17 +64,18 @@ public class TSDBPreparedStatement extends TSDBStatement implements PreparedStat
/** /**
* if the precompiled sql is insert or import * if the precompiled sql is insert or import
*
* @param sql * @param sql
* @return * @return
*/ */
private boolean isSavedSql(String sql){ private boolean isSavedSql(String sql) {
Matcher matcher = savePattern.matcher(sql); Matcher matcher = savePattern.matcher(sql);
return matcher.find(); return matcher.find();
} }
@Override @Override
public int[] executeBatch() throws SQLException { public int[] executeBatch() throws SQLException {
if (isSaved){ if (isSaved) {
return this.savedPreparedStatement.executeBatch(); return this.savedPreparedStatement.executeBatch();
} else { } else {
return super.executeBatch(); return super.executeBatch();
@ -153,6 +154,7 @@ public class TSDBPreparedStatement extends TSDBStatement implements PreparedStat
/** /**
* Populate parameters into prepared sql statements * Populate parameters into prepared sql statements
*
* @return a string of the native sql statement for TSDB * @return a string of the native sql statement for TSDB
*/ */
private String getNativeSql() { private String getNativeSql() {
@ -175,17 +177,17 @@ public class TSDBPreparedStatement extends TSDBStatement implements PreparedStat
@Override @Override
public ResultSet executeQuery() throws SQLException { public ResultSet executeQuery() throws SQLException {
if (isSaved){ if (isSaved) {
this.savedPreparedStatement.executeBatchInternal(); this.savedPreparedStatement.executeBatchInternal();
return null; return null;
}else { } else {
return super.executeQuery(getNativeSql()); return super.executeQuery(getNativeSql());
} }
} }
@Override @Override
public int executeUpdate() throws SQLException { public int executeUpdate() throws SQLException {
if (isSaved){ if (isSaved) {
return this.savedPreparedStatement.executeBatchInternal(); return this.savedPreparedStatement.executeBatchInternal();
} else { } else {
return super.executeUpdate(getNativeSql()); return super.executeUpdate(getNativeSql());
@ -289,16 +291,16 @@ public class TSDBPreparedStatement extends TSDBStatement implements PreparedStat
@Override @Override
public void setObject(int parameterIndex, Object x) throws SQLException { public void setObject(int parameterIndex, Object x) throws SQLException {
if (isSaved){ if (isSaved) {
this.savedPreparedStatement.setParam(parameterIndex, x); this.savedPreparedStatement.setParam(parameterIndex, x);
}else{ } else {
parameters.add(x); parameters.add(x);
} }
} }
@Override @Override
public boolean execute() throws SQLException { public boolean execute() throws SQLException {
if (isSaved){ if (isSaved) {
int result = this.savedPreparedStatement.executeBatchInternal(); int result = this.savedPreparedStatement.executeBatchInternal();
return result > 0; return result > 0;
} else { } else {
@ -308,9 +310,9 @@ public class TSDBPreparedStatement extends TSDBStatement implements PreparedStat
@Override @Override
public void addBatch() throws SQLException { public void addBatch() throws SQLException {
if (isSaved){ if (isSaved) {
this.savedPreparedStatement.addBatch(); this.savedPreparedStatement.addBatch();
}else { } else {
if (this.batchedArgs == null) { if (this.batchedArgs == null) {
batchedArgs = new ArrayList<String>(); batchedArgs = new ArrayList<String>();

4
src/connector/jdbc/src/main/java/com/taosdata/jdbc/bean/TSDBPreparedParam.java
View File

@ -42,7 +42,7 @@ public class TSDBPreparedParam {
this.middleParamList = middleParamList; this.middleParamList = middleParamList;
} }
public void setMiddleParam(int parameterIndex, Object x){ public void setMiddleParam(int parameterIndex, Object x) {
this.middleParamList.set(parameterIndex, x); this.middleParamList.set(parameterIndex, x);
} }
@ -55,7 +55,7 @@ public class TSDBPreparedParam {
} }
public void setValueParam(int parameterIndex, Object x){ public void setValueParam(int parameterIndex, Object x) {
this.valueList.set(parameterIndex, x); this.valueList.set(parameterIndex, x);
} }