org.mariadb.r2dbc.MariadbConnectionFactoryProvider Maven / Gradle / Ivy

Go to download
// SPDX-License-Identifier: Apache-2.0
// Copyright (c) 2020-2024 MariaDB Corporation Ab

package org.mariadb.r2dbc;

import static io.r2dbc.spi.ConnectionFactoryOptions.DRIVER;

import io.netty.handler.ssl.SslContextBuilder;
import io.r2dbc.spi.ConnectionFactoryOptions;
import io.r2dbc.spi.ConnectionFactoryProvider;
import io.r2dbc.spi.Option;
import java.util.function.UnaryOperator;
import org.mariadb.r2dbc.util.Assert;
import reactor.netty.resources.LoopResources;

public final class MariadbConnectionFactoryProvider implements ConnectionFactoryProvider {
  public static final String MARIADB_DRIVER = "mariadb";
  public static final Option SOCKET = Option.valueOf("socket");
  public static final Option ALLOW_MULTI_QUERIES = Option.valueOf("allowMultiQueries");
  public static final Option TLS_PROTOCOL = Option.valueOf("tlsProtocol");
  public static final Option SERVER_SSL_CERT = Option.valueOf("serverSslCert");
  public static final Option CLIENT_SSL_CERT = Option.valueOf("clientSslCert");
  public static final Option CLIENT_SSL_KEY = Option.valueOf("clientSslKey");
  public static final Option CLIENT_SSL_PWD = Option.valueOf("clientSslPassword");
  public static final Option COLLATION = Option.valueOf("collation");
  public static final Option TIMEZONE = Option.valueOf("timezone");

  public static final Option ALLOW_PIPELINING = Option.valueOf("allowPipelining");
  public static final Option USE_SERVER_PREPARE = Option.valueOf("useServerPrepStmts");
  public static final Option ISOLATION_LEVEL = Option.valueOf("isolationLevel");
  public static final Option AUTO_COMMIT = Option.valueOf("autocommit");
  public static final Option PERMIT_REDIRECT = Option.valueOf("permitRedirect");
  public static final Option SKIP_POST_COMMANDS = Option.valueOf("skipPostCommands");
  public static final Option TINY_IS_BIT = Option.valueOf("tinyInt1isBit");
  public static final Option PREPARE_CACHE_SIZE = Option.valueOf("prepareCacheSize");
  public static final Option SSL_MODE = Option.valueOf("sslMode");
  public static final Option TRANSACTION_REPLAY = Option.valueOf("transactionReplay");
  public static final Option HAMODE = Option.valueOf("haMode");

  public static final Option CONNECTION_ATTRIBUTES = Option.valueOf("connectionAttributes");
  public static final Option PAM_OTHER_PASSWORD = Option.valueOf("pamOtherPwd");
  public static final Option TCP_KEEP_ALIVE = Option.valueOf("tcpKeepAlive");
  public static final Option TCP_ABORTIVE_CLOSE = Option.valueOf("tcpAbortiveClose");
  public static final Option SESSION_VARIABLES = Option.valueOf("sessionVariables");
  public static final Option LOOP_RESOURCES = Option.valueOf("loopResources");
  public static final Option ALLOW_PUBLIC_KEY_RETRIEVAL =
      Option.valueOf("allowPublicKeyRetrieval");

  public static final Option CACHING_RSA_PUBLIC_KEY = Option.valueOf("cachingRsaPublicKey");
  public static final Option RSA_PUBLIC_KEY = Option.valueOf("rsaPublicKey");

  public static final Option RESTRICTED_AUTH = Option.valueOf("restrictedAuth");

  public static final Option> SSL_CONTEXT_BUILDER_CUSTOMIZER =
      Option.valueOf("sslContextBuilderCustomizer");
  public static final Option SSL_TUNNEL_DISABLE_HOST_VERIFICATION =
      Option.valueOf("sslTunnelDisableHostVerification");

  static MariadbConnectionConfiguration createConfiguration(
      ConnectionFactoryOptions connectionFactoryOptions) {
    Assert.requireNonNull(connectionFactoryOptions, "connectionFactoryOptions must not be null");
    return MariadbConnectionConfiguration.fromOptions(connectionFactoryOptions).build();
  }

  @Override
  public MariadbConnectionFactory create(ConnectionFactoryOptions connectionFactoryOptions) {
    return new MariadbConnectionFactory(createConfiguration(connectionFactoryOptions));
  }

  @Override
  public String getDriver() {
    return MARIADB_DRIVER;
  }

  @Override
  public boolean supports(ConnectionFactoryOptions connectionFactoryOptions) {
    Assert.requireNonNull(connectionFactoryOptions, "connectionFactoryOptions must not be null");

    String driver = (String) connectionFactoryOptions.getValue(DRIVER);
    return MARIADB_DRIVER.equals(driver);
  }
}