optimization: removed slow defer from worker pool handler and hijack handler

1 files changed, 4 insertions, 6 deletions

diff --git a/workerpool.go b/workerpool.go
index 04245ff..081ac65 100644
--- a/workerpool.go
+++ b/workerpool.go
@@ -201,12 +201,6 @@ func (wp *workerPool) release(ch *workerChan) bool {
 func (wp *workerPool) workerFunc(ch *workerChan) {
 	var c net.Conn
 
-	defer func() {
-		wp.lock.Lock()
-		wp.workersCount--
-		wp.lock.Unlock()
-	}()
-
 	var err error
 	for c = range ch.ch {
 		if c == nil {
@@ -230,4 +224,8 @@ func (wp *workerPool) workerFunc(ch *workerChan) {
 			break
 		}
 	}
+
+	wp.lock.Lock()
+	wp.workersCount--
+	wp.lock.Unlock()
 }