update [重大改动] 重写VO转换 支持深拷贝 将VO类抽象到 ServicePlus 泛型处理

疯狂的狮子li

2021-07-24 90d4fd8cf1db71cf51cfc2694290a9a4bc53344e

 ruoyi-common/src/main/java/com/ruoyi/common/core/mybatisplus/core/ServicePlusImpl.java

@@ -1,12 +1,21 @@
package com.ruoyi.common.core.mybatisplus.core;

import cn.hutool.core.bean.copier.BeanCopier;
import cn.hutool.core.bean.copier.CopyOptions;
import cn.hutool.core.util.ReflectUtil;
import com.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.core.toolkit.ClassUtils;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.ruoyi.common.core.page.PagePlus;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.ResolvableType;

import java.io.Serializable;
import java.util.Collection;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

/**
 * IServicePlus 实现类
@@ -15,7 +24,7 @@
 */
@Slf4j
@SuppressWarnings("unchecked")
public class ServicePlusImpl<M extends BaseMapperPlus<T>, T> extends ServiceImpl<M, T> implements IServicePlus<T> {
public class ServicePlusImpl<M extends BaseMapperPlus<T>, T, K> extends ServiceImpl<M, T> implements IServicePlus<T, K> {

   @Autowired
   protected M baseMapper;
@@ -35,6 +44,12 @@

   protected Class<T> mapperClass = currentMapperClass();

   protected Class<K> voClass = currentVoClass();

   public Class<K> getVoClass() {
      return voClass;
   }

   @Override
   protected Class<T> currentMapperClass() {
      return (Class<T>) this.getResolvableType().as(ServicePlusImpl.class).getGeneric(0).getType();
@@ -43,6 +58,10 @@
   @Override
   protected Class<T> currentModelClass() {
      return (Class<T>) this.getResolvableType().as(ServicePlusImpl.class).getGeneric(1).getType();
   }

   protected Class<K> currentVoClass() {
      return (Class<K>) this.getResolvableType().as(ServicePlusImpl.class).getGeneric(2).getType();
   }

   @Override
@@ -76,6 +95,21 @@
      return super.updateBatchById(entityList, batchSize);
   }

   @Override
   public boolean saveBatch(Collection<T> entityList) {
      return saveBatch(entityList, DEFAULT_BATCH_SIZE);
   }

   @Override
   public boolean saveOrUpdateBatch(Collection<T> entityList) {
      return saveOrUpdateBatch(entityList, DEFAULT_BATCH_SIZE);
   }

   @Override
   public boolean updateBatchById(Collection<T> entityList) {
      return updateBatchById(entityList, DEFAULT_BATCH_SIZE);
   }

   /**
    * 单sql批量插入( 全量填充 无视数据库默认值 )
    * 适用于无脑插入
@@ -85,4 +119,94 @@
      return baseMapper.insertAll(entityList) == entityList.size();
   }

   /**
    * 根据 ID 查询
    *
    * @param id 主键ID
    */
   @Override
   public K getVoById(Serializable id, CopyOptions copyOptions) {
      T t = getBaseMapper().selectById(id);
      return oneVoCopy(t, copyOptions);
   }

   /**
    * 查询（根据ID 批量查询）
    *
    * @param idList 主键ID列表
    */
   @Override
   public List<K> listVoByIds(Collection<? extends Serializable> idList, CopyOptions copyOptions) {
      List<T> list = getBaseMapper().selectBatchIds(idList);
      if (list == null) {
         return null;
      }
      return listVoCopy(list, copyOptions);
   }

   /**
    * 查询（根据 columnMap 条件）
    *
    * @param columnMap 表字段 map 对象
    */
   @Override
   public List<K> listVoByMap(Map<String, Object> columnMap, CopyOptions copyOptions) {
      List<T> list = getBaseMapper().selectByMap(columnMap);
      if (list == null) {
         return null;
      }
      return listVoCopy(list, copyOptions);
   }

   /**
    * 根据 Wrapper，查询一条记录 <br/>
    * <p>结果集，如果是多个会抛出异常，随机取一条加上限制条件 wrapper.last("LIMIT 1")</p>
    *
    * @param queryWrapper 实体对象封装操作类 {@link com.baomidou.mybatisplus.core.conditions.query.QueryWrapper}
    */
   @Override
   public K getVoOne(Wrapper<T> queryWrapper, CopyOptions copyOptions) {
      T t = getOne(queryWrapper, true);
      return oneVoCopy(t, copyOptions);
   }

   /**
    * 查询列表
    *
    * @param queryWrapper 实体对象封装操作类 {@link com.baomidou.mybatisplus.core.conditions.query.QueryWrapper}
    */
   @Override
   public List<K> listVo(Wrapper<T> queryWrapper, CopyOptions copyOptions) {
      List<T> list = getBaseMapper().selectList(queryWrapper);
      if (list == null) {
         return null;
      }
      return listVoCopy(list, copyOptions);
   }

   /**
    * 翻页查询
    *
    * @param page         翻页对象
    * @param queryWrapper 实体对象封装操作类
    */
   @Override
   public PagePlus<T, K> pageVo(PagePlus<T, K> page, Wrapper<T> queryWrapper, CopyOptions copyOptions) {
      PagePlus<T, K> result = getBaseMapper().selectPage(page, queryWrapper);
      List<K> volist = listVoCopy(result.getRecords(), copyOptions);
      result.setRecordsVo(volist);
      return result;
   }

   private K oneVoCopy(T t, CopyOptions copyOptions) {
      K k = ReflectUtil.newInstanceIfPossible(voClass);
      return BeanCopier.create(t, k, copyOptions).copy();
   }

   private List<K> listVoCopy(List<T> list, CopyOptions copyOptions) {
      return list.stream()
         .map(any -> BeanCopier.create(any, ReflectUtil.newInstanceIfPossible(voClass), copyOptions).copy())
         .collect(Collectors.toList());
   }

}