springboot 根据用户ID切换动态数据源代码实现

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpServletResponse;

/**

• file:DynamicDataSourceInterceptor

• 文件简要说明

• @author 2021-10-28 tarzan 创建初始版本

• @version V1.0 简要版本说明

*/

public class DynamicDataSourceInterceptor implements HandlerInterceptor {

private static final Logger log = LoggerFactory.getLogger(DynamicDataSourceInterceptor.class);

@Autowired

private MyDynamicDataSource dynamicDataSource;

@Autowired

private DynamicDataSourceCache dynamicDataSourceCache;

@Autowired

private DynamicDataSourceService dynamicDataSourceService;

@Override

public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) {

//获取当前登录用户信息

BladeUser user = AuthUtil.getUser();

if(user != null && user.getUserId() != null){

//如果未获取到 dsName 重新加载数据库

if(!dynamicDataSourceCache.hasDataSourceName(user.getUserId())){

dynamicDataSourceService.addUserDataSource(user.getUserId());

}

String dsName = dynamicDataSourceCache.getUserIdDataSourceName(user.getUserId());

if(!dynamicDataSource.switchDataSource(dsName)){

//如果切换数据源失败 返回错误

throw new RuntimeException(“未找到用户数据库”);

}

log.info(“数据源切换--------------用户名-----”+user.getUserName()+“------------>【{}】”, dsName);

}else{

log.info(“数据源切换------------------------------->默认数据源”);

dynamicDataSource.switchDefaultDataSource();

}

return true;

}

}

数据库设计

dynamicDataSource:

default:

url: jdbc:postgresql://$POSTGRE{S}_{H}OST:\mathrm{172.16.10.201}:${POSTGRES_PORT:5432}/${POSTGRES_DATABASE:gis_db}

username: ${POSTGRES_USERNAME:postgres}

password: ${POSTGRES_PASSWORD:postgres}

driverClassName: org.postgresql.Driver

pool:

#最小空闲连接

minimum-idle: 2

#最大连接

maximum-pool-size: 3

空闲连接存活最大时间，默认600000（10分钟）

idle-timeout: 1200000

据库连接超时时间,默认30秒

connection-timeout: 300000

初始链接一个基础数据库，放置用户表，数据源表，数据库表

数据库表建表语句

CREATE TABLE “public”.“data_server_database” (

“id” int8 NOT NULL,

“data_source_key” varchar(128) COLLATE “pg_catalog”.“default” NOT NULL,

“database_name” varchar(128) COLLATE “pg_catalog”.“default” NOT NULL,

“source_id” int8 NOT NULL,

“create_time” timestamp(6) NOT NULL DEFAULT CURRENT_TIMESTAMP,

“update_time” timestamp(6) NOT NULL DEFAULT CURRENT_TIMESTAMP,

“priority” int4 NOT NULL,

“amount” int4 NOT NULL DEFAULT 0,

“status” int2 NOT NULL DEFAULT 2

)

;

COMMENT ON COLUMN “public”.“data_server_database”.“id” IS ‘主键’;

COMMENT ON COLUMN “public”.“data_server_database”.“data_source_key” IS ‘数据源连接唯一key’;

COMMENT ON COLUMN “public”.“data_server_database”.“database_name” IS ‘数据库名’;

COMMENT ON COLUMN “public”.“data_server_database”.“source_id” IS ‘数据源id(data_server_source表主键id)’;

COMMENT ON COLUMN “public”.“data_server_database”.“create_time” IS ‘创建时间’;

COMMENT ON COLUMN “public”.“data_server_database”.“update_time” IS ‘更新时间’;

COMMENT ON COLUMN “public”.“data_server_database”.“priority” IS ‘数据库使用顺序(升序)’;

COMMENT ON COLUMN “public”.“data_server_database”.“amount” IS ‘数据使用用户数量’;

COMMENT ON COLUMN “public”.“data_server_database”.“status” IS ‘使用状态(1:正在使用；2：本库使用用户数已满)’;

COMMENT ON TABLE “public”.“data_server_database” IS ‘用户连接的数据库配置’;

---

– Uniques structure for table data_server_database

---

ALTER TABLE “public”.“data_server_database” ADD CONSTRAINT “source_key_unique” UNIQUE (“data_source_key”);

COMMENT ON CONSTRAINT “source_key_unique” ON “public”.“data_server_database” IS ‘数据源名 唯一’;

---

– Primary Key structure for table data_server_database

---

ALTER TABLE “public”.“data_server_database” ADD CONSTRAINT “data_server_source_pkey” PRIMARY KEY (“id”);

数据源表建表语句

CREATE TABLE “public”.“data_server_source” (

“id” int8 NOT NULL,

“driver_class_name” varchar(128) COLLATE “pg_catalog”.“default” NOT NULL,
自我介绍一下，小编13年上海交大毕业，曾经在小公司待过，也去过华为、OPPO等大厂，18年进入阿里一直到现在。

深知大多数Java工程师，想要提升技能，往往是自己摸索成长或者是报班学习，但对于培训机构动则几千的学费，着实压力不小。自己不成体系的自学效果低效又漫长，而且极易碰到天花板技术停滞不前！

因此收集整理了一份《2024年Java开发全套学习资料》，初衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友，同时减轻大家的负担。

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，基本涵盖了95%以上Java开发知识点，真正体系化！

由于文件比较大，这里只是将部分目录截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频，并且会持续更新！

如果你觉得这些内容对你有帮助，可以扫码获取！！（备注Java获取）

难道这样就够了吗？不，远远不够！

提前多熟悉阿里往年的面试题肯定是对面试有很大的帮助的，但是作为技术性职业，手里有实打实的技术才是你面对面试官最有用的利器，这是从内在散发出来的自信。

备战阿里时我花的最多的时间就是在学习技术上，占了我所有学习计划中的百分之70，这是一些我学习期间觉得还是很不错的一些学习笔记

我为什么要写这篇文章呢，其实我觉得学习是不能停下脚步的，在网络上和大家一起分享，一起讨论，不单单可以遇到更多一样的人，还可以扩大自己的眼界，学习到更多的技术，我还会在csdn、博客、掘金等网站上分享技术，这也是一种学习的方法。

今天就分享到这里了，谢谢大家的关注，以后会分享更多的干货给大家！

《一线大厂Java面试题解析+核心总结学习笔记+最新讲解视频+实战项目源码》，点击传送门即可获取！
到这里了，谢谢大家的关注，以后会分享更多的干货给大家！

[外链图片转存中…(img-XKm4Pqpm-1711821715838)]

[外链图片转存中…(img-icd2IlAR-1711821715838)]

[外链图片转存中…(img-pUXDwKoT-1711821715839)]

《一线大厂Java面试题解析+核心总结学习笔记+最新讲解视频+实战项目源码》，点击传送门即可获取！