BililiveRecorder/BililiveRecorder.Core/Api/PolicyWrappedApiClient.cs

using System;
using System.Threading.Tasks;
using BililiveRecorder.Core.Api.Model;
using Polly;
using Polly.Registry;

namespace BililiveRecorder.Core.Api
{
    internal class PolicyWrappedApiClient<T> : IApiClient, IDanmakuServerApiClient, IDisposable where T : class, IApiClient, IDanmakuServerApiClient, IDisposable
    {
        private readonly T client;
        private readonly IReadOnlyPolicyRegistry<string> policies;

        public PolicyWrappedApiClient(T client, IReadOnlyPolicyRegistry<string> policies)
        {
            this.client = client ?? throw new ArgumentNullException(nameof(client));
            this.policies = policies ?? throw new ArgumentNullException(nameof(policies));
        }

        public long GetUid() => this.client.GetUid();

        public async Task<BilibiliApiResponse<DanmuInfo>> GetDanmakuServerAsync(int roomid) => await this.policies
            .Get<IAsyncPolicy>(PolicyNames.PolicyDanmakuApiRequestAsync)
            .ExecuteAsync(_ => this.client.GetDanmakuServerAsync(roomid), new Context(PolicyNames.CacheKeyDanmaku + ":" + roomid))
            .ConfigureAwait(false);

        public async Task<BilibiliApiResponse<RoomInfo>> GetRoomInfoAsync(int roomid) => await this.policies
            .Get<IAsyncPolicy>(PolicyNames.PolicyRoomInfoApiRequestAsync)
            .ExecuteAsync(_ => this.client.GetRoomInfoAsync(roomid), new Context(PolicyNames.CacheKeyRoomInfo + ":" + roomid))
            .ConfigureAwait(false);

        public async Task<BilibiliApiResponse<RoomPlayInfo>> GetStreamUrlAsync(int roomid, int qn) => await this.policies
            .Get<IAsyncPolicy>(PolicyNames.PolicyStreamApiRequestAsync)
            .ExecuteAsync(_ => this.client.GetStreamUrlAsync(roomid, qn), new Context(PolicyNames.CacheKeyStream + ":" + roomid + ":" + qn))
            .ConfigureAwait(false);

        public void Dispose() => this.client.Dispose();
    }
}