Commit 05c340c5 authored by xuxin's avatar xuxin

修改UserVpsDto中id的获取方式

parent 5769d9b3
...@@ -19,6 +19,7 @@ import java.util.Arrays; ...@@ -19,6 +19,7 @@ import java.util.Arrays;
import java.util.List; import java.util.List;
import java.util.Map; import java.util.Map;
import java.util.regex.Pattern; import java.util.regex.Pattern;
import java.util.stream.Collectors;
import static org.springframework.data.mongodb.core.query.Criteria.where; import static org.springframework.data.mongodb.core.query.Criteria.where;
...@@ -85,7 +86,8 @@ public class AccountRepositoryCustomImpl implements AccountRepositoryCustom { ...@@ -85,7 +86,8 @@ public class AccountRepositoryCustomImpl implements AccountRepositoryCustom {
@Override @Override
public List<String> findIdsByParentId(String id) { public List<String> findIdsByParentId(String id) {
Query query = new Query(Criteria.where("_id").regex(Pattern.compile("^" + id + ".*$"))); Query query = new Query(Criteria.where("_id").regex(Pattern.compile("^" + id + ".*$")));
query.fields().include("_id");
return mongoTemplate.find(query, String.class, "accounts"); // query.fields().include("_id"); 这样写返回的 String 为 {_id:""}
return mongoTemplate.find(query, Account.class).stream().map(Account::getName).collect(Collectors.toList());
} }
} }
...@@ -25,7 +25,7 @@ public class UserVpsRepositoryCustomImpl implements UserVpsRepositoryCustom { ...@@ -25,7 +25,7 @@ public class UserVpsRepositoryCustomImpl implements UserVpsRepositoryCustom {
BasicQuery basicQuery = new BasicQuery(doc); BasicQuery basicQuery = new BasicQuery(doc);
basicQuery.addCriteria(where("_id").is(id)); basicQuery.addCriteria(where("_id").is(id));
Update update = new Update(); Update update = new Update();
update.pull("vpsIdList", "vpsId"); update.pull("vpsIdList", vpsId);
UpdateResult result = mongoTemplate.updateFirst(basicQuery, update, UserVpsDto.class); UpdateResult result = mongoTemplate.updateFirst(basicQuery, update, UserVpsDto.class);
return result.getModifiedCount() >= 1; return result.getModifiedCount() >= 1;
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment