1 files changed, 27 insertions, 0 deletions
diff --git a/Impostor-dev/src/Impostor.Hazel/MessageReaderPolicy.cs b/Impostor-dev/src/Impostor.Hazel/MessageReaderPolicy.cs
new file mode 100644
index 0000000..ef3939a
--- /dev/null
+++ b/Impostor-dev/src/Impostor.Hazel/MessageReaderPolicy.cs
@@ -0,0 +1,27 @@
+using System;
+using Microsoft.Extensions.DependencyInjection;
+using Microsoft.Extensions.ObjectPool;
+
+namespace Impostor.Hazel
+{
+    public class MessageReaderPolicy : IPooledObjectPolicy<MessageReader>
+    {
+        private readonly IServiceProvider _serviceProvider;
+
+        public MessageReaderPolicy(IServiceProvider serviceProvider)
+        {
+            _serviceProvider = serviceProvider;
+        }
+
+        public MessageReader Create()
+        {
+            return new MessageReader(_serviceProvider.GetRequiredService<ObjectPool<MessageReader>>());
+        }
+
+        public bool Return(MessageReader obj)
+        {
+            obj.Reset();
+            return true;
+        }
+    }
+}