From 545f0bd4b0b2f84fb54c165bc551aa08ca9c073e Mon Sep 17 00:00:00 2001
From: 抓蛙师 <770492966@qq.com>
Date: 星期三, 25 八月 2021 20:24:35 +0800
Subject: [PATCH] saveAll支持有id则更新和无id则插入
---
ruoyi-common/src/main/java/com/ruoyi/common/core/mybatisplus/core/ServicePlusImpl.java | 60 ++++++++++++++++++++++++++++++++++++++----------------------
1 files changed, 38 insertions(+), 22 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 0325431..99441b5 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,24 @@
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.annotation.TableId;
+import com.baomidou.mybatisplus.annotation.TableName;
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 com.ruoyi.common.utils.BeanCopyUtils;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.ResolvableType;
import java.io.Serializable;
+import java.lang.annotation.Annotation;
+import java.lang.reflect.Field;
+import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Map;
-import java.util.stream.Collectors;
/**
* IServicePlus 瀹炵幇绫�
@@ -95,6 +98,10 @@
return super.updateBatchById(entityList, batchSize);
}
+ /**
+ * 鍗晄ql鎵归噺鎻掑叆( 鍏ㄩ噺濉厖 鏃犺鏁版嵁搴撻粯璁ゅ�� )
+ * 閫傜敤浜庢棤鑴戞彃鍏�
+ */
@Override
public boolean saveBatch(Collection<T> entityList) {
return saveBatch(entityList, DEFAULT_BATCH_SIZE);
@@ -111,12 +118,32 @@
}
/**
- * 鍗晄ql鎵归噺鎻掑叆( 鍏ㄩ噺濉厖 鏃犺鏁版嵁搴撻粯璁ゅ�� )
+ * 鍗晄ql鎵归噺鎻掑叆( 鍏ㄩ噺濉厖 )
* 閫傜敤浜庢棤鑴戞彃鍏�
*/
@Override
public boolean saveAll(Collection<T> entityList) {
- return baseMapper.insertAll(entityList) == entityList.size();
+ ArrayList<T> list = new ArrayList<>();
+ for (T t : entityList) {
+ try {
+ //鑾峰彇灞炴�ф敞瑙ｇ殑value鍊�
+ Field f = t.getClass().getDeclaredField("id");
+ f.setAccessible( true );//璁剧疆鍙互鑼冨洿private
+ Object o = f.get(t);//鑾峰彇鍑篿d鐨勫��
+ System.out.println(o);
+ if (o == null) {
+ //濡傛灉id涓簄ull,鎻掑叆
+ list.add(t);
+ } else {
+ //鍚﹀垯鏇存柊
+ baseMapper.updateById(t);
+ }
+
+ } catch (Exception e) {
+ e.printStackTrace();
+ }
+ }
+ return baseMapper.insertAll(list) == list.size();
}
/**
@@ -127,7 +154,7 @@
@Override
public K getVoById(Serializable id, CopyOptions copyOptions) {
T t = getBaseMapper().selectById(id);
- return oneVoCopy(t, copyOptions);
+ return BeanCopyUtils.oneCopy(t, copyOptions, voClass);
}
/**
@@ -141,7 +168,7 @@
if (list == null) {
return null;
}
- return listVoCopy(list, copyOptions);
+ return BeanCopyUtils.listCopy(list, copyOptions, voClass);
}
/**
@@ -155,7 +182,7 @@
if (list == null) {
return null;
}
- return listVoCopy(list, copyOptions);
+ return BeanCopyUtils.listCopy(list, copyOptions, voClass);
}
/**
@@ -167,7 +194,7 @@
@Override
public K getVoOne(Wrapper<T> queryWrapper, CopyOptions copyOptions) {
T t = getOne(queryWrapper, true);
- return oneVoCopy(t, copyOptions);
+ return BeanCopyUtils.oneCopy(t, copyOptions, voClass);
}
/**
@@ -181,7 +208,7 @@
if (list == null) {
return null;
}
- return listVoCopy(list, copyOptions);
+ return BeanCopyUtils.listCopy(list, copyOptions, voClass);
}
/**
@@ -193,20 +220,9 @@
@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);
+ List<K> volist = BeanCopyUtils.listCopy(result.getRecords(), copyOptions, voClass);
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());
}
}
--
Gitblit v1.9.3