| | |
| | | import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; |
| | | import com.baomidou.mybatisplus.core.mapper.BaseMapper; |
| | | import com.baomidou.mybatisplus.core.metadata.IPage; |
| | | import com.baomidou.mybatisplus.core.toolkit.ReflectionKit; |
| | | import com.baomidou.mybatisplus.core.toolkit.reflect.GenericTypeUtils; |
| | | import com.baomidou.mybatisplus.extension.plugins.pagination.Page; |
| | | import com.baomidou.mybatisplus.extension.toolkit.Db; |
| | | import org.dromara.common.core.utils.MapstructUtils; |
| | | import org.apache.ibatis.logging.Log; |
| | | import org.apache.ibatis.logging.LogFactory; |
| | | import org.dromara.common.core.utils.MapstructUtils; |
| | | |
| | | import java.io.Serializable; |
| | | import java.util.Collection; |
| | |
| | | Log log = LogFactory.getLog(BaseMapperPlus.class); |
| | | |
| | | default Class<V> currentVoClass() { |
| | | return (Class<V>) ReflectionKit.getSuperClassGenericType(this.getClass(), BaseMapperPlus.class, 1); |
| | | return (Class<V>) GenericTypeUtils.resolveTypeArguments(this.getClass(), BaseMapperPlus.class)[1]; |
| | | } |
| | | |
| | | default Class<T> currentModelClass() { |
| | | return (Class<T>) ReflectionKit.getSuperClassGenericType(this.getClass(), BaseMapperPlus.class, 0); |
| | | return (Class<T>) GenericTypeUtils.resolveTypeArguments(this.getClass(), BaseMapperPlus.class)[0]; |
| | | } |
| | | |
| | | default List<T> selectList() { |
| | |
| | | * 批量插入 |
| | | */ |
| | | default boolean insertBatch(Collection<T> entityList) { |
| | | return Db.saveBatch(entityList); |
| | | Db.saveBatch(entityList); |
| | | // 临时解决 新版本 mp 插入状态判断错误问题 |
| | | return true; |
| | | } |
| | | |
| | | /** |
| | | * 批量更新 |
| | | */ |
| | | default boolean updateBatchById(Collection<T> entityList) { |
| | | return Db.updateBatchById(entityList); |
| | | Db.updateBatchById(entityList); |
| | | // 临时解决 新版本 mp 插入状态判断错误问题 |
| | | return true; |
| | | } |
| | | |
| | | /** |
| | | * 批量插入或更新 |
| | | */ |
| | | default boolean insertOrUpdateBatch(Collection<T> entityList) { |
| | | return Db.saveOrUpdateBatch(entityList); |
| | | Db.saveOrUpdateBatch(entityList); |
| | | // 临时解决 新版本 mp 插入状态判断错误问题 |
| | | return true; |
| | | } |
| | | |
| | | /** |
| | | * 批量插入(包含限制条数) |
| | | */ |
| | | default boolean insertBatch(Collection<T> entityList, int batchSize) { |
| | | return Db.saveBatch(entityList, batchSize); |
| | | Db.saveBatch(entityList, batchSize); |
| | | // 临时解决 新版本 mp 插入状态判断错误问题 |
| | | return true; |
| | | } |
| | | |
| | | /** |
| | | * 批量更新(包含限制条数) |
| | | */ |
| | | default boolean updateBatchById(Collection<T> entityList, int batchSize) { |
| | | return Db.updateBatchById(entityList, batchSize); |
| | | Db.updateBatchById(entityList, batchSize); |
| | | // 临时解决 新版本 mp 插入状态判断错误问题 |
| | | return true; |
| | | } |
| | | |
| | | /** |
| | | * 批量插入或更新(包含限制条数) |
| | | */ |
| | | default boolean insertOrUpdateBatch(Collection<T> entityList, int batchSize) { |
| | | return Db.saveOrUpdateBatch(entityList, batchSize); |
| | | } |
| | | |
| | | /** |
| | | * 插入或更新(包含限制条数) |
| | | */ |
| | | default boolean insertOrUpdate(T entity) { |
| | | return Db.saveOrUpdate(entity); |
| | | Db.saveOrUpdateBatch(entityList, batchSize); |
| | | // 临时解决 新版本 mp 插入状态判断错误问题 |
| | | return true; |
| | | } |
| | | |
| | | default V selectVoById(Serializable id) { |
| | |
| | | return selectVoOne(wrapper, this.currentVoClass()); |
| | | } |
| | | |
| | | default V selectVoOne(Wrapper<T> wrapper, boolean throwEx) { |
| | | return selectVoOne(wrapper, this.currentVoClass(), throwEx); |
| | | } |
| | | |
| | | /** |
| | | * 根据 entity 条件,查询一条记录 |
| | | */ |
| | | default <C> C selectVoOne(Wrapper<T> wrapper, Class<C> voClass) { |
| | | T obj = this.selectOne(wrapper); |
| | | return selectVoOne(wrapper, voClass, true); |
| | | } |
| | | |
| | | /** |
| | | * 根据 entity 条件,查询一条记录 |
| | | */ |
| | | default <C> C selectVoOne(Wrapper<T> wrapper, Class<C> voClass, boolean throwEx) { |
| | | T obj = this.selectOne(wrapper, throwEx); |
| | | if (ObjectUtil.isNull(obj)) { |
| | | return null; |
| | | } |
| | |
| | | * 分页查询VO |
| | | */ |
| | | default <C, P extends IPage<C>> P selectVoPage(IPage<T> page, Wrapper<T> wrapper, Class<C> voClass) { |
| | | IPage<T> pageData = this.selectPage(page, wrapper); |
| | | IPage<C> voPage = new Page<>(pageData.getCurrent(), pageData.getSize(), pageData.getTotal()); |
| | | if (CollUtil.isEmpty(pageData.getRecords())) { |
| | | List<T> list = this.selectList(page, wrapper); |
| | | IPage<C> voPage = new Page<>(page.getCurrent(), page.getSize(), page.getTotal()); |
| | | if (CollUtil.isEmpty(list)) { |
| | | return (P) voPage; |
| | | } |
| | | voPage.setRecords(MapstructUtils.convert(pageData.getRecords(), voClass)); |
| | | voPage.setRecords(MapstructUtils.convert(list, voClass)); |
| | | return (P) voPage; |
| | | } |
| | | |