BililiveRecorder/BililiveRecorder.Web/Models/Rest/Logs/WebApiLogEventSink.cs

using System;
using System.Collections.Generic;
using System.IO;
using System.Threading;
using Serilog.Core;
using Serilog.Events;
using Serilog.Formatting;

namespace BililiveRecorder.Web.Models.Rest.Logs
{
    public class WebApiLogEventSink : ILogEventSink
    {
        public static WebApiLogEventSink? Instance;

        private const int MAX_LOG = 100;

        private readonly ReaderWriterLockSlim readerWriterLock = new();
        private readonly ITextFormatter textFormatter;

        private readonly Queue<JsonLog> logs = new Queue<JsonLog>();

        private int logId = 0;

        public WebApiLogEventSink(ITextFormatter textFormatter)
        {
            this.textFormatter = textFormatter ?? throw new ArgumentNullException(nameof(textFormatter));
        }

        public void Emit(LogEvent logEvent)
        {
            using var writer = new StringWriter();
            this.textFormatter.Format(logEvent, writer);
            var json = writer.ToString();

            if (this.readerWriterLock.TryEnterWriteLock(5000))
            {
                try
                {
                    this.logs.Enqueue(new JsonLog { Id = Interlocked.Increment(ref this.logId), Log = json });

                    while (this.logs.Count > MAX_LOG)
                        this.logs.Dequeue();
                }
                finally
                {
                    this.readerWriterLock.ExitWriteLock();
                }
            }
        }

        public void ReadLogs(Action<Queue<JsonLog>> callback)
        {
            if (this.readerWriterLock.TryEnterReadLock(1000))
            {
                try
                {
                    callback(this.logs);
                }
                finally
                {
                    this.readerWriterLock.ExitReadLock();
                }
            }
        }
    }
}