com.volcengine.tos.internal.TosResponse 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
The newest version!
package com.volcengine.tos.internal;

import com.fasterxml.jackson.annotation.JsonIgnore;
import com.volcengine.tos.comm.TosHeader;
import com.volcengine.tos.internal.util.StringUtils;
import com.volcengine.tos.model.RequestInfo;
import okio.Source;

import java.io.IOException;
import java.io.InputStream;
import java.io.Serializable;
import java.util.Collections;
import java.util.Map;

public class TosResponse implements AutoCloseable, Serializable {
    private int statusCode;
    private long contentLength;

    private Map headers = Collections.emptyMap();
    @JsonIgnore
    private transient InputStream inputStream;

    private transient Source source;

    public RequestInfo RequestInfo() {
        return new RequestInfo(this.getRequesID(), this.getID2(), this.getStatusCode(), this.headers);
    }

    public String getRequesID() {
        return headers.get(TosHeader.HEADER_REQUEST_ID.toLowerCase());
    }

    public String getID2() {
        return headers.get(TosHeader.HEADER_ID_2.toLowerCase());
    }

    public int getStatusCode() {
        return statusCode;
    }

    public TosResponse setStatusCode(int statusCode) {
        this.statusCode = statusCode;
        return this;
    }

    public long getContentLength() {
        return contentLength;
    }

    public TosResponse setContentLength(long contentLength) {
        this.contentLength = contentLength;
        return this;
    }

    public InputStream getInputStream() {
        return inputStream;
    }

    public long getInputStreamContentLength() throws IOException{
        return inputStream.available() == 0 ? -1 : inputStream.available();
    }
    public TosResponse setInputStream(InputStream inputStream) {
        this.inputStream = inputStream;
        return this;
    }


    public Map getHeaders() {
        return headers;
    }

    public String getHeaderWithKeyIgnoreCase(String key) {
        if (StringUtils.isEmpty(key) || this.headers.size() == 0) {
            return null;
        }
        return this.headers.get(key.toLowerCase());
    }

    public TosResponse setHeaders(Map headers) {
        this.headers = headers;
        return this;
    }

    protected Source getSource() {
        return source;
    }

    protected TosResponse setSource(Source source) {
        this.source = source;
        return this;
    }

    @Override
    public void close() throws IOException {
        if (inputStream != null) {
            inputStream.close();
        }
    }

    @Override
    public String toString() {
        return "TosResponse{" +
                "statusCode=" + statusCode +
                ", contentLength=" + contentLength +
                ", headers=" + headers +
                '}';
    }
}