springboot干货——（十一）多数据源配置之springdata

淡淡的烟草味﹌ 2021-09-26 12:34 374阅读 0赞

废话不多说，新建项目，项目结构如图：

![SouthEast][]

## 1.pom.xml ##

<?xml version="1.0" encoding="UTF-8"?>
    <project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
    	xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
    	<modelVersion>4.0.0</modelVersion>
    
    	<groupId>spring-boot</groupId>
    	<artifactId>spring-boot-datas-jpa</artifactId>
    	<version>0.0.1-SNAPSHOT</version>
    	<packaging>jar</packaging>
    
    	<name>spring-boot-datas-jpa</name>
    	<description>基于jpa的多数据源配置</description>
    
    	<parent>
    		<groupId>org.springframework.boot</groupId>
    		<artifactId>spring-boot-starter-parent</artifactId>
    		<version>1.5.9.RELEASE</version>
    		<relativePath/> 
    	</parent>
    
    	<properties>
    		<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
    		<project.reporting.outputEncoding>UTF-8</project.reporting.outputEncoding>
    		<java.version>1.8</java.version>
    	</properties>
    
    	<dependencies>
    		<dependency>
    			<groupId>org.springframework.boot</groupId>
    			<artifactId>spring-boot-starter</artifactId>
    		</dependency>
    
    		<dependency>
    			<groupId>org.springframework.boot</groupId>
    			<artifactId>spring-boot-starter-test</artifactId>
    			<scope>test</scope>
    		</dependency>
    		<dependency>
    			<groupId>mysql</groupId>
    			<artifactId>mysql-connector-java</artifactId>
    		</dependency>
    		<dependency>
    			<groupId>org.springframework.boot</groupId>
    			<artifactId>spring-boot-starter-data-jpa</artifactId>
    		</dependency>
    
    		<dependency>
    			<groupId>org.springframework.boot</groupId>
    			<artifactId>spring-boot-starter-web</artifactId>
    		</dependency>
    		<dependency>  
                <groupId>com.alibaba</groupId>  
                <artifactId>druid</artifactId>  
                <version>1.0.29</version>  
            </dependency>  
    	</dependencies>
    
    	<build>
    		<plugins>
    			<plugin>
    				<groupId>org.springframework.boot</groupId>
    				<artifactId>spring-boot-maven-plugin</artifactId>
    			</plugin>
    		</plugins>
    	</build>
    </project>

## 2.application.properties ##

spring.datasource.primary.url=jdbc:mysql://localhost:3306/test
    spring.datasource.primary.username=root
    spring.datasource.primary.password=19940315
    spring.datasource.primary.driver-class-name=com.mysql.jdbc.Driver
    
    spring.datasource.secondary.url=jdbc:mysql://localhost:3306/myweb
    spring.datasource.secondary.username=root
    spring.datasource.secondary.password=19940315
    spring.datasource.secondary.driver-class-name=com.mysql.jdbc.Driver

## 3.DataSourceConfig ##

package com.gwd.config;
    
    import javax.sql.DataSource;
    import org.springframework.beans.factory.annotation.Qualifier;
    import org.springframework.boot.context.properties.ConfigurationProperties;
    import org.springframework.context.annotation.Bean;
    import org.springframework.context.annotation.Configuration;
    import org.springframework.context.annotation.Primary;
    /** 
    * @ProjectName spring-boot-datas-jpa
    * @author JackHisen(gu.weidong) 
    * @Date 2018年2月16日 下午1:49:13    
    * @Version 1.0 
    * @Description:
    */
    @Configuration
    public class DataSourceConfig {
    
        @Bean(name = "primaryDataSource")
        @Qualifier("primaryDataSource")
        @ConfigurationProperties(prefix="spring.datasource.primary")
        public DataSource primaryDataSource() {
        	DataSource dataSource = new com.alibaba.druid.pool.DruidDataSource();
            return dataSource;
        }
    
        @Bean(name = "secondaryDataSource")
        @Qualifier("secondaryDataSource")
        @Primary
        @ConfigurationProperties(prefix="spring.datasource.secondary")
        public DataSource secondaryDataSource() {
        	DataSource dataSource = new com.alibaba.druid.pool.DruidDataSource();
            return dataSource;
        }
    }

## 4.PrimaryConfig ##

package com.gwd.config;
    
    import java.util.Map;
    import javax.persistence.EntityManager;
    import javax.sql.DataSource;
    import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.beans.factory.annotation.Qualifier;
    import org.springframework.boot.autoconfigure.orm.jpa.JpaProperties;
    import org.springframework.boot.orm.jpa.EntityManagerFactoryBuilder;
    import org.springframework.context.annotation.Bean;
    import org.springframework.context.annotation.Configuration;
    import org.springframework.context.annotation.Primary;
    import org.springframework.data.jpa.repository.config.EnableJpaRepositories;
    import org.springframework.orm.jpa.JpaTransactionManager;
    import org.springframework.orm.jpa.LocalContainerEntityManagerFactoryBean;
    import org.springframework.transaction.PlatformTransactionManager;
    import org.springframework.transaction.annotation.EnableTransactionManagement;
    /** 
    * @ProjectName spring-boot-datas-jpa
    * @author JackHisen(gu.weidong) 
    * @Date 2018年2月16日 下午1:30:08    
    * @Version 1.0 
    * @Description:
    */
    @Configuration
    @EnableTransactionManagement
    @EnableJpaRepositories(
            entityManagerFactoryRef="entityManagerFactoryPrimary",
            transactionManagerRef="transactionManagerPrimary",
            basePackages= { "com.gwd.domain.stu" }) //设置Repository所在位置
    public class PrimaryConfig {
    
        @Autowired 
        @Qualifier("primaryDataSource")
        private DataSource primaryDataSource;
    
        @Primary
        @Bean(name = "entityManagerPrimary")
        public EntityManager entityManager(EntityManagerFactoryBuilder builder) {
            return entityManagerFactoryPrimary(builder).getObject().createEntityManager();
        }
    
        @Primary
        @Bean(name = "entityManagerFactoryPrimary")
        public LocalContainerEntityManagerFactoryBean entityManagerFactoryPrimary (EntityManagerFactoryBuilder builder) {
            return builder
                    .dataSource(primaryDataSource)
                    .properties(getVendorProperties(primaryDataSource))
                    .packages("com.gwd.domain.stu") //设置实体类所在位置
                    .persistenceUnit("primaryPersistenceUnit")
                    .build();
        }
    
        @Autowired
        private JpaProperties jpaProperties;
    
        private Map<String, String> getVendorProperties(DataSource dataSource) {
            return jpaProperties.getHibernateProperties(dataSource);
        }
    
        @Primary
        @Bean(name = "transactionManagerPrimary")
        public PlatformTransactionManager transactionManagerPrimary(EntityManagerFactoryBuilder builder) {
            return new JpaTransactionManager(entityManagerFactoryPrimary(builder).getObject());
        }
    }

## 5.SecondaryConfig ##

package com.gwd.config;
    
    import java.util.Map;
    import javax.persistence.EntityManager;
    import javax.sql.DataSource;
    import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.beans.factory.annotation.Qualifier;
    import org.springframework.boot.autoconfigure.orm.jpa.JpaProperties;
    import org.springframework.boot.orm.jpa.EntityManagerFactoryBuilder;
    import org.springframework.context.annotation.Bean;
    import org.springframework.context.annotation.Configuration;
    import org.springframework.data.jpa.repository.config.EnableJpaRepositories;
    import org.springframework.orm.jpa.JpaTransactionManager;
    import org.springframework.orm.jpa.LocalContainerEntityManagerFactoryBean;
    import org.springframework.transaction.PlatformTransactionManager;
    import org.springframework.transaction.annotation.EnableTransactionManagement;
    
    /** 
    * @ProjectName spring-boot-datas-jpa
    * @author JackHisen(gu.weidong) 
    * @Date 2018年2月16日 下午1:37:48    
    * @Version 1.0 
    * @Description:
    */
    @Configuration
    @EnableTransactionManagement
    @EnableJpaRepositories(
            entityManagerFactoryRef="entityManagerFactorySecondary",
            transactionManagerRef="transactionManagerSecondary",
            basePackages= { "com.gwd.domain.user" }) //设置Repository所在位置
    public class SecondaryConfig {
    
        @Autowired 
        @Qualifier("secondaryDataSource")
        private DataSource secondaryDataSource;
    
        @Bean(name = "entityManagerSecondary")
        public EntityManager entityManager(EntityManagerFactoryBuilder builder) {
            return entityManagerFactorySecondary(builder).getObject().createEntityManager();
        }
    
        @Bean(name = "entityManagerFactorySecondary")
        public LocalContainerEntityManagerFactoryBean entityManagerFactorySecondary (EntityManagerFactoryBuilder builder) {
            return builder
                    .dataSource(secondaryDataSource)
                    .properties(getVendorProperties(secondaryDataSource))
                    .packages("com.gwd.domain.user") //设置实体类所在位置
                    .persistenceUnit("secondaryPersistenceUnit")
                    .build();
        }
    
        @Autowired
        private JpaProperties jpaProperties;
    
        private Map<String, String> getVendorProperties(DataSource dataSource) {
            return jpaProperties.getHibernateProperties(dataSource);
        }
    
        @Bean(name = "transactionManagerSecondary")
        PlatformTransactionManager transactionManagerSecondary(EntityManagerFactoryBuilder builder) {
            return new JpaTransactionManager(entityManagerFactorySecondary(builder).getObject());
        }
    }

## 6.model层 ##

Stu:

package com.gwd.domain.stu;
    
    import javax.persistence.Column;
    import javax.persistence.Entity;
    import javax.persistence.GeneratedValue;
    import javax.persistence.Id;
    @Entity
    public class Stu {
    	@Id  
        @GeneratedValue  
        private Integer id;
    	@Column(nullable = false)
        private String name;
    
        public Integer getId() {
            return id;
        }
    
        public void setId(Integer id) {
            this.id = id;
        }
    
        public String getName() {
            return name;
        }
    
        public void setName(String name) {
            this.name = name == null ? null : name.trim();
        }
    }

User:

package com.gwd.domain.user;
    
    import java.util.Arrays;
    import java.util.Date;
    import javax.persistence.Entity;
    import javax.persistence.GeneratedValue;
    import javax.persistence.Id;
    import javax.persistence.Table;
    @Entity
    @Table(name="users")//如果pojo与表名一致，则table注解可省略
    public class User {
    	@Id  
        @GeneratedValue  
        private Integer id;
    	
        private String username;
        
        private String password;
        
        private Date birthday;
        
        private String email;
        
        private String phonenum;
        
        private String address;
        
        private String ipAddress;
        
        private String lastIpAddress;
        
        private Date lastLogTime;
        
        private String sex;
        
        private String realName;
        
        private String idNum;
        
        private byte[] imageBase64decode;
        
        public Integer getId() {
            return id;
        }
    
        public void setId(Integer id) {
            this.id = id;
        }
    
        public String getUsername() {
            return username;
        }
    
        public void setUsername(String username) {
            this.username = username == null ? null : username.trim();
        }
    
        public String getPassword() {
            return password;
        }
    
        public void setPassword(String password) {
            this.password = password == null ? null : password.trim();
        }
    
        public Date getBirthday() {
            return birthday;
        }
    
        public void setBirthday(Date birthday) {
            this.birthday = birthday;
        }
    
        public String getEmail() {
            return email;
        }
    
        public void setEmail(String email) {
            this.email = email == null ? null : email.trim();
        }
    
        public String getPhonenum() {
            return phonenum;
        }
    
        public void setPhonenum(String phonenum) {
            this.phonenum = phonenum == null ? null : phonenum.trim();
        }
    
        public String getAddress() {
            return address;
        }
    
        public void setAddress(String address) {
            this.address = address == null ? null : address.trim();
        }
    
        public String getIpAddress() {
            return ipAddress;
        }
    
        public void setIpAddress(String ipAddress) {
            this.ipAddress = ipAddress == null ? null : ipAddress.trim();
        }
    
        public String getLastIpAddress() {
            return lastIpAddress;
        }
    
        public void setLastIpAddress(String lastIpAddress) {
            this.lastIpAddress = lastIpAddress == null ? null : lastIpAddress.trim();
        }
    
        public Date getLastLogTime() {
            return lastLogTime;
        }
    
        public void setLastLogTime(Date lastLogTime) {
            this.lastLogTime = lastLogTime;
        }
    
        public String getSex() {
            return sex;
        }
    
        public void setSex(String sex) {
            this.sex = sex == null ? null : sex.trim();
        }
    
        public String getRealName() {
            return realName;
        }
    
        public void setRealName(String realName) {
            this.realName = realName == null ? null : realName.trim();
        }
    
        public String getIdNum() {
            return idNum;
        }
    
        public void setIdNum(String idNum) {
            this.idNum = idNum == null ? null : idNum.trim();
        }
    
        public byte[] getImageBase64decode() {
            return imageBase64decode;
        }
    
        public void setImageBase64decode(byte[] imageBase64decode) {
            this.imageBase64decode = imageBase64decode;
        }
    
    	@Override
    	public String toString() {
    		return "User [id=" + id + ", username=" + username + ", password=" + password + ", birthday=" + birthday
    				+ ", email=" + email + ", phonenum=" + phonenum + ", address=" + address + ", ipAddress=" + ipAddress
    				+ ", lastIpAddress=" + lastIpAddress + ", lastLogTime=" + lastLogTime + ", sex=" + sex + ", realName="
    				+ realName + ", idNum=" + idNum + ", imageBase64decode=" + Arrays.toString(imageBase64decode) + "]";
    	}
    }

## 7.Repository ##

StuRepository:

package com.gwd.domain.stu;
    
    import org.springframework.data.jpa.repository.JpaRepository;
    
    /** 
    * @ProjectName spring-boot-datas-jpa
    * @author JackHisen(gu.weidong) 
    * @Date 2018年2月16日 下午1:50:35    
    * @Version 1.0 
    * @Description:
    */
    public interface StuRepository extends JpaRepository<Stu, Integer>{
    
    }

UserRepository:

package com.gwd.domain.user;
    
    import org.springframework.data.jpa.repository.JpaRepository;
    
    /** 
    * @ProjectName spring-boot-datas-jpa
    * @author JackHisen(gu.weidong) 
    * @Date 2018年2月16日 下午1:52:01    
    * @Version 1.0 
    * @Description:
    */
    public interface UserRepository extends JpaRepository<User, Integer>{
    
    }

## 8.Controller层 ##

【注意】jpa中一般用findOne而不是getOne，之前在使用getOne的时候报错，改用findOne后解决问题，具体原因如下：

/**
         * Retrieves an entity by its id.
         * 
         * @param id must not be {@literal null}.
         * @return the entity with the given id or {@literal null} if none found
         * @throws IllegalArgumentException if {@code id} is {@literal null}
         */
        T findOne(ID id);

/**
         * Returns a reference to the entity with the given identifier.
         * 
         * @param id must not be {@literal null}.
         * @return a reference to the entity with the given identifier.
         * @see EntityManager#getReference(Class, Object)
         */
        T getOne(ID id);

getOne是“Returns a reference“

package com.gwd.controller;
    
    import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.web.bind.annotation.RequestMapping;
    import org.springframework.web.bind.annotation.RestController;
    import com.gwd.domain.stu.Stu;
    import com.gwd.domain.stu.StuRepository;
    import com.gwd.domain.user.User;
    import com.gwd.domain.user.UserRepository;
    
    /** 
    * @ProjectName spring-boot-datas-jpa
    * @author JackHisen(gu.weidong) 
    * @Date 2018年2月16日 下午1:53:38    
    * @Version 1.0 
    * @Description:
    */
    @RestController
    public class TestController {
    	@Autowired
    	StuRepository stuReponsitory;
    	@Autowired
    	UserRepository userRepository;
    	
    	@RequestMapping("/testStu")
    	public String testStu() {
    		Stu stu = stuReponsitory.findOne(1);
    		return stu.getName();
    	}
    	
    	@RequestMapping("/testUser")
    	public String testUser() {
    		User user = userRepository.findOne(6);
    		return user.getUsername();
    	}
    }

[SouthEast]: /images/20210923/1ff4dca2ea0841b4966768c518be759c.png