七牛云使用笔记

七牛云配置

进入七牛云官网七牛云 - 国内领先的企业级云服务商 (qiniu.com)
注册七牛云账号并登录,创建存储

创建SpringBoot项目

application.yml

#七牛云配置
oss:
  domain: http://******.hd-bkt.clouddn.com/  # 访问域名(默认使用七牛云测试域名) 测试域名有效期仅为30天，建议更换为自有域名
  accessKey: ***** # 公钥 AK
  secretKey: ***** # 私钥 SK
  bucket: ****  #存储空间名称 创建的存储空间名称

pom.xml

<dependency>
    <groupId>com.google.code.gson</groupId>
    <artifactId>gson</artifactId>
    <version>2.8.6</version>
</dependency>

<dependency>
    <groupId>com.qiniu</groupId>
    <artifactId>qiniu-java-sdk</artifactId>
    <version>[7.7.0, 7.7.99]</version>
</dependency>

UploadController.java

@Api(tags = "上传")
@RestController
public class UploadController {

    @Autowired
    private UploadService uploadService;

    @ApiOperation(value = "文件上传")
    @PostMapping("/upload")
    public ApiResult upload(@RequestPart MultipartFile file){
        String url = uploadService.upload(file);
        return ApiResult.success(url);
    }
}

UploadServiceImpl.java

@Service
@Data
@ConfigurationProperties(prefix = "oss")
public class UploadServiceImpl implements UploadService {
    private String accessKey;
    private String secretKey;
    private String bucket;
    private String domain;

    @Override
    public String upload(MultipartFile file) {
        //判断文件类型
        //获取原始文件名
        if (file == null){
            ApiAsserts.fail("文件上传失败！");
        }
        String originalFileName = file.getOriginalFilename();
        //对原始文件名进行判断
        assert originalFileName != null;
        if (!originalFileName.endsWith(".png") && !originalFileName.endsWith(".jpg") && !originalFileName.endsWith(".txt")){
            //TODO 抛出错误，此处只允许.png
            ApiAsserts.fail("只可以上传png格式的图片哦！");
        }
        String path = PathUtils.generateFilePath(originalFileName);
        return uploadOss(file,path);
    }

    private String uploadOss(MultipartFile file , String path) {
        Configuration cfg = new Configuration(Region.autoRegion());
        UploadManager uploadManager = new UploadManager(cfg);
        String key = path;
        try {
            InputStream inputStream = file.getInputStream();
            Auth auth = Auth.create(accessKey, secretKey);
            String upToken = auth.uploadToken(bucket);
            try {
                Response response = uploadManager.put(inputStream, key, upToken, null, null);
                DefaultPutRet putRet = new Gson().fromJson(response.bodyString(), DefaultPutRet.class);
                System.out.println(putRet.key);
                System.out.println(putRet.hash);
                return domain+key;
            } catch (QiniuException e) {
                Response r = e.response;
                System.err.println(r.toString());
                try {
                    System.err.println(r.bodyString());
                } catch (QiniuException e2) {

                }
            }
        } catch (Exception e) {

        }
        return "www";
    }
}

PathUtils.java

public class PathUtils {
    public static String generateFilePath(String fileName){
        //根据日期生成路径 2022/1/15/
        SimpleDateFormat sdf = new SimpleDateFormat("yyyy/MM/dd/");
        String datePath = sdf.format(new Date());
        //uuid作为文件名
        String uuid = UUID.randomUUID().toString().replaceAll("-", "");
        //后缀和文件后缀一致
        int index = fileName.lastIndexOf(".");
        //test.jpg -> .jpg
        String fileType = fileName.substring(index);
        return new StringBuilder().append(datePath).append(uuid).append(fileType).toString();
    }
}