2017-01-25 6 views
2

После рассмотрения this page, в частности, этот запросКак фильтровать документы на основе встроенного массива?

db.scores.find(
    { results: { $elemMatch: { $gte: 80, $lt: 85 } } } 
) 

Я использовал следующий импорт

import static com.mongodb.client.model.Filters.and; 
import static com.mongodb.client.model.Filters.elemMatch; 
import static com.mongodb.client.model.Filters.eq; 
import static com.mongodb.client.model.Projections.excludeId; 
import static com.mongodb.client.model.Projections.fields; 
import static com.mongodb.client.model.Projections.include; 

И придумал следующий код, чтобы выполнить подобную операцию (ARRAY_FIELD_NAME = "myArray")

MongoCollection<Document> collection = mongoDB.getCollection(COLLECTION_NAME); 

Bson filters = and(eq("userId", userId), elemMatch(ARRAY_FIELD_NAME, eq("id", id))); 
Bson projections = fields(include(ARRAY_FIELD_NAME), excludeId()); 

List<Document> results = (List<Document>) collection.find(filters).projection(projections).first().get(ARRAY_FIELD_NAME); 
if (CollectionUtils.isEmpty(results)) { 
    return null; 
} 
if (results.size() > 1) { 
    throw new ApiException(String.format("Multiple results matched (User ID: %s, Array item ID: %s)", userId, id)); 
} 
return results.get(0); 

Для фильтрации документов, имеющих следующую структуру:

{ 
    "_id": { 
     "$oid": "588899721bbabc26865f41cc" 
    }, 
    "userId": 55, 
    "myArray": [ 
     { 
      "id": "5888998e1bbabc26865f41d2", 
      "title": "ABC" 
     }, 
     { 
      "id": "5888aaf41bbabc3200e252aa", 
      "title": "ABC" 
     } 
    ] 
} 

Но вместо того, чтобы получить один или ничего из поля myArray, я всегда получаю оба предмета!

Единственный код, который работал для меня следующее

MongoCollection<Document> collection = mongoDB.getCollection(COLLECTION_NAME); 

List<Bson> aggregationFlags = new ArrayList<>(); 
aggregationFlags.add(new Document("$unwind", "$" + ARRAY_FIELD_NAME)); 
aggregationFlags.add(new Document("$match", new Document("userId", userId).append(ARRAY_FIELD_NAME + ".id", id))); 
aggregationFlags.add(new Document("$project", new Document("_id", 0).append(ARRAY_FIELD_NAME, "$" + ARRAY_FIELD_NAME))); 

return (Document) collection.aggregate(aggregationFlags).first().get(ARRAY_FIELD_NAME); 

Так почему же первый кусок кода, который должен вести себя так же, как запрос, показанный в начале вопроса, а не фильтровать результаты, как и ожидалось ?

Мне не нужно «агрегировать» результаты, мне нужно «фильтровать» их, используя идентификатор пользователя и идентификатор элемента массива.

ответ

1

Вам необходимо использовать $elemMatch(projection). Что-то вроде ниже должно работать.

import static com.mongodb.client.model.Projections.elemMatch; 

Bson filters = and(eq("userId", userId)); 
Bson projections = fields(elemMatch(ARRAY_FIELD_NAME, eq("id", id)), excludeId()); 
+1

Сначала я не мог видеть, как ваши ответы затрагивают мою проблему. Но я продолжал читать документацию 'elemMatch' для проекций и фильтров, и только теперь я понимаю. 'Filters.elemMatch' фильтрует документы с соответствующим фильтром массива. В то время как 'Projections.elemMatch', ограничивает элементы массива документа! Это было очень запутанно! –