Java中如何高效识别音频URL是否为流媒体

import java.io.IOException;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.Set;
import java.util.Arrays;
import java.util.HashSet;

public class AudioStreamDetector {

    // 定义常见的流媒体MIME类型集合。
    // 注意：audio/mpeg对于静态MP3和流媒体MP3都可能出现，
    // 这里将其视为流媒体的常见类型，但需结合实际场景判断。
    private static final Set<String> STREAMING_MIME_TYPES = new HashSet<>(Arrays.asList(
        "audio/mpeg",                       // MP3流 (如网络电台)
        "audio/aac",                        // AAC流
        "audio/aacp",                       // AAC+流
        "audio/ogg",                        // Ogg Vorbis/Opus流
        "audio/webm",                       // WebM音频流
        "application/vnd.apple.mpegurl",    // HLS播放列表 (.m3u8)
        "application/x-mpegurl"             // HLS播放列表 (.m3u8, 旧版或非标准)
        // 根据需要可扩展更多流媒体MIME类型
    ));

    /**
     * 判断给定的音频URL是否指向流媒体。
     * 该方法通过发送HTTP HEAD请求，检查响应的Content-Type头部信息。
     *
     * @param audioUrlStr 音频URL字符串。
     * @return 如果根据Content-Type判断为流媒体则返回true，否则返回false。
     */
    public static boolean isStreamingAudio(String audioUrlStr) {
        HttpURLConnection connection = null;
        try {
            URL url = new URL(audioUrlStr);
            connection = (HttpURLConnection) url.openConnection();
            connection.setRequestMethod("HEAD"); // 设置请求方法为HEAD
            connection.setConnectTimeout(5000);  // 设置连接超时时间为5秒
            connection.setReadTimeout(5000);     // 设置读取超时时间为5秒

            // 发送请求并获取响应码
            int responseCode = connection.getResponseCode();

            // 检查响应是否成功
            if (responseCode == HttpURLConnection.HTTP_OK) {
                String contentType = connection.getHeaderField("Content-Type");
                if (contentType != null) {
                    // 清理Content-Type字符串，移除charset等参数，并转换为小写
                    contentType = contentType.toLowerCase().split(";")[0].trim();
                    System.out.println("URL: " + audioUrlStr + ", Content-Type: " + contentType);

                    // 判断Content-Type是否在预定义的流媒体类型集合中
                    return STREAMING_MIME_TYPES.contains(contentType);
                }
            } else {
                System.err.println("无法获取URL的头部信息: " + audioUrlStr + ", 响应码: " + responseCode);
            }
        } catch (IOException e) {
            System.err.println("检查URL时发生IO错误 " + audioUrlStr + ": " + e.getMessage());
        } finally {
            // 确保关闭连接
            if (connection != null) {
                connection.disconnect();
            }
        }
        return false;
    }

    public static void main(String[] args) {
        // 示例：请替换为实际的流媒体和静态音频URL进行测试
        String streamingRadioUrl = "http://stream.psychomed.gr:8000/stream"; // 示例网络电台
        String staticMp3Url = "https://www.soundhelix.com/examples/audio/20230913_m8.mp3"; // 示例静态MP3文件
        String hlsPlaylistUrl = "https://playertest.longtailvideo.com/adaptive/oceans_aes/oceans_aes.m3u8"; // 示例HLS播放列表

        System.out.println("URL: '" + streamingRadioUrl + "' 是流媒体吗? " + isStreamingAudio(streamingRadioUrl));
        System.out.println("URL: '" + staticMp3Url + "' 是流媒体吗? " + isStreamingAudio(staticMp3Url));
        System.out.println("URL: '" + hlsPlaylistUrl + "' 是流媒体吗? " + isStreamingAudio(hlsPlaylistUrl));

        // 更多测试用例
        String anotherStreamingUrl = "http://radio.flexy.media:8000/radio.mp3";
        String nonExistentUrl = "http://this.url.does.not.exist/audio.mp3";

        System.out.println("URL: '" + anotherStreamingUrl + "' 是流媒体吗? " + isStreamingAudio(anotherStreamingUrl));
        System.out.println("URL: '" + nonExistentUrl + "' 是流媒体吗? " + isStreamingAudio(nonExistentUrl));
    }
}