FileTypeUtil.java 文件格式util 文件类型判断-CSDN博客

抽空优化了代码：

学习就是：先把书读厚，再读薄的过程。

package com.lgq.ai.Util;

import java.util.Arrays;

public class FileTypeUtil {
    public static int fileType(String fileName) {
        if (!StringUtil.isEmpty(fileName)) {
            String fileType = fileName.split("\\.")[1];
            /*图片*/
            String img[] = {"bmp", "BMP", "jpg", "JPG", "jpeg", "JPEG", "png", "PNG", "tiff", "gif", "GIF",
                    "pcx", "tga", "exif", "fpx", "svg", "SVG", "psd",
                    "cdr", "pcd", "dxf", "ufo", "eps", "ai", "raw", "wmf"};
            boolean imgs = Arrays.asList(img).contains(fileType);
            if (imgs) {
                return 1;
            }

            /*文档*/
            String document[] = {"txt", "doc", "docx", "xls", "htm", "html", "jsp", "rtf", "wpd", "pdf", "ppt"};
            boolean documents = Arrays.asList(document).contains(fileType);
            if (documents) {
                return 2;
            }

            /*视频*/
            String video[] = {"mp4", "avi", "mov", "wmv", "asf", "navi", "3 gp", "mkv", "f4v", "rmvb", "webm"};
            boolean videos = Arrays.asList(video).contains(fileType);
            if (videos) {
                return 3;
            }

            /*音乐*/
            String music[] = {"mp3", "wma", "wav", "mod", "ra", "cd", "md", "asf", "aac", "vqf", "ape", "mid", "ogg",
                    "m4a", "vqf"};
            boolean musics = Arrays.asList(music).contains(fileType);
            if (musics) {
                return 4;
            }
        }
        return 0;
    }


}

大写的自行添加。