lixiangwuxian/qq_bot
0
0
Fork 0
Code Issues 3 Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: 修改网页为远程执行

Browse Source
This commit is contained in:
lixiangwuxian 2024-10-20 01:27:15 +08:00
parent 66d0d683e8
commit 235d22a4c0
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
util/web_page_shot.go
View File

@ -15,9 +15,10 @@ import (
// ScreenshotURL 截图函数：传入网址、输出路径、宽高、四个边距和等待的元素 ID // ScreenshotURL 截图函数：传入网址、输出路径、宽高、四个边距和等待的元素 ID
func ScreenshotURL(url, output string, width, height int, marginTop, marginBottom, marginRight, marginLeft int, waitClass string) error { func ScreenshotURL(url, output string, width, height int, marginTop, marginBottom, marginRight, marginLeft int, waitClass string) error {
// 创建一个上下文，连接到 Docker 中运行的 headless-shell 实例 // 创建一个上下文，连接到 Docker 中运行的 headless-shell 实例
remoteAllocatorCtx, cancel := chromedp.NewRemoteAllocator( remoteAllocatorCtx, cancel := chromedp.NewRemoteAllocator(
context.Background(), "ws://127.0.0.1:9222/json/ws", context.Background(), "ws://100.124.180.117:9222/json/ws",
) )
defer cancel() defer cancel()
@ -69,7 +70,6 @@ func ScreenshotURL(url, output string, width, height int, marginTop, marginBotto
if err := chromedp.Run(ctx, network.SetExtraHTTPHeaders(network.Headers(headers))); err != nil { if err := chromedp.Run(ctx, network.SetExtraHTTPHeaders(network.Headers(headers))); err != nil {
return fmt.Errorf("设置请求头失败: %w", err) return fmt.Errorf("设置请求头失败: %w", err)
} }
// 执行任务：打开网页并截图 // 执行任务：打开网页并截图
err := chromedp.Run(ctx, err := chromedp.Run(ctx,
chromedp.Navigate(url), // 打开网页 chromedp.Navigate(url), // 打开网页