基于域名的动态数据源切换实现教程

概述

这是一个基于Spring Boot的多数据源动态切换方案,通过解析请求的域名自动选择对应的数据源。

核心组件实现

1. 会话上下文管理 (SessionContext)

  • 使用 TransmittableThreadLocal 实现线程间数据传递
  • 提供统一的键值对存储接口
  • 在请求开始时清理旧数据,在结束时移除数据避免内存泄漏

2. 请求拦截器 (HeaderInterceptor)

  • 在 preHandle 阶段解析请求头中的 Origin
  • 提取域名并存储到 SessionContext 中
  • 调用 SessionContext.remove() 清理线程数据

3.动态数据源实现 (DynamicDataSource)

  • 继承 AbstractRoutingDataSource
  • 重写 determineCurrentLookupKey() 方法
  • 从 SessionContext 获取域名作为数据源标识

4.数据源配置 (DataSourceConfig)

  • 配置多个实际数据源(如 bjDataSource 和 cdDataSource)
  • 构建 targetDataSources 映射表,以域名为键关联具体数据源
  • 设置默认数据源

工作流程

  1. 请求进入 → HeaderInterceptor 解析 Origin 头部
  2. 域名提取 → 将域名存入 SessionContext
  3. 数据源路由 → DynamicDataSource 根据域名选择对应数据源
  4. 执行操作 → 使用选定的数据源执行数据库操作

会话上下文管理

package com.park.context; import cn.hutool.core.convert.Convert; import com.alibaba.ttl.TransmittableThreadLocal; import com.park.constants.SecurityConstants; import org.apache.commons.lang3.StringUtils; import java.util.Map; import java.util.Set; import java.util.concurrent.ConcurrentHashMap; public class SessionContext { private static final TransmittableThreadLocal<Map<String, Object>> THREAD_LOCAL = new TransmittableThreadLocal<>(); public static void set(String key, Object value) { Map<String, Object> map = getLocalMap(); map.put(key, value == null ? StringUtils.EMPTY : value); } public static String get(String key) { Map<String, Object> map = getLocalMap(); return Convert.toStr(map.getOrDefault(key, StringUtils.EMPTY)); } public static Map<String, Object> getLocalMap() { Map<String, Object> map = THREAD_LOCAL.get(); if (map == null) { map = new ConcurrentHashMap<>(); THREAD_LOCAL.set(map); } return map; } public static void setCompanyId(String companyId) { set(SecurityConstants.COMPANY_ID, companyId); } public static String getCompanyId() { return get(SecurityConstants.COMPANY_ID); } public static void setParkingLotId(Set<String> parkingLotIds) { set(SecurityConstants.PARKINGLOT_ID, parkingLotIds); } public static Set<String> getParkingLotId() { return get(SecurityConstants.PARKINGLOT_ID, Set.class); } public static <T> T get(String key, Class<T> clazz) { Map<String, Object> map = getLocalMap(); return cast(map.getOrDefault(key, null)); } public static <T> T cast(Object obj) { if (obj == null) { return null; } return (T) obj; } public static void remove() { THREAD_LOCAL.remove(); } }

核心动态数据源类

package com.park.source; import com.park.constants.SecurityConstants; import com.park.context.SessionContext; import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource; public class DynamicDataSource extends AbstractRoutingDataSource { @Override protected Object determineCurrentLookupKey() { System.out.println("当前数据源:" + SessionContext.get(SecurityConstants.DOMAIN)); return SessionContext.get(SecurityConstants.DOMAIN); } }

请求拦截器

package com.park.constants; public class SecurityConstants { public static final String AUTHORIZATION = "Authorization"; public static final String COMPANY_ID = "companyId"; public static final String PARKINGLOT_ID = "parkingLotIds"; public static final String DOMAIN = "domain"; } package com.park.interceptor; import com.park.constants.SecurityConstants; import com.park.context.SessionContext; import jakarta.servlet.http.HttpServletRequest; import jakarta.servlet.http.HttpServletResponse; import org.springframework.stereotype.Component; import org.springframework.web.servlet.AsyncHandlerInterceptor; import java.net.URI; import java.net.URISyntaxException; @Component public class DomainInterceptor implements AsyncHandlerInterceptor { @Override public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception { // 清除之前的会话数据 SessionContext.remove(); String origin = request.getHeader("Origin"); if (origin != null) { try { URI uri = new URI(origin); String domain = uri.getHost(); SessionContext.setDomain(domain); } catch (URISyntaxException e) { // 记录错误日志 } } return true; } @Override public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception { // 请求结束后清除线程本地变量 SessionContext.remove(); } } package com.park.config; import com.park.interceptor.DomainInterceptor; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.context.annotation.Configuration; import org.springframework.web.servlet.config.annotation.InterceptorRegistry; import org.springframework.web.servlet.config.annotation.WebMvcConfigurer; @Configuration public class WebConfig implements WebMvcConfigurer { @Autowired private DomainInterceptor domainInterceptor; @Override public void addInterceptors(InterceptorRegistry registry) { registry.addInterceptor(domainInterceptor) .addPathPatterns("/**") .excludePathPatterns("/login", "/public/**"); } }

数据源配置

spring: datasource: bj: jdbc-url: jdbc:postgresql://localhost:5432/bj_db username: your_username password: your_password driver-class-name: org.postgresql.Driver cd: jdbc-url: jdbc:postgresql://localhost:5432/cd_db username: your_username password: your_password driver-class-name: org.postgresql.Driver package com.park.config; import com.park.source.DynamicDataSource; import org.springframework.boot.context.properties.ConfigurationProperties; import org.springframework.boot.jdbc.DataSourceBuilder; import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Configuration; import org.springframework.context.annotation.Primary; import javax.sql.DataSource; import java.util.HashMap; import java.util.Map; @Configuration public class DataSourceConfig { @Bean @ConfigurationProperties(prefix = "spring.datasource.bj") public DataSource bjDataSource() { return DataSourceBuilder.create().build(); } @Bean @ConfigurationProperties(prefix = "spring.datasource.cd") public DataSource cdDataSource() { return DataSourceBuilder.create().build(); } @Bean @Primary public DataSource dynamicDataSource() { Map<Object, Object> targetDataSources = new HashMap<>(); targetDataSources.put("bj.park.com", bjDataSource()); targetDataSources.put("cd.park.com", cdDataSource()); DynamicDataSource dynamicDataSource = new DynamicDataSource(); dynamicDataSource.setTargetDataSources(targetDataSources); dynamicDataSource.setDefaultTargetDataSource(primaryDataSource()); return dynamicDataSource; } }

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/news/1156335.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

SPI控制器功能验证实践:基于iverilog的端到端流程

SPI控制器功能验证实践&#xff1a;从零构建基于Icarus Verilog的开源仿真流程 你有没有遇到过这样的场景&#xff1f;手头有个SPI控制器的RTL代码&#xff0c;想快速跑个仿真看看时序对不对&#xff0c;结果发现公司没有VCS许可证&#xff0c;ModelSim又太重启动慢&#xff0c…

零基础学习指南:STLink驱动安装全过程

手把手带你搞定 STLink 驱动安装&#xff1a;从识别失败到稳定调试的完整实战指南 你有没有遇到过这样的场景&#xff1f; 刚拿到一块崭新的 Nucleo 开发板&#xff0c;兴冲冲地插上电脑&#xff0c;打开 STM32CubeIDE&#xff0c;结果弹出一条令人崩溃的提示&#xff1a; “…

【毕业设计】SpringBoot+Vue+MySQL 信息化在线教学平台平台源码+数据库+论文+部署文档

&#x1f4a1;实话实说&#xff1a;CSDN上做毕设辅导的都是专业技术服务&#xff0c;大家都要生活&#xff0c;这个很正常。我和其他人不同的是&#xff0c;我有自己的项目库存&#xff0c;不需要找别人拿货再加价。我就是个在校研究生&#xff0c;兼职赚点饭钱贴补生活费&…

手把手教程:使用esptool实现加密固件烧录

破解固件安全困局&#xff1a;用esptool构建坚不可摧的加密烧录体系你有没有遇到过这样的情况&#xff1f;产品刚上市&#xff0c;市面上就出现了功能几乎一模一样的“孪生兄弟”——电路板不同&#xff0c;但行为一致。再一深挖&#xff0c;发现对方直接从你的设备里读出了Fla…

u8g2 OLED配置教程:手把手教你写第一行代码

手把手带你用u8g2点亮OLED&#xff1a;从零写出第一行显示代码你有没有过这样的经历&#xff1f;买了一块OLED屏&#xff0c;接上ESP32或STM32&#xff0c;打开Arduino IDE&#xff0c;却卡在“怎么让它亮起来”这一步&#xff1f;查资料发现一堆术语&#xff1a;IC、SSD1306、…

【2025最新】基于SpringBoot+Vue的房屋租赁管理系统管理系统源码+MyBatis+MySQL

&#x1f4a1;实话实说&#xff1a;CSDN上做毕设辅导的都是专业技术服务&#xff0c;大家都要生活&#xff0c;这个很正常。我和其他人不同的是&#xff0c;我有自己的项目库存&#xff0c;不需要找别人拿货再加价。我就是个在校研究生&#xff0c;兼职赚点饭钱贴补生活费&…

图解说明Keil MDK中ARM Compiler 5.06的编译输出流程

深入Keil MDK的构建心脏&#xff1a;图解ARM Compiler 5.06编译全过程你有没有遇到过这样的情况&#xff1f;代码明明编译通过了&#xff0c;下载到板子上却“一上电就进HardFault”&#xff1b;或者发现RAM莫名其妙溢出&#xff0c;查来查去才发现是printf偷偷引入了浮点库&am…

基于STM32F4的GPIO初始化STM32CubeMX教程实战案例

从零开始点亮LED&#xff1a;STM32F4 STM32CubeMX实战入门指南你有没有过这样的经历&#xff1f;手头一块崭新的STM32F4开发板&#xff0c;USB线插上&#xff0c;IDE打开&#xff0c;却卡在第一步——怎么让一个最简单的LED闪烁起来&#xff1f;别急。这并不是你基础差&#x…

Multisim14.0交流小信号分析操作指南:通俗解释

深入理解Multisim14.0中的交流小信号分析&#xff1a;从原理到实战的完整指南在模拟电路设计中&#xff0c;我们常常需要回答这样一个问题&#xff1a;这个放大器到底能跑多快&#xff1f;它对高频信号会不会“听不清”&#xff1f;滤波器的截止频率真的如计算所示吗&#xff1…

I2C HID协议时序分析:实战案例解析

I2C HID协议时序实战解析&#xff1a;从波形到代码的全链路拆解一个触控失灵的早晨上周三早上&#xff0c;我刚泡好咖啡&#xff0c;测试同事就冲进办公室&#xff1a;“新批次的平板开机十分钟&#xff0c;触控突然卡死&#xff0c;日志里全是NACK错误。”我们立刻调出内核日志…

AUTOSAR经典平台入门:ECU抽象层全面讲解

AUTOSAR经典平台入门&#xff1a;深入理解ECU抽象层的“软硬桥梁”作用你有没有遇到过这样的场景&#xff1f;一个原本在英飞凌TC3xx平台上运行良好的刹车踏板检测模块&#xff0c;因为项目换用了NXP S32K芯片&#xff0c;结果整个ADC采集代码几乎要重写一遍——引脚变了、寄存…

企业级个人理财系统管理系统源码|SpringBoot+Vue+MyBatis架构+MySQL数据库【完整版】

摘要 随着社会经济的发展和人们生活水平的提高&#xff0c;个人理财需求日益增长&#xff0c;传统的理财方式已无法满足现代人对高效、便捷、安全的财务管理需求。尤其是在企业环境中&#xff0c;员工和企业的财务数据管理需要更加系统化和智能化。企业级个人理财系统能够整合个…

前后端分离论坛网站系统|SpringBoot+Vue+MyBatis+MySQL完整源码+部署教程

摘要 随着互联网技术的快速发展&#xff0c;论坛网站作为信息交流的重要平台&#xff0c;其功能需求和用户体验日益受到重视。传统的前后端耦合架构在开发效率和维护成本上存在较大局限性&#xff0c;难以满足现代论坛系统的高并发、高可扩展性需求。前后端分离架构通过将前端展…

74194双向移位时序分析:超详细版时序图讲解

74194双向移位时序图精讲&#xff1a;从波形到实战的深度拆解你有没有遇到过这样的情况&#xff1f;明明控制信号都接对了&#xff0c;时钟也稳定输出&#xff0c;可数据就是“走偏”——LED流水灯不按预期方向流动&#xff0c;或者并行加载的数据一进芯片就错位。问题很可能出…

XADC IP核在嵌入式监控中的项目应用

XADC&#xff1a;让FPGA学会“自我感知”的关键技术实战你有没有遇到过这样的情况&#xff1f;系统运行着好好的&#xff0c;突然就死机了。重启之后一切正常&#xff0c;但几天后又莫名其妙地宕机。查日志、看代码、测信号——全都对得上&#xff0c;就是找不到根因。如果你用…

什么是营销管理系统,一文说清:定义、功能、选型、产品推荐

在数字化营销成为企业标配的今天&#xff0c;“营销管理系统”已成为高频词汇&#xff0c;但很多企业对其认知仍停留在“简单的客户管理工具”层面。实际上&#xff0c;一套成熟的营销管理系统能打通“获客-培育-转化-复盘”全链路&#xff0c;让营销从“零散操作”升级为“系统…

基于SpringBoot+Vue的养老智慧服务平台管理系统设计与实现【Java+MySQL+MyBatis完整源码】

&#x1f4a1;实话实说&#xff1a;CSDN上做毕设辅导的都是专业技术服务&#xff0c;大家都要生活&#xff0c;这个很正常。我和其他人不同的是&#xff0c;我有自己的项目库存&#xff0c;不需要找别人拿货再加价。我就是个在校研究生&#xff0c;兼职赚点饭钱贴补生活费&…

BL370 为什么原生支持 Docker?这是为工业现场提前铺好的路

在工业互联网、储能 EMS、机器人、视觉检测这些场景里&#xff0c;你会发现一个趋势&#xff1a;边缘计算控制器正在变得越来越像一台“小型工业服务器”。协议要转、业务要跑、AI 要推理、前端要展示、还要远程维护……如果全塞进系统底层&#xff0c;不仅开发慢&#xff0c;后…

Java Web 游戏销售平台系统源码-SpringBoot2+Vue3+MyBatis-Plus+MySQL8.0【含文档】

摘要 近年来&#xff0c;随着互联网技术的迅猛发展和游戏产业的持续繁荣&#xff0c;游戏销售平台逐渐成为玩家获取游戏资源的重要渠道。传统的游戏销售模式受限于线下渠道和单一的交易方式&#xff0c;难以满足玩家多样化的需求。数字化游戏销售平台的兴起为玩家提供了便捷的购…

做小红书 3 年,我终于悟了:废掉你账号的不是内容,而是那张“丑封面”(附 01Agent 实操避坑指南)

这十年来&#xff0c;我见过无数个深夜对着电脑屏幕薅头发的灵魂。特别是最近两年做小红书的朋友&#xff0c;经常跟我吐槽同一个痛点&#xff1a;“明明我的内容干货满满&#xff0c;写了三个小时&#xff0c;结果发出去只有几十个小眼睛。隔壁那个号&#xff0c;内容水得要命…