From dd1a20ec9ba746dd5e0fe047c98eadce7cd0b71d Mon Sep 17 00:00:00 2001
From: yanlongqi <yuchat@163.com>
Date: Mon, 30 Dec 2024 22:46:33 +0800
Subject: [PATCH] =?UTF-8?q?feat(video):=20=E4=BC=98=E5=8C=96=E8=A7=86?=
 =?UTF-8?q?=E9=A2=91=20URL=E6=A3=80=E9=AA=8C=E5=92=8C=E8=8E=B7=E5=8F=96?=
 =?UTF-8?q?=E9=80=BB=E8=BE=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 在 HttpRequestComponent 中添加 checkUrl 方法，用于检查 URL 是否有效
- 在 MadouVideoService 中实现 getMadouVideoInfo 方法，整合视频信息获取和 URL 检查
- 修改 VideoController 中的 getMadouVideoById 方法，使用新的视频信息获取逻辑
---
 .../models/controller/VideoController.java    |  3 +--
 .../models/service/MadouVideoService.java     | 12 +++++++++
 .../video/utils/HttpRequestComponent.java     | 26 +++++++++++++++++++
 3 files changed, 39 insertions(+), 2 deletions(-)

diff --git a/src/main/java/top/yuchat/crawler/video/models/controller/VideoController.java b/src/main/java/top/yuchat/crawler/video/models/controller/VideoController.java
index ba5f3a1..7eeed59 100644
--- a/src/main/java/top/yuchat/crawler/video/models/controller/VideoController.java
+++ b/src/main/java/top/yuchat/crawler/video/models/controller/VideoController.java
@@ -44,8 +44,7 @@ public class VideoController {
 
     @GetMapping("/madou/{id}")
     public JsonResult<MadouVideoInfo> getMadouVideoById(@PathVariable("id") Long id) {
-        MadouVideoInfo madouVideoInfo = madouVideoService.getById(id);
-        madouVideoInfo.setM3u8Url("/api/video/" + id + "/index.m3u8");
+        MadouVideoInfo madouVideoInfo = madouVideoService.getMadouVideoInfo(id);
         return JsonResult.ok(madouVideoInfo);
     }
 
diff --git a/src/main/java/top/yuchat/crawler/video/models/service/MadouVideoService.java b/src/main/java/top/yuchat/crawler/video/models/service/MadouVideoService.java
index 93dbd70..77960c9 100644
--- a/src/main/java/top/yuchat/crawler/video/models/service/MadouVideoService.java
+++ b/src/main/java/top/yuchat/crawler/video/models/service/MadouVideoService.java
@@ -260,4 +260,16 @@ public class MadouVideoService extends ServiceImpl<MadouVideoMapper, MadouVideoI
         }
         httpRequestComponent.download(videoInfo.getM3u8Url().replace("index.m3u8", ts), httpServletResponse.getOutputStream());
     }
+
+    public MadouVideoInfo getMadouVideoInfo(Long id) {
+        MadouVideoInfo madouVideoInfo = getById(id);
+        if (madouVideoInfo == null) {
+            throw new RuntimeException("视频不存在");
+        }
+        String url = "https://video.yuchat.top/m3u8/" + id + "/index.m3u8";
+        if (httpRequestComponent.checkUrl(url)) {
+            madouVideoInfo.setM3u8Url(url);
+        }
+        return madouVideoInfo;
+    }
 }
diff --git a/src/main/java/top/yuchat/crawler/video/utils/HttpRequestComponent.java b/src/main/java/top/yuchat/crawler/video/utils/HttpRequestComponent.java
index d1444e1..f248d29 100644
--- a/src/main/java/top/yuchat/crawler/video/utils/HttpRequestComponent.java
+++ b/src/main/java/top/yuchat/crawler/video/utils/HttpRequestComponent.java
@@ -3,6 +3,7 @@ package top.yuchat.crawler.video.utils;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.http.HttpEntity;
 import org.apache.http.HttpStatus;
+import org.apache.http.client.ClientProtocolException;
 import org.apache.http.client.config.RequestConfig;
 import org.apache.http.client.methods.CloseableHttpResponse;
 import org.apache.http.client.methods.HttpGet;
@@ -126,6 +127,31 @@ public class HttpRequestComponent {
         IOUtils.copy(entity.getContent(), os);
     }
 
+    public boolean checkUrl(String url) {
+        HttpGet httpGet = new HttpGet(url);
+        httpGet.setConfig(requestConfig);
+        CloseableHttpResponse response = null;
+        try {
+            response = closeableHttpClient.execute(httpGet);
+            int statusCode = response.getStatusLine().getStatusCode();
+            if (HttpStatus.SC_OK == statusCode) {
+                return true;
+            }
+        } catch (Exception ignored) {
+        } finally {
+            if (response != null) {
+                try {
+                    response.close();
+                } catch (IOException e) {
+                    log.error("Failed to close response", e);
+                }
+            }
+            // 关闭 httpGet 请求
+            httpGet.releaseConnection();
+        }
+        return false;
+    }
+
 
     public long getFileSize(String url) throws IOException {
         HttpGet httpGet = new HttpGet(url);