devbox
算法创新者
这个屌丝很懒，什么也没留下！
关注作者
热门标签
热门文章
当前位置:   article > 正文

java MongoDB 列表分页查询_java mongodb分页查询

作者：算法创新者 | 2024-02-02 18:55:10

赞

踩

java mongodb分页查询

Java MongoDB分页查询

查询工具类

  1. package com.bx.utils;
  2.  
  3. import com.google.common.collect.Lists;
  4. import com.java.framework.model.WebViewModel;
  5. import org.apache.commons.lang3.StringUtils;
  6. import org.springframework.data.domain.PageRequest;
  7. import org.springframework.data.domain.Pageable;
  8. import org.springframework.data.domain.Sort;
  9. import org.springframework.data.mongodb.core.query.Criteria;
  10. import org.springframework.data.mongodb.core.query.Query;
  11.  
  12. import java.util.ArrayList;
  13. import java.util.Collection;
  14. import java.util.List;
  15. import java.util.regex.Pattern;
  16.  
  17. import static org.springframework.data.mongodb.core.query.Criteria.where;
  18.  
  19. /**
  20.  * @author: buwt
  21.  * @date: 2023/02/01
  22.  * @time: 12:45
  23.  */
  24. public class MongodbCriteriaUtils {
  25.  
  26.     public static void setQuery(Query query, WebViewModel model) {
  27.         WebViewModel.Relation relation = model.getRelationsCondition();
  28.         if (relation == null) {
  29.             List<WebViewModel.Condition> conditions = model.getCondition();
  30.             if (conditions == null || conditions.isEmpty()) {
  31.                 return;
  32.             }
  33.             relation = new WebViewModel.Relation();
  34.             relation.setRelationType("AND");
  35.             relation.setConditions(conditions);
  36.         }
  37.         List<Criteria> criteriaList = getCriteria(relation);
  38.         if (criteriaList.isEmpty()) {
  39.             return;
  40.         }
  41.         query.addCriteria(operator(relation.getRelationType(), criteriaList));
  42.     }
  43.  
  44.     public static Pageable pageable(WebViewModel viewModel) {
  45.         WebViewModel.Page page = viewModel.getPage();
  46.         int currPage = page.getCurrPage();
  47.         int pageSize = page.getPageSize();
  48.         if (pageSize < 1) {
  49.             pageSize = 10;
  50.         }
  51.         List<WebViewModel.Order> orders = viewModel.getOrder();
  52.         if (orders != null && orders.size() > 0) {
  53.             List<Sort.Order> orderList = Lists.newArrayList();
  54.             for (WebViewModel.Order order : orders) {
  55.                 try {
  56.                     if (StringUtils.isBlank(order.getColName())) {
  57.                         continue;
  58.                     }
  59.                     Sort.Order ord;
  60.                     if (StringUtils.isBlank(order.getOrderType())) {
  61.                         ord = new Sort.Order(Sort.Direction.ASC, order.getColName());
  62.                     } else {
  63.                         ord = new Sort.Order(Sort.Direction.fromString(order.getOrderType()), order.getColName());
  64.                     }
  65.                     orderList.add(ord);
  66.                 } catch (Exception e) {
  67.                     e.printStackTrace();
  68.                 }
  69.             }
  70.             if (!orders.isEmpty()) {
  71.                 Sort sort = new Sort(orderList);
  72.                 return new PageRequest(Math.max(currPage - 1, 0), pageSize, sort);
  73.             }
  74.         }
  75.         return new PageRequest(Math.max(currPage - 1, 0), pageSize);
  76.     }
  77.  
  78.  
  79.     private static List<Criteria> getCriteria(WebViewModel.Relation relation) {
  80.         List<Criteria> criteriaList = new ArrayList<>();
  81.         List<WebViewModel.Condition> conditions = relation.getConditions();
  82.         List<WebViewModel.Relation> relations = relation.getRelations();
  83.         if (isNotEmpty(conditions)) {
  84.             for (WebViewModel.Condition condition : conditions) {
  85.                 if (StringUtils.isBlank(condition.getColName())) {
  86.                     continue;
  87.                 }
  88.                 Criteria criteria = getCriteria(condition.getColName(), condition.getRuleType(), condition.getValue());
  89.                 if (criteria == null) {
  90.                     continue;
  91.                 }
  92.                 criteriaList.add(criteria);
  93.             }
  94.         }
  95.         if (isNotEmpty(relations)) {
  96.             for (WebViewModel.Relation relation1 : relations) {
  97.                 List<Criteria> critters = getCriteria(relation1);
  98.                 if (isNotEmpty(critters)) {
  99.                     Criteria criteria = operator(relation1.getRelationType(), critters);
  100.                     criteriaList.add(criteria);
  101.                 }
  102.             }
  103.         }
  104.         return criteriaList;
  105.     }
  106.  
  107.     private static boolean isEmpty(Collection<?> collection) {
  108.         return collection == null || collection.isEmpty();
  109.     }
  110.  
  111.     private static boolean isNotEmpty(Collection<?> collection) {
  112.         return !isEmpty(collection);
  113.     }
  114.  
  115.     private static Criteria getCriteria(String colName, String ruleType, Object value) {
  116.         if (StringUtils.isBlank(ruleType)) {
  117.             ruleType = "EQ";
  118.         }
  119.         switch (ruleType.toUpperCase()) {
  120.             case "NE":
  121.                 return where(colName).ne(value);
  122.             case "LIKE":
  123.                 return where(colName).regex(Pattern.compile("^.*" + value + ".*$"));
  124.             case "IN": {
  125.                 if (value == null || StringUtils.isBlank(value.toString())) {
  126.                     return null;
  127.                 }
  128.                 ArrayList<String> arrayList = Lists.newArrayList(value.toString().split(","));
  129.                 return where(colName).in(arrayList);
  130.             }
  131.             case "NIN": {
  132.                 if (value == null || StringUtils.isBlank(value.toString())) {
  133.                     return null;
  134.                 }
  135.                 ArrayList<String> arrayList = Lists.newArrayList(value.toString().split(","));
  136.                 return where(colName).nin(arrayList);
  137.             }
  138.             case "GE":
  139.                 return where(colName).gte(value);
  140.             case "LE":
  141.                 return where(colName).lte(value);
  142.             case "GT":
  143.                 return where(colName).gt(value);
  144.             case "LT":
  145.                 return where(colName).lt(value);
  146.             case "EQ":
  147.             default:
  148.                 return where(colName).is(value);
  149.         }
  150.     }
  151.  
  152.     private static Criteria operator(String ruleType, List<Criteria> criteriaList) {
  153.         Criteria criteria = new Criteria();
  154.         switch (ruleType.toUpperCase()) {
  155.             case "OR":
  156.                 criteria.orOperator(criteriaList.toArray(new Criteria[0]));
  157.                 break;
  158.             case "AND":
  159.             default:
  160.                 criteria.andOperator(criteriaList.toArray(new Criteria[0]));
  161.                 break;
  162.         }
  163.         return criteria;
  164.     }
  165. }
  166.

WebViewModel 对象（列表接口请求对象）

  1. package com.java.framework.model;
  2.  
  3. import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
  4. import com.fasterxml.jackson.core.JsonProcessingException;
  5. import com.fasterxml.jackson.databind.ObjectMapper;
  6. import io.swagger.annotations.ApiModel;
  7. import io.swagger.annotations.ApiModelProperty;
  8.  
  9. import java.io.Serializable;
  10. import java.io.StringWriter;
  11. import java.util.ArrayList;
  12. import java.util.List;
  13.  
  14. @JsonIgnoreProperties(ignoreUnknown = true)
  15. @ApiModel(value = "查询数据请求对象", description = "前端的数据查询接口请求数据封装到这个对象")
  16. public class WebViewModel implements Serializable {
  17.  
  18. 	 /**
  19. 	 *
  20. 	 */
  21. 	private static final long serialVersionUID = 1L;
  22.  
  23. 	// 查询条件
  24. 	private List<Condition> condition;
  25. 	
  26. 	//关系条件
  27. 	private Relation relationsCondition;
  28.  
  29. 	//组
  30. //	private List<Map<String,Object>> group;
  31.  
  32. 	// 排序
  33. 	private List<Order>  order;
  34.  
  35. 	// 分页
  36. 	private Page page;
  37.  
  38. 	public List<Condition> getCondition() {
  39. 		return condition;
  40. 	}
  41.  
  42. 	public void setCondition(List<Condition> condition) {
  43. 		this.condition = condition;
  44. 	}
  45.  
  46. 	public List<Order> getOrder() {
  47. 		return order;
  48. 	}
  49.  
  50. 	public void setOrder(List<Order> order) {
  51. 		this.order = order;
  52. 	}
  53.  
  54. 	public Page getPage() {
  55. 		return page;
  56. 	}
  57.  
  58. 	public void setPage(Page page) {
  59. 		this.page = page;
  60. 	}
  61.  
  62. 	public Relation getRelationsCondition() {
  63. 		return relationsCondition;
  64. 	}
  65.  
  66. 	public void setRelationsCondition(Relation relationsCondition) {
  67. 		this.relationsCondition = relationsCondition;
  68. 	}
  69.  
  70. 	// 条件
  71. //	@ApiModel(value = "查询条件", description = "查询条件组装对象")
  72. 	public static class Condition implements Serializable{
  73. 		@ApiModelProperty(value = "属性名(property)")
  74. 		private String colName;
  75. 		@ApiModelProperty(value = "匹配规则", allowableValues = "EQ, NE, LIKE, GT, LT, GE, LE, IN, NIN")
  76. 		private String ruleType;
  77. 		@ApiModelProperty(value = "属性值")
  78. 		private Object value;
  79. 		public Condition() {
  80. 			// TODO Auto-generated constructor stub
  81. 		}
  82. 		public String getColName() {
  83. 			return colName;
  84. 		}
  85. 		public void setColName(String colName) {
  86. 			this.colName = colName;
  87. 		}
  88. 		public String getRuleType() {
  89. 			return ruleType;
  90. 		}
  91. 		public void setRuleType(String ruleType) {
  92. 			this.ruleType = ruleType;
  93. 		}
  94. 		public Object getValue() {
  95. 			return value;
  96. 		}
  97. 		public void setValue(Object value) {
  98. 			this.value = value;
  99. 		}
  100. 		@Override
  101. 		public String toString() {
  102. 			return "Condition {colName=" + colName + ", ruleType=" + ruleType + ", value=" + value + "}";
  103. 		}
  104. 	}
  105.  
  106. 	public static class Relation implements Serializable {
  107. 		@ApiModelProperty(value = "条件关系符(默认AND)", allowableValues = "AND,OR")
  108. 		private String relationType = "AND";
  109. 		private List<Condition> conditions = new ArrayList<>();
  110. 		private List<Relation> relations = new ArrayList<>();
  111. 		public Relation(){}
  112.  
  113. 		public String getRelationType() {
  114. 			return relationType;
  115. 		}
  116.  
  117. 		public void setRelationType(String relationType) {
  118. 			this.relationType = relationType;
  119. 		}
  120.  
  121. 		public List<Condition> getConditions() {
  122. 			return conditions;
  123. 		}
  124.  
  125. 		public void setConditions(List<Condition> conditions) {
  126. 			this.conditions = conditions;
  127. 		}
  128.  
  129. 		public void addCondition(Condition condition) {
  130. 			this.conditions.add(condition);
  131. 		}
  132.  
  133. 		public void removeCondition(Condition condition) {
  134. 			this.conditions.remove(condition);
  135. 		}
  136.  
  137. 		public List<Relation> getRelations() {
  138. 			return relations;
  139. 		}
  140.  
  141. 		public void setRelations(List<Relation> relations) {
  142. 			this.relations = relations;
  143. 		}
  144.  
  145. 		@Override
  146. 		public String toString() {
  147. 			return "Relation{" +
  148. 					"relationType='" + relationType + '\'' +
  149. 					", conditions=" + conditions +
  150. 					", relations=" + relations +
  151. 					'}';
  152. 		}
  153. 	}
  154.  
  155. //	@ApiModel(value = "排序对象", description = "排序方式组装对象")
  156. 	public static class Order implements Serializable {
  157. 		@ApiModelProperty(value = "属性名(property)")
  158. 		private String colName;
  159. 		@ApiModelProperty(value = "排序类型", allowableValues = "asc,desc")
  160. 		private String orderType;
  161. 		public String getColName() {
  162. 			return colName;
  163. 		}
  164. 		public void setColName(String colName) {
  165. 			this.colName = colName;
  166. 		}
  167. 		public String getOrderType() {
  168. 			return orderType;
  169. 		}
  170. 		public void setOrderType(String orderType) {
  171. 			this.orderType = orderType;
  172. 		}
  173. 		@Override
  174. 		public String toString() {
  175. 			return "Order [colName=" + colName + ", orderType=" + orderType + "]";
  176. 		}
  177. 	}
  178.  
  179. 	@ApiModel(value = "分页对象", description = "分页数据组装对象")
  180. 	public static class Page implements Serializable {
  181. 		@ApiModelProperty(value = "当前页码", dataType = "int", required = true)
  182. 		private int currPage;
  183. 		@ApiModelProperty(value = "每页条数", dataType = "int", required = true)
  184. 		private int pageSize;
  185.  
  186. 		public int getCurrPage() {
  187. 				return currPage;
  188. 			}
  189.  
  190. 		public void setCurrPage(int currPage) {
  191. 			this.currPage = currPage;
  192. 		}
  193.  
  194. 		public int getPageSize() {
  195. 			return pageSize;
  196. 		}
  197.  
  198. 		public void setPageSize(int pageSize) {
  199. 			this.pageSize = pageSize;
  200. 		}
  201.  
  202. 		@Override
  203. 		public String toString() {
  204. 			return "Page [currPage=" + currPage + ", pageSize=" + pageSize + "]";
  205. 		}
  206. 	}
  207. }

代码示例

  1.     @PostMapping("taskDoListTwo")
  2.     @ApiOperation("我的已办")
  3.     public DataModel<ProcessHandleHistory> taskDoListTwo(@RequestBody WebViewModel model) throws Exception {
  4.         Pageable pageable = MongodbCriteriaUtils.pageable(model);
  5.         Query query = new Query();
  6.         MongodbCriteriaUtils.setQuery(query, model);
  7.         List<ProcessHandleHistory> handleHistories =
  8.                 mongoTemplate.find(query.with(pageable), ProcessHandleHistory.class, "process_handle_history");
  9.         long count = mongoTemplate.count(query, ProcessHandleHistory.class, "process_handle_history");
  10.         return SupplierHelper.create(DataModel.Builder<ProcessHandleHistory>::new)
  11.                 .setDatas(handleHistories)
  12.                 .setTotal(count)
  13.                 .setPageSize(pageable.getPageSize())
  14.                 .setCurrPage(pageable.getPageNumber()+1)
  15.                 .setTotalPages((int) (count / pageable.getPageSize()))
  16.                 .build();
  17.     }

查询示例

  1. {
  2. 	"page": {
  3. 		"currPage": 0,
  4. 		"pageSize": 10
  5. 	},
  6. 	"condition": [
  7. 		{
  8. 			"colName": "owner",
  9. 			"ruleType": "EQ",
  10. 			"value": "5d5a3cebcfb7e91344537723"
  11. 		},
  12. 		{
  13. 			"colName": "commentType",
  14. 			"ruleType": "NE",
  15. 			"value": "发起申请"
  16. 		},
  17. 		{
  18. 			"colName": "applyUser.bussTitle",
  19. 			"ruleType": "LIKE",
  20. 			"value": "测试"
  21. 		}
  22. 	],
  23. 	"order": [
  24. 		{
  25. 			"colName": "startTime",
  26. 			"orderType": "desc"
  27. 		}
  28. 	]
  29. }

响应示例

  1. {
  2. 	"flag": true,
  3. 	"shortMessage": null,
  4. 	"message": null,
  5. 	"condition": null,
  6. 	"datas": [
  7. 		{
  8. 			"applyUser": {
  9. 				"applyUser": "吴婷婷",
  10. 				"bussTitle": "测试27",
  11. 				"bussId": "60adece23bc08a6180fd966c"
  12. 			},
  13. 			"owner": "5d5a3cebcfb7e91344537723"
  14. 		},
  15. 		{
  16. 			"applyUser": {
  17. 				"applyUser": "administ",
  18. 				"bussTitle": "新流程测试",
  19. 				"bussId": "5efef4fbe42b5a09cd7f3205"
  20. 			}
  21. 			"owner": "5d5a3cebcfb7e91344537723"
  22. 		}
  23. 	],
  24. 	"data": null,
  25. 	"page": {
  26. 		"currPage": 1,
  27. 		"pageSize": 10,
  28. 		"total": 53,
  29. 		"totalPages": 5
  30. 	}
  31. }
声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/article/detail/56053
推荐阅读
相关标签

Copyright © 2003-2013 www.wpsshop.cn 版权所有，并保留所有权利。

  

闽ICP备14008679号