From e9ea621b93fbb58d9edfca8375918791637bbd52 Mon Sep 17 00:00:00 2001
From: chai <chaifix@163.com>
Date: Wed, 30 Dec 2020 20:59:04 +0800
Subject: +init

---
 .../Extensions/ServiceProviderExtensions.cs         | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)
 create mode 100644 Impostor-dev/src/Impostor.Hazel/Extensions/ServiceProviderExtensions.cs

(limited to 'Impostor-dev/src/Impostor.Hazel/Extensions/ServiceProviderExtensions.cs')

diff --git a/Impostor-dev/src/Impostor.Hazel/Extensions/ServiceProviderExtensions.cs b/Impostor-dev/src/Impostor.Hazel/Extensions/ServiceProviderExtensions.cs
new file mode 100644
index 0000000..56c7380
--- /dev/null
+++ b/Impostor-dev/src/Impostor.Hazel/Extensions/ServiceProviderExtensions.cs
@@ -0,0 +1,21 @@
+using Microsoft.Extensions.DependencyInjection;
+using Microsoft.Extensions.DependencyInjection.Extensions;
+using Microsoft.Extensions.ObjectPool;
+
+namespace Impostor.Hazel.Extensions
+{
+    public static class ServiceProviderExtensions
+    {
+        public static void AddHazel(this IServiceCollection services)
+        {
+            services.TryAddSingleton<ObjectPoolProvider>(new DefaultObjectPoolProvider());
+
+            services.AddSingleton(serviceProvider =>
+            {
+                var provider = serviceProvider.GetRequiredService<ObjectPoolProvider>();
+                var policy = ActivatorUtilities.CreateInstance<MessageReaderPolicy>(serviceProvider);
+                return provider.Create(policy);
+            });
+        }
+    }
+}
-- 
cgit v1.1-26-g67d0