Merge pull request #100 from bgokden/develop

Avoid infinite loop in connection close
bgokden · Aug 10, 2021 · 66937aa · 66937aa
2 parents ec1e024 + 94e21c8
commit 66937aa
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/util/connpool.go b/util/connpool.go
@@ -185,7 +185,7 @@ func (cp *ConnectionPool) Close(conn *Connection) {
 
 func (cp *ConnectionPool) CloseAll() {
 	cp.Closed = true
-	for {
+	for i := 0; i < 1000; i++ { // Maximum 1000 connections will be closed
 		connectionInterface := cp.Pool.Get()
 		if connectionInterface == nil {
 			return // Coonection pool is finished
@@ -197,6 +197,8 @@ func (cp *ConnectionPool) CloseAll() {
 					log.Printf("Connection Close Error: %v\n", err.Error())
 				}
 			}
+		} else {
+			return
 		}
 	}
 }