修正命名错误
xiyunfei authored at 2023-04-09 21:10:58
2.33 KiB
NewLife.JT808
using JT808.Data;
using NewLife;
using NewLife.Caching;
using NewLife.Log;
using NewLife.Model;
using NewLife.Serialization;

namespace JT808.Worker.Workers;

/// <summary>事件数据消费 Worker</summary>
public class EventWorker : IHostedService
{
    private readonly IProducerConsumer<Object>? _queue;
    private CancellationTokenSource? _cts;
    private volatile Boolean _running;

    public EventWorker(QueueService queueService)
    {
        _queue = queueService.Create<Object>("EventData", "Group");
    }

    public Task StartAsync(CancellationToken cancellationToken)
    {
        _running = true;
        _cts = CancellationTokenSource.CreateLinkedTokenSource(cancellationToken);
        _ = ConsumeAsync(_cts.Token);
        XTrace.WriteLine("EventWorker 已启动");
        return Task.CompletedTask;
    }

    public Task StopAsync(CancellationToken cancellationToken)
    {
        _running = false;
        _cts?.Cancel();
        XTrace.WriteLine("EventWorker 已停止");
        return Task.CompletedTask;
    }

    private async Task ConsumeAsync(CancellationToken cancellationToken)
    {
        while (_running && !cancellationToken.IsCancellationRequested)
        {
            try
            {
                if (_queue == null) return;
                var data = _queue.TakeOne();
                if (data != null)
                {
                    XTrace.WriteLine("消费事件数据:{0}", data);

                    // 解析事件数据并持久化
                    if (data is String json)
                    {
                        var alarm = json.ToJsonEntity<Alarm>();
                        if (alarm != null)
                        {
                            await alarm.InsertAsync();
                            XTrace.WriteLine("事件数据已持久化:DeviceId={0}, AlarmType={1}",
                                alarm.DeviceId, alarm.AlarmType);
                        }
                    }
                }
                else
                {
                    await Task.Delay(1000, cancellationToken);
                }
            }
            catch (TaskCanceledException) { break; }
            catch (Exception ex)
            {
                XTrace.WriteLine("EventWorker 异常:{0}", ex.Message);
                await Task.Delay(5000, cancellationToken);
            }
        }
    }
}