From 2454b7c51e2919eb79306fc7d6689b3ad5b63b60 Mon Sep 17 00:00:00 2001
From: Alexey Borodenko <kosunix@gmail.com>
Date: Mon, 3 Dec 2018 12:09:25 +0200
Subject: [PATCH] Generation of callbacks/operations separately.

---
 src/SigSpec.Core/SigSpecGenerator.cs | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

diff --git a/src/SigSpec.Core/SigSpecGenerator.cs b/src/SigSpec.Core/SigSpecGenerator.cs
index 1bd192c..ee5cf8b 100644
--- a/src/SigSpec.Core/SigSpecGenerator.cs
+++ b/src/SigSpec.Core/SigSpecGenerator.cs
@@ -36,16 +36,16 @@ public async Task<SigSpecDocument> GenerateForHubsAsync(IReadOnlyDictionary<stri
                 {
                     var operation = await GenerateOperationAsync(type, method, generator, resolver);
                     hub.Operations[method.Name] = operation;
+                }
 
-                    var baseTypeGenericArguments = type.BaseType.GetGenericArguments();
-                    if (baseTypeGenericArguments.Length == 1)
+                var baseTypeGenericArguments = type.BaseType.GetGenericArguments();
+                if (baseTypeGenericArguments.Length == 1)
+                {
+                    var callbackType = baseTypeGenericArguments[0];
+                    foreach (var callbackMethod in GetOperationMethods(callbackType))
                     {
-                        var callbackType = baseTypeGenericArguments[0];
-                        foreach (var callbackMethod in GetOperationMethods(callbackType))
-                        {
-                            var callback = await GenerateOperationAsync(type, callbackMethod, generator, resolver);
-                            hub.Callbacks[callbackMethod.Name] = callback;
-                        }
+                        var callback = await GenerateOperationAsync(type, callbackMethod, generator, resolver);
+                        hub.Callbacks[callbackMethod.Name] = callback;
                     }
                 }