在spring mongodb查询中指定单个字段投影

时间:2017-12-20 10:16:57

标签: mongodb spring-data-mongodb

我有一个正常运行的查询,它只返回我的实体(Prl)的一个属性:

@Query("{'recibido' : null ,'activo' : true}")
public List<EmpleadoIdDTO> findIdsEmpleadosPrlActivoRecibidoIsNull();

类别:

public class EmpleadoIdDTO {
private Long empleadoId;

 public Long getEmpleadoId() {
   return empleadoId;
 }

public void setEmpleadoId(Long empleadoId) {
this.empleadoId = empleadoId;
 } 
}

我需要将此查询传递给Criteria,因为它会增长:

import java.util.ArrayList;
import java.util.List;

import org.springframework.data.mongodb.core.MongoOperations;
import org.springframework.data.mongodb.core.mapping.Document;
import org.springframework.data.mongodb.core.query.BasicQuery;
import org.springframework.data.mongodb.core.query.Criteria;

import com.xxx.crm482.domain.Prl;
import com.xxxx.crm482.service.dto.FilterPrlDTO;

public class PrlRepositoryImpl implements PrlRepositoryCustom {


private final MongoOperations mongoOperations;

public PrlRepositoryImpl(MongoOperations mongoOperations) {
this.mongoOperations = mongoOperations;
}


@Override
public List<Prl> find(FilterPrlDTO filterPrlDTO) {
List<Criteria> andCriteria = new ArrayList<>();

andCriteria.add(Criteria.where("activo").is(true));
andCriteria.add(Criteria.where("recibido").is(null));
Criteria orCriteria = new Criteria().andOperator(andCriteria.toArray(new Criteria[andCriteria.size()]));
Document projection = new Document("empleadoId", 1);
return mongoOperations.find(new BasicQuery(orCriteria.getCriteriaObject(), projection), Prl.class);
}
}

在这里我需要而不是返回实体“Prl”,只需返回实体的属性“empleadoId”

2 个答案:

答案 0 :(得分:1)

您可以通过链接查询条件来简化您的方法。将字段投影添加到Query类。

添加流以映射您投影的字段。

@Override
public List<Long> find(FilterPrlDTO filterPrlDTO) {
   Criteria criteria = Criteria.where("activo").is(true).and("recibido").is(null);
   Query query = new Query(criteria);
   query.fields().include("empleadoId");
   List<String> empleadoIds = mongoOperations.find(query, Prl.class).stream().map(Prl::getEmpleadoId).collect(Collectors.toList());
   return empleadoIds;
}

答案 1 :(得分:0)

一些事情。只能获得一个属性是不可能的。但是你可以只用一个属性来获取Pr1对象(所有其他字段都是null)。您可以使用Projection进行此操作。

andCriteria.add(Criteria.where("activo").is(true));
andCriteria.add(Criteria.where("recibido").is(null));
Criteria orCriteria = new Criteria().andOperator(andCriteria.toArray(new Criteria[andCriteria.size()]));
Document projection = new Document("empleadoId", 1);

return mongoOperations.find(new BasicQuery(orCriteria.getCriteriaObject(), projection), Prl.class);