com.huaweicloud.dws.client.handler.GetActionHandler Maven / Gradle / Ivy

Go to download

package com.huaweicloud.dws.client.handler;

import com.huaweicloud.dws.client.DwsConfig;
import com.huaweicloud.dws.client.action.GetAction;
import com.huaweicloud.dws.client.exception.DwsClientException;
import com.huaweicloud.dws.client.executor.GetExecutor;
import com.huaweicloud.dws.client.model.Record;
import com.huaweicloud.dws.client.model.TableSchema;
import com.huaweicloud.dws.client.op.Get;
import com.huaweicloud.dws.client.worker.ConnectionProvider;

import lombok.extern.slf4j.Slf4j;

import java.sql.Connection;
import java.sql.SQLException;
import java.util.List;
import java.util.Map;

/**
 * @ProjectName: dws-client
 * @Description:
 * @Date: 2023/09/28
 */
@Slf4j
public class GetActionHandler extends AbstractActionHandler {
    final GetExecutor getExecutor = new GetExecutor();

    private final DwsConfig config;

    public GetActionHandler(ConnectionProvider connectionProvider, DwsConfig config) {
        super(connectionProvider);
        this.config = config;
    }

    @Override
    public void handle(GetAction getAction) throws DwsClientException {
        List gets = getAction.getGetList();
        if (gets == null || gets.isEmpty()) {
            log.debug("current action gets is empty.");
            getAction.getFuture().complete(null);
            return;
        }
        try {
            doHandleGetAction(gets.get(0).getRecord().getTableSchema(), gets);
        } catch (SQLException e) {
            throw DwsClientException.fromException(e);
        }
    }

    private void doHandleGetAction(TableSchema schema, List recordList) throws SQLException {
        try {
            Connection connection = getConnection();
            Map resultMap = getExecutor.execute(schema, recordList, connection, config);
            // 结果传递
            for (Get get : recordList) {
                Record record = get.getRecord();
                if (get.getFuture() != null) {
                    Record result = resultMap.get(record.getRecordKey());
                    get.getFuture().complete(result);
                }
            }
        } catch (DwsClientException | SQLException e) {
            for (Get get : recordList) {
                if (get.getFuture() != null && !get.getFuture().isDone()) {
                    get.getFuture().completeExceptionally(e);
                }
            }
        }
    }
}