PoolConn add IsUnusable func

nange · Nov 16, 2017 · 911064e · 911064e
1 parent 5ad858a
commit 911064e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/conn.go b/conn.go
@@ -31,6 +31,10 @@ func (pc *PoolConn) MarkUnusable() {
 	pc.unusable = true
 }
 
+func (pc *PoolConn) IsUnusable() bool {
+	return pc.unusable
+}
+
 func (pc *PoolConn) close() error {
 	return pc.Conn.Close()
 }
diff --git a/heap_test.go b/heap_test.go
@@ -117,6 +117,9 @@ func TestPoolConcurrent2(t *testing.T) {
 			time.Sleep(time.Second)
 			if i >= MaxCap-10 {
 				conn.(*PoolConn).MarkUnusable()
+				if !conn.(*PoolConn).IsUnusable() {
+					t.Errorf("after mark unusable, IsUnusable() should return true")
+				}
 			}
 			conn.Close()
 		}(conn, i)