疯狂的狮子li
2021-12-06 1b752c35cc0c7f51922351d5bcca88a043002ae8
ruoyi-common/src/main/java/com/ruoyi/common/core/mybatisplus/core/ServicePlusImpl.java
@@ -1,20 +1,18 @@
package com.ruoyi.common.core.mybatisplus.core;
import cn.hutool.core.bean.copier.CopyOptions;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.ObjectUtil;
import com.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.core.metadata.TableInfo;
import com.baomidou.mybatisplus.core.metadata.TableInfoHelper;
import com.baomidou.mybatisplus.core.toolkit.Assert;
import com.baomidou.mybatisplus.core.toolkit.ClassUtils;
import com.baomidou.mybatisplus.core.toolkit.ReflectionKit;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.ruoyi.common.core.page.PagePlus;
import com.ruoyi.common.utils.BeanCopyUtils;
import com.ruoyi.common.utils.reflect.ReflectUtils;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.ResolvableType;
import java.io.Serializable;
import java.util.ArrayList;
@@ -50,7 +48,7 @@
      return entityClass;
   }
   protected Class<T> mapperClass = currentMapperClass();
   protected Class<M> mapperClass = currentMapperClass();
   protected Class<V> voClass = currentVoClass();
@@ -59,22 +57,17 @@
   }
   @Override
   protected Class<T> currentMapperClass() {
      return (Class<T>) this.getResolvableType().as(ServicePlusImpl.class).getGeneric(0).getType();
   protected Class<M> currentMapperClass() {
      return (Class<M>) ReflectionKit.getSuperClassGenericType(this.getClass(), ServicePlusImpl.class, 0);
   }
   @Override
   protected Class<T> currentModelClass() {
      return (Class<T>) this.getResolvableType().as(ServicePlusImpl.class).getGeneric(1).getType();
      return (Class<T>) ReflectionKit.getSuperClassGenericType(this.getClass(), ServicePlusImpl.class, 1);
   }
   protected Class<V> currentVoClass() {
      return (Class<V>) this.getResolvableType().as(ServicePlusImpl.class).getGeneric(2).getType();
   }
   @Override
   protected ResolvableType getResolvableType() {
      return ResolvableType.forClass(ClassUtils.getUserClass(getClass()));
      return (Class<V>) ReflectionKit.getSuperClassGenericType(this.getClass(), ServicePlusImpl.class, 2);
   }
   /**
@@ -171,9 +164,9 @@
    * @param id 主键ID
    */
   @Override
   public V getVoById(Serializable id, CopyOptions copyOptions) {
   public V getVoById(Serializable id) {
      T t = getBaseMapper().selectById(id);
      return BeanCopyUtils.oneCopy(t, copyOptions, voClass);
      return BeanCopyUtils.copy(t, voClass);
   }
   /**
@@ -182,12 +175,12 @@
    * @param idList 主键ID列表
    */
   @Override
   public List<V> listVoByIds(Collection<? extends Serializable> idList, CopyOptions copyOptions) {
   public List<V> listVoByIds(Collection<? extends Serializable> idList) {
      List<T> list = getBaseMapper().selectBatchIds(idList);
      if (list == null) {
         return null;
      }
      return BeanCopyUtils.listCopy(list, copyOptions, voClass);
      return BeanCopyUtils.copyList(list, voClass);
   }
   /**
@@ -196,12 +189,12 @@
    * @param columnMap 表字段 map 对象
    */
   @Override
   public List<V> listVoByMap(Map<String, Object> columnMap, CopyOptions copyOptions) {
   public List<V> listVoByMap(Map<String, Object> columnMap) {
      List<T> list = getBaseMapper().selectByMap(columnMap);
      if (list == null) {
         return null;
      }
      return BeanCopyUtils.listCopy(list, copyOptions, voClass);
      return BeanCopyUtils.copyList(list, voClass);
   }
   /**
@@ -211,9 +204,9 @@
    * @param queryWrapper 实体对象封装操作类 {@link com.baomidou.mybatisplus.core.conditions.query.QueryWrapper}
    */
   @Override
   public V getVoOne(Wrapper<T> queryWrapper, CopyOptions copyOptions) {
   public V getVoOne(Wrapper<T> queryWrapper) {
      T t = getOne(queryWrapper, true);
      return BeanCopyUtils.oneCopy(t, copyOptions, voClass);
      return BeanCopyUtils.copy(t, voClass);
   }
   /**
@@ -222,12 +215,12 @@
    * @param queryWrapper 实体对象封装操作类 {@link com.baomidou.mybatisplus.core.conditions.query.QueryWrapper}
    */
   @Override
   public List<V> listVo(Wrapper<T> queryWrapper, CopyOptions copyOptions) {
   public List<V> listVo(Wrapper<T> queryWrapper) {
      List<T> list = getBaseMapper().selectList(queryWrapper);
      if (list == null) {
         return null;
      }
      return BeanCopyUtils.listCopy(list, copyOptions, voClass);
      return BeanCopyUtils.copyList(list, voClass);
   }
   /**
@@ -237,9 +230,9 @@
    * @param queryWrapper 实体对象封装操作类
    */
   @Override
   public PagePlus<T, V> pageVo(PagePlus<T, V> page, Wrapper<T> queryWrapper, CopyOptions copyOptions) {
   public PagePlus<T, V> pageVo(PagePlus<T, V> page, Wrapper<T> queryWrapper) {
      PagePlus<T, V> result = getBaseMapper().selectPage(page, queryWrapper);
      List<V> volist = BeanCopyUtils.listCopy(result.getRecords(), copyOptions, voClass);
      List<V> volist = BeanCopyUtils.copyList(result.getRecords(), voClass);
      result.setRecordsVo(volist);
      return result;
   }