chromedp自动启动为headless模式

 

// Command click is a chromedp example demonstrating how to use a selector to
// click on an element.
package main

import (
    "context"
    "fmt"
    "github.com/chromedp/cdproto/page"
    "github.com/chromedp/chromedp"
    "github.com/chromedp/chromedp/runner"
    "log"
    "os"
    "time"
)

func main() {
    var err error

    // create context
    ctxt, cancel := context.WithCancel(context.Background())
    defer cancel()

    // create chrome instance
    //c, err := chromedp.New(ctxt, chromedp.WithLog(log.Printf))
    //if err != nil {
    //    log.Fatal(err)
    //}

    c, err := NewHeadless(ctxt, `https://xxx.xx`)
    if err != nil {
        log.Fatalf("new headless: %s",err)
    }

    // run task list
    err = c.Run(ctxt, click())
    if err != nil {
        log.Fatalf("task err: %s",err)
    }


    // shutdown chrome
    err = c.Shutdown(ctxt)
    if err != nil {
        log.Fatalf("shutdown err: %s",err)
    }

    // wait for chrome to finish
    err = c.Wait()
    if err != nil {
        log.Fatalf("wait err: %s",err)
    }
}

func click() chromedp.Tasks {
    //url := `https://free-ss.site`
    //sect := "div.main > div > table > tbody >tr:nth-child(2) > td:nth-child(9) > a"
    sect := `//a[@href="v/443.json"]`
    wd,_ := os.Getwd()
    fmt.Println(wd)
    return chromedp.Tasks{
        //chromedp.Navigate(url),
        chromedp.WaitVisible(sect),
        page.SetDownloadBehavior(page.SetDownloadBehaviorBehaviorAllow).WithDownloadPath(wd),
        chromedp.Click(sect, chromedp.NodeVisible),
        chromedp.Sleep(3 * time.Second),
    }
}


// NewHeadless 创建headless chrome实例
func NewHeadless(ctx context.Context, starturl string) (*chromedp.CDP, error) {
    // runner.Flag设置启动headless chrome时的命令行参数
    // Windows用户需要设置runner.Flag("disable-gpu", true)，具体信息参见文档的FAQ
    run, err := runner.New(runner.Flag("headless", false),runner.StartURL(starturl))

    if err != nil {
        return nil, err
    }

    // run.Start启动实例
    err = run.Start(ctx)
    if err != nil {
        return nil, err
    }

    // 默认情况chromedp会输出大量log，因为是示例所以选择屏蔽，dropChromeLogs为自定义函数，形式为func(string, ...interface{}){}
    // 使用runner初始化chromedp实例
    // 实例在使用完毕后需要调用c.Shutdown()来释放资源
    //c, err := chromedp.New(ctx, chromedp.WithRunner(run), chromedp.WithErrorf(dropChromeLogs))
    c, err := chromedp.New(ctx, chromedp.WithRunner(run), chromedp.WithLog(log.Printf))
    if err != nil {
        return nil, err
    }

    return c, nil
}