com.huaweicloud.dws.client.executor.CopyUpsertExecutor Maven / Gradle / Ivy

Go to download
package com.huaweicloud.dws.client.executor;

import com.huaweicloud.dws.client.DwsConfig;
import com.huaweicloud.dws.client.exception.DwsClientException;
import com.huaweicloud.dws.client.model.Record;
import com.huaweicloud.dws.client.model.TableSchema;
import com.huaweicloud.dws.client.worker.ConnectionProvider;

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

/**
 * @ProjectName: dws-connector
 * @ClassName: CopyUpsertExecutor
 * @Description: copy + upsert 入库
 * @Date: 2023/2/6 15:57
 * @Version: 1.0
 */
public class CopyUpsertExecutor extends CopyMergeExecutor {
    private final UpsertFromExecutor upsertFromExecutor = new UpsertFromExecutor();

    public CopyUpsertExecutor(ConnectionProvider connectionProvider) {
        super(connectionProvider);
    }

    @Override
    protected void storeTarget(List records, Connection connection, DwsConfig config, TableSchema schema, String copyTable) throws DwsClientException {
        upsertFromExecutor.execute(records, connection, config, schema.getTableName().getFullName(), copyTable);
    }
}