使用动态代理实现数据源切换

动态代理可以在运行时生成代理对象，从而可以在代理对象中添加额外的逻辑，比如实现数据源的切换。下面是一个使用动态代理实现数据源切换的示例代码：

首先定义两个数据源：

public class DataSource1 {
    public void execute() {
        System.out.println("Execute query in Data Source 1");
    }
}

public class DataSource2 {
    public void execute() {
        System.out.println("Execute query in Data Source 2");
    }
}

然后定义一个接口，用于代理数据源：

public interface DataSource {
    void execute();
}

接下来实现一个动态代理类，用于在运行时生成代理对象：

import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.lang.reflect.Proxy;

public class DataSourceProxy implements InvocationHandler {
    private Object target;

    public DataSourceProxy(Object target) {
        this.target = target;
    }

    public static DataSource createProxy(DataSource dataSource1, DataSource dataSource2) {
        return (DataSource) Proxy.newProxyInstance(
                DataSourceProxy.class.getClassLoader(),
                new Class[]{DataSource.class},
                new InvocationHandler() {
                    @Override
                    public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
                        if (method.getName().equals("execute")) {
                            if (ThreadLocalUtil.getDataSource() == 1) {
                                dataSource1.execute();
                            } else {
                                dataSource2.execute();
                            }
                            return null;
                        }
                        return method.invoke(proxy, args);
                    }
                });
    }

    @Override
    public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
        if (method.getName().equals("execute")) {
            if (ThreadLocalUtil.getDataSource() == 1) {
                method.invoke(target, args);
            } else {
                System.out.println("Switch to Data Source 2");
                ThreadLocalUtil.setDataSource(2);
                method.invoke(target, args);
            }
            return null;
        }
        return method.invoke(target, args);
    }
}

在该代理类中，我们重写了 `invoke` 方法，当调用代理对象的 `execute` 方法时，根据当前的数据源决定执行哪个数据源的 `execute` 方法。在实现过程中，我们使用了一个 `ThreadLocal` 变量来存储当前使用的数据源编号，从而可以在不同的线程中使用不同的数据源。

最后，我们可以使用下面的代码来测试该代理类：

public class Application {
    public static void main(String[] args) {
        DataSource dataSource1 = new DataSource1();
        DataSource dataSource2 = new DataSource2();

        DataSource dataSource = DataSourceProxy.createProxy(dataSource1, dataSource2);

        ThreadLocalUtil.setDataSource(1);
        dataSource.execute();

        ThreadLocalUtil.setDataSource(2);
        dataSource.execute();
    }
}

在该测试代码中，我们首先使用数据源 1 执行一次查询，然后切换到数据源 2 执行一次查询。在每次查询之前，我们使用 `ThreadLocalUtil.setDataSource` 方法设置当前的数据源编号。