Java替换sql语句中的时间字段为now()
一、背景
在采用Navicat复制脚本为insert语句后,sql语句中的时间部分都复制为了具体的时间.
在别处执行这个insert语句时,想把语句中的时间字段替换为now().
二、示例代码
注意点
- java虚拟机设置的字符串常量池大小好像是65536?
- 当sql语句值中也包含()字符时,如字段值为 用户收益(%).此时未做处理,在//todo部分,需要自行完善.
- changeTimeStrToNow(目标sql语句, 需替换的字段, 具体替换值);
package cn.yang37.tools;
import com.sun.xml.internal.ws.util.StringUtils;
import java.util.*;
import java.util.stream.Collectors;
import java.util.stream.Stream;
/**
* @Description:
* @Class: TimeToNow
* @Author: Yiang37
* @Date: 2021/3/30 10:41
* @Version: 1.0
*/
public class ChangeSqlTimeToNow {
public static void main(String[] args) {
String sqlStr = "INSERT INTO `bolg`.`message`(`id`, `nickname`, `avatar`, `content`, `ip`, `city`, `user_id`, `gmt_create`, `gmt_modify`, `state`) VALUES (49, '0', '0', '嘘', '119.39.133.74', '湖南省', 0, 1590681919947, 1590681919947, 1);\n";
String changeTimeStrToNow = changeTimeStrToNow(sqlStr, "`gmt_create`, `gmt_modify`", "NOW()");
System.out.println(changeTimeStrToNow);
//得到多条sql语句List
//List<String> stringList = Arrays.asList(sqlStr.split("\n"));
}
public static String changeTimeStrToNow(String inSqlStr, String targetStr, String replaceStr) {
String target = targetStr.toUpperCase(Locale.ROOT);
//计算出sql()中的 字段名部分 和 值部分
// INSERT INTO TABLE_NAME( aa,bb ) VALUES ( xx,yy );
// stringArr[0] : aa,bb
// stringArr[1] : xx,yy
String[] stringArr = splitSqlStr(inSqlStr);
//根据","分割,转为对应的 List<String>
List<String> nameList = Arrays.asList(stringArr[0].replace(" ", "").split(","));
List<String> valueList = Arrays.asList(stringArr[1].replace(" ", "").split(","));
//记录nameList中时间部分的索引
List<Integer> indexList = new ArrayList<>();
for (int i = 0; i < nameList.size(); i++) {
//若目标字符串中包含nameList.get(i)
if (target.contains(nameList.get(i).toUpperCase(Locale.ROOT))) {
indexList.add(i);
}
}
//valueList的对应位置替换成 replaceStr eg:now()
for (Integer integer : indexList) {
valueList.set(integer, replaceStr);
}
// 至此替换完成,valueList转换回去 流中每个元素末尾追加,
List<String> valueReplaceList = valueList.stream().map((e -> e + (","))).collect(Collectors.toList());
// valueReplaceList末尾去掉","
valueReplaceList.set(valueReplaceList.size() - 1, valueReplaceList.get(valueReplaceList.size() - 1).replace(",", ""));
// 转换格式( valueReplaceList );
String resValueStr = "(" + String.join("", valueReplaceList) + ");";
//获取原sql语句的VALUES前面部分的语句
String[] split2 = inSqlStr.toUpperCase(Locale.ROOT).split("VALUES");
//获得结果str
return split2[0] + "VALUES " + resValueStr;
}
public static String[] splitSqlStr(String inSqlStr) {
String[] resStrArr = new String[2];
//输入的sql按"("分割
//INSERT INTO TABLE_NAME( xx ) VALUES ( yy );
// INSERT INTO TABLE_NAME
// xx ) VALUES
// yy );
String[] split = inSqlStr.split("\\(");
//value中不包含(
if (split.length == 3) {
//字段名部分
resStrArr[0] = split[1].split("\\)")[0];
//字段值部分 如果value中包含() 则此处不能采用
resStrArr[1] = split[2].split("\\)")[0];
} else {
//说明value部分包含()的值
//todo
}
return resStrArr;
}
}
结果
INSERT INTO `BOLG`.`MESSAGE`(`ID`, `NICKNAME`, `AVATAR`, `CONTENT`, `IP`, `CITY`, `USER_ID`, `GMT_CREATE`, `GMT_MODIFY`, `STATE`) VALUES (49,'0','0','嘘','119.39.133.74','湖南省',0,NOW(),NOW(),1);