代码中MimeTypeUtils在后续文章提供,可进行相关查看
package com.zx.test.util;
import org.springframework.web.multipart.MultipartFile;
import sun.misc.BASE64Decoder;
import javax.imageio.ImageIO;
import java.awt.image.BufferedImage;
import java.io.*;
import java.util.Base64;
public class BASE64DecodedMultipartFile implements MultipartFile {
private final byte[] imgContent;
private String header;
public void setHeader(String header){
this.header = header;
}
public BASE64DecodedMultipartFile(byte[] imgContent, String header) {
this.imgContent = imgContent;
this.header = header.split(";")[0];
}
@Override
public String getName() {
return System.currentTimeMillis() + Math.random() + "." + header.split("/")[1];
}
@Override
public String getOriginalFilename() {
return System.currentTimeMillis() + (int) Math.random() * 10000 + "." + header.split("/")[1];
}
@Override
public String getContentType() {
return header.split(":")[1];
}
@Override
public boolean isEmpty() {
return imgContent == null || imgContent.length == 0;
}
@Override
public long getSize() {
return imgContent.length;
}
@Override
public byte[] getBytes() throws IOException {
return imgContent;
}
@Override
public InputStream getInputStream() throws IOException {
return new ByteArrayInputStream(imgContent);
}
@Override
public void transferTo(File dest) throws IOException, IllegalStateException {
new FileOutputStream(dest).write(imgContent);
}
public static MultipartFile base64ToMultipart(String base64) {
try {
String[] baseStrs = base64.split(",");
BASE64Decoder decoder = new BASE64Decoder();
byte[] b = decoder.decodeBuffer(baseStrs[1]);
for (int i = 0; i < b.length; ++i) {
if (b[i] < 0) {
b[i] += 256;
}
}
return new BASE64DecodedMultipartFile(b, baseStrs[0]);
} catch (IOException e) {
e.printStackTrace();
return null;
}
}
public static String resolveMimeType(String file){
String substring = file.substring(0, 10);
//MimeTypeUtils工具类在下一篇文章提供
String mimeType = MimeTypeUtils.getMimeType(Base64.getDecoder().decode(substring));
StringBuilder builder = new StringBuilder();
builder.append("data:")
.append(mimeType)
.append(";base64,")
.append(file);
return builder.toString();
}
/**
* base64str转InputStream
* @param base64str
* @return
*/
public static InputStream base2InputStream(String base64str){
ByteArrayInputStream stream;
BASE64Decoder decoder = new BASE64Decoder();
byte[] bytes = new byte[0];
try {
bytes = decoder.decodeBuffer(base64str);
} catch (IOException e) {
e.printStackTrace();
}
stream=new ByteArrayInputStream(bytes);
return stream;
}
//BufferedImage 转base64
public static String getBase64FromImage(BufferedImage img){
ByteArrayOutputStream stream = new ByteArrayOutputStream();
try {
// 设置图片的格式
ImageIO.write(img, "jpg", stream);
} catch (IOException e) {
e.printStackTrace();
}
byte[] bytes = org.apache.commons.codec.binary.Base64.encodeBase64(stream.toByteArray());
return new String(bytes);
}
public static String fileToBase64(File file) throws IOException {
String base64 = "";
InputStream in = null;
try {
in = new FileInputStream(file);
byte[] bytes=new byte[(int)file.length()];
in.read(bytes);
base64 = Base64.getEncoder().encodeToString(bytes);
} catch (Exception e) {
throw e;
} finally {
if (in != null) {
try {
in.close();
} catch (IOException e) {
throw e;
}
}
}
return base64;
}
}