src/OneScript.DebugProtocol/TcpServer/DispatchingServer.cs - develop - smwwiki/OneScript_fork

src/OneScript.DebugProtocol/TcpServer/DispatchingServer.cs

78 строк2 KB

EvilBeaver

Перенос доработок отладчика из версии 1

17 сен 2023, 23:47

17 сен 2023, 23:47cd268ca

/*----------------------------------------------------------
This Source Code Form is subject to the terms of the
Mozilla Public License, v.2.0. If a copy of the MPL
was not distributed with this file, You can obtain one
at http://mozilla.org/MPL/2.0/.
----------------------------------------------------------*/

using System;
using OneScript.DebugProtocol.Abstractions;

namespace OneScript.DebugProtocol.TcpServer
{
    public class DispatchingService<TService>
    {
        private readonly ICommunicationServer _server;
        private readonly TService _requestService;
        private readonly MethodsDispatcher<TService> _requestProcessor;

        public DispatchingService(ICommunicationServer server, TService requestService)
        {
            _server = server;
            _requestProcessor = new MethodsDispatcher<TService>();
            _requestService = requestService;
        }

        public void Start()
        {
            _server.DataReceived += OnDataReceived;
            _server.Start();
        }

        public void Stop()
        {
            _server.DataReceived -= OnDataReceived;
            _server.Stop();
        }

        private void OnDataReceived(object sender, CommunicationEventArgs e)
        {
            if (e.Exception == null)
            {
                ProcessSuccess((RpcCall) e.Data, e.Channel);
            }
            else
            {
                if (e.Exception.StopChannel)
                {
                    Stop();
                }
            }
        }

        private void ProcessSuccess(RpcCall message, ICommunicationChannel responseChannel)
        {
            RpcCallResult callResult = null;
            try
            {
                var methodResult = _requestProcessor.Dispatch(_requestService, message.Id, message.Parameters);
                if (methodResult != null)
                    callResult = RpcCallResult.Respond(message, methodResult);
            }
            catch (StopServiceException)
            {
                Stop();
                return;
            }
            catch (Exception e)
            {
                callResult = RpcCallResult.Exception(message, e);
            }
            
            if (callResult != null)
            {
                responseChannel.Write(callResult);
            }
        }
    }
}