package com.fanruan.api.net.http.rs;

import com.fr.third.org.apache.http.HttpEntity;
import com.fr.third.org.apache.http.HttpStatus;
import com.fr.third.org.apache.http.client.methods.CloseableHttpResponse;
import com.fr.third.org.apache.http.util.EntityUtils;

import java.io.IOException;

/**
 * 上传响应解析器
 *
 * @author vito
 * @since 2019-07-14
 */
public class UploadResponseHandle extends BaseHttpResponseHandle<Void> {

    public static final UploadResponseHandle DEFAULT = new UploadResponseHandle();

    public UploadResponseHandle() {
    }

    public UploadResponseHandle(String encoding) {
        super(encoding);
    }

    @Override
    public Void parse(CloseableHttpResponse response) throws IOException {
        try {
            int statusCode = response.getStatusLine().getStatusCode();
            if (statusCode == HttpStatus.SC_OK) {
                HttpEntity entity = response.getEntity();
                if (entity != null) {
                    EntityUtils.consume(entity);
                }
            } else {
                HttpEntity entity = response.getEntity();
                String result = EntityUtils.toString(entity, getEncoding());
                throw new IOException("Connect error, error code:" + statusCode + "; message:" + result);
            }
        } finally {
            if (response != null) {
                response.close();
            }
        }
        return null;
    }
}