export GetUnhandledMessagesChannel in nats rpc server for using in li…

…bpitaya-cluster
topfreegames · Jul 18, 2018 · bb97311 · bb97311
1 parent 15498f2
commit bb97311
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/cluster/nats_rpc_server.go b/cluster/nats_rpc_server.go
@@ -194,7 +194,8 @@ func (ns *NatsRPCServer) handleMessages() {
 	}
 }
 
-func (ns *NatsRPCServer) getUnhandledRequestsChannel() chan *protos.Request {
+// GetUnhandledRequestsChannel gets the unhandled requests channel from nats rpc server
+func (ns *NatsRPCServer) GetUnhandledRequestsChannel() chan *protos.Request {
 	return ns.unhandledReqCh
 }
 
@@ -221,7 +222,7 @@ func (ns *NatsRPCServer) marshalResponse(res *protos.Response) ([]byte, error) {
 }
 
 func (ns *NatsRPCServer) processMessages(threadID int) {
-	for req := range ns.getUnhandledRequestsChannel() {
+	for req := range ns.GetUnhandledRequestsChannel() {
 		logger.Log.Debugf("(%d) processing message %v", threadID, req.GetMsg().GetId())
 		reply := req.GetMsg().GetReply()
 		var response *protos.Response

diff --git a/cluster/nats_rpc_server_test.go b/cluster/nats_rpc_server_test.go
@@ -96,8 +96,8 @@ func TestNatsRPCServerGetUnhandledRequestsChannel(t *testing.T) {
 	cfg := getConfig()
 	sv := getServer()
 	n, _ := NewNatsRPCServer(cfg, sv, nil, nil)
-	assert.NotNil(t, n.getUnhandledRequestsChannel())
-	assert.IsType(t, make(chan *protos.Request), n.getUnhandledRequestsChannel())
+	assert.NotNil(t, n.GetUnhandledRequestsChannel())
+	assert.IsType(t, make(chan *protos.Request), n.GetUnhandledRequestsChannel())
 }
 
 func TestNatsRPCServerGetBindingsChannel(t *testing.T) {