Commit 05c340c5 authored by xuxin's avatar xuxin

修改UserVpsDto中id的获取方式

parent 5769d9b3
......@@ -19,6 +19,7 @@ import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.regex.Pattern;
import java.util.stream.Collectors;
import static org.springframework.data.mongodb.core.query.Criteria.where;
......@@ -85,7 +86,8 @@ public class AccountRepositoryCustomImpl implements AccountRepositoryCustom {
@Override
public List<String> findIdsByParentId(String 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 {
BasicQuery basicQuery = new BasicQuery(doc);
basicQuery.addCriteria(where("_id").is(id));
Update update = new Update();
update.pull("vpsIdList", "vpsId");
update.pull("vpsIdList", vpsId);
UpdateResult result = mongoTemplate.updateFirst(basicQuery, update, UserVpsDto.class);
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