io.tarantool.driver.core.metadata.ProxyMetadataProvider 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.metadata;

import io.tarantool.driver.api.SingleValueCallResult;
import io.tarantool.driver.api.TarantoolCallOperations;
import io.tarantool.driver.api.metadata.TarantoolMetadataContainer;
import io.tarantool.driver.api.metadata.TarantoolMetadataProvider;
import io.tarantool.driver.exceptions.TarantoolClientException;
import io.tarantool.driver.exceptions.TarantoolMetadataRequestException;
import io.tarantool.driver.mappers.CallResultMapper;
import io.tarantool.driver.mappers.converters.ValueConverter;
import org.msgpack.value.Value;

import java.util.concurrent.CompletableFuture;

/**
 * Provides spaces and index metadata via stored function call
 *
 * @author Alexey Kuzin
 */
public class ProxyMetadataProvider implements TarantoolMetadataProvider {

    private final String metadataFunctionName;
    private final TarantoolCallOperations client;
    private final
    CallResultMapper> mapper;

    /**
     * Basic constructor
     *
     * @param client               configured Tarantool client to make requests with
     * @param metadataFunctionName the stored function name
     * @param metadataConverter    converter to {@link TarantoolMetadataContainer}
     *                             aware of the function call result format
     * @param resultClass          result class
     */
    public ProxyMetadataProvider(
        TarantoolCallOperations client,
        String metadataFunctionName,
        ValueConverter metadataConverter,
        Class> resultClass) {
        this.metadataFunctionName = metadataFunctionName;
        this.client = client;
        this.mapper = client.getResultMapperFactoryFactory()
            .singleValueResultMapperFactory()
            .withSingleValueResultConverter(metadataConverter, resultClass);
    }

    @Override
    public CompletableFuture getMetadata() {
        return client.callForSingleResult(metadataFunctionName, mapper)
            .exceptionally(ex -> {
                if (ex.getCause() != null && ex.getCause() instanceof TarantoolClientException) {
                    throw new TarantoolMetadataRequestException(metadataFunctionName, ex);
                } else if (ex instanceof RuntimeException) {
                    throw (RuntimeException) ex;
                } else {
                    throw new RuntimeException(ex);
                }
            });
    }
}