在软件开发中,我们有时会遇到一些用户只是为了修改文件后缀然后上传文件,这种操作可能会引发各种问题,有时候甚至导致系统崩溃或数据损失。为了解决这个问题,我创建了一个小工具,为了帮助大家避免这种情况。我将向大家详细介绍如何使用Java来构建一个简单而实用的文件扫描和信息录入功能,以确保我们上传的文件类型和内容都是准确的,这将有助于提高系统的可靠性和数据的完整性。以下为视频文件举例。(其他文件同理)
目录
1.我们首先来看一段Java代码,该代码负责实现文件扫描与视频信息录入功能。
一. 环境准备
1.POM导入hutool
<dependency>
<groupId>cn.hutool</groupId>
<artifactId>hutool-all</artifactId>
<version>5.8.16</version>
</dependency>
二.功能实现
1.我们首先来看一段Java代码,该代码负责实现文件扫描与视频信息录入功能。
/**
* 从指定路径扫描视频文件并录入相关信息的方法。
*
* @param path 扫描的目标文件夹路径
* @return 包含视频信息的列表
* @throws Exception 如果在操作过程中出现异常,则抛出异常
*/
public List<Vedioinfo> getfile(String path) throws Exception {
List<Vedioinfo> vedioinfoArrayList = new ArrayList<>(); // 用于存储视频信息的列表
File folder = FileUtil.file(path); // 根据指定路径创建文件对象,表示要扫描的文件夹
File[] files = folder.listFiles(); // 获取文件夹中的所有文件和子文件夹
try {
for (File file : files) { // 遍历文件夹中的每个文件
if (file.isFile() && isVideoFile(file)) { // 检查文件是否为普通文件且是视频文件
FileInputStream videoInput = new FileInputStream(file); // 创建文件输入流以读取视频文件
if (IsVedioType(file, videoInput)) { // 检查文件是否为有效的视频类型
double size = (double) file.length() / 1024 / 1024; // 计算文件大小(以MB为单位)
DecimalFormat df = new DecimalFormat("#.00"); // 格式化文件大小
Vedioinfo newvedioinfo = new Vedioinfo(); // 创建新的Vedioinfo对象
// 设置Vedioinfo对象的属性
newvedioinfo.setVedioName(file.getName()); // 获取文件名
newvedioinfo.setUpdateTime(new Date(file.lastModified())); // 获取文件更新时间
newvedioinfo.setVedioPath(file.getPath()); // 获取文件路径
newvedioinfo.setVedioSize(Double.parseDouble(df.format(size))); // 获取文件大小
newvedioinfo.setVedioTypes(getFileExtension(file)); // 获取文件格式
newvedioinfo.setVedioNewPath(file.getParentFile() + "\\jhvideo"); // 设置新路径
newvedioinfo.setVedioRelativePath(file.getPath().substring(3).replace("\\", "/")); // 获取文件的相对路径
vedioinfoArrayList.add(newvedioinfo); // 将新的Vedioinfo对象添加到列表中
}
}
}
// 返回扫描得到的视频信息列表
return vedioinfoArrayList;
} catch (IOException e) {
// 捕获IO异常并抛出异常
throw new Exception(e.getMessage(), e.getCause());
}
}
这段代码执行以下关键操作:
-
根据指定路径创建文件对象,表示要扫描的文件夹。
-
获取文件夹中的所有文件和子文件夹。
-
遍历文件夹中的每个文件,检查是否为普通文件且是视频文件。
-
创建文件输入流以读取视频文件,并检查文件是否为有效的视频类型。
-
计算文件大小并创建新的
Vedioinfo
对象,录入文件的基本信息。 -
将新的
Vedioinfo
对象添加到列表中。 -
最后,返回扫描得到的视频信息列表。
2.函数功能实现
2.1视频文件后缀判断(第一层判断)
在文件扫描与视频信息录入工具中,我们需要确保扫描到的文件是视频文件。为了实现这个目标,我们编写了一个名为isVideoFile
的方法,用于检查文件的扩展名是否与视频文件匹配。
/**
* 检查文件是否为视频文件。
*
* @param file 要检查的文件
* @return 如果文件是视频文件,则返回true,否则返回false
*/
private static boolean isVideoFile(File file) {
String extension = getFileExtension(file);
// 定义视频文件扩展名列表,可以根据实际需求扩展
List<String> videoExtensions = Arrays.asList("mp4", "ogg", "webm");
return videoExtensions.stream()
.anyMatch(extension::equalsIgnoreCase);
}
这段代码执行以下关键操作:
-
获取文件的扩展名。
-
定义了一个视频文件扩展名列表,包括常见的视频文件格式的扩展名(例如mp4、ogg、webm)。你可以根据实际需求扩展这个列表。
-
使用Stream API遍历视频文件扩展名列表,检查文件的扩展名是否与列表中的任何扩展名匹配。
-
如果匹配成功,则返回
true
,表示文件是视频文件。否则,返回false
。
2.2文件头签名判断(第二层判断)
在文件扫描与视频信息录入工具中,我们需要确保扫描到的文件是有效的视频文件。为了实现这个目标,我们编写了一个名为IsVedioType
的方法,用于检查文件的类型是否与视频文件匹配。
/**
* 检查文件是否为有效的视频类型。
*
* @param file 要检查的文件
* @param fileInputStream 文件的输入流
* @return 如果文件为有效的视频类型,则返回true,否则返回false
*/
private static Boolean IsVedioType(File file, InputStream fileInputStream) {
try {
byte[] bytes = new byte[10];
fileInputStream.read(bytes, 0, bytes.length);
String fileTypeHex = String.valueOf(bytesToHexString(bytes)); // 将文件转化成16进制的字符
// 定义视频文件类型列表,可以根据实际需求扩展
List<String> videoTypes = Arrays.asList("mp4", "ogg", "webm");
for (String type : videoTypes) {
if (fileTypeHex.toLowerCase().startsWith(type)) {
return true;
}
}
} catch (IOException e) {
e.printStackTrace();
}
return false;
}
这段代码执行以下关键操作:
-
从文件输入流中读取前10个字节的数据。
-
将这些字节数据转化为16进制字符串。
-
定义了一个视频文件类型列表,包括常见的视频文件格式(例如mp4、ogg、webm)。你可以根据实际需求扩展这个列表。
-
遍历视频文件类型列表,检查文件的前几个字节是否与列表中的任何类型匹配。
-
如果匹配成功,则返回
true
,表示文件是有效的视频类型。否则,返回false
2.3文件头签名判断文件有效性
/**
* 判断文件是否为视频类型。
*
* @param file 要检查的文件
* @param fileInputStream 文件的输入流
* @return 如果文件为视频类型,则返回 true;否则返回 false
*/
private static Boolean IsVedioType(File file, InputStream fileInputStream) {
try {
Class<?> clazz = FileConstant.class;
Field[] fields = clazz.getDeclaredFields();
InputStream inputStream = fileInputStream;
byte[] bytes = new byte[10];
inputStream.read(bytes, 0, bytes.length);
String fileTypeHex = String.valueOf(bytesToHexString(bytes));//文件转化成16进制的字符
ArrayList<String> list = new ArrayList<String>();
for (Field field : fields) {
Object value = field.get(null);
list.add((String) value);//转型添加
}
for (String type : list) {
if (type.toLowerCase().startsWith(fileTypeHex.toLowerCase().substring(0, 5)) || fileTypeHex.toLowerCase().substring(0, 5).startsWith(type.toLowerCase())) {
return true;
}
}
} catch (Exception e) {
e.printStackTrace();
}
return false;
}
这段代码执行以下关键操作:
1. 从文件输入流中读取前10个字节的数据。
2. 将这些字节数据转化为16进制的字符串。
3. 检查文件是否属于视频类型。这个方法可以用于检测文件是否是视频类型。
4. 如果文件被识别为视频类型,返回true,表示文件是有效的视频类型。否则,返回false。
2.4文件字节转16进制
在文件处理和数据转换中,有时需要将字节数据转化为16进制字符串。这种操作在处理文件头签名时特别有用。为了实现这一目标,我们编写了一个名为 bytesToHexString
的方法,用于将字节数组转化为16进制字符串。
private static String bytesToHexString(byte[] src) {
StringBuilder stringBuilder = new StringBuilder();
if (src == null || src.length <= 0) {
return null;
}
for (int i = 0; i < src.length; i++) {
int v = src[i] & 0xFF;//屏蔽符号位扩展
String hv = Integer.toHexString(v);
if (hv.length() < 2) {
stringBuilder.append(0);
}
stringBuilder.append(hv);
}
return stringBuilder.toString();
}
该方法执行以下关键操作:
- 创建一个
StringBuilder
对象,用于构建最终的16进制字符串。 - 检查输入的字节数组是否为空或长度小于等于0,如果是,则返回
null
。 - 遍历字节数组的每个元素。
- 屏蔽符号位扩展,将每个字节数据转化为16进制字符。
- 如果转化后的字符长度小于2,将其前面添加一个 '0'。
- 将转化后的字符添加到
StringBuilder
对象中。 - 返回构建好的16进制字符串。
2.5获取文件名后缀
在文件扫描与视频信息录入工具中,获取文件的扩展名是一项重要任务,因为它有助于确定文件的类型。我们编写了一个名为getFileExtension的方法,用于从文件中提取扩展名。
private static String getFileExtension(File file) {
String name = file.getName();
int lastIndex = name.lastIndexOf(".");
if (lastIndex == -1) {
return null; // 扩展名为空
}
return name.substring(lastIndex + 1);
}
这段代码执行了以下关键操作:
- 获取文件的名称。
- 查找文件名称中最后一个点(.)的位置。
- 如果找到点,则提取点之后的字符作为文件的扩展名。
- 如果未找到点,则返回null,表示扩展名为空。
三.文件头签名定义(可以拓展)
public interface FileConstant {
static final String MP4_SIGN="00000020667479706d70";
static final String OGG_SIGN="4f676753";
static final String WEBM_SIGN="1a45dfa3";
}
欢迎进行补充