From 5d795dc56dfc6ce4dfff87e6f805f6e55c74b5f2 Mon Sep 17 00:00:00 2001 From: Sammy Libre Date: Wed, 7 Dec 2016 01:59:22 +0500 Subject: [PATCH] Remove session on disconnect --- go-pool/stratum/stratum.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/go-pool/stratum/stratum.go b/go-pool/stratum/stratum.go index bc8a54c..89f3f09 100644 --- a/go-pool/stratum/stratum.go +++ b/go-pool/stratum/stratum.go @@ -186,10 +186,10 @@ func (s *StratumServer) handleClient(cs *Session, e *Endpoint) error { data, isPrefix, err := connbuff.ReadLine() if isPrefix { log.Printf("Socket flood detected") - // TODO: Ban client return errors.New("Socket flood") } else if err == io.EOF { - log.Printf("Client disconnected") + log.Println("Client disconnected", cs.ip) + s.removeSession(cs) break } else if err != nil { log.Printf("Error reading: %v", err)