用java 帮我写图片后缀和视频后缀的方法

在Java中，你可以创建一个方法来检查文件的后缀名是否为图片或视频格式。这通常涉及到获取文件的扩展名，并与一组预定义的图片和视频后缀名进行比较。以下是一个简单的示例方法，它接受一个文件名作为参数，然后返回该文件是否是图片或视频文件：

import java.io.File;

public class FileExtensionChecker {

    // 预定义的图片和视频后缀名数组
    private static final String[] IMAGE_EXTENSIONS = {".jpg", ".jpeg", ".png", ".gif", ".bmp", ".tiff", ".webp"};
    private static final String[] VIDEO_EXTENSIONS = {".mp4", ".avi", ".mov", ".wmv", ".flv", ".mkv", ".mpg", ".ogg"};

    /**
     * 检查文件后缀是否为图片或视频格式。
     *
     * @param fileName 文件名
     * @return 如果是图片或视频后缀返回true，否则返回false。
     */
    public static boolean isImageOrVideo(String fileName) {
        if (fileName == null || fileName.isEmpty()) {
            return false;
        }

        // 获取文件的扩展名
        String extension = getFileExtension(fileName);

        // 检查是否为图片后缀
        for (String imgExt : IMAGE_EXTENSIONS) {
            if (extension.equalsIgnoreCase(imgExt)) {
                return true;
            }
        }

        // 检查是否为视频后缀
        for (String videoExt : VIDEO_EXTENSIONS) {
            if (extension.equalsIgnoreCase(videoExt)) {
                return true;
            }
        }

        return false;
    }

    /**
     * 获取文件的扩展名。
     *
     * @param fileName 文件名
     * @return 文件的扩展名，如果不存在则返回空字符串。
     */
    private static String getFileExtension(String fileName) {
        int dotIndex = fileName.lastIndexOf('.');
        if (dotIndex > 0 && dotIndex < fileName.length() - 1) {
            return fileName.substring(dotIndex).toLowerCase();
        }
        return "";
    }

    public static void main(String[] args) {
        // 测试方法
        String testFileName = "example.jpg";
        System.out.println("Is " + testFileName + " an image or video file? " + isImageOrVideo(testFileName));
    }
}

在这个例子中，`isImageOrVideo` 方法用于检查文件名是否以图片或视频后缀结尾。`getFileExtension` 方法用于从文件名中提取后缀名。通过比较提取出的后缀名与预定义的图片和视频后缀数组，可以确定文件是否属于这两种类型之一。