1. pom.xml
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter</artifactId>
</dependency>
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-data-mongodb</artifactId>
</dependency>
<dependency>
<groupId>org.projectlombok</groupId>
<artifactId>lombok</artifactId>
<version>1.18.12</version>
</dependency>
2. 分页工具类
import lombok.Data;
import java.io.Serializable;
import java.util.List;
@Data
public class PageHelper<T> implements Serializable {
private long currentPage;
private long total;
private long pageSize;
private int totalPage;
private List<T> list;
public PageHelper(long pageNum, long pageSize, long total, List<T> list) {
this.currentPage = pageNum;
this.pageSize = pageSize;
this.total = total;
this.list = list;
this.totalPage = (int) Math.ceil((double) total / pageSize);
}
public PageHelper(long pageNum, long pageSize, List<T> list) {
this.currentPage = pageNum;
this.pageSize = pageSize;
this.list = list;
}
}
3. MongoDB 分页工具类
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.stereotype.Component;
import java.util.List;
@Component
public class MongoPage<T> {
private Integer pageSize;
private Integer currentPage;
public void initPage(Integer currentPage, Integer pageSize, Query query) {
pageSize = pageSize == 0 ? 20 : pageSize;
query.limit(pageSize);
query.skip((currentPage - 1) * pageSize);
this.pageSize = pageSize;
this.currentPage = currentPage;
}
public PageHelper<T> pageHelper(long total, List<T> list) {
return new PageHelper<>(this.currentPage, total, this.pageSize, list);
}
public PageHelper<T> pageHelper(List<T> list) {
return new PageHelper<>(this.currentPage, this.pageSize, list);
}
public PageHelper<T> pageHelper(long currentPage, long total, long pageSize, List<T> list) {
return new PageHelper<>(currentPage, total, pageSize, list);
}
public PageHelper<T> pageHelper(long currentPage, long pageSize, List<T> list) {
return new PageHelper<>(currentPage, pageSize, list);
}
}
4. MongoDB API
import org.springframework.data.domain.Sort;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Update;
import java.util.List;
import java.util.Map;
public interface MongoDBDao {
boolean createCollection(String collectionName);
boolean dropCollection(String collectionName);
boolean insert(Object o, String collectionName);
boolean insert(Object o);
boolean insertAll(List o, Class c);
void save(Object o, String collectionName);
Object findAll(Class c, String collectionName);
Object findById(String id, Class c);
Object findByIds(String idName, List<String> ids, Class c);
Object findOne(Criteria criteria, Class c, String collectionName);
Object findByCondition(Criteria criteria, Class c, List<Sort> sorts);
Object findByCondition(Criteria criteria, Class c, String collectionName);
Object findByCondition(Criteria criteria, Class c, int limit, String collectionName);
Object findByCondition(Criteria criteria, Class c, int limit, Sort sort, String collectionName);
Object findByCondition(Criteria criteria, Class c, Integer pageNum, Integer pageSize, List<Sort> sorts);
Object findByCondition(Criteria criteria, Class c, String collectionName, Integer pageNum, Integer pageSize, List<Sort> sorts);
Object group(Criteria criteria, Class c, String[] fields, String collectionName, long pageNum, long pageSize);
Object group(Criteria criteria, Class c, String[] fields, String collectionName);
Object findPropertyByCondition(Criteria criteria, Class c, String collectionName, String[] includes);
long count(Criteria criteria, String collectionName);
long count(Criteria criteria, Class c);
Object updateOne(Criteria criteria, Update update, String collectionName);
Object updateOne(Criteria criteria, Update update, Class c);
Object updateMany(Criteria criteria, Update update, String collectionName);
Object updateMany(Criteria criteria, Update update, Class c);
Object delete(Criteria criteria, String collectionName);
Object delete(Criteria criteria, Class c);
Criteria getCriteria(Map<String, Object> map);
Criteria getCriteria(Map<String, Object> map, List<String> skipFields);
Criteria getCriteria(Map<String, Object> map, List<String> skipFields, List<String> likeFields);
Criteria getCriteria(Map<String, Object> map, List<String> skipFields, List<String> likeFields, List<String> inFields);
}
5. MongoDB API 实现类
import org.springframework.data.domain.Sort;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.aggregation.Aggregation;
import org.springframework.data.mongodb.core.aggregation.AggregationResults;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.data.mongodb.core.query.Update;
import org.springframework.stereotype.Repository;
import org.springframework.util.StringUtils;
import javax.annotation.Resource;
import java.time.LocalDateTime;
import java.time.format.DateTimeFormatter;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
@Repository
public class MongoDBDaoImpl implements MongoDBDao {
@Resource
private MongoTemplate mongoTemplate;
@Resource
private MongoPage mongoPage;
@Override
public boolean createCollection(String collectionName) {
if (mongoTemplate.collectionExists(collectionName)) {
return true;
}
mongoTemplate.createCollection(collectionName);
return mongoTemplate.collectionExists(collectionName);
}
@Override
public boolean dropCollection(String collectionName) {
if (!mongoTemplate.collectionExists(collectionName)) {
return true;
}
mongoTemplate.getCollection(collectionName).drop();
return !mongoTemplate.collectionExists(collectionName);
}
@Override
public boolean insert(Object o, String collectionName) {
try {
mongoTemplate.insert(o, collectionName);
return true;
} catch (Exception e) {
return false;
}
}
@Override
public boolean insert(Object o) {
try {
mongoTemplate.insert(o);
return true;
} catch (Exception e) {
return false;
}
}
@Override
public boolean insertAll(List o, Class c) {
try {
mongoTemplate.insert(o, c);
return true;
} catch (Exception e) {
return false;
}
}
@Override
public void save(Object o, String collectionName) {
mongoTemplate.save(o, collectionName);
}
@Override
public Object findAll(Class c, String collectionName) {
return mongoTemplate.findAll(c, collectionName);
}
@Override
public Object findById(String id, Class c) {
return mongoTemplate.findById(id, c);
}
@Override
public Object findByIds(String idName, List<String> ids, Class c) {
Criteria in = Criteria.where(idName).in(ids);
Query query = new Query(in);
return mongoTemplate.find(query, c);
}
@Override
public Object findOne(Criteria criteria, Class c, String collectionName) {
Query query = new Query(criteria);
return mongoTemplate.findOne(query, c, collectionName);
}
@Override
public Object findByCondition(Criteria criteria, Class c, List<Sort> sorts) {
Query query = new Query(criteria);
if (sorts != null && sorts.size() > 0) {
for (Sort sort : sorts) {
query = query.with(sort);
}
}
return mongoTemplate.find(query, c);
}
@Override
public Object findByCondition(Criteria criteria, Class c, String collectionName) {
return findByCondition(criteria, c, 0, collectionName);
}
@Override
public Object findByCondition(Criteria criteria, Class c, int limit, String collectionName) {
return findByCondition(criteria, c, limit, null, collectionName);
}
@Override
public Object findByCondition(Criteria criteria, Class c, int limit, Sort sort, String collectionName) {
Query query = new Query(criteria);
if (sort != null) {
query = query.with(sort);
}
if (limit > 0) {
query.limit(limit);
}
return mongoTemplate.find(query, c, collectionName);
}
@Override
public Object findByCondition(Criteria criteria, Class c, Integer pageNum, Integer pageSize, List<Sort> sorts) {
return findByCondition(criteria, c, null, pageNum, pageSize, sorts);
}
@Override
public Object findByCondition(Criteria criteria, Class c, String collectionName, Integer pageNum, Integer pageSize, List<Sort> sorts) {
Query query = new Query(criteria);
if (sorts != null && sorts.size() > 0) {
for (Sort sort : sorts) {
query = query.with(sort);
}
}
long count = 0;
if (StringUtils.hasText(collectionName)) {
count = mongoTemplate.count(query, c);
} else {
count = mongoTemplate.count(query, collectionName);
}
mongoPage.initPage(pageNum, pageSize, query);
List list = new ArrayList<>();
if (StringUtils.hasText(collectionName)) {
list = mongoTemplate.find(query, c);
} else {
list = mongoTemplate.find(query, c, collectionName);
}
return mongoPage.pageHelper(count, list);
}
@Override
public long count(Criteria criteria, String collectionName) {
Query query = new Query(criteria);
return mongoTemplate.count(query, collectionName);
}
@Override
public Object updateOne(Criteria criteria, Update update, String collectionName) {
Query query = new Query(criteria);
return mongoTemplate.updateFirst(query, update, collectionName);
}
@Override
public Object updateMany(Criteria criteria, Update update, String collectionName) {
Query query = new Query(criteria);
return mongoTemplate.updateMulti(query, update, collectionName);
}
@Override
public Object delete(Criteria criteria, String collectionName) {
Query query = new Query(criteria);
return mongoTemplate.remove(query, collectionName);
}
@Override
public long count(Criteria criteria, Class c) {
Query query = new Query(criteria);
return mongoTemplate.count(query, c);
}
@Override
public Object updateOne(Criteria criteria, Update update, Class c) {
Query query = new Query(criteria);
return mongoTemplate.updateFirst(query, update, c);
}
@Override
public Object updateMany(Criteria criteria, Update update, Class c) {
Query query = new Query(criteria);
return mongoTemplate.updateMulti(query, update, c);
}
@Override
public Object delete(Criteria criteria, Class c) {
Query query = new Query(criteria);
return mongoTemplate.remove(query, c);
}
@Override
public Object group(Criteria criteria, Class c, String[] fields, String collectionName, long currentPage, long pageSize) {
pageSize = pageSize == 0 ? 20 : pageSize;
currentPage = currentPage == 0 ? 1 : currentPage;
Aggregation aggregation = Aggregation.newAggregation(Aggregation.match(criteria),
Aggregation.group(fields).count().as("sum"),
Aggregation.sort(Sort.by(Sort.Direction.DESC, "sum")),
Aggregation.skip((currentPage - 1) * pageSize),
Aggregation.limit(pageSize));
AggregationResults aggregate = mongoTemplate.aggregate(aggregation, collectionName, c);
return mongoPage.pageHelper(currentPage, pageSize, aggregate.getMappedResults());
}
@Override
public Object group(Criteria criteria, Class c, String[] fields, String collectionName) {
Aggregation aggregation = Aggregation.newAggregation(Aggregation.match(criteria),
Aggregation.group(fields).count().as("sum"),
Aggregation.sort(Sort.by(Sort.Direction.DESC, "sum")));
AggregationResults aggregate = mongoTemplate.aggregate(aggregation, collectionName, c);
return aggregate.getMappedResults();
}
@Override
public Object findPropertyByCondition(Criteria criteria, Class c, String collectionName, String[] includes) {
Query query = new Query(criteria);
query.fields().exclude("_id");
if (!StringUtils.hasText(Arrays.toString(includes))) {
for (String result : includes) {
query.fields().include(result);
}
}
return mongoTemplate.find(query, c, collectionName);
}
@Override
public Criteria getCriteria(Map<String, Object> map) {
return getCriteria(map, null, null, null);
}
@Override
public Criteria getCriteria(Map<String, Object> map, List<String> skipFields) {
return getCriteria(map, skipFields, null, null);
}
@Override
public Criteria getCriteria(Map<String, Object> map, List<String> skipFields, List<String> likeFields) {
return getCriteria(map, skipFields, likeFields, null);
}
@Override
public Criteria getCriteria(Map<String, Object> map, List<String> skipFields, List<String> likeFields, List<String> inFields) {
Criteria criteria = Criteria.where("1").is(1);
for (String key : map.keySet()) {
if (skipFields != null && skipFields.contains(key)) {
continue;
}
if (likeFields != null && likeFields.contains(key)) {
criteria.and(key).regex(".*" + map.get(key) + ".*");
continue;
}
if (inFields != null && inFields.contains(key)) {
criteria.and(key).in(map.get(key));
continue;
}
if ("startTime".equals(key)) {
criteria.andOperator(Criteria.where("createTime").gte(formatTime((LocalDateTime) map.get("startTime"))),
Criteria.where("createTime").lte(formatTime((LocalDateTime) map.get("endTime"))));
continue;
}
criteria.and(key).is(map.get(key));
}
return criteria;
}
private String formatTime(LocalDateTime time) {
return time.format(DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss"));
}
}
6. 测试类
import com.mongodb.client.result.DeleteResult;
import com.mongodb.client.result.UpdateResult;
import lombok.Data;
import org.springframework.data.domain.Sort;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Update;
import javax.annotation.Resource;
import java.time.LocalDateTime;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
public class MongoService {
@Resource
private MongoDBDao mongoDBDao;
public PageHelper<User> queryList(Map<String, Object> map) {
List<String> likeFields = new ArrayList<>();
likeFields.add("name");
Criteria criteria = mongoDBDao.getCriteria(map, likeFields);
List<Sort> sortList = new ArrayList<>();
sortList.add(Sort.by(Sort.Order.desc("createTime")));
return (PageHelper<User>) mongoDBDao.findByCondition(criteria, User.class,
Integer.parseInt((String) map.get("pageNum")),
Integer.parseInt((String) map.get("pageSize")), sortList);
}
public boolean insert(User user) {
user.setName("bai");
user.setCreateTime(LocalDateTime.now());
return mongoDBDao.insert(user);
}
public boolean patchInsert(List<User> list) {
for (User user : list) {
user.setName("bai");
user.setCreateTime(LocalDateTime.now());
}
return mongoDBDao.insertAll(list, User.class);
}
public UpdateResult update(Map<String, Object> map) {
Criteria criteria = Criteria.where("id").is(map.get("id"));
Update update = new Update();
for (String key : map.keySet()) {
if ("id".equals(key)) {
continue;
}
update = update.set(key, map.get(key));
}
return (UpdateResult) mongoDBDao.updateOne(criteria, update, User.class);
}
public User findById(String id) {
return (User) mongoDBDao.findById(id, User.class);
}
public Long patchDelete(List<String> ids, String field) {
Criteria criteria = Criteria.where(field).in(ids);
DeleteResult deleteCount = (DeleteResult) mongoDBDao.delete(criteria, User.class);
return deleteCount.getDeletedCount();
}
public long count(Map<String, Object> map) {
Criteria criteria = mongoDBDao.getCriteria(map);
return mongoDBDao.count(criteria, User.class);
}
@Data
public static class User {
private Integer id;
private String name;
private LocalDateTime createTime;
}
}