com.baomidou.mybatisplus.core.conditions.Wrapper Java Examples

@Override
public boolean saveOrUpdateIdempotency(T entity, DistributedLock lock, String lockKey, Wrapper<T> countWrapper, String msg) throws Exception {
    if (null != entity) {
        Class<?> cls = entity.getClass();
        TableInfo tableInfo = TableInfoHelper.getTableInfo(cls);
        if (null != tableInfo && StrUtil.isNotEmpty(tableInfo.getKeyProperty())) {
            Object idVal = ReflectionKit.getMethodValue(cls, entity, tableInfo.getKeyProperty());
            if (StringUtils.checkValNull(idVal) || Objects.isNull(getById((Serializable) idVal))) {
                if (StrUtil.isEmpty(msg)) {
                    msg = "已存在";
                }
                return this.saveIdempotency(entity, lock, lockKey, countWrapper, msg);
            } else {
                return updateById(entity);
            }
        } else {
            throw ExceptionUtils.mpe("Error:  Can not execute. Could not find @TableId.");
        }
    }
    return false;
}
 

@Override
public List<SysMenu> findByUser(String userName, Wrapper<SysMenu> queryWrapper) {
    if (userName == null || "".equals(userName) || "admin".equals(userName)) {
        return this.list(queryWrapper);
    }
    SysUser sysUser = userMapper.selectOne(new LambdaQueryWrapper<SysUser>().eq(SysUser::getName, userName));
    if (sysUser == null) {
        return new ArrayList<>();
    }
    List<SysUserRole> sysUserRoles = userRoleMapper.selectList(new LambdaQueryWrapper<SysUserRole>().eq(SysUserRole::getUserId, sysUser.getId()));
    List<Long> roldIds = sysUserRoles.stream().map(SysUserRole::getRoleId).collect(Collectors.toList());
    List<SysMenu> result = new ArrayList<>();
    for (Long roldId : roldIds) {
        result.addAll(roleService.findRoleMenus(roldId));
    }
    return result;
}
 

public static Wrapper<?> fillWrapperOrder(Page<?> page, Wrapper<?> wrapper) {
	if (null == page) {
		return wrapper;
	}
	if (ObjectUtil.isEmpty(page.orders())) {
		return wrapper;
	}
	QueryWrapper queryWrapper = null == wrapper ? new QueryWrapper() : (QueryWrapper) wrapper;
	if (ObjectUtil.isNotEmpty(page.orders())) {
		page.orders().forEach(orderItem -> queryWrapper.orderBy(true, orderItem.isAsc(),
			StringUtil.toRevertCamelCase(orderItem.getColumn(), CharUtil.UNDERLINE)));
		page.setOrders(null);
	}
	return queryWrapper;
}
 

@Override
public List<UserExcel> exportUser(Wrapper<User> queryWrapper) {
	List<UserExcel> userList = baseMapper.exportUser(queryWrapper);
	userList.forEach(user -> {
		user.setRoleName(StringUtil.join(sysClient.getRoleNames(user.getRoleId())));
		user.setDeptName(StringUtil.join(sysClient.getDeptNames(user.getDeptId())));
		user.setPostName(StringUtil.join(sysClient.getPostNames(user.getPostId())));
	});
	return userList;
}
 

@Override
public boolean saveIdempotency(T entity, DistributedLock locker, String lockKey, Wrapper<T> countWrapper, String msg) throws Exception {
    if (locker == null) {
        throw new LockException("DistributedLock is null");
    }
    if (StrUtil.isEmpty(lockKey)) {
        throw new LockException("lockKey is null");
    }
    Object lock = null;
    try {
        //加锁
        lock = locker.tryLock(lockKey, 10, 60, TimeUnit.SECONDS);
        if (lock != null) {
            //判断记录是否已存在
            int count = super.count(countWrapper);
            if (count == 0) {
                return super.save(entity);
            } else {
                if (StrUtil.isEmpty(msg)) {
                    msg = "已存在";
                }
                throw new IdempotencyException(msg);
            }
        } else {
            throw new LockException("锁等待超时");
        }
    } finally {
        locker.unlock(lock);
    }
}
 

@Override
public List<Long> getPermissionIdsByRoleId(Long roleId) throws Exception {
    Wrapper wrapper = lambdaQuery()
            .select(SysRolePermission::getPermissionId)
            .eq(SysRolePermission::getRoleId,roleId)
            .eq(SysRolePermission::getState,StateEnum.ENABLE.getCode())
            .getWrapper();
    return sysRolePermissionMapper.selectObjs(wrapper);
}
 

@Override
public boolean isExistsByPermissionIds(List<Long> permissionIds) {
    if (CollectionUtils.isEmpty(permissionIds)) {
        return false;
    }
    Wrapper wrapper = lambdaQuery().in(SysPermission::getId, permissionIds).getWrapper();
    return sysPermissionMapper.selectCount(wrapper).intValue() == permissionIds.size();
}
 

@Override
public List<SysPermissionTreeVo> getNavMenuTree() throws Exception {
    List<Integer> levels = Arrays.asList(MenuLevelEnum.ONE.getCode(), MenuLevelEnum.TWO.getCode());
    Wrapper wrapper = lambdaQuery()
            .in(SysPermission::getLevel, levels)
            .eq(SysPermission::getState, StateEnum.ENABLE.getCode())
            .getWrapper();


    List<SysPermission> list = sysPermissionMapper.selectList(wrapper);

    return convertSysPermissionTreeVoList(list);

}
 

/**
 * 分页处理service
 *
 * @param page     页码
 * @param pageSize 一页显示的大小
 */
default Response<List<T>> toPage(int page, int pageSize, Wrapper<T> wrapper) {
    int total = this.count(wrapper);
    List<T> result = Collections.emptyList();
    if (0 < total) {
        result = this.page(new Page<>(page, pageSize, false), wrapper).getRecords();
    }
    return Response.toResponse(result, total);
}
 

/**
 * 分页查询
 *
 * @param queryWrapper 条件构造器
 * @param pageNum      页码值
 * @param pageSize     每页大小
 * @param orderBy      排序字段
 * @return PageInfo 分页对象
 */
@Override
public PageInfo<T> page(Wrapper<T> queryWrapper, int pageNum, int pageSize, String orderBy) {
    if (null == orderBy) {
        PageHelper.startPage(pageNum, pageSize);
    } else {
        PageHelper.startPage(pageNum, pageSize, orderBy);
    }

    List<T> list = list(queryWrapper);
    PageInfo<T> pageInfo = new PageInfo<>(list);

    return pageInfo;
}
 

/**
 * 根据其他属性字段修改一条数据
 * @param pojo  任意类型实体类
 * @param column 具体字段
 * @param value 具体字段的值
 * @return 返回前端json数据
 */
@PostMapping("updateByColumn")
public Result update(@RequestBody T pojo, @RequestParam("column") String column, @RequestParam(name = "value") String value){
    Wrapper<T> wrapper = Wrappers.<T>update().eq(column,value);
    boolean flag = service.update(pojo, wrapper);
    if(flag){
        return Result.successJson("成功修改一条数据");
    }
    return Result.errorJson();
}
 

/**
 * 要根据哪个字段删除一条数据
 * @param column 要根据哪个字段删除
 * @param value  字段的值
 * @return 返回前端json数据
 */
@DeleteMapping("deleteByColumn")
public Result delete(@RequestParam("column") String column, @RequestParam("value") String value){
    Wrapper<T> ew = Wrappers.<T>query().eq(column,value);
    boolean flag = service.remove(ew);
    if(flag){
        return Result.successJson("成功删除一条数据");
    }
    return Result.errorJson();
}
 

@Override
public IPage getSchemeVoPage(PageModel pm, SchemeQueryCriteria schemeQueryCriteria) {
	Wrapper wrapper = QueryWrapperUtil.getWrapper(pm, schemeQueryCriteria);
	pm.addOrder(OrderItem.desc("a." + Scheme.F_SQL_CREATEDDATE));
	IPage<List<SchemeVo>> userVosPage = repository.getSchemeVoPage(pm, wrapper);
	return userVosPage;
}
 

@Override
public <V> List<V> listObjs(Wrapper<T> queryWrapper, Function<? super Object, V> mapper) {
    return serviceImpl.listObjs(queryWrapper, mapper);
}
 

@Override
public int count(Wrapper<T> queryWrapper) {
    Integer total = baseMapper.selectCount(queryWrapper);
    return null != total ? total : 0;
}
 

@Override
User selectOne(@Param("ew") Wrapper<User> wrapper);
 

@Override
public IPage<Map<String, Object>> pageMaps(IPage<T> page, Wrapper<T> queryWrapper) {
    return serviceImpl.pageMaps(page, queryWrapper);
}
 

@Override
public boolean saveOrUpdateIdempotency(T entity, DistributedLock lock, String lockKey, Wrapper<T> countWrapper) throws Exception {
    return this.saveOrUpdateIdempotency(entity, lock, lockKey, countWrapper, null);
}
 

@Override
public List<T> list(Wrapper<T> queryWrapper) {
    return serviceImpl.list(queryWrapper);
}
 

@Override
public int count(Wrapper<T> queryWrapper) {
    return serviceImpl.count(queryWrapper);
}
 

@Override
public <V> V getObj(Wrapper<T> queryWrapper, Function<? super Object, V> mapper) {
    return serviceImpl.getObj(queryWrapper, mapper);
}
 

@Override
public Map<String, Object> getMap(Wrapper<T> queryWrapper) {
    return serviceImpl.getMap(queryWrapper);
}
 

@Override
public List<Map<String, Object>> listMaps(Wrapper<T> queryWrapper) {
    return serviceImpl.listMaps(queryWrapper);
}
 

@Override
public List<PluginData> getByName(String name){
    Wrapper<PluginData> wrapper = Wrappers.<PluginData>query()
            .eq("name", name);
    return list(wrapper);
}
 

@Override
public List<JSONObject> queryAllDepart(Wrapper wrapper) {
    //noinspection unchecked
    return JSON.parseArray(JSON.toJSONString(sysDepartService.list(wrapper))).toJavaList(JSONObject.class);
}
 

@Override
public List<T> list(Wrapper<T> queryWrapper) {
    return baseMapper.selectList(queryWrapper);
}
 

@Override
public List<JSONObject> queryAllUser(Wrapper wrapper) {
    //noinspection unchecked
    return JSON.parseArray(JSON.toJSONString(userMapper.selectList(wrapper))).toJavaList(JSONObject.class);
}
 

@Override
public List<JSONObject> queryAllDepart(Wrapper wrapper) {
    //noinspection unchecked
    return JSON.parseArray(JSON.toJSONString(sysDepartService.list(wrapper))).toJavaList(JSONObject.class);
}
 

@Override
public List<JSONObject> queryAllUser(Wrapper wrapper) {
    //noinspection unchecked
    return JSON.parseArray(JSON.toJSONString(userMapper.selectList(wrapper))).toJavaList(JSONObject.class);
}
 

@Override
public T getOne(Wrapper<T> queryWrapper) {
    return baseMapper.selectOne(queryWrapper);
}