using MariSocketMiddleware.Entities.MariEventArgs;
using MariSocketMiddleware.Services;
using Microsoft.AspNetCore.Http;
using System.Threading.Tasks;

namespace MyApi
{
    public class MyWebSocketService : MariWebSocketService
    {
        public MyWebSocketService()
        {
            OnOpen += OpenAsync;
            OnDisconnected += OnDisconnectedAsync;
            OnError += OnErrorAsync;
            OnMessage += OnMessageAsync;
        }

        private Task OnMessageAsync(MessageEventArgs arg)
            => Task.CompletedTask;

        private Task OnErrorAsync(ErrorEventArgs arg)
            => Task.CompletedTask;

        private Task OnDisconnectedAsync(DisconnectEventArgs arg)
            => Task.CompletedTask;

        private Task OpenAsync(OpenEventArgs arg)
            => Task.CompletedTask;

        public override string Path { get; set; } = "/Gateway";

        public override Task<bool> AuthorizeAsync(HttpContext context)
            => Task.FromResult(true);
    }
}

using MariSocketMiddleware.Utils;
using Microsoft.AspNetCore.Builder;
using Microsoft.AspNetCore.Hosting;
using Microsoft.AspNetCore.Mvc;
using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.DependencyInjection;

namespace MyApi
{
    public class Startup
    {
        public Startup(IConfiguration configuration)
        {
            Configuration = configuration;
        }

        public IConfiguration Configuration { get; }

        public void ConfigureServices(IServiceCollection services)
        {
            // Here
            // The generic type or the instance.
            services.AddMariWebSocketService<MyWebSocketService>();

            // ASP.NET Core template.
            services.AddMvc().SetCompatibilityVersion(CompatibilityVersion.Version_2_2);
        }

        public void Configure(IApplicationBuilder app, IHostingEnvironment env)
        {
            // ASP.NET Core template.
            if (env.IsDevelopment())
            {
                app.UseDeveloperExceptionPage();
            }
            else
            {
                app.UseHsts();
            }

            // Here
            app.UseMariWebSockets();

            // ASP.NET Core template.
            app.UseHttpsRedirection();
            app.UseMvc();
        }
    }
}

using MariSocketMiddleware.Entities.MariEventArgs;
using MariSocketMiddleware.Services;
using Microsoft.AspNetCore.Http;
using Microsoft.Extensions.Logging;
using System.Threading.Tasks;

namespace MyApi
{
    public class MyWebSocketService : MariWebSocketService
    {
        private readonly ILogger<MyWebSocketService> _logger;

        // ASP.NET Core will pass the dependencies in your class ctor.
        public MyWebSocketService(ILogger<MyWebSocketService> logger)
        {
            _logger = logger;
            OnOpen += OpenAsync;
            OnDisconnected += OnDisconnectedAsync;
            OnError += OnErrorAsync;
            OnMessage += OnMessageAsync;
        }

        private Task OnMessageAsync(MessageEventArgs arg)
            => Task.CompletedTask;

        private Task OnErrorAsync(ErrorEventArgs arg)
            => Task.CompletedTask;

        private Task OnDisconnectedAsync(DisconnectEventArgs arg)
            => Task.CompletedTask;

        private Task OpenAsync(OpenEventArgs arg)
            => Task.CompletedTask;

        public override string Path { get; set; } = "/Gateway";

        public override Task<bool> AuthorizeAsync(HttpContext context)
            => Task.FromResult(true);
    }
}

using System;
using Microsoft.AspNetCore.Mvc;

namespace MyApi.Controllers
{
    [Route("api/[controller]")]
    [ApiController]
    public class ValuesController : ControllerBase
    {
        private readonly MyWebSocketService _service1;
        private readonly MyOtherWebSocketService _service2;

        public ValuesController(MyWebSocketService service1, IServiceProvider provider)
        {
            // Direct requesting your service or
            _service1 = service1;

            // Requesting for your provider.
            _service2 = provider.GetService<MyOtherWebSocketService>();
        }

        // GET api/values
        [HttpGet]
        public ActionResult Get()
            => Ok(new string[] { "value1", "value2" });
    }
}