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

[fix]:下掉bson

Browse Source
This commit is contained in:
luojun
2024-05-31 20:51:32 +08:00
parent 9d87a42a3b
commit 35d99b1c0b
1 changed files with 15 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
chat2db-server/chat2db-spi/src/main/java/ai/chat2db/spi/sql/DocumentUtils.java
View File

@ -3,8 +3,7 @@ package ai.chat2db.spi.sql;
import java.util.LinkedHashMap;
import java.util.Map;
import com.alibaba.fastjson2.JSON;
import com.alibaba.fastjson2.JSONObject;
import org.apache.commons.lang3.ClassUtils;
/**
* @author luojun
@ -15,15 +14,21 @@ import com.alibaba.fastjson2.JSONObject;
public class DocumentUtils {
public static LinkedHashMap<String, Object> convertToMap(Object obj) {
if (obj == null) {
return new LinkedHashMap<>();
}
LinkedHashMap<String, Object> map = new LinkedHashMap<>();
if (obj != null) {
String json = JSON.toJSONString(obj);
JSONObject jsonObject = JSON.parseObject(json);
JSONObject m = jsonObject.getJSONObject("documentAsMap");
if (m != null) {
for (Map.Entry<String, Object> entry : m.entrySet()) {
map.put(entry.getKey(), entry.getValue());
}
for (Map.Entry<String, Object> entry : ((Map<String, Object>) obj).entrySet()) {
Object value = entry.getValue();
if (value == null) {
map.put(entry.getKey(), null);
} else if (ClassUtils.isPrimitiveOrWrapper(value.getClass()) || String.class.equals(value.getClass())) {
map.put(entry.getKey(), value);
} else if (entry.getValue() instanceof Map) {
LinkedHashMap<String, Object> mmp = convertToMap(entry.getValue());
map.put(entry.getKey(), mmp);
} else {
map.put(entry.getKey(), entry.getValue().toString());
}
}
return map;