com.qiniu.process.qdora.QiniuPfop 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.JsonArray;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import com.qiniu.config.JsonFile;
import com.qiniu.process.Base;
import com.qiniu.processing.OperationManager;
import com.qiniu.storage.Configuration;
import com.qiniu.util.*;

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

public class QiniuPfop extends Base> {

    private StringMap pfopParams;
    private List pfopConfigs;
    private String fopsIndex;
    private Configuration configuration;
    private OperationManager operationManager;

    public QiniuPfop(String accessKey, String secretKey, Configuration configuration, String bucket, String pipeline,
                     String pfopJsonPath, List pfopConfigs, String fopsIndex, String savePath,
                     int saveIndex) throws IOException {
        super("pfop", accessKey, secretKey, bucket, savePath, saveIndex);
        set(configuration, pipeline, pfopJsonPath, pfopConfigs, fopsIndex);
        this.operationManager = new OperationManager(Auth.create(accessKey, secretKey), configuration.clone());
        CloudApiUtils.checkQiniu(accessKey, secretKey, configuration, bucket);
    }

    public QiniuPfop(String accessKey, String secretKey, Configuration configuration, String bucket, String pipeline,
                     String pfopJsonPath, List pfopConfigs, String fopsIndex) throws IOException {
        super("pfop", accessKey, secretKey, bucket);
        set(configuration, pipeline, pfopJsonPath, pfopConfigs, fopsIndex);
        this.operationManager = new OperationManager(Auth.create(accessKey, secretKey), configuration.clone());
        CloudApiUtils.checkQiniu(accessKey, secretKey, configuration, bucket);
    }

    public QiniuPfop(String accessKey, String secretKey, Configuration configuration, String bucket, String pipeline,
                     String pfopJsonPath, List pfopConfigs, String fopsIndex, String savePath)
            throws IOException {
        this(accessKey, secretKey, configuration, bucket, pipeline, pfopJsonPath, pfopConfigs, fopsIndex, savePath, 0);
    }

    private void set(Configuration configuration, String pipeline, String pfopJsonPath, List pfopConfigs,
                     String fopsIndex) throws IOException {
        this.configuration = configuration;
        this.pfopParams = new StringMap().putNotEmpty("pipeline", pipeline);
        if (pfopConfigs != null && pfopConfigs.size() > 0) {
            this.pfopConfigs = pfopConfigs;
        } else if (pfopJsonPath != null && !"".equals(pfopJsonPath)) {
            this.pfopConfigs = new ArrayList<>();
            JsonFile jsonFile = new JsonFile(pfopJsonPath);
            JsonArray array = jsonFile.getElement("pfop").getAsJsonArray();
            for (JsonElement jsonElement : array) {
                JsonObject jsonObject = PfopUtils.checkPfopJson(jsonElement.getAsJsonObject(), false);
                this.pfopConfigs.add(jsonObject);
            }
        } else if (fopsIndex != null && !"".equals(fopsIndex)) {
            this.fopsIndex = fopsIndex;
        } else {
            throw new IOException("please set the pfop-config or fops-index.");
        }
    }

    public QiniuPfop clone() throws CloneNotSupportedException {
        QiniuPfop qiniuPfop = (QiniuPfop)super.clone();
        qiniuPfop.operationManager = new OperationManager(Auth.create(accessId, secretKey), configuration.clone());
        return qiniuPfop;
    }

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

    @Override
    protected String singleResult(Map line) throws Exception {
        String key = line.get("key");
        if (key == null) throw new IOException("key is not exists or empty in " + line);
        if (pfopConfigs != null && pfopConfigs.size() > 0) {
            StringBuilder cmdBuilder = new StringBuilder();
            for (JsonObject pfopConfig : pfopConfigs) {
                cmdBuilder.append(pfopConfig.get("cmd").getAsString())
                        .append("|saveas/")
                        .append(UrlSafeBase64.encodeToString(PfopUtils.generateFopSaveAs(key, pfopConfig)))
                        .append(";");
            }
            cmdBuilder.deleteCharAt(cmdBuilder.length() - 1);
            return String.join("\t", key, operationManager.pfop(bucket, key, cmdBuilder.toString(), pfopParams));
        } else {
            String fops = line.get(fopsIndex);
            if (fops == null) throw new IOException("fops is not exists or empty in " + line);
            return String.join("\t", key, operationManager.pfop(bucket, key, line.get(fopsIndex), pfopParams));
        }
    }

    @Override
    public void closeResource() {
        super.closeResource();
        pfopParams = null;
        pfopConfigs = null;
        fopsIndex = null;
        configuration = null;
        operationManager = null;
    }
}