Fix issue on Flow Provider

Author: mauroa

src/Client/Flows/ClientProtocolFlowProvider.cs

@@ -11,7 +11,7 @@ public ClientProtocolFlowProvider (ITopicEvaluator topicEvaluator, IRepositoryPr
 		{
 		}
 
-		protected override IDictionary<ProtocolFlowType, IProtocolFlow> GetFlows ()
+		protected override IDictionary<ProtocolFlowType, IProtocolFlow> InitializeFlows ()
 		{
 			var flows = new Dictionary<ProtocolFlowType, IProtocolFlow>();
 

src/Portable/Flows/ProtocolFlowProvider.cs

@@ -8,23 +8,22 @@ namespace Hermes.Flows
 {
 	public abstract class ProtocolFlowProvider : IProtocolFlowProvider
 	{
-		protected readonly IDictionary<ProtocolFlowType, IProtocolFlow> flows;
 		protected readonly ITopicEvaluator topicEvaluator;
 		protected readonly IRepositoryProvider repositoryProvider;
 		protected readonly ProtocolConfiguration configuration;
 
+		IDictionary<ProtocolFlowType, IProtocolFlow> flows;
+
 		protected ProtocolFlowProvider (ITopicEvaluator topicEvaluator, 
 			IRepositoryProvider repositoryProvider, 
 			ProtocolConfiguration configuration)
 		{
 			this.topicEvaluator = topicEvaluator;
 			this.repositoryProvider = repositoryProvider;
 			this.configuration = configuration;
-
-			this.flows = this.GetFlows ();
 		}
 
-		protected abstract IDictionary<ProtocolFlowType, IProtocolFlow> GetFlows ();
+		protected abstract IDictionary<ProtocolFlowType, IProtocolFlow> InitializeFlows ();
 
 		protected abstract bool IsValidPacketType (PacketType packetType);
 
@@ -40,7 +39,7 @@ public IProtocolFlow GetFlow (PacketType packetType)
 			var flow = default (IProtocolFlow);
 			var flowType = packetType.ToFlowType();
 
-			if (!this.flows.TryGetValue (flowType, out flow)) {
+			if (!this.GetFlows().TryGetValue (flowType, out flow)) {
 				var error = string.Format (Resources.ProtocolFlowProvider_UnknownPacketType, packetType);
 
 				throw new ProtocolException (error);
@@ -51,13 +50,22 @@ public IProtocolFlow GetFlow (PacketType packetType)
 
 		public T GetFlow<T> () where T : class
 		{
-			var pair = this.flows.FirstOrDefault (f => f.Value is T);
+			var pair = this.GetFlows().FirstOrDefault (f => f.Value is T);
 
 			if (pair.Equals (default (KeyValuePair<ProtocolFlowType, IProtocolFlow>))) {
 				return default (T);
 			}
 
 			return pair.Value as T;
 		}
+
+		private IDictionary<ProtocolFlowType, IProtocolFlow> GetFlows()
+		{
+			if (this.flows == default (IDictionary<ProtocolFlowType, IProtocolFlow>)) {
+				this.flows = this.InitializeFlows ();
+			}
+
+			return this.flows;
+		}
 	}
 }

src/Server/Flows/ServerProtocolFlowProvider.cs

@@ -17,7 +17,7 @@ public ServerProtocolFlowProvider (IConnectionProvider connectionProvider,
 			this.connectionProvider = connectionProvider;
 		}
 
-		protected override IDictionary<ProtocolFlowType, IProtocolFlow> GetFlows ()
+		protected override IDictionary<ProtocolFlowType, IProtocolFlow> InitializeFlows ()
 		{
 			var flows = new Dictionary<ProtocolFlowType, IProtocolFlow>();