From dc18841b475bc61da9f0a6df55a74ed7029602e9 Mon Sep 17 00:00:00 2001 From: Sergey Starosek Date: Tue, 5 Apr 2022 10:03:47 +0300 Subject: [PATCH] Add `sslMode` option lowercase alias For easier JDBC URL adaption [resolves #486] --- .../PostgresqlConnectionFactoryProvider.java | 18 +++++++++++++++++- ...esqlConnectionFactoryProviderUnitTests.java | 15 +++++++++++++++ 2 files changed, 32 insertions(+), 1 deletion(-) diff --git a/src/main/java/io/r2dbc/postgresql/PostgresqlConnectionFactoryProvider.java b/src/main/java/io/r2dbc/postgresql/PostgresqlConnectionFactoryProvider.java index 80b54f4f..867a8805 100644 --- a/src/main/java/io/r2dbc/postgresql/PostgresqlConnectionFactoryProvider.java +++ b/src/main/java/io/r2dbc/postgresql/PostgresqlConnectionFactoryProvider.java @@ -169,6 +169,11 @@ public final class PostgresqlConnectionFactoryProvider implements ConnectionFact */ public static final Option SSL_MODE = Option.valueOf("sslMode"); + /** + * Ssl mode alias (JDBC style). Default: disabled + */ + public static final Option SSL_MODE_ALIAS = Option.valueOf("sslmode"); + /** * SSL key password */ @@ -288,7 +293,18 @@ private static void setupSsl(PostgresqlConnectionConfiguration.Builder builder, return (SSLMode) it; - }).to(builder::sslMode); + }).to(builder::sslMode).otherwise(() -> { + + mapper.from(SSL_MODE_ALIAS).map(it -> { + + if (it instanceof String) { + return SSLMode.fromValue(it.toString()); + } + + return (SSLMode) it; + + }).to(builder::sslMode); + }); mapper.from(SSL_CERT).to(builder::sslCert); mapper.from(SSL_CONTEXT_BUILDER_CUSTOMIZER).to(builder::sslContextBuilderCustomizer); diff --git a/src/test/java/io/r2dbc/postgresql/PostgresqlConnectionFactoryProviderUnitTests.java b/src/test/java/io/r2dbc/postgresql/PostgresqlConnectionFactoryProviderUnitTests.java index 76d67523..d2def429 100644 --- a/src/test/java/io/r2dbc/postgresql/PostgresqlConnectionFactoryProviderUnitTests.java +++ b/src/test/java/io/r2dbc/postgresql/PostgresqlConnectionFactoryProviderUnitTests.java @@ -219,6 +219,21 @@ void supportsSslMode() { assertThat(sslConfig.getSslMode()).isEqualTo(SSLMode.DISABLE); } + @Test + void supportsSslModeAlias() { + PostgresqlConnectionFactory factory = this.provider.create(builder() + .option(DRIVER, POSTGRESQL_DRIVER) + .option(HOST, "test-host") + .option(PASSWORD, "test-password") + .option(USER, "test-user") + .option(Option.valueOf("sslmode"), "require") + .build()); + + SSLConfig sslConfig = factory.getConfiguration().getSslConfig(); + + assertThat(sslConfig.getSslMode()).isEqualTo(SSLMode.REQUIRE); + } + @Test void shouldCreateConnectionFactoryWithoutPassword() { assertThat(this.provider.create(ConnectionFactoryOptions.builder()