chromedpでのタイムアウトする方法がわからない

Question

実現したいこと

golangのchromedpについて質問させてください。
chromedp.Runの中で、WaitVisibleなどで何か要素を待っている場合、
指定時間が経過しても待っている要素が表示されない場合に、
タイムアウトして、chromedp.Runの処理を抜け、その先の処理が進むようにしたいです。

発生している問題・分からないこと

何も考えず一生現れない要素を指定するとWaitVisibleで永遠待たれる事になります・・・
（厳密には何分かすればタイムアウトする？？）

そこでタイムアウト機能をもたせようと、context.WithTimeoutを使ってみたのですが、
処理が正常に進んでもしなくても、指定時間でブラウザ全体が終了してしまいます。。
ブラウザは終了させず、残したままにしておきたいのです。

該当のソースコード

package main

import (
        "context"
        "fmt"
        "log"
        "time"

        "github.com/chromedp/chromedp"
)

func main() {
        // ブラウザ起動オプションの設定
        opts := append(chromedp.DefaultExecAllocatorOptions[:],
                chromedp.Flag("headless", false), // ヘッドレスモードを解除
        )

        allocCtx, cancel := chromedp.NewExecAllocator(context.Background(), opts...)
        defer cancel()

        ctx, cancel := chromedp.NewContext(allocCtx)
        defer cancel()

        // タイムアウトを設定
        ctx, cancel := context.WithTimeout(ctx, 7*time.Second)
        defer cancel()

        // タスクの実行
        var res string
        err := chromedp.Run(ctx,
                chromedp.Navigate("https://www.yahoo.co.jp/"), // Yahooのページを開く
                chromedp.Title(&res),                          // ページタイトルを取得
                chromedp.WaitVisible("#hoge"),           // 出てこないであろう要素を待ってみる
        )
        if err != nil {
                // ここでタイムアウトを検知して何かの処理をする。
                // ブラウザは終了させたくない。
                fmt.Printf("エラーがでた: %s\n", err.Error())
        }

        log.Printf("Page Title: %s", res)

        fmt.Println("ブラウザは開いたままです。Enterキーを押して終了してください...")
        fmt.Scanln() // ユーザーがEnterを押すまで待機
}

試したこと・調べたこと

• teratailやGoogle等で検索した
• ソースコードを自分なりに変更した
• 知人に聞いた
• その他

上記の詳細・結果

調べてはみたのですが、解決ができませんでした・・・
自分が欲しい情報にたどり着けなかったです。。

補足

特になし

Answer 1

https://github.com/chromedp/chromedp/issues/37#issuecomment-548271460

を試してみました。

package main

import (
	"context"
	"fmt"
	"log"
	"time"

	"github.com/chromedp/chromedp"
)

func RunWithTimeOut(ctx *context.Context, timeout time.Duration, tasks chromedp.Tasks) chromedp.ActionFunc {
	return func(ctx context.Context) error {
		timeoutContext, cancel := context.WithTimeout(ctx, timeout*time.Second)
		defer cancel()
		return tasks.Do(timeoutContext)
	}
}

func main() {
	// ブラウザ起動オプションの設定
	opts := append(chromedp.DefaultExecAllocatorOptions[:],
		chromedp.Flag("headless", false), // ヘッドレスモードを解除
	)

	allocCtx, cancel := chromedp.NewExecAllocator(context.Background(), opts...)
	defer cancel()

	ctx, cancel := chromedp.NewContext(allocCtx)
	defer cancel()


	// タスクの実行
	var res string
	err := chromedp.Run(ctx,
		chromedp.Navigate("https://www.yahoo.co.jp/"), // Yahooのページを開く
		chromedp.Title(&res),                          // ページタイトルを取得
		// chromedp.WaitVisible("#hoge"),                 // 出てこないであろう要素を待ってみる
		RunWithTimeOut(&ctx, 3, chromedp.Tasks{
			chromedp.WaitVisible("#hoge"),
		}),
	)
	if err != nil {
		// ここでタイムアウトを検知して何かの処理をする。
		// ブラウザは終了させたくない。
		fmt.Printf("エラーがでた: %s\n", err.Error())
	}

	log.Printf("Page Title: %s", res)

	fmt.Println("ブラウザは開いたままです。Enterキーを押して終了してください...")
	fmt.Scanln() // ユーザーがEnterを押すまで待機
}

でどうでしょう。

エラーがでた: context deadline exceeded
2024/12/31 11:38:48 Page Title: Yahoo! JAPAN
ブラウザは開いたままです。Enterキーを押して終了してください...

のような動きでした。Enterを押すブラウザーを閉じます。

Comments

[night]

ご回答ありがとうございます！
期待通りに動作しました
大変助かりました、ありがとうございました！