telboon/ktm-booking-bot
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Feat(booking-job): Auto timeout close browser
All checks were successful
ktm-booking-bot/ktm-booking-bot/pipeline/head This commit looks good
Details

Browse Source
This commit is contained in:
Samuel Pua
2022-10-02 00:19:58 +08:00
parent 635352edbd
commit 05126fa110
1 changed files with 13 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
backend/internal/ktmtrainbot/backgroundbookingjob.go
View File

@@ -169,6 +169,19 @@ func (env *Env) startBooking(job *Booking, username string, password string, cre
defaultDevice.UserAgent = "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/102.0.0.0 Safari/537.36" defaultDevice.UserAgent = "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/102.0.0.0 Safari/537.36"
browser := rod.New().ControlURL(u).MustConnect().DefaultDevice(defaultDevice) browser := rod.New().ControlURL(u).MustConnect().DefaultDevice(defaultDevice)
// Close browser when context is done
go func() {
defer func() {
if r := recover(); r != nil {
log.Println("Recovering from browser panic...")
}
}()
<-timerCtx.Done()
log.Println("Timed out. Closing browser.")
browser.MustClose()
}()
// Defer closing browser // Defer closing browser
defer browser.MustClose() defer browser.MustClose()