BaseUploader.java

package com.qiniu.storage;

import com.qiniu.common.QiniuException;
import com.qiniu.http.Client;
import com.qiniu.http.Response;

public abstract class BaseUploader {

    protected final Client client;
    protected final String key;
    protected final String upToken;
    protected final ConfigHelper configHelper;
    protected final Configuration config;

    BaseUploader(Client client, String upToken, String key, Configuration config) {
        this.client = client;
        this.key = key;
        this.upToken = upToken;
        if (config == null) {
            this.config = Configuration.create();
        } else {
            this.config = config.clone();
        }
        this.configHelper = new ConfigHelper(this.config);
    }

    public Response upload() throws QiniuException {
        if (this.config == null) {
            throw QiniuException.unrecoverable("config can't be empty");
        }
        return uploadWithRegionRetry();
    }

    private Response uploadWithRegionRetry() throws QiniuException {
        Response response = null;
        QiniuException exception = null;
        while (true) {
            response = null;
            exception = null;

            try {
                response = uploadFlows();
            } catch (QiniuException e) {
                exception = e;
            }

            if (!Retry.canSwitchRegionAndRetry(response, exception)
                    || !couldReloadSource() || !reloadSource()) {
                break;
            }

            // context 过期，不需要切换 region
            if (response != null && response.isContextExpiredError()
                    || exception != null && exception.response != null && exception.response.isContextExpiredError()) {
                continue;
            }

            // 切换 region
            if (config.region == null || !config.region.switchRegion(new UploadToken(upToken))) {
                break;
            }
        }

        if (exception != null) {
            throw exception;
        }

        return response;
    }

    abstract Response uploadFlows() throws QiniuException;

    abstract boolean couldReloadSource();

    abstract boolean reloadSource();
}