Merge pull request linkedin#102 from jjyao/master

Give a chance for DarkCanaryResponseHandler to initialize
maxid · Jun 17, 2015 · 78c3652 · 78c3652
2 parents 46d6c2e + 7c6243e
commit 78c3652
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 1 deletion.
diff --git a/gradle.properties b/gradle.properties
@@ -1,4 +1,4 @@
-version=0.6.98
+version=0.6.99
 defaultScalaVersion=2.10.3
 targetScalaVersions=2.10.3
 crossBuild=false
diff --git a/network/src/main/scala/com/linkedin/norbert/network/client/ResponseHandlerComponent.scala b/network/src/main/scala/com/linkedin/norbert/network/client/ResponseHandlerComponent.scala
@@ -98,5 +98,6 @@ class ResponseProcessorMBeanImpl(clientName: Option[String], serviceName: String
 
 abstract class DarkCanaryResponseHandler()
 {
+  def initialize() {}
   def upstreamCallback(isDark: Boolean, requestId: UUID, request: Request[Any, Any], response: NorbertProtos.NorbertMessage);
 }
diff --git a/network/src/main/scala/com/linkedin/norbert/network/netty/DarkCanaryChannelHandler.scala b/network/src/main/scala/com/linkedin/norbert/network/netty/DarkCanaryChannelHandler.scala
@@ -138,6 +138,13 @@ class DarkCanaryChannelHandler extends Logging {
           clientConfig.clientName,
           serviceName
         ))
+
+        darkCanaryResponseHandler match {
+          case Some(responseHandler) => {
+            responseHandler.initialize()
+          }
+          case None =>
+        }
       }
     }
   }