一、增加插件
<dependency>
<groupId>com.github.pagehelper</groupId>
<artifactId>pagehelper-spring-boot-starter</artifactId>
<version>1.2.10</version>
</dependency>
二、封装插件
加服务类
package com.ouer.hetao.service.base;
import com.github.pagehelper.PageHelper;
import com.ouer.hetao.dal.mapper.BaseMapper;
import com.ouer.hetao.model.base.BaseEntity;
import org.apache.commons.collections4.CollectionUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageImpl;
import org.springframework.data.domain.Pageable;
import org.springframework.data.domain.Sort;
import java.io.Serializable;
import java.util.List;
/**
* @author hetao
*/
public abstract class AbstractBaseService<T extends BaseEntity, ID extends Serializable> implements BaseService<T, ID> {
protected Logger log = LoggerFactory.getLogger(getClass());
@Autowired
protected BaseMapper<T, ID> mapper;
@Override
public T get(ID id) {
return mapper.get(id);
}
@Override
public int insert(T entity) {
return mapper.insert(entity);
}
@Override
public int delete(ID id) {
return mapper.delete(id);
}
@Override
public int update(T entity) {
return mapper.update(entity);
}
@Override
public List<T> list() {
return mapper.list(null);
}
@Override
public List<T> list(T entity) {
return mapper.list(entity);
}
public T selectOne(T entity) {
PageHelper.startPage(0, 1);
return limitOne(mapper.list(entity));
}
protected T limitOne(List<T> list) {
return CollectionUtils.isNotEmpty(list)?list.get(0):null;
}
@Override
public Page<T> page(T entity, Pageable pageable) {
Sort sort = pageable.getSort();
if(sort != null) {
PageHelper.startPage(pageable.getPageNumber(), pageable.getPageSize(), getSort(sort));
} else {
PageHelper.startPage(pageable.getPageNumber(), pageable.getPageSize());
}
List<T> content = mapper.list(entity);
Long total = mapper.countList(entity);
return new PageImpl<T>(content, pageable, total);
}
@Override
public List<T> list(T entity, Pageable pageable) {
Sort sort = pageable.getSort();
if(sort != null) {
PageHelper.startPage(pageable.getPageNumber(), pageable.getPageSize(), getSort(sort));
} else {
PageHelper.startPage(pageable.getPageNumber(), pageable.getPageSize());
}
return mapper.list(entity);
}
private String getSort(Sort sort) {
StringBuilder sb = new StringBuilder();
for (Sort.Order order : sort) {
sb.append(order.getProperty()).append(" ").append(order.getDirection().name());
}
return sb.toString();
}
}
对UserMapper封装
@Service
public class UserServiceImpl extends AbstractBaseService<User, Long> implements UserService {
}
三、测试
修改测试代码
@RequestMapping("/getUser")
public ResponseObject<List<User>> getUser() {
return new ResponseObject<>(userService.list(new User(), PageRequest.of(3,2, Sort.Direction.DESC, "id")));
}