| | |
| | | package com.lunhan.xxx.repository; |
| | | |
| | | import com.baomidou.mybatisplus.annotation.TableName; |
| | | import com.baomidou.mybatisplus.core.conditions.Wrapper; |
| | | import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; |
| | | import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper; |
| | | import com.baomidou.mybatisplus.core.mapper.BaseMapper; |
| | | import com.baomidou.mybatisplus.core.metadata.IPage; |
| | | import com.baomidou.mybatisplus.core.toolkit.StringUtils; |
| | | import com.baomidou.mybatisplus.core.toolkit.Wrappers; |
| | | import com.lunhan.xxx.common.PagerResult; |
| | | import com.lunhan.xxx.common.enums.ELogger; |
| | | import com.lunhan.xxx.common.enums.EResultCode; |
| | | import com.lunhan.xxx.common.enums.EYesOrNo; |
| | | import com.lunhan.xxx.common.exceptions.BusinessException; |
| | | import com.lunhan.xxx.common.util.LoggerUtil; |
| | | import com.lunhan.xxx.entity.dto.SearchBasicDTO; |
| | | import org.apache.ibatis.executor.BatchResult; |
| | | import org.apache.ibatis.session.ResultHandler; |
| | | import org.slf4j.Logger; |
| | | import org.springframework.dao.EmptyResultDataAccessException; |
| | | import org.springframework.transaction.annotation.Transactional; |
| | | |
| | | import javax.persistence.Table; |
| | | import java.io.Serializable; |
| | | import java.lang.reflect.ParameterizedType; |
| | | import java.util.Collection; |
| | |
| | | * |
| | | * @param <T> 数据实体 |
| | | */ |
| | | public class BasicMapper<T> { |
| | | public abstract class BasicMapperImpl<T> implements BasicMapper<T> { |
| | | protected BaseMapper<T> DB; |
| | | protected Class<T> clazz; |
| | | protected String tableName; |
| | |
| | | /** |
| | | * 构造方法 |
| | | */ |
| | | public BasicMapper(BaseMapper<T> db) { |
| | | public BasicMapperImpl(BaseMapper<T> db) { |
| | | this.clazz = (Class<T>) ( |
| | | (ParameterizedType) getClass().getGenericSuperclass() |
| | | ).getActualTypeArguments()[0]; |
| | | this.DB = db; |
| | | Table annotation = clazz.getAnnotation(Table.class); |
| | | TableName annotation = clazz.getAnnotation(TableName.class); |
| | | if (null == annotation) { |
| | | this.tableName = StringUtils.camelToUnderline(clazz.getSimpleName().replace("PO", "")); |
| | | return; |
| | | } |
| | | this.tableName = annotation.name(); |
| | | this.tableName = annotation.value(); |
| | | } |
| | | |
| | | public LambdaQueryWrapper<T> query() { |
| | |
| | | return Wrappers.update(); |
| | | } |
| | | |
| | | /** |
| | | * 插入一条记录 |
| | | * |
| | | * @param entity 实体对象 |
| | | */ |
| | | public int insert(T entity) { |
| | | return DB.insert(entity); |
| | | } |
| | | |
| | | /** |
| | | * 根据实体(ID)删除 |
| | | * |
| | | * @param entity 实体对象 |
| | | * @since 3.4.4 |
| | | */ |
| | | public int deleteById(T entity) { |
| | | return DB.deleteById(entity); |
| | | } |
| | | |
| | | /** |
| | | * 根据 entity 条件,删除记录 |
| | | * |
| | | * @param queryWrapper 实体对象封装操作类(可以为 null,里面的 entity 用于生成 where 语句) |
| | | */ |
| | | public int delete(Wrapper<T> queryWrapper) { |
| | | return DB.delete(queryWrapper); |
| | | } |
| | | |
| | | /** |
| | | * 根据 ID 修改 |
| | | * |
| | | * @param entity 实体对象 |
| | | */ |
| | | public int updateById(T entity) { |
| | | return DB.updateById(entity); |
| | | } |
| | | |
| | | /** |
| | | * 根据 whereEntity 条件,更新记录 |
| | | * |
| | | * @param entity 实体对象 (set 条件值,可以为 null,当entity为null时,无法进行自动填充) |
| | | * @param updateWrapper 实体对象封装操作类(可以为 null,里面的 entity 用于生成 where 语句) |
| | | */ |
| | | public int update(T entity, Wrapper<T> updateWrapper) { |
| | | return DB.update(entity, updateWrapper); |
| | | } |
| | | |
| | | /** |
| | | * 根据 ID 查询 |
| | | * |
| | | * @param id 主键ID |
| | | */ |
| | | public T selectById(Serializable id) { |
| | | return DB.selectById(id); |
| | | } |
| | | |
| | | /** |
| | | * 查询(根据ID 批量查询) |
| | | * |
| | | * @param idList 主键ID列表(不能为 null 以及 empty) |
| | | */ |
| | | public List<T> selectBatchIds(Collection<? extends Serializable> idList) { |
| | | return DB.selectBatchIds(idList); |
| | | } |
| | | |
| | | /** |
| | | * 查询(根据ID 批量查询) |
| | | * |
| | | * @param idList idList 主键ID列表(不能为 null 以及 empty) |
| | | * @param resultHandler resultHandler 结果处理器 {@link ResultHandler} |
| | | * @since 3.5.4 |
| | | */ |
| | | public void selectBatchIds(Collection<? extends Serializable> idList, ResultHandler<T> resultHandler) { |
| | | DB.selectBatchIds(idList, resultHandler); |
| | | } |
| | | |
| | | /** |
| | | * 根据 Wrapper 条件,查询总记录数 |
| | | * |
| | | * @param queryWrapper 实体对象封装操作类(可以为 null) |
| | | */ |
| | | public Long selectCount(Wrapper<T> queryWrapper) { |
| | | return DB.selectCount(queryWrapper); |
| | | } |
| | | |
| | | /** |
| | | * 根据 entity 条件,查询全部记录 |
| | | * |
| | | * @param queryWrapper 实体对象封装操作类(可以为 null) |
| | | */ |
| | | public List<T> selectList(Wrapper<T> queryWrapper) { |
| | | return DB.selectList(queryWrapper); |
| | | } |
| | | |
| | | /** |
| | | * 根据 entity 条件,查询全部记录 |
| | | * |
| | | * @param queryWrapper 实体对象封装操作类(可以为 null) |
| | | * @param resultHandler 结果处理器 {@link ResultHandler} |
| | | * @since 3.5.4 |
| | | */ |
| | | public void selectList(Wrapper<T> queryWrapper, ResultHandler<T> resultHandler) { |
| | | DB.selectList(queryWrapper, resultHandler); |
| | | } |
| | | |
| | | /** |
| | | * 根据 entity 条件,查询全部记录(并翻页) |
| | | * |
| | | * @param page 分页查询条件 |
| | | * @param queryWrapper 实体对象封装操作类(可以为 null) |
| | | * @since 3.5.3.2 |
| | | */ |
| | | public List<T> selectList(IPage<T> page, Wrapper<T> queryWrapper) { |
| | | return DB.selectList(page, queryWrapper); |
| | | return DB.selectList(queryWrapper); |
| | | } |
| | | |
| | | /** |
| | | * 根据 entity 条件,查询全部记录(并翻页) |
| | | * |
| | | * @param page 分页查询条件 |
| | | * @param queryWrapper 实体对象封装操作类(可以为 null) |
| | | * @param resultHandler 结果处理器 {@link ResultHandler} |
| | | * @since 3.5.4 |
| | | */ |
| | | public void selectList(IPage<T> page, Wrapper<T> queryWrapper, ResultHandler<T> resultHandler) { |
| | | DB.selectList(page, queryWrapper, resultHandler); |
| | | } |
| | | |
| | | /** |
| | | * 根据 Wrapper 条件,查询全部记录 |
| | | * |
| | | * @param queryWrapper 实体对象封装操作类 |
| | | */ |
| | | public List<Map<String, Object>> selectMaps(Wrapper<T> queryWrapper) { |
| | | return DB.selectMaps(queryWrapper); |
| | | } |
| | | |
| | | /** |
| | | * 根据 Wrapper 条件,查询全部记录 |
| | | * |
| | | * @param queryWrapper 实体对象封装操作类 |
| | | * @param resultHandler 结果处理器 {@link ResultHandler} |
| | | * @since 3.5.4 |
| | | */ |
| | | public void selectMaps(Wrapper<T> queryWrapper, ResultHandler<Map<String, Object>> resultHandler) { |
| | | DB.selectMaps(queryWrapper, resultHandler); |
| | | } |
| | | |
| | | /** |
| | | * 根据 Wrapper 条件,查询全部记录(并翻页) |
| | | * |
| | | * @param page 分页查询条件 |
| | | * @param queryWrapper 实体对象封装操作类 |
| | | * @since 3.5.3.2 |
| | | */ |
| | | public List<Map<String, Object>> selectMaps(IPage<? extends Map<String, Object>> page, Wrapper<T> queryWrapper) { |
| | | return DB.selectMaps(page, queryWrapper); |
| | | } |
| | | |
| | | /** |
| | | * 根据 Wrapper 条件,查询全部记录(并翻页) |
| | | * |
| | | * @param page 分页查询条件 |
| | | * @param queryWrapper 实体对象封装操作类 |
| | | * @param resultHandler 结果处理器 {@link ResultHandler} |
| | | * @since 3.5.4 |
| | | */ |
| | | public void selectMaps(IPage<? extends Map<String, Object>> page, Wrapper<T> queryWrapper, ResultHandler<Map<String, Object>> resultHandler) { |
| | | DB.selectMaps(page, queryWrapper, resultHandler); |
| | | } |
| | | |
| | | /** |
| | | * 根据 Wrapper 条件,查询全部记录 |
| | | * <p>注意: 只返回第一个字段的值</p> |
| | | * |
| | | * @param queryWrapper 实体对象封装操作类(可以为 null) |
| | | */ |
| | | public <E> List<E> selectObjs(Wrapper<T> queryWrapper) { |
| | | return DB.selectObjs(queryWrapper); |
| | | } |
| | | |
| | | /** |
| | | * 根据 Wrapper 条件,查询全部记录 |
| | | * <p>注意: 只返回第一个字段的值</p> |
| | | * |
| | | * @param queryWrapper 实体对象封装操作类(可以为 null) |
| | | * @param resultHandler 结果处理器 {@link ResultHandler} |
| | | * @since 3.5.4 |
| | | */ |
| | | public <E> void selectObjs(Wrapper<T> queryWrapper, ResultHandler<E> resultHandler) { |
| | | DB.selectObjs(queryWrapper, resultHandler); |
| | | } |
| | | |
| | | |
| | | /** |
| | | * 批量新增 |
| | | * |
| | | * @param list 数据实体列表 |
| | | */ |
| | | public Boolean addList(List<T> list) { |
| | | List<BatchResult> batchResult = DB.insert(list); |
| | | // TODO 判断成功 |
| | | return Boolean.TRUE; |
| | | } |
| | | |
| | | /** |
| | | * 批量更新 |
| | | * |
| | | * @param list 数据实体列表 |
| | | */ |
| | | public Boolean modifyList(List<T> list) { |
| | | List<BatchResult> batchResult = DB.updateById(list); |
| | | // TODO 判断成功 |
| | | return Boolean.TRUE; |
| | | } |
| | | |
| | | /** |
| | | * 批量查询其他实体列表 |
| | | * |
| | | * @param queryWrapper 查询条件 |
| | | */ |
| | | public <E> List<E> selectListObject(Wrapper<T> queryWrapper) { |
| | | return DB.selectObjs(queryWrapper); |
| | | } |
| | | |
| | | /** |
| | | * 批量查询其他实体列表 |
| | | * |
| | | * @param queryWrapper 查询条件 |
| | | * @param resultHandler 查询结果处理器 |
| | | */ |
| | | public <E> void selectListObject(Wrapper<T> queryWrapper, ResultHandler<E> resultHandler) { |
| | | DB.selectObjs(queryWrapper, resultHandler); |
| | | } |
| | | |
| | | /** |
| | | * 根据条件获取数据实体列表 |
| | | * 根据主键查询单个数据实体 |
| | | * |
| | | * @param queryWrapper 查询条件 |
| | | * @param id 主键值 |
| | | */ |
| | | @Transactional(readOnly = true) |
| | | protected List<T> getList(Wrapper<T>... queryWrapper) { |
| | | return DB.selectList(queryWrapper.length > 0 ? queryWrapper[0] : null); |
| | | public T get(Serializable id) { |
| | | return DB.selectById(id); |
| | | } |
| | | |
| | | /** |
| | |
| | | * |
| | | * @param queryWrapper 查询条件 |
| | | */ |
| | | @Transactional(readOnly = true) |
| | | protected T get(Wrapper<T> queryWrapper) { |
| | | public T get(Wrapper<T> queryWrapper) { |
| | | String method = "BasicDao.Wrapper<T> queryWrapper"; |
| | | try { |
| | | return DB.selectOne(queryWrapper); |
| | |
| | | } |
| | | } |
| | | |
| | | /** |
| | | * 根据主键批量获取 |
| | | * @param listId 主键列表 |
| | | */ |
| | | public <E extends Serializable> List<T> getList(List<E> listId) { |
| | | return DB.selectBatchIds(listId); |
| | | } |
| | | |
| | | /** |
| | | * 获取全量数据(慎重) |
| | | */ |
| | | public List<T> getList() { |
| | | return DB.selectList(null); |
| | | } |
| | | |
| | | /** |
| | | * 根据传入sql查询数据 |
| | |
| | | * @param id 主键id |
| | | * @return 是否成功 |
| | | */ |
| | | public Boolean remove(Long id) { |
| | | public Boolean remove(Serializable id) { |
| | | int rowCount = DB.deleteById(id); |
| | | if (rowCount > 0) { |
| | | return Boolean.TRUE; |
| | |
| | | * @param listId id列表 |
| | | * @return 是否成功 |
| | | */ |
| | | public Boolean remove(List<Long> listId) { |
| | | public <E extends Serializable> Boolean remove(List<E> listId) { |
| | | int rowCount = DB.deleteByIds(listId); |
| | | if (rowCount > 0) { |
| | | return Boolean.TRUE; |
| | |
| | | * |
| | | * @param queryWrapper 查询条件 |
| | | */ |
| | | protected Boolean remove(Wrapper<T> queryWrapper) { |
| | | public Boolean remove(Wrapper<T> queryWrapper) { |
| | | int rowCount = DB.delete(queryWrapper); |
| | | if (rowCount > 0) { |
| | | return Boolean.TRUE; |
| | |
| | | * @param id 主键id |
| | | * @return 是否成功 |
| | | */ |
| | | public Boolean deleteLogic(Long id) { |
| | | public Boolean deleteLogic(Serializable id) { |
| | | UpdateWrapper<T> where = Wrappers.update(); |
| | | where.set("is_delete", EYesOrNo.YES.getValue()); |
| | | where.eq("id", id); |
| | |
| | | * @param listId id列表 |
| | | * @return 是否成功 |
| | | */ |
| | | public Boolean deleteLogic(List<Long> listId) { |
| | | public <E extends Serializable> Boolean deleteLogic(List<E> listId) { |
| | | UpdateWrapper<T> where = Wrappers.update(); |
| | | where.set("is_delete", EYesOrNo.YES.getValue()); |
| | | where.in("id", listId); |
| | |
| | | } |
| | | return Boolean.FALSE; |
| | | } |
| | | |
| | | /** |
| | | * 分页查询 |
| | | * |
| | | * @param search 查询参数 |
| | | */ |
| | | @Override |
| | | public abstract PagerResult<T> search(SearchBasicDTO search); |
| | | } |