Spring全家桶--單資料來源的配置

程式設計師阿牛發表於2021-08-22
Spring

前言

spring資料來源的配置網路上有很多例子，這裡我也來介紹一下單資料來源配置的例子，基於SpringBoot的方式和原生的Spring的方式。

一、生成專案骨架（SpringBoot），執行一個簡單的程式

訪問：https://start.spring.io/ ，選擇必要的依賴
file

下面我們先看下Application類的程式碼：

@SpringBootApplication
@Slf4j
public class SpringDatasourceApplication implements CommandLineRunner {

	@Autowired
	private DataSource dataSource;

	@Autowired
	private JdbcTemplate jdbcTemplate;

	public static void main(String[] args) {
		SpringApplication.run(SpringDatasourceApplication.class, args);
	}

	@Override
	public void run(String... args) throws Exception {
		showConnection();
		showData();
	}

	private void showConnection() throws SQLException {
		log.info("資料來源："+dataSource.toString());
		Connection conn = dataSource.getConnection();
		log.info("連線："+conn.toString());
		conn.close();
	}

	private void showData() {
		jdbcTemplate.queryForList("SELECT * FROM user")
				.forEach(row -> log.info("記錄："+row.toString()));
	}
}

application.properties檔案的配置項，我們可以看到我們使用的h2資料庫

management.endpoints.web.exposure.include=*
spring.output.ansi.enabled=ALWAYS

spring.datasource.url=jdbc:h2:mem:demodb
spring.datasource.username=sa
spring.datasource.password=

在資原始檔目錄，寫入兩個檔案，一個是data.sql、一個是schema.sql
file

schema.sql內容是：

CREATE TABLE user (ID INT IDENTITY, name VARCHAR(64),age INT);

data.sql內容是：

INSERT INTO user (ID,name,age) VALUES (1, '張三',18);
INSERT INTO user (ID, name,age) VALUES (2, '李四',19);

執行程式碼，結果如下：
file
其實我們並沒有去對DataSource進行bean配置，只是指定了資料庫的型別，載入了建表語句和初始化資料語句，可以看到連線池是Hikari，這也是springboot預設的連線池。
由於是使用的內建資料庫，我們可以在程式碼中

這也是因為springboot給我們自動裝配了我們所需要的資訊，由於我們引入了actuator，我們可以通過http://localhost:8080/actuator/beans 看到springboot幫我們裝載了很多的bean，有些可能是我們根本用不到的。下面我們講一下原生Spring方式怎麼實現配置資料來源。

二、選擇原生Spring方式配置資料來源

pom檔案配置內容：

  <dependency>
            <groupId>org.springframework</groupId>
            <artifactId>spring-context</artifactId>
            <version>${spring.version}</version>
        </dependency>
        <dependency>
            <groupId>org.springframework</groupId>
            <artifactId>spring-jdbc</artifactId>
            <version>${spring.version}</version>
        </dependency>
        <dependency>
            <groupId>org.apache.commons</groupId>
            <artifactId>commons-dbcp2</artifactId>
            <version>2.8.0</version>
        </dependency>
        <dependency>
            <groupId>com.h2database</groupId>
            <artifactId>h2</artifactId>
            <version>1.4.200</version>
            <scope>runtime</scope>
        </dependency>
		```
		
		**建立applicationContext.xml檔案，內容如下：**  
```xml
<?xml version="1.0" encoding="UTF-8"?>
<beans xmlns="http://www.springframework.org/schema/beans"
       xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
       xmlns:context="http://www.springframework.org/schema/context"
       xsi:schemaLocation="http://www.springframework.org/schema/beans
        http://www.springframework.org/schema/beans/spring-beans.xsd
        http://www.springframework.org/schema/context
        http://www.springframework.org/schema/context/spring-context.xsd">

    <context:component-scan base-package="com.xxx.xxxx" />
    <!--
    <bean id="dataSource" class="org.apache.commons.dbcp.BasicDataSource"
          destroy-method="close">
        <property name="driverClassName" value="org.h2.Driver"/>
        <property name="url" value="jdbc:h2:mem:testdb"/>
        <property name="username" value="SA"/>
        <property name="password" value=""/>
    </bean>
    -->
</beans>

** 自定義DataSource，這裡使用註解來實現（使用dbcp連線池） **

@Configuration
@EnableTransactionManagement
public class DataSourceDemo {
    @Autowired
    private DataSource dataSource;

    public static void main(String[] args) throws SQLException {
        ApplicationContext applicationContext =
                new ClassPathXmlApplicationContext("applicationContext*.xml");
        showBeans(applicationContext);
        dataSourceDemo(applicationContext);
    }

    @Bean(destroyMethod = "close")
    public DataSource dataSource() throws Exception {
        Properties properties = new Properties();
        properties.setProperty("driverClassName", "org.h2.Driver");
        properties.setProperty("url", "jdbc:h2:mem:testdb");
        properties.setProperty("username", "sa");
        return BasicDataSourceFactory.createDataSource(properties);
    }

    @Bean
    public PlatformTransactionManager transactionManager() throws Exception {
        return new DataSourceTransactionManager(dataSource());
    }

    private static void showBeans(ApplicationContext applicationContext) {
        System.out.println(Arrays.toString(applicationContext.getBeanDefinitionNames()));
    }

    private static void dataSourceDemo(ApplicationContext applicationContext) throws SQLException {
        DataSourceDemo demo = applicationContext.getBean("dataSourceDemo", DataSourceDemo.class);
        demo.showDataSource();
    }

    public void showDataSource() throws SQLException {
        System.out.println(dataSource.toString());
        Connection conn = dataSource.getConnection();
        System.out.println(conn.toString());
        conn.close();
    }
}

執行main方法：
file
可以看到可以實現和springboot一樣的效果

通過上面的兩個例子，我們可以看出SpringBoot幫我們實現瞭如下功能：

  • 通過DataSourceAutoConfiguration 配置 DataSource
  • 通過DataSourceTransactionManagerAutoConfiguration 配置 DataSourceTransactionManager
  • 通過JdbcTemplateAutoConfiguration 配置 JdbcTemplate
    當然上面是按需來配置的，如果我們在程式碼中已經配置了一個DataSource，SpringBoot不會再幫我們配置一個DataSource

在實際情況下，我們可能需要在應用中配置多個資料來源，下篇文章我將介紹多個資料來源的配置方式。

相關文章