每天一个注解之@DataSource、 @DS

本文主要是介绍每天一个注解之@DataSource、 @DS，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

在Java中，@DataSource 注解通常用于标记数据源（DataSource）相关的信息。数据源是一个用于获取数据库连接的对象，它通常用于与数据库进行交互。@DataSource 注解的详细说明可能会因不同的框架或库而有所不同，但通常用于配置数据源的一些属性，例如数据库的连接URL、用户名、密码等。

方案一

不同的包具体实现是不一样的，举例这个注解，主要是为了引出多数据源的配置方法。

依赖

       <dependency><groupId>com.baomidou</groupId><artifactId>dynamic-datasource-spring-boot-starter</artifactId><version>3.4.1</version></dependency>

配置：

在这里插入图片描述

使用不同的数据源

@DS(“master”), master 就是配置的数据源名称。
在这里插入图片描述

说明

@DS 可以注解在方法上和类上。如果同时存在，方法注解优先于类上注解。强烈建议注解在 service 实现或 mapper 接口方法上。

@DS(“xxx”) 指定使用 xxx 这个数据源，xxx 可以为组名也可以为具体某个库的名称。如果是组名则切换时采用负载均衡算法切换。如果指定的组名或者库不存在，则自动使用默认数据源（主库）

如果没有 @DS，则使用默认数据源（主库）

如果设置了 @DS 但没有指定某个组或者库，则根据 DynamicDataSourceStrategy 策略，选择一个从库。默认负载均衡策略。

方案二：借助com.alibaba.druid 自己实现多数据源

依赖

       <dependency><groupId>com.alibaba</groupId><artifactId>druid</artifactId><version>${druid.version}</version></dependency>

spring:profiles: devdatasource:url: jdbc:mysql://127.0.0.1:3306/pos?serverTimezone=UTC&useSSL=false&autoReconnect=true&tinyInt1isBit=false&useUnicode=true&characterEncoding=utf8username: rootpassword: 123456#多数据源
biz:datasource:url: jdbc:mysql://127.0.0.1:3306/biz?serverTimezone=UTC&useSSL=false&autoReconnect=true&tinyInt1isBit=false&useUnicode=true&characterEncoding=utf8username: rootpassword: 123456

java config 类

package com.xncoding.pos.config;import com.alibaba.druid.pool.DruidDataSource;
import com.baomidou.mybatisplus.plugins.PaginationInterceptor;
import com.xncoding.pos.common.mutidatesource.DSEnum;
import com.xncoding.pos.common.mutidatesource.DynamicDataSource;
import com.xncoding.pos.config.properties.DruidProperties;
import com.xncoding.pos.config.properties.MutiDataSourceProperties;
import org.mybatis.spring.annotation.MapperScan;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.transaction.annotation.EnableTransactionManagement;import java.sql.SQLException;
import java.util.HashMap;/*** MybatisPlus配置** @author xiongneng* @since 2017/5/20 21:58*/
@Configuration
@EnableTransactionManagement(order = 2)
@MapperScan(basePackages = {"com.xncoding.pos.common.dao.repository"})
public class MybatisPlusConfig {@AutowiredDruidProperties druidProperties;@AutowiredMutiDataSourceProperties mutiDataSourceProperties;/*** 核心数据源*/private DruidDataSource coreDataSource() {DruidDataSource dataSource = new DruidDataSource();druidProperties.config(dataSource);return dataSource;}/*** 另一个数据源*/private DruidDataSource bizDataSource() {DruidDataSource dataSource = new DruidDataSource();druidProperties.config(dataSource);mutiDataSourceProperties.config(dataSource);return dataSource;}/*** 单数据源连接池配置*/@Bean@ConditionalOnProperty(prefix = "xncoding", name = "muti-datasource-open", havingValue = "false")public DruidDataSource singleDatasource() {return coreDataSource();}/*** 多数据源连接池配置*/@Bean@ConditionalOnProperty(prefix = "xncoding", name = "muti-datasource-open", havingValue = "true")public DynamicDataSource mutiDataSource() {DruidDataSource coreDataSource = coreDataSource();DruidDataSource bizDataSource = bizDataSource();try {coreDataSource.init();bizDataSource.init();} catch (SQLException sql) {sql.printStackTrace();}DynamicDataSource dynamicDataSource = new DynamicDataSource();HashMap<Object, Object> hashMap = new HashMap<>();hashMap.put("dataSourceCore", coreDataSource);hashMap.put("dataSourceBiz", bizDataSource);dynamicDataSource.setTargetDataSources(hashMap);dynamicDataSource.setDefaultTargetDataSource(coreDataSource);return dynamicDataSource;}/*** mybatis-plus分页插件*/@Beanpublic PaginationInterceptor paginationInterceptor() {return new PaginationInterceptor();}
}

自定义一个注解：


/*** 多数据源标识** @author xiongneng* @since 2017年3月5日 上午9:44:24*/
@Inherited
@Retention(RetentionPolicy.RUNTIME)
@Target({ElementType.METHOD})
public @interface DataSource {String name() default "";
}

aop 捕获注解，切换数据源：

package com.xncoding.pos.common.aop;import com.xncoding.pos.common.annotion.DataSource;
import com.xncoding.pos.common.mutidatesource.DSEnum;
import com.xncoding.pos.common.mutidatesource.DataSourceContextHolder;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.Signature;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.core.Ordered;
import org.springframework.stereotype.Component;import java.lang.reflect.Method;/*** 多数据源切换的aop** @author xiongneng* @since 2017年3月5日 上午10:22:16*/
@Aspect
@Component
@ConditionalOnProperty(prefix = "xncoding", name = "muti-datasource-open", havingValue = "true")
public class MultiSourceExAop implements Ordered {private Logger log = LoggerFactory.getLogger(this.getClass());@Pointcut(value = "@annotation(com.xncoding.pos.common.annotion.DataSource)")private void cut() {}@Around("cut()")public Object around(ProceedingJoinPoint point) throws Throwable {Signature signature = point.getSignature();MethodSignature methodSignature = null;if (!(signature instanceof MethodSignature)) {throw new IllegalArgumentException("该注解只能用于方法");}methodSignature = (MethodSignature) signature;Object target = point.getTarget();Method currentMethod = target.getClass().getMethod(methodSignature.getName(), methodSignature.getParameterTypes());DataSource datasource = currentMethod.getAnnotation(DataSource.class);if (datasource != null) {DataSourceContextHolder.setDataSourceType(datasource.name());log.debug("设置数据源为：" + datasource.name());} else {DataSourceContextHolder.setDataSourceType("dataSourceCore");log.debug("设置数据源为：dataSourceCore");}try {return point.proceed();} finally {log.debug("清空数据源信息！");DataSourceContextHolder.clearDataSourceType();}}/*** aop的顺序要早于spring的事务*/@Overridepublic int getOrder() {return 1;}}