如何在Spring Boot的application.yml中正确使用Postgres风格URL

当然可以实现，以下是几种靠谱的方法，帮你像Python/Django、GoLang那样用postgres://username:secret@localhost/somedb这类单字符串URL配置Spring数据源：

方法1：利用PostgreSQL驱动自带的URL解析工具 #

PostgreSQL JDBC驱动提供了PGURL类，专门用来解析postgres://格式的URL，能直接提取出JDBC URL、用户名和密码，不用自己手动拆分字符串，避免处理端口默认值、特殊字符等边缘情况。

示例配置类：

import org.postgresql.util.PGURL;
import org.springframework.boot.jdbc.DataSourceBuilder;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import javax.sql.DataSource;

@Configuration
public class CustomPostgresDataSourceConfig {

    @Bean
    public DataSource dataSource() {
        // 可通过@Value("${spring.datasource.custom-url}")从配置文件注入
        String dbUrl = "postgres://username:secret@localhost/somedb";
        PGURL pgUrl = PGURL.parse(dbUrl);

        return DataSourceBuilder.create()
                .url(pgUrl.getJdbcUrl())
                .username(pgUrl.getUser())
                .password(pgUrl.getPassword())
                .driverClassName("org.postgresql.Driver")
                .build();
    }
}

然后在application.properties里定义自定义属性即可：

spring.datasource.custom-url=postgres://username:secret@localhost/somedb

方法2：自定义URL解析逻辑 #

如果不想依赖驱动的工具类，也可以自己用Java的URI类解析URL，手动转换为JDBC格式并提取凭证：

import org.springframework.boot.jdbc.DataSourceBuilder;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import javax.sql.DataSource;
import java.net.URI;
import java.net.URISyntaxException;

@Configuration
public class ManualDataSourceConfig {

    @Bean
    public DataSource dataSource() throws URISyntaxException {
        String dbUrl = "postgres://username:secret@localhost/somedb";
        URI dbUri = new URI(dbUrl);

        String[] userInfo = dbUri.getUserInfo().split(":");
        String username = userInfo[0];
        String password = userInfo[1];
        // 处理PostgreSQL默认端口5432
        int port = dbUri.getPort() == -1 ? 5432 : dbUri.getPort();
        String jdbcUrl = String.format("jdbc:postgresql://%s:%d%s", dbUri.getHost(), port, dbUri.getPath());

        return DataSourceBuilder.create()
                .url(jdbcUrl)
                .username(username)
                .password(password)
                .driverClassName("org.postgresql.Driver")
                .build();
    }
}

注意：如果密码里包含冒号，这种拆分方式会失效，此时推荐用方法1。

方法3：纯配置文件实现（不写代码） #

如果不想编写Java类，可以在配置文件中用Spring表达式（SpEL）直接解析URL，但这种方式仅适用于密码不含冒号的简单场景：

spring.datasource.custom-url=postgres://username:secret@localhost/somedb
spring.datasource.url=jdbc:postgresql://#{T(java.net.URI).create('${spring.datasource.custom-url}').host + ':' + (T(java.net.URI).create('${spring.datasource.custom-url}').port == -1 ? 5432 : T(java.net.URI).create('${spring.datasource.custom-url}').port) + T(java.net.URI).create('${spring.datasource.custom-url}').path}
spring.datasource.username=#{T(java.net.URI).create('${spring.datasource.custom-url}').userInfo.split(':')[0]}
spring.datasource.password=#{T(java.net.URI).create('${spring.datasource.custom-url}').userInfo.split(':')[1]}

以上三种方法都能满足你的需求，其中方法1最简洁可靠，推荐优先使用。

内容的提问来源于stack exchange，提问作者Mikayel