From eb2be90e74adfa06ee251f80c08243d0301ce467 Mon Sep 17 00:00:00 2001 From: 疯狂的狮子li <15040126243@163.com> Date: 星期三, 28 七月 2021 10:49:16 +0800 Subject: [PATCH] Merge remote-tracking branch 'origin/dev' --- ruoyi-common/src/main/java/com/ruoyi/common/core/mybatisplus/core/ServicePlusImpl.java | 117 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 116 insertions(+), 1 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..e00076e 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,12 +1,19 @@ package com.ruoyi.common.core.mybatisplus.core; +import cn.hutool.core.bean.copier.CopyOptions; +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.util.Collection; +import java.util.List; +import java.util.Map; /** * IServicePlus 瀹炵幇绫� @@ -15,7 +22,7 @@ */ @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, K> extends ServiceImpl<M, T> implements IServicePlus<T, K> { @Autowired protected M baseMapper; @@ -35,6 +42,12 @@ protected Class<T> mapperClass = currentMapperClass(); + protected Class<K> voClass = currentVoClass(); + + public Class<K> getVoClass() { + return voClass; + } + @Override protected Class<T> currentMapperClass() { return (Class<T>) this.getResolvableType().as(ServicePlusImpl.class).getGeneric(0).getType(); @@ -43,6 +56,10 @@ @Override protected Class<T> currentModelClass() { return (Class<T>) this.getResolvableType().as(ServicePlusImpl.class).getGeneric(1).getType(); + } + + protected Class<K> currentVoClass() { + return (Class<K>) this.getResolvableType().as(ServicePlusImpl.class).getGeneric(2).getType(); } @Override @@ -81,8 +98,106 @@ * 閫傜敤浜庢棤鑴戞彃鍏� */ @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) { return baseMapper.insertAll(entityList) == entityList.size(); } + /** + * 鏍规嵁 ID 鏌ヨ + * + * @param id 涓婚敭ID + */ + @Override + public K getVoById(Serializable id, CopyOptions copyOptions) { + T t = getBaseMapper().selectById(id); + return BeanCopyUtils.oneCopy(t, copyOptions, voClass); + } + + /** + * 鏌ヨ锛堟牴鎹甀D 鎵归噺鏌ヨ锛� + * + * @param idList 涓婚敭ID鍒楄〃 + */ + @Override + public List<K> listVoByIds(Collection<? extends Serializable> idList, CopyOptions copyOptions) { + List<T> list = getBaseMapper().selectBatchIds(idList); + if (list == null) { + return null; + } + return BeanCopyUtils.listCopy(list, copyOptions, voClass); + } + + /** + * 鏌ヨ锛堟牴鎹� columnMap 鏉′欢锛� + * + * @param columnMap 琛ㄥ瓧娈� map 瀵硅薄 + */ + @Override + public List<K> listVoByMap(Map<String, Object> columnMap, CopyOptions copyOptions) { + List<T> list = getBaseMapper().selectByMap(columnMap); + if (list == null) { + return null; + } + return BeanCopyUtils.listCopy(list, copyOptions, voClass); + } + + /** + * 鏍规嵁 Wrapper锛屾煡璇竴鏉¤褰� <br/> + * <p>缁撴灉闆嗭紝濡傛灉鏄涓細鎶涘嚭寮傚父锛岄殢鏈哄彇涓�鏉″姞涓婇檺鍒舵潯浠� wrapper.last("LIMIT 1")</p> + * + * @param queryWrapper 瀹炰綋瀵硅薄灏佽鎿嶄綔绫� {@link com.baomidou.mybatisplus.core.conditions.query.QueryWrapper} + */ + @Override + public K getVoOne(Wrapper<T> queryWrapper, CopyOptions copyOptions) { + T t = getOne(queryWrapper, true); + return BeanCopyUtils.oneCopy(t, copyOptions, voClass); + } + + /** + * 鏌ヨ鍒楄〃 + * + * @param queryWrapper 瀹炰綋瀵硅薄灏佽鎿嶄綔绫� {@link com.baomidou.mybatisplus.core.conditions.query.QueryWrapper} + */ + @Override + public List<K> listVo(Wrapper<T> queryWrapper, CopyOptions copyOptions) { + List<T> list = getBaseMapper().selectList(queryWrapper); + if (list == null) { + return null; + } + return BeanCopyUtils.listCopy(list, copyOptions, voClass); + } + + /** + * 缈婚〉鏌ヨ + * + * @param page 缈婚〉瀵硅薄 + * @param queryWrapper 瀹炰綋瀵硅薄灏佽鎿嶄綔绫� + */ + @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 = BeanCopyUtils.listCopy(result.getRecords(), copyOptions, voClass); + result.setRecordsVo(volist); + return result; + } + } -- Gitblit v1.9.3