OssUploadResult.java
2.86 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
package com.lhcredit.common.utils.oss;
import com.alibaba.fastjson.annotation.JSONField;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
import org.springframework.web.multipart.MultipartFile;
import java.io.InputStream;
import java.io.Serializable;
import java.util.Arrays;
import java.util.List;
import java.util.UUID;
import java.util.stream.Collectors;
@Data
@AllArgsConstructor
@NoArgsConstructor
@Builder
public class OssUploadResult implements Serializable {
@JSONField(serialize = false)
public static List<String> mediaList= Arrays.asList("jpg","png","pdf","doc","txt","docx","xlsx","xls","ppt");
private Integer code;
private String msg;
private String originalFilename;
private String fileName;
private String saveName;
private String ex;
private String url;
@JSONField(serialize = false)
private Boolean media=false; //校验规范媒体文件
@JSONField(serialize = false)
private MultipartFile file;
@JSONField(serialize = false)
private Boolean useOriginalFilename=false;//使用原文件名称
public static OssUploadResult of(MultipartFile file){
return of(file,false,false);
}
public static OssUploadResult of(MultipartFile file,Boolean useOriginalFilename,Boolean noRestrict){
String filename = file.getOriginalFilename();
String time = String.valueOf(System.currentTimeMillis());
String name = OssService.getFileNameNoEx(filename);
String ex = OssService.getExtensionName(filename);
String uuid= UUID.randomUUID().toString().replaceAll("-","");
String saveName = uuid.concat("_").concat(time).concat(".").concat(ex);
return OssUploadResult.builder().originalFilename(filename).ex(ex).fileName(name).saveName(saveName).file(file).media(noRestrict).useOriginalFilename(useOriginalFilename).build();
}
public OssUploadResult media(){
this.media=true;
return this;
}
public OssUploadResult useOriginalFilename(){
this.useOriginalFilename=true;
return this;
}
public OssUploadResult upload(){
if (media&&!mediaList.contains(ex)){
error("请上传"+mediaList.stream().collect(Collectors.joining(","))+"类型");
}else {
try{
InputStream inputStream = file.getInputStream();
this.url=OssService.uploadStream(this.useOriginalFilename==false?saveName:originalFilename, inputStream, Long.valueOf(inputStream.available()));
success();
}catch (Exception e){
e.printStackTrace();
error(e.getMessage());
}
}
return this;
}
protected void success(){
this.code=2000;
}
protected void error(String msg){
this.code=5000;
this.msg=msg;
}
}