超实用的mybatis-plus工具类-工具盒子

每人心中都应有两盏灯光，一盏是希望的灯光;一盏是勇气的灯光。有了这两盏灯光，我们就不怕海上的黑暗和风涛的险恶了。------罗兰

分享给大家

需要引入mybatis-plus

|------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------| | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 | hljs java package com.kuang.common.util; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.mapper.BaseMapper; import com.baomidou.mybatisplus.core.toolkit.CollectionUtils; import com.baomidou.mybatisplus.core.toolkit.support.SFunction; import java.util.Collections; import java.util.List; import java.util.Map; import java.util.Optional; import java.util.function.Function; import java.util.function.Supplier; import java.util.stream.Collectors; /** * mp工具类 * * @author <achao1441470436@gmail.com> * @date 2021/5/4 0004 16:35 */ public class MybatisPlusUtils { private MybatisPlusUtils() { // Do not new me! } /** * 传入mapper,paramList,源属性，新属性得到 List<新属性> 用于关联表等 * * @param mapper 持久层操作类 * @param param 查询参数 * @param origin 源属性 * @param target 新属性 * @param <T> PO * @param T中的源属性 * @param <R> T中的新属性 * @return java.util.Map<I, T> * @author <achao1441470436@gmail.com> * @date 2021/5/4 0004 20:37 */ public static <T, I, R> List<R> getList(BaseMapper<T> mapper, Supplier param, SFunction<T, I> origin, SFunction<T, R> target) { return Optional.ofNullable(param).map(Supplier::get).map(p -> mapper.selectList(new LambdaQueryWrapper<T>().eq(origin, p).select(target))).map(List::parallelStream).map(s -> s.map(target).collect(Collectors.toList())).orElseGet(Collections::emptyList); } /** * 传入mapper,paramList,对应的属性得到 Map<实体属性, 实体> 用于一对一 * * @param mapper 持久层操作类 * @param paramList 查询参数 * @param sFunction 条件 * @param <T> PO * @param T中的属性 * @return java.util.Map<I, T> * @author <achao1441470436@gmail.com> * @date 2021/5/4 0004 20:37 */ public static <T, I> Map<I, T> getMapBy(BaseMapper<T> mapper, List paramList, SFunction<T, I> sFunction) { return Optional.ofNullable(paramList).filter(CollectionUtils::isNotEmpty).map(p -> mapper.selectList(new LambdaQueryWrapper<T>().in(sFunction, p))).map(List::parallelStream).map(s -> s.collect(Collectors.toMap(sFunction, Function.identity(), (a, b) -> b))).orElseGet(Collections::emptyMap); } /** * 传入mapper,paramList,对应的属性得到 Map<实体属性, List<实体>> 用于一对多 * * @param mapper 持久层操作类 * @param paramList 查询参数 * @param sFunction 条件 * @param <T> PO * @param T中的属性 * @return java.util.Map<I, List < T> > * @author <achao1441470436@gmail.com> * @date 2021/5/4 0004 20:37 */ public static <T, I> Map<I, List<T>> groupBy(BaseMapper<T> mapper, List paramList, SFunction<T, I> sFunction) { return Optional.ofNullable(paramList).filter(CollectionUtils::isNotEmpty).map(p -> mapper.selectList(new LambdaQueryWrapper<T>().in(sFunction, p))).map(List::parallelStream).map(s -> s.collect(Collectors.groupingBy(sFunction))).orElseGet(Collections::emptyMap); } } |

|-----------|-------------------------------------------------------------| | 1 | hljs shell ##2021-05-08 13:10:54 很多小伙伴说看不懂，这里就补充补充啦~ |

第一个函数`getList` {#第一个函数getlist}

一般用于关联表处，例如我这里需要实现一个已购课程的功能，我们首先要根据用户id去查询用户的课程购买记录表

|-----------------|-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------| | 1 2 3 4 | hljs java // 通过userId拿到CourseId User current = UserThreadLocal.getNoneNullUser(); // 从关联表获取课程ids 【使用方式在这里】 List<String> courseIds = MybatisPlusUtils.getList(this.baseMapper, current::getUserid, UserCourse::getUserid, UserCourse::getCourseid); |

这里三个参数，baseMapper是继承mybatis-plus中com.baomidou.mybatisplus.extension.service.impl.ServiceImpl获取的

然后第二个参数current::getUserid就是一个java.util.function.Supplier

是作为查询条件参数值

你也可以这么写() -> current.getUserid()

第三个参数是用于查询的条件，这里UserCourse::getUserid表示根据userId查询数据

结合上个参数，这里是使用userId = current.getUserid()去查询数据

第四个参数是你要返回的数据，例如这里我是通过用户id获取课程id,我们就给个UserCourse::getCourseid

最后实现的效果就下面这段差不多的

|-------------------|------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------| | 1 2 3 4 5 | hljs java List<UserCourse> userCourses = this.baseMapper.selectList(Wrappers.lambdaQuery(new UserCourse()).in(UserCourse::getUserid, current.getUserid())); List<String> courseIds = new ArrayList<>(userCourses.size()); for (int i = 0; i < userCourses.size(); i++) { courseIds.add(userCourses.get(i).getCourseid()); } |

第二个函数`getMapBy` {#第二个函数getmapby}

这里描述写的多用于一对一，不是指表之间关联关系是一对一，而是key和value一对一

三个参数分别为

继承com.baomidou.mybatisplus.core.mapper.BaseMapper的mapper接口

数据参数集合

查询条件

例如我这里，查询一个用户信息列表，但我需要带一个用户的信息，放在用户信息表

|---------------------------------------|-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------| | 1 2 3 4 5 6 7 8 9 10 11 12 13 | hljs java // 用户信息 List<User> list = userService.list(); // 用户ids List<String> userIds = list.parallelStream().map(User::getUserid).collect(Collectors.toList()); // 根据用户ids获取 Map<userId,UserInfo> 【使用方式在这里】 Map<String, UserInfo> userIdInfoMap = MybatisPlusUtils.getMapBy(userInfoMapper, userIds, UserInfo::getUserid); List<UserVO> collect = list.parallelStream().map(user -> { // 两种方式获取到 user 关联的 UserInfo UserInfo fans = userIdInfoMap.getOrDefault(user.getUserid(), new UserInfo()); UserInfo fans1 = Optional.ofNullable(user.getUserid()).map(userIdInfoMap::get).orElseGet(UserInfo::new); /* 封装vo返回...这里省略 */ return vo; }).collect(Collectors.toList()); |

第三个函数`groupBy` {#第三个函数groupby}

这里用于key对多个value场景

三个参数分别为

继承com.baomidou.mybatisplus.core.mapper.BaseMapper的mapper接口

数据参数集合

查询条件

例如我这里，查询一个用户信息列表，并携带出用户的粉丝列表，用户和粉丝列表是一对多的关系

|---------------------------------------|---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------| | 1 2 3 4 5 6 7 8 9 10 11 12 13 | hljs java // 用户信息 List<User> list = userService.list(); // 用户ids List<String> userIds = list.parallelStream().map(User::getUserid).collect(Collectors.toList()); // 根据用户ids获取 Map<userId,List<UserFans>> 【使用方式在这里】 Map<String, List<UserFans>> userIdFansMap = MybatisPlusUtils.groupBy(userFansMapper, userIds, UserFans::getUserid); List<UserVO> collect = list.parallelStream().map(user -> { // 两种方式获取到 user 关联的 UserFans List<UserFans> fans = userIdFansMap.getOrDefault(user.getUserid(), Collections.emptyList()); List<UserFans> fans = Optional.ofNullable(user.getUserid()).map(userIdFansMap::get).orElseGet(Collections::emptyList); /* 封装vo返回...这里省略 */ return vo; }).collect(Collectors.toList()); |