当前位置：首页 > news >正文

在Spring Boot中实现图片上传和修改

news 来源：原创 2025/6/19 6:45:50

1. 图片上传实现步骤

1.1 添加依赖

确保 spring-boot-starter-web 和 spring-boot-starter-validation 已存在：

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-web</artifactId>
</dependency>
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-validation</artifactId>
</dependency>

1.2 配置文件上传限制

在 application.properties 中配置最大文件大小：

spring.servlet.multipart.max-file-size=10MB
spring.servlet.multipart.max-request-size=10MB

1.3 编写上传接口

import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import java.io.File;
import java.io.IOException;
import java.util.UUID;

@RestController
public class ImageController {

    // 指定图片存储目录（示例路径，建议使用配置类或环境变量）
    private final String UPLOAD_DIR = "src/main/resources/static/images/";

    @PostMapping("/upload")
    public String uploadImage(@RequestParam("file") MultipartFile file) throws IOException {
        if (file.isEmpty()) {
            return "文件不能为空";
        }

        // 生成唯一文件名（防止重复）
        String originalFilename = file.getOriginalFilename();
        String fileExtension = originalFilename.substring(originalFilename.lastIndexOf("."));
        String newFileName = UUID.randomUUID() + fileExtension;

        // 创建目标文件
        File dest = new File(UPLOAD_DIR + newFileName);
        if (!dest.getParentFile().exists()) {
            dest.getParentFile().mkdirs(); // 创建目录
        }

        // 保存文件
        file.transferTo(dest);
        return "上传成功，访问地址: /images/" + newFileName;
    }
}

1.4 配置静态资源访问

默认情况下，static 目录下的文件可以直接访问。如果需要自定义路径，添加配置类：

import org.springframework.context.annotation.Configuration;
import org.springframework.web.servlet.config.annotation.ResourceHandlerRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;

@Configuration
public class WebConfig implements WebMvcConfigurer {
    @Override
    public void addResourceHandlers(ResourceHandlerRegistry registry) {
        registry.addResourceHandler("/images/**")
                .addResourceLocations("file:src/main/resources/static/images/");
    }
}

2. 图片修改实现步骤

修改图片通常需要先删除旧图片，再上传新图片，并更新数据库记录（如果有）。

2.1 添加删除旧图片的逻辑

public class ImageController {

    @PostMapping("/update")
    public String updateImage(
            @RequestParam("oldFileName") String oldFileName,
            @RequestParam("newFile") MultipartFile newFile) throws IOException {
        
        // 删除旧图片
        File oldFile = new File(UPLOAD_DIR + oldFileName);
        if (oldFile.exists()) {
            oldFile.delete();
        }

        // 上传新图片
        return uploadImage(newFile);
    }
}

2.2 结合数据库操作（示例）

如果图片路径存储在数据库中，需结合JPA/Hibernate更新记录：

@Entity
public class User {
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;
    private String avatar; // 图片路径字段
    // getter/setter
}

@RestController
public class UserController {

    @Autowired
    private UserRepository userRepository;

    @PostMapping("/user/{userId}/avatar")
    public String updateAvatar(
            @PathVariable Long userId,
            @RequestParam("file") MultipartFile file) throws IOException {
        
        User user = userRepository.findById(userId).orElseThrow();
        
        // 删除旧头像（如果有）
        if (user.getAvatar() != null) {
            File oldFile = new File(UPLOAD_DIR + user.getAvatar());
            if (oldFile.exists()) oldFile.delete();
        }

        // 上传新头像并更新数据库
        String newFileName = UUID.randomUUID() + getFileExtension(file);
        file.transferTo(new File(UPLOAD_DIR + newFileName));
        user.setAvatar(newFileName);
        userRepository.save(user);

        return "头像更新成功";
    }

    private String getFileExtension(MultipartFile file) {
        String name = file.getOriginalFilename();
        return name.substring(name.lastIndexOf("."));
    }
}

3. 前端调用示例（HTML）

<form action="/upload" method="post" enctype="multipart/form-data">
    <input type="file" name="file">
    <button type="submit">上传</button>
</form>

<form action="/update" method="post" enctype="multipart/form-data">
    <input type="hidden" name="oldFileName" value="old-image.jpg">
    <input type="file" name="newFile">
    <button type="submit">修改图片</button>
</form>