提交 2add95e9 编写于 作者: X xieyinglin

format the code

上级 76f82481
......@@ -33,9 +33,9 @@ public class SavedPreparedStatement {
/**
* insert or import sql template pattern, the template are the following:
*
* <p>
* insert/import into tableName [(field1, field2, ...)] [using stables tags(?, ?, ...) ] values(?, ?, ...) (?, ?, ...)
*
* <p>
* we split it to three part:
* 1. prefix, insert/import
* 2. middle, tableName [(field1, field2, ...)] [using stables tags(?, ?, ...) ]
......@@ -70,9 +70,9 @@ public class SavedPreparedStatement {
/**
* 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;
......@@ -81,7 +81,7 @@ public class SavedPreparedStatement {
*/
private boolean isAddBatch;
public SavedPreparedStatement(String sql, TSDBPreparedStatement tsdbPreparedStatement)throws SQLException {
public SavedPreparedStatement(String sql, TSDBPreparedStatement tsdbPreparedStatement) throws SQLException {
this.sql = sql;
this.tsdbPreparedStatement = tsdbPreparedStatement;
this.sqlParamList = new ArrayList<>();
......@@ -91,17 +91,18 @@ public class SavedPreparedStatement {
/**
* parse the init param according the sql param
*
* @param sql
*/
private void parsePreparedParam(String sql)throws SQLException{
private void parsePreparedParam(String sql) throws SQLException {
Matcher matcher = sqlPattern.matcher(sql);
if(matcher.find()){
if (matcher.find()) {
tableName = matcher.group("tablename");
if (tableName != null && PLACEHOLDER.equals(tableName)){
if (tableName != null && PLACEHOLDER.equals(tableName)) {
// the table name is dynamic
this.isTableNameDynamic = true;
}
......@@ -110,24 +111,24 @@ public class SavedPreparedStatement {
middle = matcher.group(2);
valueList = matcher.group("valueList");
if(middle != null && !"".equals(middle)){
if (middle != null && !"".equals(middle)) {
middleParamSize = parsePlaceholder(middle);
}
if(valueList != null && !"".equals(valueList)){
if (valueList != null && !"".equals(valueList)) {
valueListSize = parsePlaceholder(valueList);
}
initPreparedParam = initDefaultParam(tableName, middleParamSize, valueListSize);
}else{
} else {
// not match
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);
......@@ -140,14 +141,15 @@ public class SavedPreparedStatement {
/**
* generate the default param value list
*
* @param paramSize
* @return
*/
private List<Object> getDefaultParamList(int paramSize){
private List<Object> getDefaultParamList(int paramSize) {
List<Object> paramList = new ArrayList<>(paramSize);
if (paramSize > 0){
for (int i = 0; i < paramSize; i++){
if (paramSize > 0) {
for (int i = 0; i < paramSize; i++) {
paramList.add(i, DEFAULT_VALUE);
}
}
......@@ -157,17 +159,18 @@ public class SavedPreparedStatement {
/**
* calculate the placeholder num
*
* @param value
* @return
*/
private int parsePlaceholder(String value){
private int parsePlaceholder(String value) {
Pattern pattern = Pattern.compile("[?]");
Matcher matcher = pattern.matcher(value);
int result = 0;
while (matcher.find()){
while (matcher.find()) {
result++;
}
return result;
......@@ -175,34 +178,35 @@ public class SavedPreparedStatement {
/**
* set current row params
*
* @param parameterIndex the first parameter is 1, the second is 2, ...
* @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;
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));
}
this.isAddBatch = false; //set isAddBatch to false
if (x == null){
if (x == null) {
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);
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);
return;
......@@ -220,9 +224,9 @@ public class SavedPreparedStatement {
/**
* 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.isAddBatch = true;
......@@ -231,6 +235,7 @@ public class SavedPreparedStatement {
/**
* execute the sql with batch sql
*
* @return
* @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
}
......@@ -262,26 +267,27 @@ public class SavedPreparedStatement {
/**
* generate the batch sql
*
* @return
*/
private String generateExecuteSql(){
private String generateExecuteSql() {
StringBuilder stringBuilder = new StringBuilder();
stringBuilder.append(prefix);
stringBuilder.append(" into ");
if (!isTableNameDynamic){
if (!isTableNameDynamic) {
// tablename will not need to be replaced
stringBuilder.append(middle);
stringBuilder.append(" values");
stringBuilder.append(replaceValueListParam(valueList, sqlParamList));
} else{
} else {
// need to replace tablename
if (sqlParamList.size() > 0 ){
if (sqlParamList.size() > 0) {
TSDBPreparedParam firstPreparedParam = sqlParamList.get(0);
......@@ -292,12 +298,12 @@ public class SavedPreparedStatement {
//the first param in the middleParamList is the tableName
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);
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
String values = replaceTemplateParam(valueList, currentParam.getValueList());
......@@ -313,7 +319,7 @@ public class SavedPreparedStatement {
}
}
}else{
} else {
stringBuilder.append(middle);
stringBuilder.append(" values");
......@@ -327,10 +333,11 @@ public class SavedPreparedStatement {
/**
* replace the middle and value part
*
* @param tsdbPreparedParam
* @return
*/
private String replaceMiddleAndValuePart(TSDBPreparedParam tsdbPreparedParam){
private String replaceMiddleAndValuePart(TSDBPreparedParam tsdbPreparedParam) {
StringBuilder stringBuilder = new StringBuilder(" ");
......@@ -348,17 +355,18 @@ public class SavedPreparedStatement {
/**
* replace the placeholder of the template with TSDBPreparedParam list
*
* @param template
* @param sqlParamList
* @return
*/
private String replaceValueListParam(String template, List<TSDBPreparedParam> sqlParamList){
private String replaceValueListParam(String template, List<TSDBPreparedParam> sqlParamList) {
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());
......@@ -374,13 +382,14 @@ public class SavedPreparedStatement {
/**
* replace the placeholder of the template with paramList
*
* @param template
* @param paramList
* @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;
......@@ -402,10 +411,11 @@ public class SavedPreparedStatement {
/**
* get the string of param object
*
* @param paramObj
* @return
*/
private String getParamString(Object paramObj){
private String getParamString(Object paramObj) {
String paraStr = paramObj.toString();
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);
}
......
......@@ -47,16 +47,16 @@ public class TSDBPreparedStatement extends TSDBStatement implements PreparedStat
init(sql);
}
private void init(String sql){
private void init(String sql) {
this.rawSql = sql;
preprocessSql();
this.isSaved = isSavedSql(this.rawSql);
if (this.isSaved){
if (this.isSaved) {
try {
this.savedPreparedStatement = new SavedPreparedStatement(this.rawSql, this);
} catch (SQLException e){
} catch (SQLException e) {
e.printStackTrace();
}
}
......@@ -64,17 +64,18 @@ public class TSDBPreparedStatement extends TSDBStatement implements PreparedStat
/**
* if the precompiled sql is insert or import
*
* @param sql
* @return
*/
private boolean isSavedSql(String sql){
private boolean isSavedSql(String sql) {
Matcher matcher = savePattern.matcher(sql);
return matcher.find();
}
@Override
public int[] executeBatch() throws SQLException {
if (isSaved){
if (isSaved) {
return this.savedPreparedStatement.executeBatch();
} else {
return super.executeBatch();
......@@ -153,6 +154,7 @@ public class TSDBPreparedStatement extends TSDBStatement implements PreparedStat
/**
* Populate parameters into prepared sql statements
*
* @return a string of the native sql statement for TSDB
*/
private String getNativeSql() {
......@@ -175,17 +177,17 @@ public class TSDBPreparedStatement extends TSDBStatement implements PreparedStat
@Override
public ResultSet executeQuery() throws SQLException {
if (isSaved){
if (isSaved) {
this.savedPreparedStatement.executeBatchInternal();
return null;
}else {
} else {
return super.executeQuery(getNativeSql());
}
}
@Override
public int executeUpdate() throws SQLException {
if (isSaved){
if (isSaved) {
return this.savedPreparedStatement.executeBatchInternal();
} else {
return super.executeUpdate(getNativeSql());
......@@ -289,16 +291,16 @@ public class TSDBPreparedStatement extends TSDBStatement implements PreparedStat
@Override
public void setObject(int parameterIndex, Object x) throws SQLException {
if (isSaved){
if (isSaved) {
this.savedPreparedStatement.setParam(parameterIndex, x);
}else{
} else {
parameters.add(x);
}
}
@Override
public boolean execute() throws SQLException {
if (isSaved){
if (isSaved) {
int result = this.savedPreparedStatement.executeBatchInternal();
return result > 0;
} else {
......@@ -308,9 +310,9 @@ public class TSDBPreparedStatement extends TSDBStatement implements PreparedStat
@Override
public void addBatch() throws SQLException {
if (isSaved){
if (isSaved) {
this.savedPreparedStatement.addBatch();
}else {
} else {
if (this.batchedArgs == null) {
batchedArgs = new ArrayList<String>();
......
......@@ -42,7 +42,7 @@ public class TSDBPreparedParam {
this.middleParamList = middleParamList;
}
public void setMiddleParam(int parameterIndex, Object x){
public void setMiddleParam(int parameterIndex, Object 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);
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册