抽空优化了代码:
学习就是:先把书读厚,再读薄的过程。
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;
}
}
大写的自行添加。