疯狂的狮子li
2022-01-11 26fc652d33fb82a75d6ec7771ca50ee414b21723
ruoyi-common/src/main/java/com/ruoyi/common/core/mybatisplus/core/IServicePlus.java
@@ -1,10 +1,8 @@
package com.ruoyi.common.core.mybatisplus.core;
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 com.ruoyi.common.core.page.PagePlus;
import java.io.Serializable;
import java.util.Collection;
@@ -15,104 +13,104 @@
/**
 * 自定义 Service 接口, 实现 数据库实体与 vo 对象转换返回
 *
 * @param <T> 数据实体类
 * @param <V> vo类
 * @author Lion Li
 * @since 2021-05-13
 */
public interface IServicePlus<T, K> extends IService<T> {
public interface IServicePlus<T, V> extends IService<T> {
   K getVoById(Serializable id, CopyOptions copyOptions);
    /**
     * @param id 主键id
     * @return V对象
     */
    V getVoById(Serializable id);
   default K getVoById(Serializable id) {
      return getVoById(id, new CopyOptions());
   }
    /**
     * @param convertor 自定义转换器
     */
    default V getVoById(Serializable id, Function<T, V> convertor) {
        return convertor.apply(getById(id));
    }
   default K getVoById(Serializable id, Function<T, K> convertor) {
      return convertor.apply(getById(id));
   }
    /**
     * @param idList id列表
     * @return V对象
     */
    List<V> listVoByIds(Collection<? extends Serializable> idList);
   List<K> listVoByIds(Collection<? extends Serializable> idList, CopyOptions copyOptions);
    /**
     * @param convertor 自定义转换器
     */
    default List<V> listVoByIds(Collection<? extends Serializable> idList,
                                Function<Collection<T>, List<V>> convertor) {
        List<T> list = getBaseMapper().selectBatchIds(idList);
        if (list == null) {
            return null;
        }
        return convertor.apply(list);
    }
   default List<K> listVoByIds(Collection<? extends Serializable> idList) {
      return listVoByIds(idList, new CopyOptions());
   }
    /**
     * @param columnMap 表字段 map 对象
     * @return V对象
     */
    List<V> listVoByMap(Map<String, Object> columnMap);
   default List<K> listVoByIds(Collection<? extends Serializable> idList,
                        Function<Collection<T>, List<K>> convertor) {
      List<T> list = getBaseMapper().selectBatchIds(idList);
      if (list == null) {
         return null;
      }
      return convertor.apply(list);
   }
    /**
     * @param convertor 自定义转换器
     */
    default List<V> listVoByMap(Map<String, Object> columnMap,
                                Function<Collection<T>, List<V>> convertor) {
        List<T> list = getBaseMapper().selectByMap(columnMap);
        if (list == null) {
            return null;
        }
        return convertor.apply(list);
    }
   List<K> listVoByMap(Map<String, Object> columnMap, CopyOptions copyOptions);
    /**
     * @param queryWrapper 查询条件
     * @return V对象
     */
    V getVoOne(Wrapper<T> queryWrapper);
   default List<K> listVoByMap(Map<String, Object> columnMap) {
      return listVoByMap(columnMap, new CopyOptions());
   }
    /**
     * @param convertor 自定义转换器
     */
    default V getVoOne(Wrapper<T> queryWrapper, Function<T, V> convertor) {
        return convertor.apply(getOne(queryWrapper, true));
    }
    /**
     * @param queryWrapper 查询条件
     * @return V对象
     */
    List<V> listVo(Wrapper<T> queryWrapper);
   default List<K> listVoByMap(Map<String, Object> columnMap,
                        Function<Collection<T>, List<K>> convertor) {
      List<T> list = getBaseMapper().selectByMap(columnMap);
      if (list == null) {
         return null;
      }
      return convertor.apply(list);
   }
    /**
     * @param convertor 自定义转换器
     */
    default List<V> listVo(Wrapper<T> queryWrapper, Function<Collection<T>, List<V>> convertor) {
        List<T> list = getBaseMapper().selectList(queryWrapper);
        if (list == null) {
            return null;
        }
        return convertor.apply(list);
    }
   K getVoOne(Wrapper<T> queryWrapper, CopyOptions copyOptions);
    default List<V> listVo() {
        return listVo(Wrappers.emptyWrapper());
    }
   default K getVoOne(Wrapper<T> queryWrapper) {
      return getVoOne(queryWrapper, new CopyOptions());
   }
    /**
     * @param convertor 自定义转换器
     */
    default List<V> listVo(Function<Collection<T>, List<V>> convertor) {
        return listVo(Wrappers.emptyWrapper(), convertor);
    }
   default K getVoOne(Wrapper<T> queryWrapper, Function<T, K> convertor) {
      return convertor.apply(getOne(queryWrapper, true));
   }
    boolean saveAll(Collection<T> entityList);
   List<K> listVo(Wrapper<T> queryWrapper, CopyOptions copyOptions);
   default List<K> listVo(Wrapper<T> queryWrapper) {
      return listVo(queryWrapper, new CopyOptions());
   }
   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);
   }
   default List<K> listVo() {
      return listVo(Wrappers.emptyWrapper());
   }
   default List<K> listVo(Function<Collection<T>, List<K>> convertor) {
      return listVo(Wrappers.emptyWrapper(), convertor);
   }
   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());
   }
   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()));
   }
   default PagePlus<T, K> pageVo(PagePlus<T, K> page) {
      return pageVo(page, Wrappers.emptyWrapper());
   }
   default PagePlus<T, K> pageVo(PagePlus<T, K> page, Function<Collection<T>, List<K>> convertor) {
      return pageVo(page, Wrappers.emptyWrapper(), convertor);
   }
   boolean saveAll(Collection<T> entityList);
    boolean saveOrUpdateAll(Collection<T> entityList);
}