| | |
| | | package com.ruoyi.common.core.mybatisplus.core; |
| | | |
| | | import cn.hutool.core.bean.BeanUtil; |
| | | import cn.hutool.core.bean.copier.CopyOptions; |
| | | import com.baomidou.mybatisplus.core.conditions.Wrapper; |
| | | import com.baomidou.mybatisplus.core.toolkit.Wrappers; |
| | | import com.baomidou.mybatisplus.extension.service.IService; |
| | |
| | | import java.util.List; |
| | | import java.util.Map; |
| | | import java.util.function.Function; |
| | | import java.util.stream.Collectors; |
| | | |
| | | /** |
| | | * 自定义 Service 接口, 实现 数据库实体与 vo 对象转换返回 |
| | |
| | | * @author Lion Li |
| | | * @since 2021-05-13 |
| | | */ |
| | | public interface IServicePlus<T> extends IService<T> { |
| | | public interface IServicePlus<T, K> extends IService<T> { |
| | | |
| | | /** |
| | | * 根据 ID 查询 |
| | | * |
| | | * @param kClass vo类型 |
| | | * @param id 主键ID |
| | | */ |
| | | default <K> K getVoById(Serializable id, Class<K> kClass) { |
| | | T t = getBaseMapper().selectById(id); |
| | | return BeanUtil.toBean(t, kClass); |
| | | K getVoById(Serializable id, CopyOptions copyOptions); |
| | | |
| | | default K getVoById(Serializable id) { |
| | | return getVoById(id, new CopyOptions()); |
| | | } |
| | | |
| | | /** |
| | | * 根据 ID 查询 |
| | | * |
| | | * @param id 主键ID |
| | | * @param convertor 转换函数 |
| | | * @param <K> vo类型 |
| | | */ |
| | | default <K> K getVoById(Serializable id, Function<T, K> convertor) { |
| | | T t = getBaseMapper().selectById(id); |
| | | return convertor.apply(t); |
| | | default K getVoById(Serializable id, Function<T, K> convertor) { |
| | | return convertor.apply(getById(id)); |
| | | } |
| | | |
| | | /** |
| | | * 查询(根据ID 批量查询) |
| | | * |
| | | * @param kClass vo类型 |
| | | * @param idList 主键ID列表 |
| | | */ |
| | | default <K> List<K> listVoByIds(Collection<? extends Serializable> idList, Class<K> kClass) { |
| | | List<T> list = getBaseMapper().selectBatchIds(idList); |
| | | if (list == null) { |
| | | return null; |
| | | } |
| | | return list.stream() |
| | | .map(any -> BeanUtil.toBean(any, kClass)) |
| | | .collect(Collectors.toList()); |
| | | List<K> listVoByIds(Collection<? extends Serializable> idList, CopyOptions copyOptions); |
| | | |
| | | default List<K> listVoByIds(Collection<? extends Serializable> idList) { |
| | | return listVoByIds(idList, new CopyOptions()); |
| | | } |
| | | |
| | | /** |
| | | * 查询(根据ID 批量查询) |
| | | * |
| | | * @param convertor 转换函数 |
| | | * @param idList 主键ID列表 |
| | | */ |
| | | default <K> List<K> listVoByIds(Collection<? extends Serializable> idList, |
| | | default List<K> listVoByIds(Collection<? extends Serializable> idList, |
| | | Function<Collection<T>, List<K>> convertor) { |
| | | List<T> list = getBaseMapper().selectBatchIds(idList); |
| | | if (list == null) { |
| | |
| | | return convertor.apply(list); |
| | | } |
| | | |
| | | /** |
| | | * 查询(根据 columnMap 条件) |
| | | * |
| | | * @param kClass vo类型 |
| | | * @param columnMap 表字段 map 对象 |
| | | */ |
| | | default <K> List<K> listVoByMap(Map<String, Object> columnMap, Class<K> kClass) { |
| | | List<T> list = getBaseMapper().selectByMap(columnMap); |
| | | if (list == null) { |
| | | return null; |
| | | } |
| | | return list.stream() |
| | | .map(any -> BeanUtil.toBean(any, kClass)) |
| | | .collect(Collectors.toList()); |
| | | List<K> listVoByMap(Map<String, Object> columnMap, CopyOptions copyOptions); |
| | | |
| | | default List<K> listVoByMap(Map<String, Object> columnMap) { |
| | | return listVoByMap(columnMap, new CopyOptions()); |
| | | } |
| | | |
| | | /** |
| | | * 查询(根据 columnMap 条件) |
| | | * |
| | | * @param convertor 转换函数 |
| | | * @param columnMap 表字段 map 对象 |
| | | */ |
| | | default <K> List<K> listVoByMap(Map<String, Object> columnMap, |
| | | |
| | | default List<K> listVoByMap(Map<String, Object> columnMap, |
| | | Function<Collection<T>, List<K>> convertor) { |
| | | List<T> list = getBaseMapper().selectByMap(columnMap); |
| | | if (list == null) { |
| | |
| | | return convertor.apply(list); |
| | | } |
| | | |
| | | /** |
| | | * 根据 Wrapper,查询一条记录 <br/> |
| | | * <p>结果集,如果是多个会抛出异常,随机取一条加上限制条件 wrapper.last("LIMIT 1")</p> |
| | | * |
| | | * @param kClass vo类型 |
| | | * @param queryWrapper 实体对象封装操作类 {@link com.baomidou.mybatisplus.core.conditions.query.QueryWrapper} |
| | | */ |
| | | default <K> K getVoOne(Wrapper<T> queryWrapper, Class<K> kClass) { |
| | | return BeanUtil.toBean(getOne(queryWrapper, true), kClass); |
| | | K getVoOne(Wrapper<T> queryWrapper, CopyOptions copyOptions); |
| | | |
| | | default K getVoOne(Wrapper<T> queryWrapper) { |
| | | return getVoOne(queryWrapper, new CopyOptions()); |
| | | } |
| | | |
| | | /** |
| | | * 根据 Wrapper,查询一条记录 <br/> |
| | | * <p>结果集,如果是多个会抛出异常,随机取一条加上限制条件 wrapper.last("LIMIT 1")</p> |
| | | * |
| | | * @param convertor 转换函数 |
| | | * @param queryWrapper 实体对象封装操作类 {@link com.baomidou.mybatisplus.core.conditions.query.QueryWrapper} |
| | | */ |
| | | default <K> K getVoOne(Wrapper<T> queryWrapper, Function<T, K> convertor) { |
| | | default K getVoOne(Wrapper<T> queryWrapper, Function<T, K> convertor) { |
| | | return convertor.apply(getOne(queryWrapper, true)); |
| | | } |
| | | |
| | | /** |
| | | * 查询列表 |
| | | * |
| | | * @param kClass vo类型 |
| | | * @param queryWrapper 实体对象封装操作类 {@link com.baomidou.mybatisplus.core.conditions.query.QueryWrapper} |
| | | */ |
| | | default <K> List<K> listVo(Wrapper<T> queryWrapper, Class<K> kClass) { |
| | | List<T> list = getBaseMapper().selectList(queryWrapper); |
| | | if (list == null) { |
| | | return null; |
| | | } |
| | | return list.stream() |
| | | .map(any -> BeanUtil.toBean(any, kClass)) |
| | | .collect(Collectors.toList()); |
| | | List<K> listVo(Wrapper<T> queryWrapper, CopyOptions copyOptions); |
| | | |
| | | default List<K> listVo(Wrapper<T> queryWrapper) { |
| | | return listVo(queryWrapper, new CopyOptions()); |
| | | } |
| | | |
| | | /** |
| | | * 查询列表 |
| | | * |
| | | * @param convertor 转换函数 |
| | | * @param queryWrapper 实体对象封装操作类 {@link com.baomidou.mybatisplus.core.conditions.query.QueryWrapper} |
| | | */ |
| | | default <K> List<K> listVo(Wrapper<T> queryWrapper, Function<Collection<T>, List<K>> convertor) { |
| | | default List<K> listVo(Wrapper<T> queryWrapper, Function<Collection<T>, List<K>> convertor) { |
| | | List<T> list = getBaseMapper().selectList(queryWrapper); |
| | | if (list == null) { |
| | | return null; |
| | |
| | | return convertor.apply(list); |
| | | } |
| | | |
| | | /** |
| | | * 查询所有 |
| | | * |
| | | * @param kClass vo类型 |
| | | * @see Wrappers#emptyWrapper() |
| | | */ |
| | | default <K> List<K> listVo(Class<K> kClass) { |
| | | return listVo(Wrappers.emptyWrapper(), kClass); |
| | | default List<K> listVo() { |
| | | return listVo(Wrappers.emptyWrapper()); |
| | | } |
| | | |
| | | /** |
| | | * 查询所有 |
| | | * |
| | | * @param convertor 转换函数 |
| | | * @see Wrappers#emptyWrapper() |
| | | */ |
| | | default <K> List<K> listVo(Function<Collection<T>, List<K>> convertor) { |
| | | default List<K> listVo(Function<Collection<T>, List<K>> convertor) { |
| | | return listVo(Wrappers.emptyWrapper(), convertor); |
| | | } |
| | | |
| | | /** |
| | | * 翻页查询 |
| | | * |
| | | * @param page 翻页对象 |
| | | * @param queryWrapper 实体对象封装操作类 |
| | | */ |
| | | default <K> PagePlus<T, K> pageVo(PagePlus<T, K> page, Wrapper<T> queryWrapper, Class<K> kClass) { |
| | | PagePlus<T, K> result = getBaseMapper().selectPage(page, queryWrapper); |
| | | List<K> volist = result.getRecords().stream() |
| | | .map(any -> BeanUtil.toBean(any, kClass)) |
| | | .collect(Collectors.toList()); |
| | | result.setRecordsVo(volist); |
| | | return result; |
| | | PagePlus<T, K> pageVo(PagePlus<T, K> page, Wrapper<T> queryWrapper, CopyOptions copyOptions); |
| | | |
| | | default PagePlus<T, K> pageVo(PagePlus<T, K> page, Wrapper<T> queryWrapper) { |
| | | return pageVo(page, queryWrapper, new CopyOptions()); |
| | | } |
| | | |
| | | /** |
| | | * 翻页查询 |
| | | * |
| | | * @param page 翻页对象 |
| | | * @param queryWrapper 实体对象封装操作类 |
| | | * @param convertor 转换函数 |
| | | */ |
| | | default <K> PagePlus<T, K> pageVo(PagePlus<T, K> page, Wrapper<T> queryWrapper, |
| | | default PagePlus<T, K> pageVo(PagePlus<T, K> page, Wrapper<T> queryWrapper, |
| | | Function<Collection<T>, List<K>> convertor) { |
| | | PagePlus<T, K> result = getBaseMapper().selectPage(page, queryWrapper); |
| | | return result.setRecordsVo(convertor.apply(result.getRecords())); |
| | | } |
| | | |
| | | /** |
| | | * 无条件翻页查询 |
| | | * |
| | | * @param page 翻页对象 |
| | | */ |
| | | default <K> PagePlus<T, K> pageVo(PagePlus<T, K> page, Class<K> kClass) { |
| | | return pageVo(page, Wrappers.emptyWrapper(), kClass); |
| | | default PagePlus<T, K> pageVo(PagePlus<T, K> page) { |
| | | return pageVo(page, Wrappers.emptyWrapper()); |
| | | } |
| | | |
| | | /** |
| | | * 无条件翻页查询 |
| | | * |
| | | * @param page 翻页对象 |
| | | * @param convertor 转换函数 |
| | | */ |
| | | default <K> PagePlus<T, K> pageVo(PagePlus<T, K> page, Function<Collection<T>, List<K>> convertor) { |
| | | default PagePlus<T, K> pageVo(PagePlus<T, K> page, Function<Collection<T>, List<K>> convertor) { |
| | | return pageVo(page, Wrappers.emptyWrapper(), convertor); |
| | | } |
| | | |
| | | @Override |
| | | default boolean saveBatch(Collection<T> entityList) { |
| | | return saveBatch(entityList, DEFAULT_BATCH_SIZE); |
| | | } |
| | | |
| | | @Override |
| | | default boolean saveOrUpdateBatch(Collection<T> entityList) { |
| | | return saveOrUpdateBatch(entityList, DEFAULT_BATCH_SIZE); |
| | | } |
| | | |
| | | @Override |
| | | default boolean updateBatchById(Collection<T> entityList) { |
| | | return updateBatchById(entityList, DEFAULT_BATCH_SIZE); |
| | | } |
| | | |
| | | boolean saveAll(Collection<T> entityList); |