fixes from Bucky's and Emmanuel's reviews

This commit is contained in:
Anton Kaliaev
2017-10-30 11:12:01 -05:00
parent 6d18e2f447
commit 61d76a273f
10 changed files with 76 additions and 56 deletions

View File

@ -478,8 +478,13 @@ func (wsc *wsConnection) TryWriteRPCResponse(resp types.RPCResponse) bool {
}
}
func (wsc *wsConnection) AddSubscription(query string, data interface{}) {
func (wsc *wsConnection) AddSubscription(query string, data interface{}) error {
if _, ok := wsc.subscriptions[query]; ok {
return errors.New("Already subscribed")
}
wsc.subscriptions[query] = data
return nil
}
func (wsc *wsConnection) DeleteSubscription(query string) (interface{}, bool) {