疯狂的狮子li
2022-01-11 26fc652d33fb82a75d6ec7771ca50ee414b21723
ruoyi-common/src/main/java/com/ruoyi/common/core/mybatisplus/core/ServicePlusImpl.java
@@ -31,172 +31,172 @@
@SuppressWarnings("unchecked")
public class ServicePlusImpl<M extends BaseMapperPlus<T>, T, V> extends ServiceImpl<M, T> implements IServicePlus<T, V> {
   @Autowired
   protected M baseMapper;
    @Autowired
    protected M baseMapper;
   @Override
   public M getBaseMapper() {
      return baseMapper;
   }
    @Override
    public M getBaseMapper() {
        return baseMapper;
    }
   protected Class<T> entityClass = currentModelClass();
    protected Class<T> entityClass = currentModelClass();
   @Override
   public Class<T> getEntityClass() {
      return entityClass;
   }
    @Override
    public Class<T> getEntityClass() {
        return entityClass;
    }
   protected Class<M> mapperClass = currentMapperClass();
    protected Class<M> mapperClass = currentMapperClass();
   protected Class<V> voClass = currentVoClass();
    protected Class<V> voClass = currentVoClass();
   public Class<V> getVoClass() {
      return voClass;
   }
    public Class<V> getVoClass() {
        return voClass;
    }
   @Override
   protected Class<M> currentMapperClass() {
      return (Class<M>) ReflectionKit.getSuperClassGenericType(this.getClass(), ServicePlusImpl.class, 0);
   }
    @Override
    protected Class<M> currentMapperClass() {
        return (Class<M>) ReflectionKit.getSuperClassGenericType(this.getClass(), ServicePlusImpl.class, 0);
    }
   @Override
   protected Class<T> currentModelClass() {
      return (Class<T>) ReflectionKit.getSuperClassGenericType(this.getClass(), ServicePlusImpl.class, 1);
   }
    @Override
    protected Class<T> currentModelClass() {
        return (Class<T>) ReflectionKit.getSuperClassGenericType(this.getClass(), ServicePlusImpl.class, 1);
    }
   protected Class<V> currentVoClass() {
      return (Class<V>) ReflectionKit.getSuperClassGenericType(this.getClass(), ServicePlusImpl.class, 2);
   }
    protected Class<V> currentVoClass() {
        return (Class<V>) ReflectionKit.getSuperClassGenericType(this.getClass(), ServicePlusImpl.class, 2);
    }
   /**
    * 单条执行性能差 适用于列表对象内容不确定
    */
   @Override
   public boolean saveBatch(Collection<T> entityList, int batchSize) {
      return super.saveBatch(entityList, batchSize);
   }
    /**
     * 单条执行性能差 适用于列表对象内容不确定
     */
    @Override
    public boolean saveBatch(Collection<T> entityList, int batchSize) {
        return super.saveBatch(entityList, batchSize);
    }
   @Override
   public boolean saveOrUpdate(T entity) {
      return super.saveOrUpdate(entity);
   }
    @Override
    public boolean saveOrUpdate(T entity) {
        return super.saveOrUpdate(entity);
    }
   /**
    * 单条执行性能差 适用于列表对象内容不确定
    */
   @Override
   public boolean saveOrUpdateBatch(Collection<T> entityList, int batchSize) {
      return super.saveOrUpdateBatch(entityList, batchSize);
   }
    /**
     * 单条执行性能差 适用于列表对象内容不确定
     */
    @Override
    public boolean saveOrUpdateBatch(Collection<T> entityList, int batchSize) {
        return super.saveOrUpdateBatch(entityList, batchSize);
    }
   @Override
   public boolean updateBatchById(Collection<T> entityList, int batchSize) {
      return super.updateBatchById(entityList, batchSize);
   }
    @Override
    public boolean updateBatchById(Collection<T> entityList, int batchSize) {
        return super.updateBatchById(entityList, batchSize);
    }
   /**
    * 单sql批量插入( 全量填充 无视数据库默认值 )
    * 适用于无脑插入
    */
   @Override
   public boolean saveBatch(Collection<T> entityList) {
      return saveBatch(entityList, DEFAULT_BATCH_SIZE);
   }
    /**
     * 单sql批量插入( 全量填充 无视数据库默认值 )
     * 适用于无脑插入
     */
    @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 saveOrUpdateBatch(Collection<T> entityList) {
        return saveOrUpdateBatch(entityList, DEFAULT_BATCH_SIZE);
    }
   @Override
   public boolean updateBatchById(Collection<T> entityList) {
      return updateBatchById(entityList, DEFAULT_BATCH_SIZE);
   }
    @Override
    public boolean updateBatchById(Collection<T> entityList) {
        return updateBatchById(entityList, DEFAULT_BATCH_SIZE);
    }
   /**
    * 单sql批量插入( 全量填充 )
    */
   @Override
   public boolean saveAll(Collection<T> entityList) {
      if (CollUtil.isEmpty(entityList)) {
         return false;
      }
      return baseMapper.insertAll(entityList) == entityList.size();
   }
    /**
     * 单sql批量插入( 全量填充 )
     */
    @Override
    public boolean saveAll(Collection<T> entityList) {
        if (CollUtil.isEmpty(entityList)) {
            return false;
        }
        return baseMapper.insertAll(entityList) == entityList.size();
    }
   /**
    * 全量保存或更新 ( 按主键区分 )
    */
   @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();
      Assert.notEmpty(keyProperty, "error: can not execute. because can not find column for id from entity!");
      List<T> addList = new ArrayList<>();
      List<T> updateList = new ArrayList<>();
      int row = 0;
      for (T entity : entityList) {
         Object id = ReflectUtils.invokeGetter(entity, keyProperty);
         if (ObjectUtil.isNull(id)) {
            addList.add(entity);
         } else {
            updateList.add(entity);
         }
      }
      if (CollUtil.isNotEmpty(updateList) && updateBatchById(updateList)) {
         row += updateList.size();
      }
    /**
     * 全量保存或更新 ( 按主键区分 )
     */
    @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();
        Assert.notEmpty(keyProperty, "error: can not execute. because can not find column for id from entity!");
        List<T> addList = new ArrayList<>();
        List<T> updateList = new ArrayList<>();
        int row = 0;
        for (T entity : entityList) {
            Object id = ReflectUtils.invokeGetter(entity, keyProperty);
            if (ObjectUtil.isNull(id)) {
                addList.add(entity);
            } else {
                updateList.add(entity);
            }
        }
        if (CollUtil.isNotEmpty(updateList) && updateBatchById(updateList)) {
            row += updateList.size();
        }
        if (CollUtil.isNotEmpty(addList)) {
            row += baseMapper.insertAll(addList);
        }
      return row == entityList.size();
   }
        return row == entityList.size();
    }
   /**
    * 根据 ID 查询
    */
   @Override
   public V getVoById(Serializable id) {
    /**
     * 根据 ID 查询
     */
    @Override
    public V getVoById(Serializable id) {
        return getBaseMapper().selectVoById(id, voClass);
   }
    }
   /**
    * 查询(根据ID 批量查询)
    */
   @Override
   public List<V> listVoByIds(Collection<? extends Serializable> idList) {
    /**
     * 查询(根据ID 批量查询)
     */
    @Override
    public List<V> listVoByIds(Collection<? extends Serializable> idList) {
        return getBaseMapper().selectVoBatchIds(idList, voClass);
   }
    }
   /**
    * 查询(根据 columnMap 条件)
    */
   @Override
   public List<V> listVoByMap(Map<String, Object> columnMap) {
    /**
     * 查询(根据 columnMap 条件)
     */
    @Override
    public List<V> listVoByMap(Map<String, Object> columnMap) {
        return getBaseMapper().selectVoByMap(columnMap, voClass);
   }
    }
   /**
    * 根据 Wrapper,查询一条记录 <br/>
    * <p>结果集,如果是多个会抛出异常,随机取一条加上限制条件 wrapper.last("LIMIT 1")</p>
    */
   @Override
   public V getVoOne(Wrapper<T> queryWrapper) {
    /**
     * 根据 Wrapper,查询一条记录 <br/>
     * <p>结果集,如果是多个会抛出异常,随机取一条加上限制条件 wrapper.last("LIMIT 1")</p>
     */
    @Override
    public V getVoOne(Wrapper<T> queryWrapper) {
        return getBaseMapper().selectVoOne(queryWrapper, voClass);
   }
    }
   /**
    * 查询列表
    */
   @Override
   public List<V> listVo(Wrapper<T> queryWrapper) {
    /**
     * 查询列表
     */
    @Override
    public List<V> listVo(Wrapper<T> queryWrapper) {
        return getBaseMapper().selectVoList(queryWrapper, voClass);
   }
    }
    /**
     * 翻页查询