Merge cbbffa7 into 3c5fc68

stan.go

@@ -50,6 +50,9 @@ type Conn interface {
 
 	// Close
 	Close() error
+
+	// Status returns the status of the underlying NATS conn.
+	Status() nats.Status
 }
 
 // Errors
@@ -271,6 +274,14 @@ func (sc *conn) Close() error {
 	return nil
 }
 
+// Status returns the status of the underlying NATS conn.
+func (sc *conn) Status() nats.Status {
+	if sc.nc == nil {
+		return nats.CLOSED
+	}
+	return sc.nc.Status()
+}
+
 // Process a heartbeat from the NATS Streaming cluster
 func (sc *conn) processHeartBeat(m *nats.Msg) {
 	// No payload assumed, just reply.

stan_test.go

@@ -1853,3 +1853,19 @@ func TestRaceAckOnClose(t *testing.T) {
 	time.Sleep(10 * time.Millisecond)
 	sc.Close()
 }
+
+func TestStatus(t *testing.T) {
+	s := RunServer(clusterName)
+	defer s.Shutdown()
+
+	sc := NewDefaultConnection(t)
+	defer sc.Close()
+
+	if sc.Status() != nats.CONNECTED {
+		t.Fatal("Should have status set to CONNECTED")
+	}
+	sc.Close()
+	if sc.Status() != nats.CLOSED {
+		t.Fatal("Should have status set to CLOSED")
+	}
+}