Wino-Mail/Wino.Server/Core/ServerMessageHandlerBase.cs

using System;
using System.Text.Json;
using System.Threading;
using System.Threading.Tasks;
using Windows.ApplicationModel.AppService;
using Windows.Foundation.Collections;
using Wino.Core.Domain.Interfaces;
using Wino.Core.Domain.Models.Server;
using Wino.Messaging;

namespace Wino.Server.Core
{
    public abstract class ServerMessageHandlerBase
    {
        public string HandlingRequestType { get; }

        public abstract Task ExecuteAsync(IClientMessage message, AppServiceRequest request = null, CancellationToken cancellationToken = default);
    }

    public abstract class ServerMessageHandler<TClientMessage, TResponse> : ServerMessageHandlerBase where TClientMessage : IClientMessage
    {
        /// <summary>
        /// Response to return when server encounters and exception while executing code.
        /// </summary>
        /// <param name="ex">Exception that target threw.</param>
        /// <returns>Default response on failure object.</returns>
        public abstract WinoServerResponse<TResponse> FailureDefaultResponse(Exception ex);


        /// <summary>
        /// Safely executes the handler code and returns the response.
        /// This call will never crash the server. Exceptions encountered will be handled and returned as response.
        /// </summary>
        /// <param name="message">IClientMessage that client asked the response for from the server.</param>
        /// <param name="request">optional AppServiceRequest to return response for.</param>
        /// <param name="cancellationToken">Cancellation token.</param>
        /// <returns>Response object that server executes for the given method.</returns>
        public override async Task ExecuteAsync(IClientMessage message, AppServiceRequest request = null, CancellationToken cancellationToken = default)
        {
            WinoServerResponse<TResponse> response = default;

            try
            {
                response = await HandleAsync((TClientMessage)message, cancellationToken).ConfigureAwait(false);
            }
            catch (Exception ex)
            {
                response = FailureDefaultResponse(ex);
            }
            finally
            {
                // No need to send response if request is null.
                // Handler might've been called directly from the server itself.
                if (request != null)
                {
                    var valueSet = new ValueSet()
                    {
                        { MessageConstants.MessageDataKey, JsonSerializer.Serialize(response) }
                    };

                    await request.SendResponseAsync(valueSet);
                }
            }
        }

        /// <summary>
        /// Code that will be executed directly on the server.
        /// All handlers must implement this method.
        /// Response is wrapped with WinoServerResponse.
        /// </summary>
        /// <param name="message">IClientMessage that client asked the response for from the server.</param>
        /// <param name="cancellationToken">Cancellation token.</param>
        protected abstract Task<WinoServerResponse<TResponse>> HandleAsync(TClientMessage message, CancellationToken cancellationToken = default);
    }
}