From 014497a8c856e0a8235beeb22987ef328d707cf5 Mon Sep 17 00:00:00 2001
From: "lilong.129" <lilong.129@bytedance.com>
Date: Wed, 20 Sep 2023 22:35:33 +0800
Subject: [PATCH] feat: enter/exit live room by chance

---
 hrp/pkg/uixt/video_crawler.go | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/hrp/pkg/uixt/video_crawler.go b/hrp/pkg/uixt/video_crawler.go
index 0c1f6c0f..f7b92599 100644
--- a/hrp/pkg/uixt/video_crawler.go
+++ b/hrp/pkg/uixt/video_crawler.go
@@ -2,6 +2,7 @@ package uixt
 
 import (
 	"math"
+	"math/rand"
 	"time"
 
 	"github.com/pkg/errors"
@@ -222,6 +223,13 @@ func (dExt *DriverExt) VideoCrawler(configs *VideoCrawlerConfigs) (err error) {
 					break
 				} else {
 					time.Sleep(1 * time.Second)
+
+					if rand.Float64() >= 0.75 {
+						// 75% chance enter live room
+						log.Info().Msg("skip enter live room by 25% chance")
+						break
+					}
+
 					// live target not achieved, enter live
 					entryPoint := PointF{
 						X: float64(dExt.windowSize.Width / 2),
@@ -278,9 +286,17 @@ func (dExt *DriverExt) VideoCrawler(configs *VideoCrawlerConfigs) (err error) {
 				screenResultFromOCR.SwipeFinishTime = screenResult.SwipeFinishTime
 				screenResultFromOCR.TotalElapsed = time.Since(swipeFinishTime).Milliseconds()
 
+				var exitLive bool
 				if crawler.isLiveTargetAchieved() {
 					log.Info().Interface("live", screenResult.Video).
 						Msg("live count achieved, exit live house")
+					exitLive = true
+				} else if rand.Float64() <= 0.25 {
+					// 25% chance exit live room
+					log.Info().Msg("exit live room by 25% chance")
+					exitLive = true
+				}
+				if exitLive {
 					err = crawler.exitLiveRoom()
 					if err != nil {
 						if errors.Is(err, code.TimeoutError) || errors.Is(err, code.InterruptError) {