From e136a0566317888e5f3743e147ff2a25c729400d Mon Sep 17 00:00:00 2001
From: yangxiaodong <yangxiaodong1214@126.com>
Date: Mon, 5 Jun 2017 11:36:52 +0800
Subject: [PATCH] refactor

---
 .../BuilderExtensions.cs                      | 39 +++++++++++++++++++
 1 file changed, 39 insertions(+)
 create mode 100644 src/Cap.Consistency/Microsoft.AspNetCore.Builder/BuilderExtensions.cs

diff --git a/src/Cap.Consistency/Microsoft.AspNetCore.Builder/BuilderExtensions.cs b/src/Cap.Consistency/Microsoft.AspNetCore.Builder/BuilderExtensions.cs
new file mode 100644
index 0000000..da6058f
--- /dev/null
+++ b/src/Cap.Consistency/Microsoft.AspNetCore.Builder/BuilderExtensions.cs
@@ -0,0 +1,39 @@
+using System;
+using Cap.Consistency;
+using Microsoft.Extensions.DependencyInjection;
+
+// ReSharper disable once CheckNamespace
+namespace Microsoft.AspNetCore.Builder
+{
+    /// <summary>
+    /// Consistence extensions for <see cref="IApplicationBuilder"/>
+    /// </summary>
+    public static class BuilderExtensions
+    {
+        ///<summary>
+        /// Enables Consistence for the current application
+        /// </summary>
+        /// <param name="app">The <see cref="IApplicationBuilder"/> instance this method extends.</param>
+        /// <returns>The <see cref="IApplicationBuilder"/> instance this method extends.</returns>
+        public static IApplicationBuilder UseConsistency(this IApplicationBuilder app) {
+            if (app == null) {
+                throw new ArgumentNullException(nameof(app));
+            }
+            
+            var marker = app.ApplicationServices.GetService<ConsistencyMarkerService>();
+         
+            if (marker == null) {
+                throw new InvalidOperationException("Add Consistency must be called on the service collection.");
+            }
+
+            var router = app.ApplicationServices.GetService<ITopicRouteHandler>();
+
+            var context = new TopicRouteContext();
+
+            router.RouteAsync(context);
+
+            return app;
+        }
+
+    }
+}
\ No newline at end of file