io.gridgo.connector.jdbc.JdbcProducer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of gridgo-jdbc Show documentation
There is a newer version: 0.8.1
package io.gridgo.connector.jdbc;

import static io.gridgo.connector.jdbc.JdbcConstants.OPERATION;
import static io.gridgo.connector.support.transaction.TransactionConstants.HEADER_CREATE_TRANSACTION;

import org.jdbi.v3.core.ConnectionFactory;
import org.jdbi.v3.core.Handle;
import org.jdbi.v3.core.Jdbi;
import org.joo.promise4j.Promise;
import org.joo.promise4j.impl.CompletableDeferredObject;

import io.gridgo.connector.jdbc.support.Helper;
import io.gridgo.connector.support.config.ConnectorContext;
import io.gridgo.framework.execution.impl.ExecutorExecutionStrategy;
import io.gridgo.framework.support.Message;
import lombok.Getter;
import lombok.extern.slf4j.Slf4j;

@Slf4j
@Getter
class JdbcProducer extends JdbcClient {

    private static final ExecutorExecutionStrategy DEFAULT_STRATEGY = new ExecutorExecutionStrategy(8);

    private Jdbi jdbiClient;

    JdbcProducer(ConnectorContext context, ConnectionFactory connectionFactory) {
        super(context);
        this.jdbiClient = Jdbi.create(connectionFactory);
    }

    protected Promise doCall(Message request,
            CompletableDeferredObject deferred, boolean isRPC) {
        var operationParam = request.headers().remove(OPERATION);
        var sqlStatement = request.body().asValue().getString();
        String operation = operationParam == null ? Helper.getOperation(sqlStatement).toLowerCase()
                : operationParam.asValue().getString();
        if (HEADER_CREATE_TRANSACTION.equals(operation)) {
            var result = beginTransaction(jdbiClient.open(), this.getContext());
            ack(deferred, result);
            return deferred == null ? null : deferred.promise();
        }
        var handler = findHandler(operation);
        var strategy = getContext().getProducerExecutionStrategy().orElse(DEFAULT_STRATEGY);
        strategy.execute(() -> {
            try (var handle = jdbiClient.open()) {
                var result = handler.handle(request, handle);
                ack(deferred, result);
            } catch (Exception ex) {
                log.error("Error while processing JDBC request", ex);
                ack(deferred, ex);
            }
        });
        return deferred == null ? null : deferred.promise();
    }

    private JdbcClientHandler findHandler(String operation) {
        var handler = operationsMap.get(operation);
        if (handler != null)
            return handler;
        return JdbcOperator::execute;
    }

    private Message beginTransaction(Handle handle, ConnectorContext context) {
        handle.begin();
        JdbcTransaction jdbcTransaction = new JdbcTransaction(handle, context);
        return Message.ofAny(jdbcTransaction);
    }

    @Override
    protected String generateName() {
        return "producer.jdbc";
    }
}