com.qiniu.process.qdora.QueryAvinfo Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of qsuits Show documentation
qiniu-suits is a efficient tools for qiniu api implemented by java8.
There is a newer version: 8.4.8
package com.qiniu.process.qdora;

import com.google.gson.JsonParseException;
import com.qiniu.common.QiniuException;
import com.qiniu.process.Base;
import com.qiniu.util.*;

import java.io.IOException;
import java.util.Map;

public class QueryAvinfo extends Base {

    private MediaManager mediaManager;
    private String domain;
    private String protocol;
    private String urlIndex;

    public QueryAvinfo(String domain, String protocol, String urlIndex, String rmPrefix, String savePath, int saveIndex)
            throws IOException {
        super("avinfo", "", "", null, null, rmPrefix, savePath, saveIndex);
        this.mediaManager = new MediaManager(configuration, protocol);
        set(protocol, domain, urlIndex);
    }

    public void updateQuery(String protocol, String domain, String urlIndex, String rmPrefix) throws IOException {
        set(protocol, domain, urlIndex);
        this.rmPrefix = rmPrefix;
    }

    private void set(String protocol, String domain, String urlIndex) throws IOException {
        if (urlIndex == null || "".equals(urlIndex)) {
            this.urlIndex = null;
            if (domain == null || "".equals(domain)) {
                throw new IOException("please set one of domain and urlIndex.");
            } else {
                RequestUtils.checkHost(domain);
                this.domain = domain;
                this.protocol = protocol == null || !protocol.matches("(http|https)") ? "http" : protocol;
            }
        } else {
            this.urlIndex = urlIndex;
        }
    }

    public QueryAvinfo(String domain, String protocol, String urlIndex, String rmPrefix, String savePath) throws IOException {
        this(domain, protocol, urlIndex, rmPrefix, savePath, 0);
    }

    public QueryAvinfo clone() throws CloneNotSupportedException {
        QueryAvinfo queryAvinfo = (QueryAvinfo)super.clone();
        queryAvinfo.mediaManager = new MediaManager(configuration, protocol);
        return queryAvinfo;
    }

    @Override
    protected Map formatLine(Map line) throws IOException {
        if (urlIndex == null) {
            line.put("key", FileNameUtils.rmPrefix(rmPrefix, line.get("key")));
            urlIndex = "url";
            line.put(urlIndex, protocol + "://" + domain + "/" + line.get("key").replaceAll("\\?", "%3F"));
        }
        return line;
    }

    @Override
    protected String resultInfo(Map line) {
        return line.get(urlIndex);
    }

    protected String singleResult(Map line) throws QiniuException {
        String avinfo = mediaManager.getAvinfoBody(line.get(urlIndex));
        if (avinfo != null && !"".equals(avinfo)) {
            // 由于响应的 body 为多行需经过格式化处理为一行字符串
            try {
                return JsonConvertUtils.toJson(avinfo);
            } catch (JsonParseException e) {
                throw new QiniuException(e);
            }
        } else {
            throw new QiniuException(null, "empty_result");
        }
    }
}