Austin/BililiveRecorder
1
0
Fork 0
mirror of https://github.com/BililiveRecorder/BililiveRecorder.git synced 2024-11-15 19:22:19 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat(core): hold danmaku connect until stream info fetched

Browse Source
This commit is contained in:
genteure 2023-07-21 18:51:11 +08:00
parent 071087bea1
commit b826ebf611
1 changed files with 17 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
BililiveRecorder.Core/Room.cs
View File

@ -55,10 +55,10 @@ namespace BililiveRecorder.Core
private bool autoRecordForThisSession = true; private bool autoRecordForThisSession = true;
private IRecordTask? recordTask; private IRecordTask? recordTask;
private DateTimeOffset recordTaskStartTime;
private DateTimeOffset danmakuClientConnectTime; private DateTimeOffset danmakuClientConnectTime;
private static readonly TimeSpan danmakuClientReconnectNoDelay = TimeSpan.FromMinutes(1); private readonly ManualResetEventSlim danmakuConnectHoldOff = new();
private static readonly TimeSpan danmakuClientReconnectNoDelay = TimeSpan.FromMinutes(1);
private static readonly HttpClient coverDownloadHttpClient = new HttpClient(); private static readonly HttpClient coverDownloadHttpClient = new HttpClient();
static Room() static Room()
@ -220,6 +220,9 @@ namespace BililiveRecorder.Core
this.Name = room.User.BaseInfo.Name; this.Name = room.User.BaseInfo.Name;
this.RawBilibiliApiJsonData = room.RawBilibiliApiJsonData; this.RawBilibiliApiJsonData = room.RawBilibiliApiJsonData;
// allow danmaku client to connect
this.danmakuConnectHoldOff.Set();
} }
} }
@ -244,7 +247,6 @@ namespace BililiveRecorder.Core
task.RecordFileClosed += this.RecordTask_RecordFileClosed; task.RecordFileClosed += this.RecordTask_RecordFileClosed;
task.RecordSessionEnded += this.RecordTask_RecordSessionEnded; task.RecordSessionEnded += this.RecordTask_RecordSessionEnded;
this.recordTask = task; this.recordTask = task;
this.recordTaskStartTime = DateTimeOffset.UtcNow;
this.Stats.Reset(); this.Stats.Reset();
this.OnPropertyChanged(nameof(this.Recording)); this.OnPropertyChanged(nameof(this.Recording));
@ -370,6 +372,18 @@ namespace BililiveRecorder.Core
} }
} }
// 至少要等到获取到一次房间信息后才能连接弹幕服务器。
// 理论上不专门写这么个逻辑也应该没有问题的，但为了保证连接时肯定有主播 uid 信息还是加上吧。
// 这里同步堵塞等待了，因为 ManualResetEventSlim 没有 Async Wait
// 并且这里运行在一个新的 Task 里，同步等待的影响也不是很大。
if (!this.danmakuConnectHoldOff.Wait(TimeSpan.FromSeconds(2)))
{
// 如果 2 秒后还没有获取到房间信息直接返回
// 房间信息获取成功后会自动再次尝试连接弹幕服务器所以不用担心会导致不连接弹幕服务器
this.logger.Debug("暂无房间信息，不连接弹幕服务器");
return;
}
await this.danmakuClient.ConnectAsync(this.RoomConfig.RoomId, this.RoomConfig.DanmakuTransport, this.ct).ConfigureAwait(false); await this.danmakuClient.ConnectAsync(this.RoomConfig.RoomId, this.RoomConfig.DanmakuTransport, this.ct).ConfigureAwait(false);
} }
catch (Exception ex) catch (Exception ex)