疯狂的狮子li
2021-10-20 82f1f5d0cf1b51a5d81915e842e01760f404fa74
ruoyi-common/src/main/java/com/ruoyi/common/core/mybatisplus/core/ServicePlusImpl.java
@@ -1,19 +1,19 @@
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;
@@ -49,7 +49,7 @@
      return entityClass;
   }
   protected Class<T> mapperClass = currentMapperClass();
   protected Class<M> mapperClass = currentMapperClass();
   protected Class<V> voClass = currentVoClass();
@@ -58,22 +58,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);
   }
   /**
@@ -126,6 +121,9 @@
    */
   @Override
   public boolean saveAll(Collection<T> entityList) {
      if (CollUtil.isEmpty(entityList)) {
         return false;
      }
      return baseMapper.insertAll(entityList) == entityList.size();
   }
@@ -134,6 +132,9 @@
    */
   @Override
   public boolean saveOrUpdateAll(Collection<T> entityList) {
      if (CollUtil.isEmpty(entityList)) {
         return false;
      }
      TableInfo tableInfo = TableInfoHelper.getTableInfo(entityClass);
      Assert.notNull(tableInfo, "error: can not execute. because can not find cache of TableInfo for entity!");
      String keyProperty = tableInfo.getKeyProperty();
@@ -149,10 +150,10 @@
            updateList.add(entity);
         }
      }
      if (updateList.size()>0 && updateBatchById(updateList)) {
      if (CollUtil.isNotEmpty(updateList) && updateBatchById(updateList)) {
         row += updateList.size();
      }
        if (addList.size() > 0) {
        if (CollUtil.isNotEmpty(addList)) {
            row += baseMapper.insertAll(addList);
        }
      return row == entityList.size();