io.tarantool.driver.core.TarantoolConnectionFactory Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of cartridge-driver Show documentation

Tarantool Cartridge driver for Tarantool versions 1.10+ based on Netty framework

There is a newer version: 0.14.0

package io.tarantool.driver.core;

import io.netty.bootstrap.Bootstrap;
import io.netty.channel.Channel;
import io.netty.channel.ChannelFuture;
import io.netty.channel.ChannelFutureListener;
import io.tarantool.driver.TarantoolClientConfig;
import io.tarantool.driver.TarantoolVersionHolder;
import io.tarantool.driver.exceptions.TarantoolClientException;

import java.net.InetSocketAddress;
import java.util.Collection;
import java.util.concurrent.CompletableFuture;
import java.util.stream.Collectors;
import java.util.stream.Stream;

/**
 * Manages instantiation and connection procedures for Tarantool server connections
 *
 * @author Alexey Kuzin
 */
public class TarantoolConnectionFactory {

    private final TarantoolClientConfig config;
    private final Bootstrap bootstrap;

    /**
     * Basic constructor.
     * @param config Tarantool client config
     * @param bootstrap prepared Netty's bootstrap
     */
    public TarantoolConnectionFactory(TarantoolClientConfig config, Bootstrap bootstrap) {
        this.config = config;
        this.bootstrap = bootstrap;
    }

    /**
     * Create single connection and return connection future
     * @param serverAddress Tarantool server address to connect
     * @return connection future
     */
    public CompletableFuture singleConnection(InetSocketAddress serverAddress) {
        CompletableFuture connectionFuture = new CompletableFuture<>();
        RequestFutureManager requestManager = new RequestFutureManager(config);
        TarantoolVersionHolder versionHolder = new TarantoolVersionHolder();
        ChannelFuture future = bootstrap.clone()
                .handler(new TarantoolChannelInitializer(config, requestManager, versionHolder, connectionFuture))
                .remoteAddress(serverAddress).connect();
        future.addListener((ChannelFutureListener) f -> {
            //TODO reconnect when failed
            if (!f.isSuccess()) {
                connectionFuture.completeExceptionally(new TarantoolClientException(
                        "Failed to connect to the Tarantool server", f.cause()));
            }
        });
        return connectionFuture.thenApply(ch -> new TarantoolConnectionImpl(requestManager, versionHolder, ch));
    }

    /**
     * Create several connections and return their futures
     * @param serverAddress Tarantool server address to connect
     * @param connections number of connections to create
     * @return a collection with specified number of connection futures
     */
    public Collection> multiConnection(InetSocketAddress serverAddress,
                                                                              int connections) {
        return Stream.generate(() -> serverAddress)
                .map(this::singleConnection)
                .limit(connections)
                .collect(Collectors.toList());
    }
}