SenderSkeleton.cs

using System;
using System.Collections.Generic;
using System.Net.Sockets;
using System.Threading;
using System.Threading.Tasks;

namespace ZabbixSender.Async
{
    /// <summary>
    /// A basic class for sending datato Zabbix, which allows custom setup for TcpClient and Formatter.
    /// </summary>
    public class SenderSkeleton
    {
        private readonly Func<CancellationToken, Task<TcpClient>> tcpClientFactory;
        private readonly Func<IFormatter> formatterFactory;

        /// <summary>
        /// Initializes a new instance of the ZabbixSender.Async.SenderSkeleton class.
        /// </summary>
        /// <param name="tcpClientFactory">An async factory delegate for TcpClient.</param>
        /// <param name="formatterFactory">A factory delegate for the data formatter.</param>
        public SenderSkeleton(
            Func<CancellationToken, Task<TcpClient>> tcpClientFactory,
            Func<IFormatter> formatterFactory)
        {
            this.tcpClientFactory = tcpClientFactory;
            this.formatterFactory = formatterFactory;
        }

        /// <summary>
        /// Send an array of data items.
        /// </summary>
        /// <param name="data">Data items to send.</param>
        /// <returns>Zabbix Trapper response.</returns>
        public Task<SenderResponse> Send(params SendData[] data) =>
            Send(data, CancellationToken.None);

        /// <summary>
        /// Send a single data item for specified host.
        /// </summary>
        /// <param name="host"></param>
        /// <param name="key">An item key (see https://www.zabbix.com/documentation/4.4/manual/config/items/item/key).</param>
        /// <param name="value">An item value</param>
        /// <param name="cancellationToken">A CancellationToken for an overall request processing.</param>
        public Task<SenderResponse> Send(string host, string key, string value, CancellationToken cancellationToken = default) =>
            Send(new[]
            {
                new SendData
                {
                    Host = host,
                    Key = key,
                    Value = value
                }
            }, cancellationToken);

        /// <summary>
        /// Send multiple data items.
        /// </summary>
        /// <param name="data">Data items to send</param>
        /// <param name="cancellationToken">A CancellationToken for an overall request processing.</param>
        public async Task<SenderResponse> Send(IEnumerable<SendData> data, CancellationToken cancellationToken = default)
        {
            using (var tcpClient = await tcpClientFactory(cancellationToken))
            using (var networkStream = tcpClient.GetStream())
            {
                var formatter = formatterFactory();

                await formatter.WriteRequestAsync(networkStream, data, cancellationToken);
                await networkStream.FlushAsync();
                await networkStream.WaitForDataAvailable(tcpClient.ReceiveTimeout, cancellationToken);

                return await formatter.ReadResponseAsync(networkStream, cancellationToken);
            }
        }
    }
}