opensource/Chat2DB
1
0
Fork 0
mirror of https://github.com/CodePhiliaX/Chat2DB.git synced 2025-07-29 18:53:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

MySQL移动列问题修复。

Browse Source 
修复跨多连续数据移动不能识别的问题。
This commit is contained in:
tmlx1990
2024-04-12 11:47:44 +08:00
parent d7667c6f82
commit 9d2af9c6e2
1 changed files with 11 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
chat2db-server/chat2db-plugins/chat2db-mysql/src/main/java/ai/chat2db/plugin/mysql/builder/MysqlSqlBuilder.java
View File

@ -12,6 +12,7 @@ import cn.hutool.core.util.ArrayUtil;
import org.apache.commons.lang3.StringUtils;
import java.util.*;
import java.util.concurrent.atomic.AtomicInteger;
public class MysqlSqlBuilder extends DefaultSqlBuilder {
@ -177,7 +178,7 @@ public class MysqlSqlBuilder extends DefaultSqlBuilder {
if (!originalArray[0].equals(targetArray[0])) {
int a = findIndex(originalArray, targetArray[0]);
TableColumn column = oldTable.getColumnList().stream().filter(col -> StringUtils.equals(col.getName(), originalArray[a])).findFirst().get();
String[] newArray = moveElement(originalArray, a, 0, targetArray);
String[] newArray = moveElement(originalArray, a, 0, targetArray, new AtomicInteger(0));
sql.append(" MODIFY COLUMN ");
MysqlColumnTypeEnum typeEnum = MysqlColumnTypeEnum.getByType(column.getColumnType());
sql.append(typeEnum.buildColumn(column));
@ -201,7 +202,7 @@ public class MysqlSqlBuilder extends DefaultSqlBuilder {
int a = findIndex(originalArray, targetArray[max]);
//System.out.println("Move " + originalArray[a] + " after " + (a > 0 ? originalArray[max] : "start"));
TableColumn column = oldTable.getColumnList().stream().filter(col -> StringUtils.equals(col.getName(), originalArray[a])).findFirst().get();
String[] newArray = moveElement(originalArray, a, max, targetArray);
String[] newArray = moveElement(originalArray, a, max, targetArray, new AtomicInteger(0));
if (n > 0) {
sql.append("ALTER TABLE ");
if (StringUtils.isNotBlank(oldTable.getDatabaseName())) {
@ -249,15 +250,17 @@ public class MysqlSqlBuilder extends DefaultSqlBuilder {
sql.append(" COMMENT '").append(column.getComment()).append("'");
}
sql.append(" AFTER ");
AtomicInteger continuousDataCount = new AtomicInteger(0);
String[] newArray = moveElement(originalArray, i, a, targetArray, continuousDataCount);
if (i < a) {
sql.append(originalArray[a]);
sql.append(originalArray[a + continuousDataCount.get()]);
} else {
sql.append(originalArray[a - 1]);
}
sql.append(";\n");
n++;
String[] newArray = moveElement(originalArray, i, a, targetArray);
if (Arrays.equals(newArray, targetArray)) {
return newArray;
}
@ -285,23 +288,22 @@ public class MysqlSqlBuilder extends DefaultSqlBuilder {
|| (i > 0 && a > 0 && !originalArray[i - 1].equals(targetArray[a - 1]));
}
private static String[] moveElement(String[] originalArray, int from, int to, String[] targetArray) {
private static String[] moveElement(String[] originalArray, int from, int to, String[] targetArray, AtomicInteger continuousDataCount) {
String[] newArray = new String[originalArray.length];
System.arraycopy(originalArray, 0, newArray, 0, originalArray.length);
String temp = newArray[from];
// 是否有连续移动数据
boolean isContinuousData = false;
// 连续数据数量
int continuousDataCount = 0;
if (from < to) {
for (int i = to; i < originalArray.length - 1; i++) {
if (originalArray[i+1].equals(targetArray[findIndex(targetArray, originalArray[i]) +1])) {
continuousDataCount++;
continuousDataCount.set(continuousDataCount.incrementAndGet());
} else {
break;
}
}
if (continuousDataCount > 0) {
if (continuousDataCount.get() > 0) {
System.arraycopy(originalArray, from + 1, newArray, from, to - from +1);
isContinuousData = true;
} else {
@ -311,7 +313,7 @@ public class MysqlSqlBuilder extends DefaultSqlBuilder {
System.arraycopy(originalArray, to, newArray, to + 1, from - to);
}
if (isContinuousData){
newArray[to+continuousDataCount] = temp;
newArray[to+continuousDataCount.get()] = temp;
} else {
newArray[to] = temp;
}