Springboot中利用aop和注解实现动态数据源

本篇文章将介绍如何使用AOP和注解来实现动态数据源．

使用ThreadLocal存储当前线程使用的数据源的key

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;


/**
 * 用于保存当前线程使用的数据源名称的工具类，多数据源动态切换的工具类
 * @author lishouyu
 * @version 1.0
 * @since 1.0
 */
public class DataSourceContextHolder {
   

  /**
   * 用于日志记录的对象
   */
  public static final Logger logger = LoggerFactory.getLogger(DataSourceContextHolder.class);

  /**
   * 进程内数据存储
   */
  private static final ThreadLocal<String> contextHolder = new ThreadLocal<>();

  /**
   * 默认的数据源的名称
   */
  public static final String DEFAULT_DATDASOURCE_NAME = "master";

  /**
   * 设置数据源名
   * @param datasourceName 数据源的名字
   */
  public static void setDatasourceName(String datasourceName){
    logger.info("切换到{}数据源",datasourceName);
    contextHolder.set(datasourceName);
  }

  /**
   * 获取数据源名
   * @return
   */
  public static String getDatdasourceName(){
    return contextHolder.get();
  }

  /**
   * 清除数据源名称
   */
  public static void clearDatasourceName(){
    contextHolder.remove();
  }
}

创建动态数据源类

继承spring框架为我提供的数据源路由抽象类AbstractRoutingDataSource，创建我们自己的动态数据源类