springboot实现图片上传至服务器本地保存
作者：安久1
这篇文章主要为大家详细介绍了如何基于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()));
}
}
到此这篇关于springboot实现图片上传至服务器本地保存的文章就介绍到这了,更多相关springboot图片上传服务器内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！