org.panteleyev.mysqlapi.DataSourceBuilder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of java-api-for-mysql Show documentation
Annotation based Java API for MySQL database.
The newest version!
/*
 Copyright (c) Petr Panteleyev. All rights reserved.
 Licensed under the BSD license. See LICENSE file in the project root for full license information.
 */
package org.panteleyev.mysqlapi;

import javax.sql.DataSource;
import java.util.TimeZone;

/**
 * This class provides MySQL data source builder.
 */
public class DataSourceBuilder {
    private String host = "localhost";
    private int port = 3306;
    private String dbName;
    private String user;
    private String password;
    private String characterEncoding = "utf8";
    private String serverTimeZone = TimeZone.getDefault().getID();
    private boolean useSsl = false;
    private boolean allowPublicKeyRetrieval = true;

    /**
     * Builds datasource.
     *
     * @return datasource instance
     */
    public DataSource build() {
        try {
            // Using reflection to avoid hard dependency on MySQL connector jar. At least until it becomes a module.
            var dsClass = Class.forName("com.mysql.cj.jdbc.MysqlDataSource");
            var ds = (DataSource) dsClass.getDeclaredConstructor().newInstance();

            dsClass.getDeclaredMethod("setDatabaseName", String.class).invoke(ds, dbName);
            dsClass.getDeclaredMethod("setPort", int.class).invoke(ds, port);
            dsClass.getDeclaredMethod("setServerName", String.class).invoke(ds, host);
            dsClass.getDeclaredMethod("setUser", String.class).invoke(ds, user);
            dsClass.getDeclaredMethod("setPassword", String.class).invoke(ds, password);
            dsClass.getDeclaredMethod("setCharacterEncoding", String.class).invoke(ds, characterEncoding);
            dsClass.getDeclaredMethod("setUseSSL", boolean.class).invoke(ds, useSsl);
            dsClass.getDeclaredMethod("setAllowPublicKeyRetrieval", boolean.class).invoke(ds, allowPublicKeyRetrieval);
            dsClass.getDeclaredMethod("setServerTimezone", String.class).invoke(ds, serverTimeZone);
            return ds;
        } catch (Exception ex) {
            throw new RuntimeException(ex);
        }
    }

    /**
     * Sets database host. Default value is "localhost".
     *
     * @param host database host
     * @return {@code this}
     */
    public DataSourceBuilder host(String host) {
        this.host = host;
        return this;
    }

    /**
     * Sets database port. Default value is 3306.
     *
     * @param port database port
     * @return {@code this}
     */
    public DataSourceBuilder port(int port) {
        this.port = port;
        return this;
    }

    /**
     * Sets database name
     *
     * @param name database name
     * @return {@code this}
     */
    public DataSourceBuilder name(String name) {
        this.dbName = name;
        return this;
    }

    /**
     * Sets database user name.
     *
     * @param user database user
     * @return {@code this}
     */
    public DataSourceBuilder user(String user) {
        this.user = user;
        return this;
    }

    /**
     * Sets database password.
     *
     * @param password database password
     * @return {@code this}
     */
    public DataSourceBuilder password(String password) {
        this.password = password;
        return this;
    }

    /**
     * Sets character encoding. Default value is "utf8".
     *
     * @param characterEncoding character encoding
     * @return {@code this}
     */
    public DataSourceBuilder characterEncoding(String characterEncoding) {
        this.characterEncoding = characterEncoding;
        return this;
    }

    /**
     * Sets server timezone. Default value is {@code TimeZone.getDefault().getID()}.
     *
     * @param serverTimeZone server timezone
     * @return {@code this}
     */
    public DataSourceBuilder serverTimeZone(String serverTimeZone) {
        this.serverTimeZone = serverTimeZone;
        return this;
    }

    /**
     * Sets allowPublicKeyRetrieval option. Default value is {@code true}.
     *
     * @param allowPublicKeyRetrieval allowPublicKeyRetrieval option value
     * @return {@code this}
     */
    public DataSourceBuilder allowPublicKeyRetrieval(boolean allowPublicKeyRetrieval) {
        this.allowPublicKeyRetrieval = allowPublicKeyRetrieval;
        return this;
    }

    /**
     * Sets useSSL option. Default value is {@code false}.
     *
     * @param useSsl useSSL option value
     * @return {@code this}
     */
    public DataSourceBuilder useSsl(boolean useSsl) {
        this.useSsl = useSsl;
        return this;
    }
}