如何访问数据库，而运行一个webdriver使用chromedp?

Question

我想在银行页面上自动提交OTP。只有在webdriver在银行页面上点击确认后，我才能在我的数据库中获得OTP。确认后，我需要从数据库中获取OTP，然后自动提交OTP。

ctx, cancel := chromedp.NewContext(context.Background(),      chromedp.WithDebugf(log.Printf))
defer cancel()
// run chromedp tasks
err := chromedp.Run(ctx,
chromedp.Navigate(bankUrl),
chromedp.WaitVisible(`#username`),
chromedp.SendKeys(`#username`, `usernameXXX`),
chromedp.WaitVisible(`#label2`, ),
chromedp.SendKeys(`#label2`, `passwordxxx` ),
chromedp.Click(`//input[@title="Login"]`),
chromedp.WaitVisible(`#Go`),
chromedp.Click(`#Go`),
chromedp.WaitVisible(`#confirmButton`),
chromedp.Click(`#confirmButton`),
chromedp.WaitVisible(`//input[@type="password"]`),
// perform  fetch OTP below, this raise error
otp := fetchOTPFromDb()
chromedp.SendKeys(`//input[@type="password"]`, otp),
chromedp.WaitVisible(`#confirmButton`),
chromedp.Click(`#confirmButton`))
if err != nil {
log.Fatal(err)
}

问题是chromedp。Run期望所有参数都是chromedp类型。任务，所以我不能在那里调用自定义函数，我得到错误，而从数据库获取OTP。我怎么绕过这个?

Answer 1

解决方案是在Action.Do调用中包装otp获取，然后将调用结果返回给chromdp.SendKeys以设置HTML输入值。

需要这样工作，因为一次性密码在获取页面之前不存在，因此，它的读取必须在操作资源时进行。

这样的

package main
import "context"
type OTPAction struct {
// DB ....
}
func (a OTPAction) Do(ctx context.Context) error {
// fetch OTP here
otp := "otp test"
return chromedp.SendKeys(`//input[@id="user-message"]`, otp).Do(ctx)
}