feat: 备注完善

yanghb

2024-12-24 fe6e43d5e1144156d0ca4e9d6080c9821c25d97c

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
package com.zzg.common.utils.enums;
 
import lombok.AllArgsConstructor;
import lombok.NoArgsConstructor;
 
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
 
/**
 * 文件类型
 */
@NoArgsConstructor
@AllArgsConstructor
public enum FileTypeEnum {
    //表格
    EXCEL(1, Arrays.asList("xls", "xlsx")),
    //文档
    WORD(2, Arrays.asList("doc", "docx")),
    //PDF
    PDF(3, Arrays.asList("pdf")),
    //媒体
    MEDIA(4, Arrays.asList("mp4", "mkv", "avi", "jpg", "png", "jpeg", "bmp")),
    //其他
    OTHER(5, Arrays.asList("ppt", "shp", "dwg", "wmv")),
    ;
 
    public static List<String> getSuffixByCode(Integer type) {
        for (FileTypeEnum v : FileTypeEnum.values()) {
            if (type.equals(v.type)) {
                return v.suffixList;
            }
        }
        return new ArrayList<>();
    }
 
    public static Integer getCodeBySuffix(String suffix) {
        for (FileTypeEnum v : FileTypeEnum.values()) {
            if (v.suffixList.contains(suffix)) {
                return v.type;
            }
        }
        return 5;
    }
 
    public static List<String> getSuffixOther(Integer type) {
        List<String> list = new ArrayList<>();
        for (FileTypeEnum v : FileTypeEnum.values()) {
            if (!type.equals(v.type)) {
                list.addAll(v.suffixList);
            }
        }
        return list;
    }
 
    Integer type;
    List<String> suffixList;
 
}