From c1b1509e07c964d9c94299e8ad82523b8a539654 Mon Sep 17 00:00:00 2001
From: yangxiaodong <yangxiaodong1214@126.com>
Date: Mon, 15 May 2017 16:55:53 +0800
Subject: [PATCH] refactor

---
 .../Builder/ConsistencyBuilder.cs                | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/src/Cap.Consistency/Builder/ConsistencyBuilder.cs b/src/Cap.Consistency/Builder/ConsistencyBuilder.cs
index 53dc333..1ee559b 100644
--- a/src/Cap.Consistency/Builder/ConsistencyBuilder.cs
+++ b/src/Cap.Consistency/Builder/ConsistencyBuilder.cs
@@ -5,7 +5,10 @@ using System.Linq;
 using Microsoft.Extensions.DependencyInjection;
 using System.Collections.Generic;
 using Cap.Consistency.Consumer;
-using Cap.Consistency.Consumer.Kafka;
+using Cap.Consistency.Routing;
+using Cap.Consistency.Infrastructure;
+using Cap.Consistency.Internal;
+using Cap.Consistency.Abstractions;
 
 namespace Cap.Consistency
 {
@@ -24,6 +27,8 @@ namespace Cap.Consistency
             Services = service;
 
             AddConsumerServices();
+
+            AddKafkaServices();
         }
 
         /// <summary>
@@ -60,14 +65,17 @@ namespace Cap.Consistency
                     Services.AddSingleton(typeof(IConsumerService), type);
                 }
             }
+
+            Services.AddSingleton<IConsumerExcutorSelector, ConsumerExcutorSelector>();        
+            Services.AddSingleton<IConsumerInvokerFactory, ConsumerInvokerFactory>();
+            Services.AddSingleton<MethodMatcherCache>();
+
             return this;
         }
 
         public virtual ConsistencyBuilder AddKafkaServices() {
 
-
-            Services.AddSingleton<IConsumerHandler, KafkaConsumerHandler>();
-            return this;
+            return AddScoped(typeof(ITopicRoute), typeof(ConsumerHandler<>).MakeGenericType(MessageType));
         }