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

在Java中，获取文件后缀名通常可以通过文件的名称来处理。以下是一个简单的Java方法，用于根据文件名获取图片和视频的后缀名：

public class FileSuffixHelper {

    /**
     * 获取文件后缀名
     *
     * @param fileName 文件名
     * @return 文件后缀名（包含点号）或null，如果文件名不符合常规命名规则
     */
    public static String getFileSuffix(String fileName) {
        if (fileName != null) {
            int dotIndex = fileName.lastIndexOf('.');
            if (dotIndex > 0 && dotIndex < fileName.length() - 1) {
                return fileName.substring(dotIndex);
            }
        }
        return null;
    }

    /**
     * 获取图片后缀名
     *
     * @param fileName 文件名
     * @return 图片文件后缀名或null，如果文件不是图片格式
     */
    public static String getImageSuffix(String fileName) {
        String suffix = getFileSuffix(fileName);
        if (suffix != null) {
            // 常见的图片格式后缀
            String[] imageSuffixes = {".jpg", ".jpeg", ".png", ".gif", ".bmp", ".tiff", ".webp"};
            for (String imageSuffix : imageSuffixes) {
                if (suffix.equalsIgnoreCase(imageSuffix)) {
                    return imageSuffix;
                }
            }
        }
        return null;
    }

    /**
     * 获取视频后缀名
     *
     * @param fileName 文件名
     * @return 视频文件后缀名或null，如果文件不是视频格式
     */
    public static String getVideoSuffix(String fileName) {
        String suffix = getFileSuffix(fileName);
        if (suffix != null) {
            // 常见的视频格式后缀
            String[] videoSuffixes = {".mp4", ".avi", ".mov", ".flv", ".mkv", ".wmv", ".rmvb"};
            for (String videoSuffix : videoSuffixes) {
                if (suffix.equalsIgnoreCase(videoSuffix)) {
                    return videoSuffix;
                }
            }
        }
        return null;
    }
    
    public static void main(String[] args) {
        // 测试获取后缀名的方法
        String fileName1 = "example.jpg";
        String fileName2 = "example.mp4";
        System.out.println("图片文件后缀：" + getImageSuffix(fileName1));
        System.out.println("视频文件后缀：" + getVideoSuffix(fileName2));
    }
}

这段代码中定义了三个方法：

1. `getFileSuffix`：用于获取文件的基本后缀名。
2. `getImageSuffix`：用于获取图片文件的后缀名。
3. `getVideoSuffix`：用于获取视频文件的后缀名。

在`main`方法中，我们用两个示例文件名测试了这些方法，并打印出相应的后缀名。