From c82afc8c461956e45be0aa29ca2a7fe3f18ee6f7 Mon Sep 17 00:00:00 2001
From: 疯狂的狮子li <15040126243@163.com>
Date: 星期二, 28 十二月 2021 11:23:33 +0800
Subject: [PATCH] update RedisUtils 迁移到 utils/redis 包下
---
ruoyi-common/src/main/java/com/ruoyi/common/core/mybatisplus/core/ServicePlusImpl.java | 158 +++++++++++++++++++++++++++++++++++++++++++++++++---
1 files changed, 148 insertions(+), 10 deletions(-)
diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/core/mybatisplus/core/ServicePlusImpl.java b/ruoyi-common/src/main/java/com/ruoyi/common/core/mybatisplus/core/ServicePlusImpl.java
index fb0998b..0650cfa 100644
--- a/ruoyi-common/src/main/java/com/ruoyi/common/core/mybatisplus/core/ServicePlusImpl.java
+++ b/ruoyi-common/src/main/java/com/ruoyi/common/core/mybatisplus/core/ServicePlusImpl.java
@@ -1,21 +1,37 @@
package com.ruoyi.common.core.mybatisplus.core;
-import com.baomidou.mybatisplus.core.toolkit.ClassUtils;
+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.IPage;
+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.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;
import java.util.Collection;
+import java.util.List;
+import java.util.Map;
/**
* IServicePlus 瀹炵幇绫�
*
+ * @param <M> Mapper绫�
+ * @param <T> 鏁版嵁瀹炰綋绫�
+ * @param <V> vo绫�
* @author Lion Li
*/
@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, V> extends ServiceImpl<M, T> implements IServicePlus<T, V> {
@Autowired
protected M baseMapper;
@@ -33,21 +49,26 @@
return entityClass;
}
- protected Class<T> mapperClass = currentMapperClass();
+ protected Class<M> mapperClass = currentMapperClass();
+
+ protected Class<V> voClass = currentVoClass();
+
+ public Class<V> getVoClass() {
+ return voClass;
+ }
@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);
}
- @Override
- protected ResolvableType getResolvableType() {
- return ResolvableType.forClass(ClassUtils.getUserClass(getClass()));
+ protected Class<V> currentVoClass() {
+ return (Class<V>) ReflectionKit.getSuperClassGenericType(this.getClass(), ServicePlusImpl.class, 2);
}
/**
@@ -81,8 +102,125 @@
* 閫傜敤浜庢棤鑴戞彃鍏�
*/
@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);
+ }
+
+ /**
+ * 鍗晄ql鎵归噺鎻掑叆( 鍏ㄩ噺濉厖 )
+ */
+ @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();
+ }
+ if (CollUtil.isNotEmpty(addList)) {
+ row += baseMapper.insertAll(addList);
+ }
+ return row == entityList.size();
+ }
+
+ /**
+ * 鏍规嵁 ID 鏌ヨ
+ */
+ @Override
+ public V getVoById(Serializable id) {
+ return getBaseMapper().selectVoById(id, voClass);
+ }
+
+ /**
+ * 鏌ヨ锛堟牴鎹甀D 鎵归噺鏌ヨ锛�
+ */
+ @Override
+ public List<V> listVoByIds(Collection<? extends Serializable> idList) {
+ return getBaseMapper().selectVoBatchIds(idList, voClass);
+ }
+
+ /**
+ * 鏌ヨ锛堟牴鎹� 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) {
+ return getBaseMapper().selectVoOne(queryWrapper, voClass);
+ }
+
+ /**
+ * 鏌ヨ鍒楄〃
+ */
+ @Override
+ public List<V> listVo(Wrapper<T> queryWrapper) {
+ return getBaseMapper().selectVoList(queryWrapper, voClass);
+ }
+
+ /**
+ * 缈婚〉鏌ヨ
+ * @deprecated 3.6.0 绉婚櫎 璇蜂娇鐢� {@link #pageVo(IPage, Wrapper)}
+ */
+ @Override
+ @Deprecated
+ public PagePlus<T, V> pageVo(PagePlus<T, V> page, Wrapper<T> queryWrapper) {
+ PagePlus<T, V> result = getBaseMapper().selectPage(page, queryWrapper);
+ List<V> volist = BeanCopyUtils.copyList(result.getRecords(), voClass);
+ result.setRecordsVo(volist);
+ return result;
+ }
+
+ /**
+ * 缈婚〉鏌ヨ
+ *
+ * @param page 缈婚〉瀵硅薄
+ * @param queryWrapper 瀹炰綋瀵硅薄灏佽鎿嶄綔绫�
+ */
+ public <P extends IPage<V>> P pageVo(IPage<T> page, Wrapper<T> queryWrapper) {
+ return getBaseMapper().selectVoPage(page, queryWrapper, voClass);
+ }
+
}
--
Gitblit v1.9.3