鍍金池/ 問答/Java  數(shù)據(jù)庫/ Java MongoTemplate查詢返回指定字段及指定數(shù)量的數(shù)據(jù)

Java MongoTemplate查詢返回指定字段及指定數(shù)量的數(shù)據(jù)

假設(shè)我有一個(gè)實(shí)體類Article

@Data
@Document(collection = "articles")
public final class Article {
    @Id
    private String id;
    private String name;
    private String url;
    private String author;
    private List<Comment> comments;
}

實(shí)體類Comment

@Data
public final class Comment {
    private String userId;
    private String content;
}

已經(jīng)通過測試用例添加進(jìn)去幾條數(shù)據(jù),假設(shè)其中一個(gè)Article的_id為5b5694f38ec9b636c4e52d69
可以通過MongoRepository獲取到這個(gè)id的Article,也可以通過MongoTemplate獲取,

final String id = "5b5694f38ec9b636c4e52d69";
Query query = new Query()
              .addCriteria(Criteria.where("id").is(id));
Article article = mongoTemplate.findOne(query, Article.class);

這樣可以獲取到,現(xiàn)在我想只獲取這個(gè)Article的List<Comment>,

Document queryObject = new Document();
queryObject.put("id", id);

Document fieldsObject = new Document();
fieldsObject.put("id", false);
fieldsObject.put("comments", true);

Query query = new BasicQuery(queryObject, fieldsObject);

如果我find時(shí)返回List

List comments = mongoTemplate.findOne(query, List.class);

這樣毫無疑問會出現(xiàn)異常,返回Article

Article article = mongoTemplate.findOne(query, Article.class);
List<Comment> comments = Optional.of(article).orElseThrow(ArticleNotFoundException::new)
                                            .getComments();

這樣可以獲取List<Comment>,但是這樣的獲取意義不是很大,也無法獲取指定數(shù)量的Commit(subList()也沒有什么意義)。
有沒有什么方式直接返回List<Commit>呢,這樣便可以使用limit()skip();
如果沒有,那有沒有一種方式可以實(shí)現(xiàn)limit() skip()同樣的效果?

回答
編輯回答
伐木累

通過查找文檔找到了答案

MongoCollection<Document> collection =
    mongoClient.getDatabase(YOUR_DATABASE_NAME).getCollection(YOUR_COLLECTION_NAME);
FindIterable<Document> findIterable = collection.find()
                                                .filter(eq("_id", new ObjectId(id)))
                                                .projection(fields(include("comments"), excludeId(), slice("comments", 2)));
Document document = findIterable.first();
if (document != null) {
    Object o = document.get("comments");
    if (o instanceof List) {
        List comments = (List) o;
        assert comments.size() == 2;
    }
}

MongoDB文檔

2018年1月5日 04:01