From 110301a975e43739192577166d089e28c22ae266 Mon Sep 17 00:00:00 2001 From: Keuin Date: Sat, 29 Jul 2023 21:20:12 +0800 Subject: Add API server --- recording/runner.go | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) (limited to 'recording/runner.go') diff --git a/recording/runner.go b/recording/runner.go index 9be78c9..7bf3790 100644 --- a/recording/runner.go +++ b/recording/runner.go @@ -175,7 +175,9 @@ func tryRunTask(t *RunningTask) error { var err error run := true for run { - err = record(t.ctx, bi, &t.TaskConfig, t.logger) + err = record(t.ctx, bi, &t.TaskConfig, t.logger, func(resp types.RoomProfileResponse) { + t.roomTitle.Store(&resp.Data.Title) + }) if err == nil { // live is ended t.logger.Info("The live is ended. Restarting current task...") @@ -247,6 +249,7 @@ func record( bi *bilibili.Bilibili, task *TaskConfig, logger logging.Logger, + profileConsumer func(types.RoomProfileResponse), ) error { logger.Info("Getting room profile...") @@ -265,6 +268,8 @@ func record( return errs.NewError(errs.GetRoomInfo, err) } + profileConsumer(profile) + logger.Info("Getting stream url...") urlInfo, err := AutoRetryWithConfig( ctx, -- cgit v1.2.3