写一个Java完成文件后缀判断的程序

import java.util.Scanner;

public class FileExtensionChecker {
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
System.out.print("请输入文件名: ");
String fileName = scanner.nextLine();
String[] fileParts = fileName.split("\\."); //使用"."分割文件名
String extension = fileParts[fileParts.length - 1]; //获取文件后缀

//判断文件后缀
if (extension.equals("txt")) {
System.out.println("这是一个文本文件。");
} else if (extension.equals("jpg") || extension.equals("png") || extension.equals("gif")) {
System.out.println("这是一个图片文件。");
} else if (extension.equals("mp3") || extension.equals("wav")) {
System.out.println("这是一个音频文件。");
} else if (extension.equals("mp4") || extension.equals("avi") || extension.equals("mov")) {
System.out.println("这是一个视频文件。");
} else {
System.out.println("这是一个未知类型的文件。");
}
}
}