io.vertx.ext.asyncsql.impl.BaseSQLClient.java Source code

Introduction

Here is the source code for io.vertx.ext.asyncsql.impl.BaseSQLClient.java

Source

/*
 *  Copyright 2015 Red Hat, Inc.
 *
 *  All rights reserved. This program and the accompanying materials
 *  are made available under the terms of the Eclipse Public License v1.0
 *  and Apache License v2.0 which accompanies this distribution.
 *
 *  The Eclipse Public License is available at
 *  http://www.eclipse.org/legal/epl-v10.html
 *
 *  The Apache License v2.0 is available at
 *  http://www.opensource.org/licenses/apache2.0.php
 *
 *  You may elect to redistribute this code under either of these licenses.
 */

package io.vertx.ext.asyncsql.impl;

import com.github.mauricio.async.db.Configuration;
import com.github.mauricio.async.db.Connection;
import io.netty.buffer.PooledByteBufAllocator;
import io.vertx.core.AsyncResult;
import io.vertx.core.Future;
import io.vertx.core.Handler;
import io.vertx.core.Vertx;
import io.vertx.core.json.JsonObject;
import io.vertx.core.logging.Logger;
import io.vertx.core.logging.LoggerFactory;
import io.vertx.ext.asyncsql.impl.pool.AsyncConnectionPool;
import io.vertx.ext.sql.SQLConnection;
import scala.Option;
import scala.concurrent.ExecutionContext;
import scala.concurrent.duration.Duration;

import java.nio.charset.Charset;
import java.util.concurrent.TimeUnit;

/**
 * Base class for the SQL client.
 *
 * @author <a href="http://escoffier.me">Clement Escoffier</a>
 */
public abstract class BaseSQLClient {

    protected final Logger log = LoggerFactory.getLogger(this.getClass());
    protected final Vertx vertx;

    protected int maxPoolSize;
    protected int transactionTimeout;
    protected String registerAddress;

    public BaseSQLClient(Vertx vertx, JsonObject config) {
        this.vertx = vertx;
        this.maxPoolSize = config.getInteger("maxPoolSize", 10);
        this.transactionTimeout = config.getInteger("transactionTimeout", 500);
        this.registerAddress = config.getString("address");
    }

    protected abstract AsyncConnectionPool pool();

    protected abstract SQLConnection createFromPool(Connection conn, AsyncConnectionPool pool, ExecutionContext ec);

    public void getConnection(Handler<AsyncResult<SQLConnection>> handler) {
        pool().take(ar -> {
            if (ar.succeeded()) {
                final AsyncConnectionPool pool = pool();
                ExecutionContext ec = VertxEventLoopExecutionContext.create(vertx);
                handler.handle(Future.succeededFuture(createFromPool(ar.result(), pool, ec)));
            } else {
                handler.handle(Future.failedFuture(ar.cause()));
            }
        });
    }

    public void close(Handler<AsyncResult<Void>> handler) {
        log.info("Stopping async SQL client " + this);
        pool().close(ar -> {
            if (ar.succeeded()) {
                if (handler != null) {
                    handler.handle(Future.succeededFuture());
                }
            } else {
                if (handler != null) {
                    handler.handle(Future.failedFuture(ar.cause()));
                }
            }
        });
    }

    public void close() {
        close(null);
    }

    protected Configuration getConfiguration(String defaultHost, int defaultPort, String defaultDatabase,
            String defaultUser, String defaultPassword, String defaultCharset, long defaultConnectTimeout,
            long defaultTestTimeout, JsonObject config) {

        String host = config.getString("host", defaultHost);
        int port = config.getInteger("port", defaultPort);
        String username = config.getString("username", defaultUser);
        String password = config.getString("password", defaultPassword);
        String database = config.getString("database", defaultDatabase);
        Charset charset = Charset.forName(config.getString("charset", defaultCharset));
        long connectTimeout = config.getLong("connectTimeout", defaultConnectTimeout);
        long testTimeout = config.getLong("testTimeout", defaultTestTimeout);
        Long queryTimeout = config.getLong("queryTimeout");
        Option<Duration> queryTimeoutOption = (queryTimeout == null) ? Option.empty()
                : Option.apply(Duration.apply(queryTimeout, TimeUnit.MILLISECONDS));

        log.info("Creating configuration for " + host + ":" + port);
        return new Configuration(username, host, port, Option.apply(password), Option.apply(database), charset,
                16777216, PooledByteBufAllocator.DEFAULT, Duration.apply(connectTimeout, TimeUnit.MILLISECONDS),
                Duration.apply(testTimeout, TimeUnit.MILLISECONDS), queryTimeoutOption);
    }

}