疯狂的狮子li
2021-07-26 11c73d887736cd0e6b5208580a81b53c0663f05e
ruoyi-common/src/main/java/com/ruoyi/common/core/mybatisplus/core/ServicePlusImpl.java
@@ -125,9 +125,9 @@
    * @param id 主键ID
    */
   @Override
   public K getVoById(Serializable id, CopyOptions copyOptions) {
   public K getVoById(Serializable id, CopyOptions copyOptions, Class<K> kClass) {
      T t = getBaseMapper().selectById(id);
      return oneVoCopy(t, copyOptions);
      return oneVoCopy(t, copyOptions, kClass);
   }
   /**
@@ -136,12 +136,12 @@
    * @param idList 主键ID列表
    */
   @Override
   public List<K> listVoByIds(Collection<? extends Serializable> idList, CopyOptions copyOptions) {
   public List<K> listVoByIds(Collection<? extends Serializable> idList, CopyOptions copyOptions, Class<K> kClass) {
      List<T> list = getBaseMapper().selectBatchIds(idList);
      if (list == null) {
         return null;
      }
      return listVoCopy(list, copyOptions);
      return listVoCopy(list, copyOptions, kClass);
   }
   /**
@@ -150,12 +150,12 @@
    * @param columnMap 表字段 map 对象
    */
   @Override
   public List<K> listVoByMap(Map<String, Object> columnMap, CopyOptions copyOptions) {
   public List<K> listVoByMap(Map<String, Object> columnMap, CopyOptions copyOptions, Class<K> kClass) {
      List<T> list = getBaseMapper().selectByMap(columnMap);
      if (list == null) {
         return null;
      }
      return listVoCopy(list, copyOptions);
      return listVoCopy(list, copyOptions, kClass);
   }
   /**
@@ -165,9 +165,9 @@
    * @param queryWrapper 实体对象封装操作类 {@link com.baomidou.mybatisplus.core.conditions.query.QueryWrapper}
    */
   @Override
   public K getVoOne(Wrapper<T> queryWrapper, CopyOptions copyOptions) {
   public K getVoOne(Wrapper<T> queryWrapper, CopyOptions copyOptions, Class<K> kClass) {
      T t = getOne(queryWrapper, true);
      return oneVoCopy(t, copyOptions);
      return oneVoCopy(t, copyOptions, kClass);
   }
   /**
@@ -176,12 +176,12 @@
    * @param queryWrapper 实体对象封装操作类 {@link com.baomidou.mybatisplus.core.conditions.query.QueryWrapper}
    */
   @Override
   public List<K> listVo(Wrapper<T> queryWrapper, CopyOptions copyOptions) {
   public List<K> listVo(Wrapper<T> queryWrapper, CopyOptions copyOptions, Class<K> kClass) {
      List<T> list = getBaseMapper().selectList(queryWrapper);
      if (list == null) {
         return null;
      }
      return listVoCopy(list, copyOptions);
      return listVoCopy(list, copyOptions, kClass);
   }
   /**
@@ -191,21 +191,30 @@
    * @param queryWrapper 实体对象封装操作类
    */
   @Override
   public PagePlus<T, K> pageVo(PagePlus<T, K> page, Wrapper<T> queryWrapper, CopyOptions copyOptions) {
   public PagePlus<T, K> pageVo(PagePlus<T, K> page, Wrapper<T> queryWrapper, CopyOptions copyOptions, Class<K> kClass) {
      PagePlus<T, K> result = getBaseMapper().selectPage(page, queryWrapper);
      List<K> volist = listVoCopy(result.getRecords(), copyOptions);
      List<K> volist = listVoCopy(result.getRecords(), copyOptions, kClass);
      result.setRecordsVo(volist);
      return result;
   }
   private K oneVoCopy(T t, CopyOptions copyOptions) {
   private K oneVoCopy(T t, CopyOptions copyOptions, Class<K> kClass) {
      Class<K> voClass = this.voClass;
      if (kClass != null) {
         voClass = kClass;
      }
      K k = ReflectUtil.newInstanceIfPossible(voClass);
      return BeanCopier.create(t, k, copyOptions).copy();
   }
   private List<K> listVoCopy(List<T> list, CopyOptions copyOptions) {
   private List<K> listVoCopy(List<T> list, CopyOptions copyOptions, Class<K> kClass) {
      Class<K> voClass = this.voClass;
      if (kClass != null) {
         voClass = kClass;
      }
      final Class<K> finalVoClass = voClass;
      return list.stream()
         .map(any -> BeanCopier.create(any, ReflectUtil.newInstanceIfPossible(voClass), copyOptions).copy())
         .map(any -> BeanCopier.create(any, ReflectUtil.newInstanceIfPossible(finalVoClass), copyOptions).copy())
         .collect(Collectors.toList());
   }