45 lines
1,016 B
Go
45 lines
1,016 B
Go
package utils
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/chromedp/cdproto/page"
|
|
"github.com/chromedp/chromedp"
|
|
)
|
|
|
|
// WaitForNetworkIdle waits for the network to be idle (no pending requests)
|
|
// for the specified duration. This is useful for pages that load content dynamically.
|
|
func WaitForNetworkIdle(timeout time.Duration) chromedp.ActionFunc {
|
|
return func(ctx context.Context) error {
|
|
// Enable lifecycle events
|
|
if err := page.SetLifecycleEventsEnabled(true).Do(ctx); err != nil {
|
|
return err
|
|
}
|
|
|
|
ch := make(chan struct{}, 1)
|
|
cctx, cancel := context.WithCancel(ctx)
|
|
defer cancel()
|
|
|
|
chromedp.ListenTarget(cctx, func(ev interface{}) {
|
|
if e, ok := ev.(*page.EventLifecycleEvent); ok {
|
|
if e.Name == "networkIdle" {
|
|
select {
|
|
case ch <- struct{}{}:
|
|
default:
|
|
}
|
|
}
|
|
}
|
|
})
|
|
|
|
select {
|
|
case <-ch:
|
|
return nil
|
|
case <-time.After(timeout):
|
|
// Timeout reached, but don't error - page may still be usable
|
|
return nil
|
|
case <-ctx.Done():
|
|
return ctx.Err()
|
|
}
|
|
}
|
|
}
|