package com.ruoyi.oss.service.impl;
|
|
import com.qiniu.http.Response;
|
import com.qiniu.storage.BucketManager;
|
import com.qiniu.storage.Configuration;
|
import com.qiniu.storage.Region;
|
import com.qiniu.storage.UploadManager;
|
import com.qiniu.util.Auth;
|
import com.ruoyi.oss.entity.UploadResult;
|
import com.ruoyi.oss.enumd.CloudServiceEnumd;
|
import com.ruoyi.oss.exception.OssException;
|
import com.ruoyi.oss.factory.OssFactory;
|
import com.ruoyi.oss.properties.CloudStorageProperties;
|
import com.ruoyi.oss.properties.CloudStorageProperties.QiniuProperties;
|
import com.ruoyi.oss.service.abstractd.AbstractCloudStorageService;
|
import org.springframework.beans.factory.InitializingBean;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.context.annotation.Lazy;
|
import org.springframework.stereotype.Service;
|
|
import java.io.InputStream;
|
|
/**
|
* 七牛云存储
|
*
|
* @author Lion Li
|
*/
|
@Lazy
|
@Service
|
public class QiniuCloudStorageServiceImpl extends AbstractCloudStorageService implements InitializingBean {
|
|
private final UploadManager uploadManager;
|
private final BucketManager bucketManager;
|
private final String token;
|
private final QiniuProperties properties;
|
|
@Autowired
|
public QiniuCloudStorageServiceImpl(CloudStorageProperties properties) {
|
this.properties = properties.getQiniu();
|
try {
|
// z0 z1 z2
|
Configuration config = new Configuration(Region.autoRegion());
|
// 默认不使用https
|
config.useHttpsDomains = false;
|
uploadManager = new UploadManager(config);
|
Auth auth = Auth.create(
|
this.properties.getAccessKey(),
|
this.properties.getSecretKey());
|
token = auth.uploadToken(this.properties.getBucketName());
|
bucketManager = new BucketManager(auth, config);
|
} catch (Exception e) {
|
throw new IllegalArgumentException("七牛云存储配置错误! 请检查系统配置!");
|
}
|
}
|
|
|
@Override
|
public String getServiceType() {
|
return CloudServiceEnumd.QINIU.getValue();
|
}
|
|
@Override
|
public UploadResult upload(byte[] data, String path, String contentType) {
|
try {
|
Response res = uploadManager.put(data, path, token);
|
if (!res.isOK()) {
|
throw new RuntimeException("上传七牛出错：" + res.toString());
|
}
|
} catch (Exception e) {
|
throw new OssException("上传文件失败，请核对七牛配置信息");
|
}
|
return new UploadResult().setUrl(properties.getDomain() + "/" + path).setFilename(path);
|
}
|
|
@Override
|
public void delete(String path) {
|
try {
|
path = path.replace(this.properties.getDomain() + "/", "");
|
Response res = bucketManager.delete(this.properties.getBucketName(), path);
|
if (!res.isOK()) {
|
throw new RuntimeException("删除七牛文件出错：" + res.toString());
|
}
|
} catch (Exception e) {
|
throw new OssException(e.getMessage());
|
}
|
}
|
|
@Override
|
public UploadResult uploadSuffix(byte[] data, String suffix, String contentType) {
|
return upload(data, getPath(this.properties.getPrefix(), suffix), contentType);
|
}
|
|
@Override
|
public UploadResult uploadSuffix(InputStream inputStream, String suffix, String contentType) {
|
return upload(inputStream, getPath(this.properties.getPrefix(), suffix), contentType);
|
}
|
|
@Override
|
public void afterPropertiesSet() throws Exception {
|
OssFactory.register(getServiceType(),this);
|
}
|
|
}
|
