com.aliyun.datahub.model.PutRecordsRequest Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of aliyun-sdk-datahub Show documentation
There is a newer version: 2.25.6
package com.aliyun.datahub.model;

import com.aliyun.datahub.DatahubConstants;

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

public class PutRecordsRequest {

    private String projectName;
    private String topicName;
    private List records;
    private long requestLimitSize;
    private long approxSize;

    public PutRecordsRequest(String projectName, String topicName) {
        this.projectName = projectName;
        this.topicName = topicName;
        this.records = new ArrayList();
        this.requestLimitSize = DatahubConstants.DefaultLimitSize;
        this.approxSize = 0;
    }

    public PutRecordsRequest(String projectName, String topicName, List records) {
        this.projectName = projectName;
        this.topicName = topicName;
        this.records = records;
        this.requestLimitSize = DatahubConstants.DefaultLimitSize;
        this.approxSize = getRecordSize(records);
    }

    public PutRecordsRequest(String projectName, String topicName, List records, long limitSize) {
        this.projectName = projectName;
        this.topicName = topicName;
        this.records = records;
        this.requestLimitSize = limitSize;
        this.approxSize = getRecordSize(records);
    }

    private long getRecordSize(List recordEntries) {
        long size = 0;
        for (RecordEntry recordEntry: recordEntries) {
            size += recordEntry.getRecordSize();
        }
        return size;
    }

    public PutRecordsRequest(List records) {
        this.records = records;
    }

    public String getProjectName() {
        return projectName;
    }

    public String getTopicName() {
        return topicName;
    }

    public List getRecords() {
        return records;
    }

    public void setRecords(List records) {
        this.records = records;
    }

    @Deprecated
    public void addRecord(RecordEntry record) {
        this.records.add(record);
    }

    public boolean appendRecord(RecordEntry record) {
        long recordSize = record.getRecordSize();
        if (approxSize + recordSize > requestLimitSize * DatahubConstants.DefaultLimitPercent) {
            return false;
        }
        records.add(record);
        approxSize += recordSize;
        return true;
    }

    public long getRequestLimitSize() {
        return requestLimitSize;
    }

    public void setRequestLimitSize(long requestLimitSize) {
        this.requestLimitSize = requestLimitSize;
    }

    public long getApproxSize() {
        return approxSize;
    }
}