cn.woodwhales.common.mybatisplus.MybatisPlusExecutor Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of woodwhales-common Show documentation
Show all versions of woodwhales-common Show documentation
https://github.com/woodwhales
package cn.woodwhales.common.mybatisplus;
import cn.woodwhales.common.business.DataTool;
import cn.woodwhales.common.model.field.PageQueryInterface;
import cn.woodwhales.common.model.vo.PageRespVO;
import com.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import java.io.Serializable;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.function.Consumer;
import java.util.function.Function;
import static java.util.Objects.isNull;
import static org.apache.commons.collections4.CollectionUtils.isEmpty;
/**
* @author woodwhales on 2021-01-05 14:02
* mybatisPlus sql 执行器
*/
public class MybatisPlusExecutor {
/**
* 查询数据集合
* @param mapper mapper
* @param wrapper wrapper
* @param mapping mapping
* @param 数据实体泛型
* @param mapper泛型
* @param DTO
* @return list
*/
public static , DTO> List executeQueryList(Mapper mapper,
Wrapper wrapper,
Function mapping) {
List entityList = mapper.selectList(wrapper);
if (isEmpty(entityList)) {
return new ArrayList<>();
}
return DataTool.toList(entityList, mapping);
}
/**
* 查询数据集合
* @param mapper mapper
* @param lambdaQueryWrapperConsumer lambdaQueryWrapperConsumer
* @param mapping mapping
* @param 数据实体泛型
* @param mapper泛型
* @param DTO泛型
* @return list
*/
public static , DTO> List executeQueryList(Mapper mapper,
Consumer> lambdaQueryWrapperConsumer,
Function mapping) {
LambdaQueryWrapper wrapper = Wrappers.lambdaQuery();
return executeQueryList(mapper, wrapper, mapping);
}
/**
* 查询数据集合
* @param mapper mapper
* @param wrapper wrapper
* @param 数据实体泛型
* @param mapper泛型
* @return list
*/
public static > List executeQueryList(Mapper mapper,
Wrapper wrapper) {
List entityList = mapper.selectList(wrapper);
if (isEmpty(entityList)) {
return new ArrayList<>();
}
return entityList;
}
/**
* 查询数据集合
* @param mapper mapper
* @param lambdaQueryWrapperConsumer lambdaQueryWrapperConsumer
* @param 数据实体泛型
* @param mapper泛型
* @return list
*/
public static > List executeQueryList(Mapper mapper,
Consumer> lambdaQueryWrapperConsumer) {
LambdaQueryWrapper wrapper = Wrappers.lambdaQuery();
lambdaQueryWrapperConsumer.accept(wrapper);
return executeQueryList(mapper, wrapper);
}
/**
* 查询数据并按照 keyMapping 规则转成 map 集合
* @param mapper mapper
* @param wrapper wrapper
* @param mapping mapping
* @param keyMapping keyMapping
* @param 数据实体泛型
* @param mapper泛型
* @param DTO泛型
* @param key泛型
* @return list
*/
public static , DTO, Key> Map executeQueryListMap(Mapper mapper,
Wrapper wrapper,
Function mapping,
Function keyMapping) {
return DataTool.toMap(executeQueryList(mapper, wrapper, mapping), keyMapping);
}
/**
* 查询数据并按照 keyMapping 规则转成 map 集合
* @param mapper mapper
* @param lambdaQueryWrapperConsumer lambdaQueryWrapperConsumer
* @param mapping mapping
* @param keyMapping keyMapping
* @param 数据实体泛型
* @param mapper泛型
* @param DTO泛型
* @param key泛型
* @return list
*/
public static , DTO, Key> Map executeQueryListMap(Mapper mapper,
Consumer> lambdaQueryWrapperConsumer,
Function mapping,
Function keyMapping) {
LambdaQueryWrapper wrapper = Wrappers.lambdaQuery();
lambdaQueryWrapperConsumer.accept(wrapper);
return DataTool.toMap(executeQueryList(mapper, wrapper, mapping), keyMapping);
}
/**
* 查询单条数据
* @param mapper mapper
* @param wrapper wrapper
* @param mapping mapping
* @param 数据实体泛型
* @param mapper泛型
* @param DTO泛型
* @return DTO实例
*/
public static , DTO> DTO executeQueryOne(Mapper mapper,
Wrapper wrapper,
Function mapping) {
Entity entity = mapper.selectOne(wrapper);
if (isNull(entity)) {
return null;
}
return mapping.apply(entity);
}
/**
* 查询单条数据
* @param mapper mapper
* @param lambdaQueryWrapperConsumer lambdaQueryWrapperConsumer
* @param mapping mapping
* @param 数据实体泛型
* @param mapper泛型
* @param DTO泛型
* @return DTO实例
*/
public static , DTO> DTO executeQueryOne(Mapper mapper,
Consumer> lambdaQueryWrapperConsumer,
Function mapping) {
LambdaQueryWrapper wrapper = Wrappers.lambdaQuery();
lambdaQueryWrapperConsumer.accept(wrapper);
return executeQueryOne(mapper, lambdaQueryWrapperConsumer, mapping);
}
/**
* 查询单条数据
* @param mapper mapper
* @param wrapper wrapper
* @param 数据实体泛型
* @param mapper泛型
* @return 数据实体
*/
public static > Entity executeQueryOne(Mapper mapper,
Wrapper wrapper) {
Entity entity = mapper.selectOne(wrapper);
if (isNull(entity)) {
return null;
}
return entity;
}
/**
* 分页查询数据
* @param mapper mapper
* @param queryParam queryParam
* @param lambdaQueryWrapperConsumer lambdaQueryWrapperConsumer
* @param mapping mapping
* @param 数据实体泛型
* @param DTO泛型
* @param mapper泛型
* @return PageRespVO
*/
public static > PageRespVO executeQueryPage(Mapper mapper,
PageQueryInterface queryParam,
Consumer> lambdaQueryWrapperConsumer,
Function mapping) {
IPage pageResult = executeSelectPage(mapper, queryParam, lambdaQueryWrapperConsumer);
return PageRespVO.buildPageRespVO(pageResult, mapping);
}
/**
* 分页查询数据
* @param mapper mapper
* @param queryParam queryParam
* @param lambdaQueryWrapperConsumer lambdaQueryWrapperConsumer
* @param 数据实体泛型
* @param mapper泛型
* @return PageRespVO
*/
public static > PageRespVO executeQueryPage(Mapper mapper,
PageQueryInterface queryParam,
Consumer> lambdaQueryWrapperConsumer) {
IPage pageResult = executeSelectPage(mapper, queryParam, lambdaQueryWrapperConsumer);
return PageRespVO.buildPageRespVO(pageResult);
}
/**
* 分页查询数据
* @param mapper mapper
* @param queryParam queryParam
* @param wrapper wrapper
* @param 数据实体泛型
* @param mapper泛型
* @return PageRespVO
*/
public static > PageRespVO executeQueryPage(Mapper mapper,
PageQueryInterface queryParam,
Wrapper wrapper) {
Page page = new Page<>(queryParam.getPage(), queryParam.getLimit());
IPage pageResult = mapper.selectPage(page, wrapper);
return PageRespVO.buildPageRespVO(pageResult);
}
/**
* 分页查询数据
* @param mapper mapper
* @param queryParam queryParam
* @param lambdaQueryWrapperConsumer lambdaQueryWrapperConsumer
* @param 数据实体泛型
* @param mapper泛型
* @return IPage
*/
private static > IPage executeSelectPage(Mapper mapper,
PageQueryInterface queryParam,
Consumer> lambdaQueryWrapperConsumer) {
Page page = new Page<>(queryParam.getPage(), queryParam.getLimit());
LambdaQueryWrapper wrapper = Wrappers.lambdaQuery();
lambdaQueryWrapperConsumer.accept(wrapper);
return mapper.selectPage(page, wrapper);
}
/**
* 插入数据
* @param mapper mapper
* @param entity entity
* @param 数据实体泛型
* @param mapper泛型
* @return 响应行数
*/
public static > int insert(Mapper mapper, Entity entity) {
if (Objects.nonNull(entity)) {
return mapper.insert(entity);
}
return -1;
}
/**
* 批量插入数据
* @param batchMapper batchMapper
* @param entityList entityList
* @param mapper泛型
* @param 数据实体泛型
* @return 影响行数
*/
public static , Entity> boolean batchInsert(ServiceImpl batchMapper, List entityList) {
if(isEmpty(entityList)) {
return true;
}
return batchMapper.saveBatch(entityList);
}
/**
* 批量插入数据
* @param batchMapper batchMapper
* @param mapping mapping
* @param dtoList dtoList
* @param mapper泛型
* @param 数据实体泛型
* @param DTO泛型
* @return 是否成功
*/
public static , Entity, DTO> boolean batchInsert(ServiceImpl batchMapper,
Function, List> mapping,
List dtoList) {
return batchMapper.saveBatch(mapping.apply(dtoList));
}
/**
* 按照 entityIdFunction 更新数据
* @param mapper mapper
* @param entity entity
* @param entityIdFunction entityIdFunction
* @param function function
* @param mapper泛型
* @param 数据实体泛型
* @param key泛型
* @return 影响行数
*/
public static , Entity, K extends Serializable> int update(Mapper mapper,
Entity entity,
Function entityIdFunction,
Function function) {
Entity entityFromDB = mapper.selectById(entityIdFunction.apply(entity));
if(isNull(entityFromDB)) {
return 0;
}
return mapper.updateById(function.apply(entityFromDB));
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy