springboot图片上传至服务器本地保存

定义config映射文件：本地存储路径为x:xxx,映射路径为"/uploads/avatar/**"

package com.example.demo.config;

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

import java.nio.file.Path;
import java.nio.file.Paths;

@Configuration
public class WebConfig implements WebMvcConfigurer {

    @Override
    public void addResourceHandlers(ResourceHandlerRegistry registry) {
        Path uploadPath = Paths.get("F:", "aaa");
        registry.addResourceHandler("/uploads/avatar/**")
                .addResourceLocations(uploadPath.toUri().toString());
    }
}

编写mapper

package com.example.demo.mapper;

import com.example.demo.pojo.Member;
import com.example.demo.pojo.MemberQuery;
import org.apache.ibatis.annotations.Mapper;
import org.apache.ibatis.annotations.Param;

import java.util.List;

@Mapper
public interface UserMapper {
    
     Integer insertUserProfile(
             @Param("userId") Integer userId,
             @Param("userName") String userName,
             @Param("userGender") String userGender,
             @Param("userAvatar") String userAvatar
     );
}

实现UserMapper.xml文件作为Sql查询

<insert id="insertUserProfile">
    INSERT INTO t_user
    <trim prefix="(" suffix=")" suffixOverrides=",">
        user_id,
        <if test="userName != null and userName != ''">
            user_name,
        </if>
        <if test="userGender != null and userGender != ''">
            user_gender,
        </if>
        <if test="userAvatar != null and userAvatar != ''">
            user_avatar,
        </if>
    </trim>
    <trim prefix="VALUES (" suffix=")" suffixOverrides=",">
        #{userId},
        <if test="userName != null and userName != ''">
            #{userName},
        </if>
        <if test="userGender != null and userGender != ''">
            #{userGender},
        </if>
        <if test="userAvatar != null and userAvatar != ''">
            #{userAvatar},
        </if>
    </trim>
</insert>

编写service层接口

package com.example.demo.service;

import com.example.demo.common.PageResult;
import com.example.demo.common.UploadImageResult;
import com.example.demo.pojo.Member;
import com.example.demo.pojo.MemberQuery;
import com.example.demo.pojo.UserAvatarBase64Request;
import org.springframework.web.multipart.MultipartFile;

public interface UserService {
    UploadImageResult uploadUserAvatarBase64(UserAvatarBase64Request request);
}

实现接口uploadUserAvatarBase64方法，parseBase64Image方法判断图片格式，saveUserProfile方法保存或更新用户的基本信息到数据库

package com.example.demo.service.impl;

import com.example.demo.common.PageResult;
import com.example.demo.common.UploadImageResult;
import com.example.demo.mapper.UserMapper;
import com.example.demo.pojo.Member;
import com.example.demo.pojo.MemberQuery;
import com.example.demo.pojo.UserAvatarBase64Request;
import com.example.demo.service.UserService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.util.StringUtils;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.Base64;
import java.util.List;
import java.util.Locale;
import java.util.Set;
import java.util.UUID;

@Service
public class UserServiceImpl implements UserService {

    private static final Set<String> ALLOWED_IMAGE_EXTENSIONS = Set.of("jpg", "jpeg", "png", "gif", "webp");

    @Autowired
    private UserMapper userMapper;

    @Override
    public PageResult<Member> getUserList(MemberQuery query) {
        MemberQuery safeQuery = query == null ? new MemberQuery() : query;
        int safeIndex = (safeQuery.getIndex() == null || safeQuery.getIndex() < 1) ? 1 : safeQuery.getIndex();
        int safePageSize = (safeQuery.getPageSize() == null || safeQuery.getPageSize() < 1) ? 10 : safeQuery.getPageSize();
        int offset = (safeIndex - 1) * safePageSize;
        List<Member> users = userMapper.selectUserList(safeQuery, offset, safePageSize);
        Long total = userMapper.selectUserCount(safeQuery);
        return new PageResult<>(users, total, safeIndex, safePageSize);
    }

    

    @Override
    public UploadImageResult uploadUserAvatarBase64(UserAvatarBase64Request request) {
        if (request == null) {
            throw new IllegalArgumentException("请求参数不能为空");
        }
        Integer userId = request.getUserId();
        String userName = request.getUserName();
        String userGender = request.getUserGender();
        String userAvatar = request.getUserAvatar();

        if (userId == null) {
            throw new IllegalArgumentException("用户id不能为空");
        }
        String avatarPath = null;
        if (StringUtils.hasText(userAvatar)) {
            Base64Image base64Image = parseBase64Image(userAvatar);
            String fileName = UUID.randomUUID() + "." + base64Image.extension;
            Path uploadDir = Paths.get("F:", "aaa");
            Path targetPath = uploadDir.resolve(fileName);

            try {
                Files.createDirectories(uploadDir);
                Files.write(targetPath, base64Image.bytes);
            } catch (IOException e) {
                throw new RuntimeException("图片保存失败", e);
            }

            avatarPath = "/uploads/avatar/" + fileName;
        }

        return saveUserProfile(userId, userName, userGender, avatarPath);
    }

    private UploadImageResult saveUserProfile(Integer userId, String userName, String userGender, String userAvatar) {
        if (!StringUtils.hasText(userName) && !StringUtils.hasText(userGender) && !StringUtils.hasText(userAvatar)) {
            throw new IllegalArgumentException("用户名称、用户性别、用户头像至少传一个");
        }

        Integer exists = userMapper.selectUserById(userId);
        Integer affectedRows;
        if (exists != null && exists > 0) {
            affectedRows = userMapper.updateUserProfile(userId, userName, userGender, userAvatar);
        } else {
            affectedRows = userMapper.insertUserProfile(userId, userName, userGender, userAvatar);
        }

        if (affectedRows == null || affectedRows == 0) {
            throw new IllegalArgumentException("用户信息保存失败");
        }
        return new UploadImageResult(userId, userName, userGender, userAvatar);
    }

    private String getFileExtension(String originalFilename) {
        String filename = StringUtils.cleanPath(originalFilename == null ? "" : originalFilename);
        int dotIndex = filename.lastIndexOf('.');
        if (dotIndex < 0 || dotIndex == filename.length() - 1) {
            throw new IllegalArgumentException("图片文件后缀不能为空");
        }
        return filename.substring(dotIndex + 1).toLowerCase(Locale.ROOT);
    }

    private Base64Image parseBase64Image(String userAvatar) {
        String base64Content = userAvatar.trim();
        String extension = "png";

        if (base64Content.startsWith("data:image/")) {
            int mimeEndIndex = base64Content.indexOf(";base64,");
            if (mimeEndIndex < 0) {
                throw new IllegalArgumentException("图片base64格式错误");
            }
            extension = base64Content.substring("data:image/".length(), mimeEndIndex).toLowerCase(Locale.ROOT);
            if ("jpg".equals(extension)) {
                extension = "jpeg";
            }
            base64Content = base64Content.substring(mimeEndIndex + ";base64,".length());
        }

        if (!ALLOWED_IMAGE_EXTENSIONS.contains(extension)) {
            throw new IllegalArgumentException("只支持 jpg、jpeg、png、gif、webp 格式图片");
        }

        try {
            byte[] bytes = Base64.getDecoder().decode(base64Content);
            if (bytes.length == 0) {
                throw new IllegalArgumentException("用户头像base64不能为空");
            }
            return new Base64Image(extension, bytes);
        } catch (IllegalArgumentException e) {
            throw new IllegalArgumentException("图片base64解析失败");
        }
    }

    private static class Base64Image {
        private final String extension;
        private final byte[] bytes;

        private Base64Image(String extension, byte[] bytes) {
            this.extension = extension;
            this.bytes = bytes;
        }
    }
}

contrller层调用

@PostMapping("/uploadAvatarBase64")
public org.springframework.http.ResponseEntity<ResponseEntity<UploadImageResult>> uploadAvatarBase64(
        @RequestBody UserAvatarBase64Request request
) {
    try {
        UploadImageResult result = userService.uploadUserAvatarBase64(request);
        return org.springframework.http.ResponseEntity.ok(
                ResponseEntity.success(200, result, "用户信息保存成功")
        );
    } catch (IllegalArgumentException e) {
        return org.springframework.http.ResponseEntity
                .status(HttpStatus.BAD_REQUEST)
                .body(ResponseEntity.fail(400, e.getMessage()));
    } catch (RuntimeException e) {
        return org.springframework.http.ResponseEntity
                .status(HttpStatus.INTERNAL_SERVER_ERROR)
                .body(ResponseEntity.fail(500, e.getMessage()));
    }
}