OmronFinsUdpSimulatorServer.cs

이 파일의 문서화 페이지로 가기

using System.Net;
using System.Net.Sockets;
using Dreamine.PLC.Core.Memory;
 
namespace Dreamine.PLC.Omron.Fins.Simulation;

public sealed class OmronFinsUdpSimulatorServer : IAsyncDisposable
{
    private readonly OmronFinsSimulatorServerOptions _options;
    private readonly InMemoryPlcMemory _memory;
    private readonly OmronFinsSimulatorProtocol _protocol;
    private UdpClient? _udpClient;
    private CancellationTokenSource? _cts;
    private Task? _receiveTask;

    public OmronFinsUdpSimulatorServer(OmronFinsSimulatorServerOptions options)
        : this(options, new InMemoryPlcMemory())
    {
    }

    public OmronFinsUdpSimulatorServer(OmronFinsSimulatorServerOptions options, InMemoryPlcMemory memory)
    {
        _options = options ?? throw new ArgumentNullException(nameof(options));
        _memory = memory ?? throw new ArgumentNullException(nameof(memory));
        _protocol = new OmronFinsSimulatorProtocol(_memory, _options);
        _protocol.StatusChanged += (_, message) => StatusChanged?.Invoke(this, message);
    }

    public event EventHandler<string>? StatusChanged;

    public bool IsRunning => _udpClient is not null;

    public Task StartAsync()
    {
        if (IsRunning)
        {
            return Task.CompletedTask;
        }
 
        var address = ParseAddress(_options.Host);
 
        _cts = new CancellationTokenSource();
        _udpClient = new UdpClient(new IPEndPoint(address, _options.Port));
        _receiveTask = ReceiveLoopAsync(_cts.Token);
        StatusChanged?.Invoke(this, $"FINS UDP simulator listening on {_options.Host}:{_options.Port}.");
        return Task.CompletedTask;
    }

    public async Task StopAsync()
    {
        if (!IsRunning)
        {
            return;
        }
 
        _cts?.Cancel();
        _udpClient?.Dispose();
        _udpClient = null;
 
        if (_receiveTask is not null)
        {
            try
            {
                await _receiveTask.ConfigureAwait(false);
            }
            catch (OperationCanceledException)
            {
            }
            catch (ObjectDisposedException)
            {
            }
            catch (SocketException)
            {
            }
        }
 
        _cts?.Dispose();
        _cts = null;
        _receiveTask = null;
        StatusChanged?.Invoke(this, "FINS UDP simulator stopped.");
    }

    public async ValueTask DisposeAsync()
    {
        await StopAsync().ConfigureAwait(false);
    }

    private static IPAddress ParseAddress(string host)
    {
        if (string.IsNullOrWhiteSpace(host) || host == "0.0.0.0" || host == "*" || host == "+")
        {
            return IPAddress.Any;
        }
 
        return IPAddress.TryParse(host, out var address) ? address : IPAddress.Any;
    }

    private async Task ReceiveLoopAsync(CancellationToken cancellationToken)
    {
        while (!cancellationToken.IsCancellationRequested && _udpClient is not null)
        {
            try
            {
                var request = await _udpClient.ReceiveAsync(cancellationToken).ConfigureAwait(false);
                var response = _protocol.HandleRequest(request.Buffer);
                await _udpClient.SendAsync(response, response.Length, request.RemoteEndPoint).ConfigureAwait(false);
            }
            catch (OperationCanceledException)
            {
                break;
            }
            catch (ObjectDisposedException)
            {
                break;
            }
            catch (SocketException ex)
            {
                StatusChanged?.Invoke(this, $"FINS UDP socket error: {ex.Message}");
            }
            catch (Exception ex)
            {
                StatusChanged?.Invoke(this, $"FINS UDP server error: {ex.Message}");
            }
        }
    }
}