6 changed files with 253 additions and 64 deletions
@ -0,0 +1,108 @@ |
|||
package org.dromara.im.util; |
|||
|
|||
import cn.hutool.core.util.ObjectUtil; |
|||
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; |
|||
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper; |
|||
import com.baomidou.mybatisplus.core.toolkit.support.SFunction; |
|||
import lombok.AccessLevel; |
|||
import lombok.NoArgsConstructor; |
|||
import org.dromara.im.service.IImAgentService; |
|||
import org.springframework.beans.factory.annotation.Autowired; |
|||
import org.springframework.stereotype.Component; |
|||
|
|||
/** |
|||
* LambdaQueryWrapper 通用工具类 |
|||
* |
|||
* @author Blue |
|||
* @date 2026-04-08 |
|||
*/ |
|||
@NoArgsConstructor(access = AccessLevel.PRIVATE) |
|||
public class LambdaQueryWrapperHelper { |
|||
|
|||
private static IImAgentService imAgentService; |
|||
|
|||
/** |
|||
* 初始化服务（由 Spring 调用） |
|||
*/ |
|||
@Component |
|||
public static class Initializer { |
|||
@Autowired |
|||
public void setImAgentService(IImAgentService imAgentService) { |
|||
LambdaQueryWrapperHelper.imAgentService = imAgentService; |
|||
} |
|||
} |
|||
|
|||
/** |
|||
* 为 LambdaQueryWrapper 追加 uniqueToken 查询条件 |
|||
* <p> |
|||
* 根据当前登录用户ID获取对应的 uniqueToken，并添加到查询条件中 |
|||
* 如果用户ID为1（超级管理员），则不添加该条件 |
|||
* |
|||
* @param wrapper 查询条件包装器 |
|||
* @param tokenField 实体类中 uniqueToken 字段的 getter 方法引用 |
|||
* @param <T> 实体类型 |
|||
* @return 添加了 uniqueToken 条件的 LambdaQueryWrapper |
|||
*/ |
|||
public static <T> LambdaQueryWrapper<T> appendToken(LambdaQueryWrapper<T> wrapper, SFunction<T, ?> tokenField) { |
|||
if (wrapper == null || tokenField == null) { |
|||
return wrapper; |
|||
} |
|||
|
|||
String token = getTokenByUserId(); |
|||
|
|||
// 如果 token 为 null，表示是超级管理员，不添加过滤条件
|
|||
if (token == null) { |
|||
return wrapper; |
|||
} |
|||
|
|||
// 添加 uniqueToken 查询条件
|
|||
wrapper.eq(ObjectUtil.isNotEmpty(token), tokenField, token); |
|||
|
|||
return wrapper; |
|||
} |
|||
|
|||
/** |
|||
* 为 LambdaUpdateWrapper 追加 uniqueToken 修改/删除条件 |
|||
* <p> |
|||
* 根据当前登录用户ID获取对应的 uniqueToken，并添加到更新或删除的 WHERE 条件中 |
|||
* 如果用户ID为1（超级管理员），则不添加该条件 |
|||
* <p> |
|||
* 使用场景： |
|||
* - 更新数据时限制只能更新自己的数据 |
|||
* - 删除数据时限制只能删除自己的数据 |
|||
* |
|||
* @param wrapper 更新条件包装器 |
|||
* @param tokenField 实体类中 uniqueToken 字段的 getter 方法引用 |
|||
* @param <T> 实体类型 |
|||
* @return 添加了 uniqueToken 条件的 LambdaUpdateWrapper |
|||
*/ |
|||
public static <T> LambdaUpdateWrapper<T> appendToken(LambdaUpdateWrapper<T> wrapper, SFunction<T, ?> tokenField) { |
|||
if (wrapper == null || tokenField == null) { |
|||
return wrapper; |
|||
} |
|||
|
|||
String token = getTokenByUserId(); |
|||
|
|||
// 如果 token 为 null，表示是超级管理员，不添加过滤条件
|
|||
if (token == null) { |
|||
return wrapper; |
|||
} |
|||
|
|||
// 添加 uniqueToken 修改/删除条件
|
|||
wrapper.eq(ObjectUtil.isNotEmpty(token), tokenField, token); |
|||
|
|||
return wrapper; |
|||
} |
|||
|
|||
/** |
|||
* 获取当前用户的 uniqueToken |
|||
* |
|||
* @return uniqueToken，超级管理员返回 null，未找到返回 "error" |
|||
*/ |
|||
private static String getTokenByUserId() { |
|||
if (imAgentService == null) { |
|||
throw new IllegalStateException("IImAgentService 未初始化，请确保 Spring 容器已启动"); |
|||
} |
|||
return imAgentService.getTokenByUserId(); |
|||
} |
|||
} |
|||
Loading…
Reference in new issue