com.volcengine.tos.internal.taskman.ResumableCopyObjectTask Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of ve-tos-java-sdk Show documentation
The VolcEngine TOS SDK for Java
There is a newer version: 2.8.3
package com.volcengine.tos.internal.taskman;

import com.volcengine.tos.TosException;
import com.volcengine.tos.comm.event.*;
import com.volcengine.tos.internal.TosObjectRequestHandler;
import com.volcengine.tos.internal.util.ParamsChecker;
import com.volcengine.tos.model.object.*;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.util.concurrent.Callable;

public class ResumableCopyObjectTask extends TosTask implements TaskOutput {
    private ResumableCopyObjectCheckpoint checkpoint;
    private String copySourceSSECKey;
    private CopyPartInfo partInfo;
    private boolean enableCheckpoint;
    private String checkpointFile;
    private CopyEventListener copyEventListener;
    private TosObjectRequestHandler handler;
    private UploadPartCopyV2Output output;
    private ObjectMetaRequestOptions options;
    private long trafficLimit;

    public ResumableCopyObjectTask(ResumableCopyObjectCheckpoint checkpoint, int taskIdx) {
        ParamsChecker.ensureNotNull(checkpoint, "ResumableCopyObjectCheckpoint");
        ParamsChecker.ensureNotNull(checkpoint.getCopySourceObjectInfo(), "CopySourceObjectInfo");
        ParamsChecker.ensureNotNull(checkpoint.getCopyPartInfoList(), "CopyPartInfo list");
        this.checkpoint = checkpoint;
        this.partInfo = checkpoint.getCopyPartInfoList().get(taskIdx);
    }

    @Override
    public Callable> getCallableTask() {
        return () -> {
            if (partInfo.getPartNumber() == 1 && partInfo.getCopySourceRangeStart() == 0
            && partInfo.getCopySourceRangeEnd() == 0) {
                UploadPartV2Input input = UploadPartV2Input.builder()
                        .content(new ByteArrayInputStream("".getBytes())).contentLength(0)
                        .bucket(checkpoint.getBucket()).key(checkpoint.getKey()).partNumber(1)
                        .uploadID(checkpoint.getUploadID()).options(this.options).build();
                uploadNullPart(input);
                return this;
            }
            if (trafficLimit != 0) {
                if (options == null) {
                    options = new ObjectMetaRequestOptions();
                }
                options.setTrafficLimit(trafficLimit);
            }
            UploadPartCopyV2Input input = UploadPartCopyV2Input.builder()
                    .bucket(checkpoint.getBucket())
                    .key(checkpoint.getKey())
                    .sourceBucket(checkpoint.getSrcBucket())
                    .sourceKey(checkpoint.getSrcKey())
                    .sourceVersionID(checkpoint.getSrcVersionID())
                    .uploadID(checkpoint.getUploadID())
                    .partNumber(partInfo.getPartNumber())
                    .copySourceRange(partInfo.getCopySourceRangeStart(), partInfo.getCopySourceRangeEnd())
                    .copySourceIfMatch(checkpoint.getCopySourceIfMatch())
                    .copySourceIfModifiedSinceDate(checkpoint.getCopySourceIfModifiedSince())
                    .copySourceIfNoneMatch(checkpoint.getCopySourceIfNoneMatch())
                    .copySourceIfUnmodifiedSinceDate(checkpoint.getCopySourceIfUnModifiedSince())
                    .copySourceSSECAlgorithm(checkpoint.getCopySourceSSECAlgorithm())
                    .copySourceSSECKeyMD5(checkpoint.getCopySourceSSECKeyMD5())
                    .copySourceSSECKey(copySourceSSECKey)
                    .options(this.options).build();
            CopyEvent event = new CopyEvent().setBucket(checkpoint.getBucket()).setKey(checkpoint.getKey())
                    .setSrcBucket(checkpoint.getSrcBucket()).setSrcKey(checkpoint.getSrcKey())
                    .setSrcVersionID(checkpoint.getSrcVersionID()).setCheckpointFile(checkpointFile)
                    .setUploadID(checkpoint.getUploadID());
            try{
                this.output = this.handler.uploadPartCopy(input);
                this.partInfo.setCompleted(true);
                this.partInfo.setEtag(this.output.getEtag());
                if (this.enableCheckpoint) {
                    this.checkpoint.writeToFile(this.checkpointFile);
                }
                Util.postCopyEvent(this.copyEventListener, event.setCopyPartInfo(this.partInfo)
                        .setType(CopyEventType.CopyEventUploadPartCopySucceed));
            } catch (TosException e) {
                if (Util.needAbortTask(e.getStatusCode())) {
                    Util.postCopyEvent(this.copyEventListener, event.setException(e)
                            .setType(CopyEventType.CopyEventUploadPartCopyAborted));
                    throw e;
                } else {
                    Util.postCopyEvent(this.copyEventListener, event.setException(e)
                            .setType(CopyEventType.CopyEventUploadPartCopyFailed));
                }
            }
            return this;
        };
    }

    private void uploadNullPart(UploadPartV2Input input) throws IOException {
        CopyEvent event = new CopyEvent().setBucket(checkpoint.getBucket()).setKey(checkpoint.getKey())
                .setSrcBucket(checkpoint.getSrcBucket()).setSrcKey(checkpoint.getSrcKey())
                .setSrcVersionID(checkpoint.getSrcVersionID()).setCheckpointFile(checkpointFile)
                .setUploadID(checkpoint.getUploadID());
        try{
            UploadPartV2Output output = this.handler.uploadPart(input);
            this.output = new UploadPartCopyV2Output().etag(output.getEtag()).partNumber(1).requestInfo(output.getRequestInfo());
            this.partInfo.setCompleted(true);
            this.partInfo.setEtag(this.output.getEtag());
            if (this.enableCheckpoint) {
                this.checkpoint.writeToFile(this.checkpointFile);
            }
            Util.postCopyEvent(this.copyEventListener, event.setCopyPartInfo(this.partInfo)
                    .setType(CopyEventType.CopyEventUploadPartCopySucceed));
        } catch (TosException e) {
            if (Util.needAbortTask(e.getStatusCode())) {
                Util.postCopyEvent(this.copyEventListener, event.setException(e)
                        .setType(CopyEventType.CopyEventUploadPartCopyAborted));
                throw e;
            } else {
                Util.postCopyEvent(this.copyEventListener, event.setException(e)
                        .setType(CopyEventType.CopyEventUploadPartCopyFailed));
            }
        }
    }

    @Override
    public UploadPartCopyV2Output getOutput() {
        return this.output;
    }

    public ResumableCopyObjectCheckpoint getCheckpoint() {
        return checkpoint;
    }

    public ResumableCopyObjectTask setCheckpoint(ResumableCopyObjectCheckpoint checkpoint) {
        this.checkpoint = checkpoint;
        return this;
    }

    public boolean isEnableCheckpoint() {
        return enableCheckpoint;
    }

    public ResumableCopyObjectTask setEnableCheckpoint(boolean enableCheckpoint) {
        this.enableCheckpoint = enableCheckpoint;
        return this;
    }

    public String getCheckpointFile() {
        return checkpointFile;
    }

    public ResumableCopyObjectTask setCheckpointFile(String checkpointFile) {
        this.checkpointFile = checkpointFile;
        return this;
    }

    public CopyEventListener getCopyEventListener() {
        return copyEventListener;
    }

    public ResumableCopyObjectTask setCopyEventListener(CopyEventListener copyEventListener) {
        this.copyEventListener = copyEventListener;
        return this;
    }

    public TosObjectRequestHandler getHandler() {
        return handler;
    }

    public ResumableCopyObjectTask setHandler(TosObjectRequestHandler handler) {
        this.handler = handler;
        return this;
    }

    public ResumableCopyObjectTask setOutput(UploadPartCopyV2Output output) {
        this.output = output;
        return this;
    }

    public ObjectMetaRequestOptions getOptions() {
        return options;
    }

    public ResumableCopyObjectTask setOptions(ObjectMetaRequestOptions options) {
        this.options = options;
        return this;
    }

    public String getCopySourceSSECKey() {
        return copySourceSSECKey;
    }

    public ResumableCopyObjectTask setCopySourceSSECKey(String copySourceSSECKey) {
        this.copySourceSSECKey = copySourceSSECKey;
        return this;
    }

    public long getTrafficLimit() {
        return trafficLimit;
    }

    public ResumableCopyObjectTask setTrafficLimit(long trafficLimit) {
        this.trafficLimit = trafficLimit;
        return this;
    }
}