smart.git - Gitblit

package com.qianwen.core.oss;
 
import java.io.IOException;
import java.io.InputStream;
import java.util.Date;
import java.util.List;
 
import org.springframework.web.multipart.MultipartFile;
 
import com.qianwen.core.oss.model.BladeFile;
import com.qianwen.core.oss.model.OssFile;
import com.qianwen.core.oss.props.OssProperties;
import com.qianwen.core.oss.rule.OssRule;
import com.qianwen.core.tool.utils.CollectionUtil;
import com.qianwen.core.tool.utils.Func;
import com.qiniu.common.QiniuException;
import com.qiniu.common.Zone;
import com.qiniu.http.Response;
import com.qiniu.storage.BucketManager;
import com.qiniu.storage.UploadManager;
import com.qiniu.storage.model.FileInfo;
import com.qiniu.util.Auth;
import com.qiniu.util.StringMap;
 
public class QiniuTemplate implements OssTemplate {
    private final Auth auth;
    private final UploadManager uploadManager;
    private final BucketManager bucketManager;
    private final OssProperties ossProperties;
    private final OssRule ossRule;
 
    public QiniuTemplate(final Auth auth, final UploadManager uploadManager, final BucketManager bucketManager, final OssProperties ossProperties, final OssRule ossRule) {
        this.auth = auth;
        this.uploadManager = uploadManager;
        this.bucketManager = bucketManager;
        this.ossProperties = ossProperties;
        this.ossRule = ossRule;
    }
 
    @Override 
    public void makeBucket(String bucketName) {
        try {
            if (!CollectionUtil.contains(this.bucketManager.buckets(), getBucketName(bucketName))) {
                this.bucketManager.createBucket(getBucketName(bucketName), Zone.autoZone().getRegion());
            }
        }catch(QiniuException ex) {
            throw new RuntimeException(ex);
        }
    }
 
    @Override 
    public void removeBucket(String bucketName) {
    }
 
    @Override 
    public boolean bucketExists(String bucketName) {
        try {
            return CollectionUtil.contains(this.bucketManager.buckets(), getBucketName(bucketName));
        }catch(QiniuException ex) {
            throw new RuntimeException(ex);
        }
    }
 
    @Override 
    public void copyFile(String bucketName, String fileName, String destBucketName) {
        try {
            this.bucketManager.copy(getBucketName(bucketName), fileName, getBucketName(destBucketName), fileName);
        }catch(QiniuException ex) {
            throw new RuntimeException(ex);
        }
    }
 
    @Override
    public void copyFile(String bucketName, String fileName, String destBucketName, String destFileName) {
        try {
            this.bucketManager.copy(getBucketName(bucketName), fileName, getBucketName(destBucketName), destFileName);
        }catch(QiniuException ex) {
            throw new RuntimeException(ex);
        }
    }
 
    @Override 
    public OssFile statFile(String fileName) {
        return statFile(this.ossProperties.getBucketName(), fileName);
    }
 
    @Override
    public OssFile statFile(String bucketName, String fileName) {
        try {
            FileInfo stat = this.bucketManager.stat(getBucketName(bucketName), fileName);
            OssFile ossFile = new OssFile();
            ossFile.setName(Func.isEmpty(stat.key) ? fileName : stat.key);
            ossFile.setLink(fileLink(ossFile.getName()));
            ossFile.setHash(stat.hash);
            ossFile.setLength(stat.fsize);
            ossFile.setPutTime(new Date(stat.putTime / 10000));
            ossFile.setContentType(stat.mimeType);
            return ossFile;
        }catch(QiniuException ex) {
            throw new RuntimeException(ex);
        }
    }
 
    @Override
    public String filePath(String fileName) {
        return getBucketName().concat("/").concat(fileName);
    }
 
    @Override
    public String filePath(String bucketName, String fileName) {
        return getBucketName(bucketName).concat("/").concat(fileName);
    }
 
    @Override
    public String fileLink(String fileName) {
        return this.ossProperties.getEndpoint().concat("/").concat(fileName);
    }
 
    @Override
    public String fileLink(String bucketName, String fileName) {
        return this.ossProperties.getEndpoint().concat("/").concat(fileName);
    }
 
    @Override 
    public InputStream getObject(String bucketName, String fileName, Long offset, Long length) {
        return get(bucketName, fileName, offset, length);
    }
 
    @Override
    public InputStream getObject(String fileName, Long offset, Long length) {
        return get(this.ossProperties.getBucketName(), fileName, offset, length);
    }
 
    @Override
    public InputStream getObject(String bucketName, String fileName) {
        return get(bucketName, fileName);
    }
 
    @Override
    public InputStream getObject(String fileName) {
        return get(this.ossProperties.getBucketName(), fileName);
    }
 
    @Override 
    public BladeFile putFile(MultipartFile file) {
        return putFile(this.ossProperties.getBucketName(), file.getOriginalFilename(), file);
    }
 
    @Override 
    public BladeFile putFile(String fileName, MultipartFile file) {
        return putFile(this.ossProperties.getBucketName(), fileName, file);
    }
 
    @Override 
    public BladeFile putFile(String bucketName, String fileName, MultipartFile file) {
        
        try {
            return putFile(bucketName, fileName, file.getInputStream());
        } catch (IOException ex) {
            throw new RuntimeException(ex);
        }
    }
 
    @Override 
    public BladeFile putFile(String fileName, InputStream stream) {
        return putFile(this.ossProperties.getBucketName(), fileName, stream);
    }
 
    @Override 
    public BladeFile putFile(String bucketName, String fileName, InputStream stream) {
        return put(bucketName, stream, fileName, false);
    }
 
    public InputStream get(String bucketName, String fileName) {
        throw new UnsupportedOperationException();
    }
 
    public InputStream get(String bucketName, String fileName, Long offset, Long length) {
        throw new UnsupportedOperationException();
    }
 
    public BladeFile put(String bucketName, InputStream stream, String key, boolean cover) {
        makeBucket(bucketName);
        String key2 = getFileName(key);
        try {
            if (cover) {
                this.uploadManager.put(stream, key2, getUploadToken(bucketName, key2), (StringMap) null, (String) null);
            } else {
                Response response = this.uploadManager.put(stream, key2, getUploadToken(bucketName), (StringMap) null, (String) null);
                for (int retry = 0; response.needRetry() && retry < 5; retry++) {
                    response = this.uploadManager.put(stream, key2, getUploadToken(bucketName), (StringMap) null, (String) null);
                }
            }
            BladeFile file = new BladeFile();
            file.setOriginalName(key);
            file.setName(key2);
            file.setDomain(getOssHost());
            file.setLink(fileLink(bucketName, key2));
            return file;
        }catch(QiniuException ex) {
            throw new RuntimeException(ex);
        }
    }
 
    @Override 
    public void removeFile(String fileName) {
        try {
            this.bucketManager.delete(getBucketName(), fileName);
        }catch(QiniuException ex) {
            throw new RuntimeException(ex);
        }
    }
 
    @Override 
    public void removeFile(String bucketName, String fileName) {
        try {
            this.bucketManager.delete(getBucketName(bucketName), fileName);
        }catch(QiniuException ex) {
            throw new RuntimeException(ex);
        }
    }
 
    @Override 
    public void removeFiles(List<String> fileNames) {
        fileNames.forEach(this::removeFile);
    }
 
    @Override 
    public void removeFiles(String bucketName, List<String> fileNames) {
        fileNames.forEach(fileName -> {
            removeFile(getBucketName(bucketName), fileName);
        });
    }
 
    private String getBucketName() {
        return getBucketName(this.ossProperties.getBucketName());
    }
 
    private String getBucketName(String bucketName) {
        return this.ossRule.bucketName(bucketName);
    }
 
    private String getFileName(String originalFilename) {
        return this.ossRule.fileName(originalFilename);
    }
 
    public String getUploadToken(String bucketName) {
        return this.auth.uploadToken(getBucketName(bucketName));
    }
 
    private String getUploadToken(String bucketName, String key) {
        return this.auth.uploadToken(getBucketName(bucketName), key);
    }
 
    public String getOssHost() {
        return this.ossProperties.getEndpoint();
    }
}