com.alicloud.openservices.tablestore.model.BulkImportRequest Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of tablestore Show documentation
There is a newer version: 5.17.5
package com.alicloud.openservices.tablestore.model;

import com.alicloud.openservices.tablestore.core.utils.Preconditions;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;

public class BulkImportRequest implements Request {
    private String tableName;
    private List rowChanges = new ArrayList();

    public BulkImportRequest(String tableName){
        Preconditions.checkArgument(
                tableName != null && !tableName.isEmpty(),
                "The name of table should not be null or empty.");
        this.tableName = tableName;
    }

    public void setTableName(String tableName) {
        Preconditions.checkArgument(
                tableName != null && !tableName.isEmpty(),
                "The name of table should not be null or empty.");
        this.tableName = tableName;
    }

    public String getTableName(){
        return tableName;
    }

    public String getOperationName() {
        return OperationNames.OP_BULK_IMPORT;
    }

    public void addRowChange(RowChange rowChange) {
        Preconditions.checkNotNull(rowChange, "The rowPutChange should not be null.");
        this.rowChanges.add(rowChange);
    }

    public void addRowChanges(List rowChanges){
        Preconditions.checkNotNull(rowChanges, "The rowPutChanges should not be null.");
        for (RowChange rowChange : rowChanges){
            Preconditions.checkNotNull(rowChange, String.format("The rowPutChange should not be null."));
            this.rowChanges.add(rowChange);
        }
    }

    public List getRowChange(){
        return rowChanges;
    }

    public RowChange getRowChange(int index){
        return rowChanges.get(index);
    }

    /**
     * 通过一次BulkImportRequest的返回结果，生成一个新的BulkImportRequest
     *
     * @return 新的BulkImportRequest请求
     */
    public BulkImportRequest createRequestForRetry(List failedRows) {
        Preconditions.checkArgument((failedRows != null) && !failedRows.isEmpty(), "failedRows can't be null or empty.");
        BulkImportRequest request = new BulkImportRequest(tableName);
        for (BulkImportResponse.RowResult rowResult : failedRows) {
            RowChange rowChange = getRowChange(rowResult.getIndex());
            if (rowChange == null) {
                throw new IllegalArgumentException("Can not find item in table '" + tableName + "' " +
                        "with index " + rowResult.getIndex());
            }
            request.addRowChange(rowChange);
        }
        return request;
    }
    /**
     * 判断该BulkImport请求中包含的总的行数是否为空。
     *
     * @return 行数是否为空
     */
    public boolean isEmpty() {
        return rowChanges.isEmpty();
    }

    /**
     * 获取该BulkImport请求中包含的总的行数。
     *
     * @return 总的行数
     */
    public int getRowsCount() {
        return rowChanges.size();
    }

}