Java Code Examples for org.springframework.data.mongodb.core.query.Query#with()

@Override
public Page<User> queryPage(SysRole role, UserQueryRequest queryRequest) {
	int start = queryRequest.getStart();
	int limit = queryRequest.getLimit();
	Query query = new Query();
	query.addCriteria(Criteria.where("roles").in(role));
	if (!StringUtils.isEmpty(queryRequest.getUsername())) {
		query.addCriteria(Criteria.where("username").regex(queryRequest.getUsername()));
	}
	query.addCriteria(Criteria.where("deleted").ne(true));

	PageRequest pageable = new PageRequest(start, limit, new Sort(Sort.Direction.DESC, "username"));
	query.with(pageable);
	long count = mongoTemplate.count(query, User.class);
	List<User> list = mongoTemplate.find(query, User.class);

	return new PageImpl<>(list, pageable, count);
}
 

@Test
public void givenUsersExist_whenFindingByPage_thenUsersAreFoundByPage() {
    User user = new User();
    user.setName("Eric");
    user.setAge(45);
    mongoTemplate.insert(user);

    user = new User();
    user.setName("Antony");
    user.setAge(33);
    mongoTemplate.insert(user);

    user = new User();
    user.setName("Alice");
    user.setAge(35);
    mongoTemplate.insert(user);

    final Pageable pageableRequest = PageRequest.of(0, 2);
    Query query = new Query();
    query.with(pageableRequest);

    List<User> users = mongoTemplate.find(query, User.class);

    assertThat(users.size(), is(2));
}
 

@Override
public List<Artifact> getSimilarProjects(Artifact prj1, ISimilarityCalculator similarityCalculator, int numResult) {

	Query q1 = new Query(Criteria.where("type.name").is(similarityCalculator.getSimilarityName()).orOperator(
			Criteria.where("toArtifact.$id").is(new ObjectId(prj1.getId())),
			Criteria.where("fromArtifact.$id").is(new ObjectId(prj1.getId()))));
	q1.with(new Sort(Sort.Direction.DESC, "value"));
	q1.limit(numResult);
	List<Relation> r1 = mongoOperations.find(q1, Relation.class);
	List<Artifact> results = new ArrayList<>();
	for (Relation rel : r1) {
		if (rel.getFromProject().getId().equals(prj1.getId()))
			// results.add(rel.getToProject());
			results.add(artifactRepository.findOne(rel.getToProject().getId()));
		else
			// results.add(rel.getFromProject());
			results.add(artifactRepository.findOne(rel.getFromProject().getId()));
	}
	return results;
}
 

@Override
public List<EventEntity> getLastEvent(String cluster, String address, String version) {
    Query query = new Query();
    query.addCriteria(Criteria.where("cluster").is(cluster));
    query.addCriteria(Criteria.where("address").is(address));
    query.addCriteria(Criteria.where("version").is(version));
    query.with(new Sort(Sort.Direction.DESC, "dateCreated"));
    query.limit(3);
    return super.selectList(query);
}
 

@Test
public void givenUsersExist_whenFindingByPage_thenUsersAreFoundByPage() {
    User user = new User();
    user.setName("Eric");
    user.setAge(45);
    mongoTemplate.insert(user);

    user = new User();
    user.setName("Antony");
    user.setAge(33);
    mongoTemplate.insert(user);

    user = new User();
    user.setName("Alice");
    user.setAge(35);
    mongoTemplate.insert(user);

    final Pageable pageableRequest = PageRequest.of(0, 2);
    Query query = new Query();
    query.with(pageableRequest);

    List<User> users = mongoTemplate.find(query, User.class);
    assertThat(users.size(), is(2));
}
 

/**
 * 글 보기에서 앞 글, 뒷 글의 정보를 가져온다.
 */
@Override
public ArticleSimple findByIdAndCategoryWithOperator(ObjectId id, String category, Constants.CRITERIA_OPERATOR operator) {
    Query query = new Query();

    if (StringUtils.isNotBlank(category))
        query.addCriteria(Criteria.where("category").is(category));

    switch (operator) {
        case GT:
            query.addCriteria(Criteria.where("_id").gt(id));
            query.with(new Sort(Sort.Direction.ASC, "_id"));
            break;
        case LT:
            query.addCriteria(Criteria.where("_id").lt(id));
            query.with(new Sort(Sort.Direction.DESC, "_id"));
            break;
    }

    return mongoTemplate.findOne(query, ArticleSimple.class);
}
 

@Test
public void givenUsersExist_whenFindingUsersAndSortThem_thenUsersAreFoundAndSorted() {
    User user = new User();
    user.setName("Eric");
    user.setAge(45);
    mongoTemplate.insert(user);

    user = new User();
    user.setName("Antony");
    user.setAge(33);
    mongoTemplate.insert(user);

    user = new User();
    user.setName("Alice");
    user.setAge(35);
    mongoTemplate.insert(user);

    Query query = new Query();
    query.with(new Sort(Sort.Direction.ASC, "age"));

    List<User> users = mongoTemplate.find(query, User.class);
    assertThat(users.size(), is(3));
}
 

@Override
public Page<User> queryPage(UsernameQueryRequest queryRequest) {
	int start = queryRequest.getStart();
	int limit = queryRequest.getLimit();
	Query query = new Query();
	if (!StringUtils.isEmpty(queryRequest.getUsername())) {
		query.addCriteria(Criteria.where("username").regex(queryRequest.getUsername()));
	}
	query.addCriteria(Criteria.where("deleted").ne(true));

	PageRequest pageable = new PageRequest(start, limit, new Sort(Sort.Direction.ASC, "rank", "username"));
	query.with(pageable);
	long count = mongoTemplate.count(query, User.class);
	List<User> list = mongoTemplate.find(query, User.class);

	return new PageImpl<>(list, pageable, count);
}
 

@Override
public List<ProxyDataDTO> findLimitProxy(int count) {
    Query query = new Query();
    query.limit(count);
    query.with(new Sort(Sort.Direction.DESC, "lastSuccessfulTime"));
    query.with(new Sort(Sort.Direction.ASC, "proxyPort"));
    return mongoTemplate.find(query, ProxyDataDTO.class,Constant.COL_NAME_PROXY);
}
 

@Override
public Page<AuthEvent> queryPage(AuthEventQueryRequest queryRequest) {

	int start = queryRequest.getStart();
	int limit = queryRequest.getLimit();
	Query query = buildQuery(queryRequest);

	PageRequest pageable = new PageRequest(start, limit, new Sort(Sort.Direction.DESC, "loginAt"));
	query.with(pageable);
	long count = mongoTemplate.count(query, AuthEvent.class);
	List<AuthEvent> list = mongoTemplate.find(query, AuthEvent.class);

	return new PageImpl<>(list, pageable, count);
}
 

@Override
public <K extends Comparable, T extends IdentityObject<K>> K maximumIdentity(Class<T> clazz, K from, K to) {
	MongoMetadata metadata = metadatas.get(clazz);
	Query query = Query.query(Criteria.where(MongoMetadata.mongoId).gte(from).lte(to));
	query.fields().include(MongoMetadata.mongoId);
	query.with(Sort.by(Direction.DESC, MongoMetadata.mongoId));
	T instance = template.findOne(query, clazz, metadata.getOrmName());
	return instance.getId();
}
 

private void addSort(Query query, DataTablesInput input) {
    query.skip(input.getStart());
    query.limit(input.getLength());

    if (isEmpty(input.getOrder())) return;

    List<Sort.Order> orders = input.getOrder().stream()
            .filter(order -> isOrderable(input, order))
            .map(order -> toOrder(input, order)).collect(toList());
    query.with(by(orders));
}
 

@Override
public List<EventEntity> getAllEvent(String cluster, String address, String version) {
    Query query = new Query();
    query.addCriteria(Criteria.where("cluster").is(cluster));
    query.addCriteria(Criteria.where("address").is(address));
    query.addCriteria(Criteria.where("version").is(version));
    query.with(new Sort(Sort.Direction.DESC, "dateCreated"));
    return super.selectList(query);
}
 

@Override
public List<TaskEntity> selectTaskInfoByCluster(String cluster) {
    Query query = new Query();
    query.addCriteria(Criteria.where("cluster").is(cluster));
    query.with(new Sort(Sort.Direction.DESC, "dateUpdated"));
    return selectList(query);
}
 

/**
 * mongodb分页排序查询
 * @param pageParam：分页参数（pageNo、pageSize、order、direction）
 * @param query
 * @param cla
 * @param collectionName
 * @return
 */
public Pagenation page(PageParam pageParam , Query query, Class<T> cla, String collectionName ){
    if(pageParam.getOrders()!=null&&pageParam.getDirection()!=null) {
        query.with(new Sort(pageParam.getDirection(),pageParam.getOrders()));
    }
    long total  = mongoTemplate.count(query,cla,collectionName);
    query.limit(pageParam.getPageSize()).skip((pageParam.getPageNo()-1)*pageParam.getPageSize());
    List<T> list = mongoTemplate.find(query,cla,collectionName);
    long consult = total/pageParam.getPageSize() ;
    long residue = total%pageParam.getPageSize();
    long totalPage = residue == 0 ?consult :(consult+1) ;
    Pagenation pagenation = new Pagenation(pageParam);
    return pagenation.setData(list,total,totalPage);
}
 

public List<Docs> pageNoCount(String collectionName, Document queryDoc, Pageable pageable) {
	Query query = new BasicQuery(new Document() , queryDoc); // 组装query
	query.with(pageable); // 组装分页
	List<Docs> dtos = (List<Docs>) mongoTemplate.find(query, clasz, collectionName); // 映射结果集
	return dtos;
}
 

@Override
public List<ResourcePlan> findAllResourcePlan() {
    Query query = new Query();
    query.with(new Sort(Sort.Direction.DESC, "modTime"));
    return mongoTemplate.find(query, ResourcePlan.class, Constant.COL_NAME_RESOURCE_PLAN);
}
 

@Override
public Order findLastOrder() {
    Query query = query(where("orderDate").exists(true));
    query.with(new Sort(Sort.Direction.DESC, "orderDate"));
    return mongoOperations.findOne(query, Order.class);
}
 

@Override
public Order findFirstOrder() {
    Query query = query(where("orderDate").exists(true));
    query.with(new Sort(Sort.Direction.ASC, "orderDate"));
    return mongoOperations.findOne(query, Order.class);
}
 

/**
 * 排序
 *
 * @param criteriaDefinition CriteriaDefinition
 * @return Query
 */
private Query descQuery(CriteriaDefinition criteriaDefinition) {
    Query query = new Query(criteriaDefinition);
    query.with(Sort.by(Sort.Direction.DESC, "originTime"));
    return query;
}