Java中如何实现在每个查询中都自动填充某个查询条件

在Java中实现在每个查询中都自动填充某个查询条件，可以使用`AOP`（面向切面编程）技术和`ThreadLocal`线程局部变量的方式来实现。下面分别介绍一下这两种方式的实现。

## 1. AOP方式实现

`AOP`技术可以在方法执行前或执行后插入一段代码，可以使用它来实现在每个查询中都自动填充某个查询条件的功能。具体实现步骤如下：

1. 定义一个注解，用于标记需要自动填充的查询条件。

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface AutoFill {
    String value();
}

2. 定义一个切面类，在切面类中实现自动填充查询条件的逻辑。

@Aspect
@Component
public class AutoFillAspect {
    @Before("@annotation(autoFill)")
    public void before(JoinPoint joinPoint, AutoFill autoFill) {
        String fieldName = autoFill.value();
        Object[] args = joinPoint.getArgs();
        for (Object arg : args) {
            if (arg instanceof Wrapper) {
                ((Wrapper) arg).eq(fieldName, getCurrentUserId());
            }
        }
    }
    private Long getCurrentUserId() {
        // 获取当前用户的ID
        return 1L;
    }
}

在切面类中，我们使用`@Before`注解来标记需要在方法执行前插入代码的方法。使用`@annotation(autoFill)`来指定切点，表示当被注解`@AutoFill`标记的方法被调用时，会执行`before`方法中的逻辑。

在`before`方法中，我们首先获取被调用方法的参数列表，然后遍历参数列表，如果参数类型是`Wrapper`，则将自动填充的查询条件添加到查询条件中。`getCurrentUserId`方法用于获取当前用户的ID，这个方法可以根据实际情况进行实现。

3. 在需要自动填充查询条件的方法上添加`@AutoFill`注解。

@AutoFill("user_id")
public List<User> findAll(Wrapper<User> wrapper) {
    // 查询所有用户
}

这样，在调用`findAll`方法时，会自动将`user_id`这个查询条件添加到查询条件中，从而实现了自动填充查询条件的功能。

## 2. ThreadLocal方式实现

`ThreadLocal`线程局部变量可以在当前线程中存储一个对象，可以使用它来实现在每个查询中都自动填充某个查询条件的功能。具体实现步骤如下：

1. 定义一个静态的`ThreadLocal`变量，用于存储需要自动填充的查询条件。

public class QueryCondition {
    private static final ThreadLocal<Map<String, Object>> THREAD_LOCAL = new ThreadLocal<>();
    public static void set(String key, Object value) {
        Map<String, Object> map = THREAD_LOCAL.get();
        if (map == null) {
            map = new HashMap<>();
            THREAD_LOCAL.set(map);
        }
        map.put(key, value);
    }
    public static Object get(String key) {
        Map<String, Object> map = THREAD_LOCAL.get();
        if (map == null) {
            return null;
        }
        return map.get(key);
    }
    public static void clear() {
        THREAD_LOCAL.remove();
    }
}

在`QueryCondition`类中，我们使用`ThreadLocal`来存储查询条件，提供了三个方法：`set`、`get`和`clear`，分别用于设置查询条件、获取查询条件和清空查询条件。

2. 在需要自动填充查询条件的地方调用`QueryCondition.set`方法设置查询条件。

QueryCondition.set("user_id", getCurrentUserId());
List<User> userList = userMapper.selectList(wrapper);
QueryCondition.clear();

在调用`selectList`方法之前，我们需要先调用`QueryCondition.set`方法设置需要自动填充的查询条件。在设置完查询条件后，调用`selectList`方法进行查询。查询完成后，需要调用`QueryCondition.clear`方法清空查询条件，以免影响下一次查询。

3. 在查询条件中使用`QueryCondition.get`方法获取需要自动填充的查询条件。

QueryWrapper<User> wrapper = new QueryWrapper<>();
wrapper.eq("name", "test")
       .eq("age", 20)
       .eq("user_id", QueryCondition.get("user_id"));
List<User> userList = userMapper.selectList(wrapper);

在查询条件中，我们使用`QueryCondition.get`方法获取需要自动填充的查询条件，从而实现了自动填充查询条件的功能。