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

Question

我有一个正常的查询，它只返回我的实体（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”

Answer 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;
}

Answer 2

几件事。只能获得一个属性是不可能的。但是你可以只用一个属性来获取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);