admin/MilvusPlus
1
0
Fork 0
mirror of https://gitee.com/dromara/MilvusPlus.git synced 2025-12-06 17:08:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

解决更新必须需要全部字段数据的bug

Browse Source
This commit is contained in:
xgc 2024-06-06 18:17:04 +08:00
parent 1564b28b9a
commit 338110858b
1 changed files with 32 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
milvus-plus-core/src/main/java/org/dromara/milvus/plus/core/conditions/LambdaUpdateWrapper.java
View File

@ -470,7 +470,38 @@ public class LambdaUpdateWrapper<T> extends AbstractChainWrapper<T> implements W
}
jsonObjects.add(jsonObject);
}
return update(jsonObjects);
// 准备更新的数据列表
List<JSONObject> updateDataList = new ArrayList<>();
for (JSONObject updateObject : jsonObjects) {
boolean isBuild=false;
for (Map.Entry<String, String> property : propertyCache.functionToPropertyMap.entrySet()) {
if (updateObject.get(property.getValue()) == null) {
//缺少数据需要
isBuild=true;
}
}
if(isBuild){
QueryReq.QueryReqBuilder<?, ?> builder = QueryReq.builder()
.collectionName(collectionName).filter(pk+" == "+updateObject.get(pk));
QueryResp queryResp= client.query(builder.build());
if (queryResp != null) {
for (QueryResp.QueryResult result : queryResp.getQueryResults()) {
Map<String, Object> existingEntity = result.getEntity();
JSONObject existingData = new JSONObject();
for (Map.Entry<String, Object> existingEntry : existingEntity.entrySet()) {
String existingField = existingEntry.getKey();
Object existingValue = existingEntry.getValue();
Object updateValue = updateObject.get(existingField);
existingData.put(existingField, updateValue != null ? updateValue : existingValue);
}
updateDataList.add(existingData);
}
}
}else {
updateDataList.add(updateObject);
}
}
return update(updateDataList);
}
@Override