From b0d2032488a16ce1497edd47e7c2e0dda246723f Mon Sep 17 00:00:00 2001
From: Anton Kaliaev <anton.kalyaev@gmail.com>
Date: Tue, 28 Mar 2017 14:01:22 +0400
Subject: [PATCH] use BaseService.OnReset method to recreate channels

---
 client/ws_client.go | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/client/ws_client.go b/client/ws_client.go
index 99737ca8f..fb359bdaf 100644
--- a/client/ws_client.go
+++ b/client/ws_client.go
@@ -47,16 +47,21 @@ func (wsc *WSClient) String() string {
 	return wsc.Address + ", " + wsc.Endpoint
 }
 
+// OnStart implements cmn.BaseService interface
 func (wsc *WSClient) OnStart() error {
 	wsc.BaseService.OnStart()
 	err := wsc.dial()
 	if err != nil {
 		return err
 	}
+	go wsc.receiveEventsRoutine()
+	return nil
+}
 
+// OnReset implements cmn.BaseService interface
+func (wsc *WSClient) OnReset() error {
 	wsc.ResultsCh = make(chan json.RawMessage, wsResultsChannelCapacity)
 	wsc.ErrorsCh = make(chan error, wsErrorsChannelCapacity)
-	go wsc.receiveEventsRoutine()
 	return nil
 }
 
@@ -86,6 +91,7 @@ func (wsc *WSClient) dial() error {
 	return nil
 }
 
+// OnStop implements cmn.BaseService interface
 func (wsc *WSClient) OnStop() {
 	wsc.BaseService.OnStop()
 	// ResultsCh/ErrorsCh is closed in receiveEventsRoutine.