com.taosdata.jdbc.ws.Transport Maven / Gradle / Ivy

Go to download
package com.taosdata.jdbc.ws;

import com.taosdata.jdbc.ws.entity.Request;
import com.taosdata.jdbc.ws.entity.Response;

import java.sql.SQLException;
import java.util.concurrent.CompletableFuture;

/**
 * send message
 */
public class Transport implements AutoCloseable {

    public static final int DEFAULT_MAX_REQUEST = 100;
    public static final int DEFAULT_MESSAGE_WAIT_TIMEOUT = 3_000;

    private final WSClient client;
    private final InFlightRequest inFlightRequest;

    public Transport(WSClient client, InFlightRequest inFlightRequest) {
        this.client = client;
        this.inFlightRequest = inFlightRequest;
    }

    public CompletableFuture send(Request request) {
        CompletableFuture completableFuture = new CompletableFuture<>();
        try {
            inFlightRequest.put(new ResponseFuture(request.getAction(), request.id(), completableFuture));
            client.send(request.toString());
        } catch (Throwable t) {
            inFlightRequest.remove(request.getAction(), request.id());
            completableFuture.completeExceptionally(t);
        }
        return completableFuture;
    }

    public void sendWithoutRep(Request request) {
        client.send(request.toString());
    }

    public boolean isClosed() throws SQLException {
        return client.isClosed();
    }

    @Override
    public void close() {
        inFlightRequest.close();
        client.close();
    }

}