如何在Spring Boot中实现图片上传至本地和阿里云OSS

news2024/11/29 22:29:18

在开发Web应用时，处理文件上传是常见的需求之一，尤其是在涉及到图片、视频等多媒体数据时。本文将详细介绍如何使用Spring Boot实现图片上传至本地服务器以及阿里云OSS存储服务，并提供完整的代码示例。

一、上传图片至本地

首先，我们来看如何在Spring Boot项目中实现图片上传至本地的功能。这通常涉及接收前端发送的文件流，然后将其保存到服务器上的指定目录。

代码示例:

@RestController
public class FileController {

    @PostMapping("/upload")
    public Result upload(MultipartFile file) throws IOException {
        String originalFilename = file.getOriginalFilename();
        String filename = UUID.randomUUID() + originalFilename.substring(originalFilename.lastIndexOf("."));
        file.transferTo(new File("D:\\img\\" + filename));
        return Result.success();
    }
}

上述代码中，MultipartFile用于接收上传的文件，通过transferTo()方法将文件保存到本地磁盘。

二、上传图片至阿里云OSS

当应用需要处理大量或大体积的文件时，使用云存储服务如阿里云OSS可以显著提高系统的稳定性和可扩展性。下面是如何在Spring Boot中集成阿里云OSS并上传文件的步骤。

引入依赖:

在pom.xml中添加以下依赖以支持阿里云OSS和JDK 9以上的环境。

<dependency>
    <groupId>com.aliyun.oss</groupId>
    <artifactId>aliyun-sdk-oss</artifactId>
    <version>3.15.1</version>
</dependency>
<dependency>
    <groupId>javax.xml.bind</groupId>
    <artifactId>jaxb-api</artifactId>
    <version>2.3.1</version>
</dependency>
<dependency>
    <groupId>javax.activation</groupId>
    <artifactId>activation</artifactId>
    <version>1.1.1</version>
</dependency>
<dependency>
    <groupId>org.glassfish.jaxb</groupId>
    <artifactId>jaxb-runtime</artifactId>
    <version>2.3.3</version>
</dependency>

封装工具类:

创建一个工具类AliOssUtil来处理与OSS的交互，包括上传文件。

public class AliOssUtil {

    private static final String ENDPOINT = "https://oss-cn-beijing.aliyuncs.com";
    private static final String ACCESS_KEY_ID="XXXXXXXXXXXXXXXX";
    private static final String ACCESS_KEY_SECRET="XXXXXXXXXXXXXXXXXXXXXX";
    private static final String BUCKET_NAME = "XXXXXXXXXX";

    public static String uploadFile(String objectName, InputStream in) throws Exception {
        OSS ossClient = new OSSClientBuilder().build(ENDPOINT, ACCESS_KEY_ID, ACCESS_KEY_SECRET);
        try {
            PutObjectRequest putObjectRequest = new PutObjectRequest(BUCKET_NAME, objectName, in);
            PutObjectResult result = ossClient.putObject(putObjectRequest);
            String url = "https://" + BUCKET_NAME + "." + ENDPOINT.substring(ENDPOINT.lastIndexOf("/") + 1) + "/" + objectName;
            return url;
        } finally {
            ossClient.shutdown();
        }
    }
}

Controller代码:

修改FileController，使用AliOssUtil上传文件。

@RestController
public class FileController {

    @PostMapping("/upload")
    public Result upload(MultipartFile file) throws Exception {
        String originalFilename = file.getOriginalFilename();
        String filename = UUID.randomUUID() + originalFilename.substring(originalFilename.lastIndexOf("."));
        String url = AliOssUtil.uploadFile(filename, file.getInputStream());
        return Result.success(url);
    }
}