com.qiniu.entry.QSuitsEntry 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.entry;

import com.qiniu.common.Zone;
import com.qiniu.config.ParamsConfig;
import com.qiniu.datasource.FileInput;
import com.qiniu.datasource.IDataSource;
import com.qiniu.datasource.BucketList;
import com.qiniu.interfaces.IEntryParam;
import com.qiniu.storage.Configuration;

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

public class QSuitsEntry {

    private IEntryParam entryParam;
    private Configuration configuration;
    private CommonParams commonParams;
    private String source;
    private int unitLen;
    private int retryTimes;
    private int threads;
    private String savePath;
    private String saveTag;
    private boolean saveTotal;
    private String saveFormat;
    private String saveSeparator;
    private List rmFields;

    public QSuitsEntry(String[] args) throws IOException {
        setEntryParam(args);
        setConfiguration();
        setCommonParams(new CommonParams(entryParam));
        setCommons();
    }

    public QSuitsEntry(IEntryParam entryParam) throws IOException {
        this.entryParam = entryParam;
        setConfiguration();
        setCommonParams(new CommonParams(entryParam));
        setCommons();
    }

    public QSuitsEntry(IEntryParam entryParam, Configuration configuration) throws IOException {
        this.entryParam = entryParam;
        this.configuration = configuration;
        setCommonParams(new CommonParams(entryParam));
        setCommons();
    }

    private void setCommons() {
        source = commonParams.getSource();
        unitLen = commonParams.getUnitLen();
        retryTimes = commonParams.getRetryTimes();
        threads = commonParams.getThreads();
        savePath = commonParams.getSavePath();
        saveTotal = commonParams.getSaveTotal();
        saveFormat = commonParams.getSaveFormat();
        saveSeparator = commonParams.getSaveSeparator();
        rmFields = commonParams.getRmFields();
    }


    private void setEntryParam(String[] args) throws IOException {
        List configFiles = new ArrayList(){{
            add("resources" + System.getProperty("file.separator") + "qiniu.properties");
            add("resources" + System.getProperty("file.separator") + ".qiniu.properties");
        }};
        boolean paramFromConfig = true;
        if (args != null && args.length > 0) {
            if (args[0].startsWith("-config=")) configFiles.add(args[0].split("=")[1]);
            else paramFromConfig = false;
        }
        String configFilePath = null;
        if (paramFromConfig) {
            for (int i = configFiles.size() - 1; i >= 0; i--) {
                File file = new File(configFiles.get(i));
                if (file.exists()) {
                    configFilePath = configFiles.get(i);
                    break;
                }
            }
            if (configFilePath == null) throw new IOException("there is no config file detected.");
            else paramFromConfig = true;
        }
        entryParam = paramFromConfig ? new ParamsConfig(configFilePath) : new ParamsConfig(args);
    }

    private void setConfiguration() {
        this.configuration = new Configuration(Zone.autoZone());
        // 自定义超时时间
        configuration.connectTimeout = Integer.valueOf(entryParam.getValue("connect-timeout", "60"));
        configuration.readTimeout = Integer.valueOf(entryParam.getValue("read-timeout", "120"));
        configuration.writeTimeout = Integer.valueOf(entryParam.getValue("write-timeout", "60"));
    }

    public void setCommonParams(CommonParams commonParams) {
        this.commonParams = commonParams;
    }

    public IEntryParam getEntryParam() {
        return entryParam;
    }

    public Configuration getConfiguration() {
        return configuration;
    }

    public CommonParams getCommonParams() {
        return commonParams;
    }

    public IDataSource getDataSource() {
        if ("list".equals(source)) {
            return getBucketList();
        } else if ("file".equals(source)) {
            return getFileInput();
        } else {
            return null;
        }
    }

    public FileInput getFileInput() {
        String filePath = commonParams.getPath();
        String parseType = commonParams.getParse();
        String separator = commonParams.getSeparator();
        HashMap indexMap = commonParams.getIndexMap();
        FileInput fileInput = new FileInput(filePath, parseType, separator, indexMap, unitLen, threads, savePath);
        fileInput.setRetryTimes(retryTimes);
        fileInput.setResultOptions(saveTotal, saveFormat, saveSeparator, rmFields);
        fileInput.setSaveTag(saveTag);
        return fileInput;
    }

    public BucketList getBucketList() {
        String accessKey = commonParams.getAccessKey();
        String secretKey = commonParams.getSecretKey();
        String bucket = commonParams.getBucket();
        Map prefixesMap = commonParams.getPrefixesMap();
        List antiPrefixes = commonParams.getAntiPrefixes();
        boolean prefixLeft = commonParams.getPrefixLeft();
        boolean prefixRight = commonParams.getPrefixRight();
        BucketList bucketList = new BucketList(accessKey, secretKey, configuration, bucket, unitLen, prefixesMap,
                antiPrefixes, prefixLeft, prefixRight, threads, savePath);
        bucketList.setResultOptions(saveTotal, saveFormat, saveSeparator, rmFields);
        bucketList.setRetryTimes(retryTimes);
        bucketList.setSaveTag(saveTag);
        return bucketList;
    }
}