30个类手写Spring核心原理之动态数据源切换（8）

/**

• 还原当前切面的数据源

*/

public void restore() {

local.set(DEFAULT_SOURCE);

}

/**

• 设置已知名字的数据源

• @param dataSource

*/

public void set(String source) {

local.set(source);

}

/**

• 根据年份动态设置数据源

• @param year

*/

public void set(int year) {

local.set(“DB_” + year);

}

}

5 运行效果演示

---

5.1 创建Member实体类

创建Member实体类代码如下：

package com.tom.orm.demo.entity;

import lombok.Data;

import javax.persistence.Entity;

import javax.persistence.Id;

import javax.persistence.Table;

import java.io.Serializable;

@Entity

@Table(name=“t_member”)

@Data

public class Member implements Serializable {

@Id private Long id;

private String name;

private String addr;

private Integer age;

@Override

public String toString() {

return “Member{” +

“id=” + id +

“, name='” + name + ‘’’ +

“, addr='” + addr + ‘’’ +

“, age=” + age +

‘}’;

}

}

5.2 创建Order实体类

创建Order实体类代码如下：

package com.tom.orm.demo.entity;

import lombok.Data;

import javax.persistence.Column;

import javax.persistence.Entity;

import javax.persistence.Table;

import java.io.Serializable;

@Entity

@Table(name=“t_order”)

@Data

public class Order implements Serializable {

private Long id;

@Column(name=“mid”)

private Long memberId;

private String detail;

private Long createTime;

private String createTimeFmt;

@Override

public String toString() {

return “Order{” +

“id=” + id +

“, memberId=” + memberId +

“, detail='” + detail + ‘’’ +

“, createTime=” + createTime +

“, createTimeFmt='” + createTimeFmt + ‘’’ +

‘}’;

}

}

5.3 创建MemberDao

创建MemberDao代码如下：

package com.tom.orm.demo.dao;

import com.tom.orm.demo.entity.Member;

import com.tom.orm.framework.BaseDaoSupport;

import com.tom.orm.framework.QueryRule;

import org.springframework.stereotype.Repository;

import javax.annotation.Resource;

import javax.sql.DataSource;

import java.util.List;

@Repository

public class MemberDao extends BaseDaoSupport<Member,Long> {

@Override

protected String getPKColumn() {

return “id”;

}

@Resource(name=“dataSource”)

public void setDataSource(DataSource dataSource){

super.setDataSourceReadOnly(dataSource);

super.setDataSourceWrite(dataSource);

}

public List selectAll() throws Exception{

QueryRule queryRule = QueryRule.getInstance();

queryRule.andLike(“name”,“Tom%”);

return super.select(queryRule);

}

}

5.4 创建OrderDao

创建OrderDao代码如下：

package com.tom.orm.demo.dao;

import com.tom.orm.demo.entity.Order;

import com.tom.orm.framework.BaseDaoSupport;

import org.springframework.stereotype.Repository;

import javax.annotation.Resource;

import javax.core.common.jdbc.datasource.DynamicDataSource;

import javax.sql.DataSource;

import java.text.SimpleDateFormat;

import java.util.Date;

@Repository

public class OrderDao extends BaseDaoSupport<Order, Long> {

private SimpleDateFormat yearFormat = new SimpleDateFormat(“yyyy”);

private SimpleDateFormat fullDataFormat = new SimpleDateFormat(“yyyy-MM-dd HH:mm:ss”);

private DynamicDataSource dataSource;

@Override

protected String getPKColumn() {return “id”;}

@Resource(name=“dynamicDataSource”)

public void setDataSource(DataSource dataSource) {

this.dataSource = (DynamicDataSource)dataSource;

this.setDataSourceReadOnly(dataSource);

this.setDataSourceWrite(dataSource);

}

/**

• @throws Exception

*/

public boolean insertOne(Order order) throws Exception{

//约定优于配置

Date date = null;

if(order.getCreateTime() == null){

date = new Date();

order.setCreateTime(date.getTime());

}else {

date = new Date(order.getCreateTime());

}

Integer dbRouter = Integer.valueOf(yearFormat.format(date));

System.out.println(“自动分配到【DB_” + dbRouter + “】数据源”);

this.dataSource.getDataSourceEntry().set(dbRouter);

order.setCreateTimeFmt(fullDataFormat.format(date));

Long orderId = super.insertAndReturnId(order);

order.setId(orderId);

return orderId > 0;

}

}

5.5 修改db.properties文件

修改db.properties文件代码如下：

#sysbase database mysql config

#mysql.jdbc.driverClassName=com.mysql.jdbc.Driver

#mysql.jdbc.url=jdbc:mysql://127.0.0.1:3306/gp-vip-spring-db-demo?characterEncoding=UTF-8&rewriteBatchedStatements=true

#mysql.jdbc.username=root

#mysql.jdbc.password=123456

db2018.mysql.jdbc.driverClassName=com.mysql.jdbc.Driver

db2018.mysql.jdbc.url=jdbc:mysql://127.0.0.1:3306/gp-vip-spring-db-2018?characterEncoding=UTF-8&rewriteBatchedStatements=true

db2018.mysql.jdbc.username=root

db2018.mysql.jdbc.password=123456

db2019.mysql.jdbc.driverClassName=com.mysql.jdbc.Driver

db2019.mysql.jdbc.url=jdbc:mysql://127.0.0.1:3306/gp-vip-spring-db-2019?characterEncoding=UTF-8&rewriteBatchedStatements=true

db2019.mysql.jdbc.username=root

db2019.mysql.jdbc.password=123456

#alibaba druid config

dbPool.initialSize=1

dbPool.minIdle=1

dbPool.maxActive=200

dbPool.maxWait=60000

dbPool.timeBetweenEvictionRunsMillis=60000

dbPool.minEvictableIdleTimeMillis=300000

dbPool.validationQuery=SELECT ‘x’

dbPool.testWhileIdle=true

dbPool.testOnBorrow=false

dbPool.testOnReturn=false

dbPool.poolPreparedStatements=false

dbPool.maxPoolPreparedStatementPerConnectionSize=20

dbPool.filters=stat,log4j,wall

5.6 修改application-db.xml文件

修改application-db.xml文件代码如下：

5.7 编写测试用例

编写测试用例代码如下：

package com.tom.orm.test;

import com.tom.orm.demo.dao.MemberDao;

import com.tom.orm.demo.dao.OrderDao;

import com.tom.orm.demo.entity.Member;

import com.tom.orm.demo.entity.Order;

import org.junit.Ignore;

import org.junit.Test;

import org.junit.runner.RunWith;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.test.context.ContextConfiguration;

import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;

import java.text.SimpleDateFormat;

import java.util.Arrays;

import java.util.Date;

import java.util.List;

@ContextConfiguration(locations = {“classpath:application-context.xml”})

@RunWith(SpringJUnit4ClassRunner.class)

public class OrmTest {

private SimpleDateFormat sdf = new SimpleDateFormat(“yyyyMMddHHmmdd”);

@Autowired private MemberDao memberDao;

@Autowired private OrderDao orderDao;

@Test

public void testSelectAllForMember(){

try {

List result = memberDao.selectAll();

System.out.println(Arrays.toString(result.toArray()));

} catch (Exception e) {

e.printStackTrace();

}

}

@Test

@Ignore

public void testInsertMember(){

try {

for (int age = 25; age < 35; age++) {

Member member = new Member();

member.setAge(age);

member.setName(“Tom”);

member.setAddr(“Hunan Changsha”);

memberDao.insert(member);

}

}catch (Exception e){

e.printStackTrace();

}

}

@Test

// @Ignore

最后

自我介绍一下，小编13年上海交大毕业，曾经在小公司待过，也去过华为、OPPO等大厂，18年进入阿里一直到现在。

深知大多数Java工程师，想要提升技能，往往是自己摸索成长，自己不成体系的自学效果低效漫长且无助。

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

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，基本涵盖了95%以上Java开发知识点，不论你是刚入门Java开发的新手，还是希望在技术上不断提升的资深开发者，这些资料都将为你打开新的学习之门！

如果你觉得这些内容对你有帮助，需要这份全套学习资料的朋友可以戳我获取！！

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

memberDao.insert(member);

}

}catch (Exception e){

e.printStackTrace();

}

}

@Test

// @Ignore

最后

自我介绍一下，小编13年上海交大毕业，曾经在小公司待过，也去过华为、OPPO等大厂，18年进入阿里一直到现在。

深知大多数Java工程师，想要提升技能，往往是自己摸索成长，自己不成体系的自学效果低效漫长且无助。

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

[外链图片转存中…(img-xkSC3rrC-1715606566339)]

[外链图片转存中…(img-zvo61azQ-1715606566339)]

[外链图片转存中…(img-uW6m06Bx-1715606566340)]

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，基本涵盖了95%以上Java开发知识点，不论你是刚入门Java开发的新手，还是希望在技术上不断提升的资深开发者，这些资料都将为你打开新的学习之门！

如果你觉得这些内容对你有帮助，需要这份全套学习资料的朋友可以戳我获取！！

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