From 6064aedd6d5e28f808ab4b394451a0974b0da815 Mon Sep 17 00:00:00 2001
From: Bruno Paulo <bruno@werules.com>
Date: Thu, 13 May 2021 17:23:24 +0200
Subject: [PATCH] allowing Client to have its connection closed and checked

---
 client.go | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/client.go b/client.go
index 93daf2f..ec326cd 100644
--- a/client.go
+++ b/client.go
@@ -99,6 +99,21 @@ func (client *Client) Emit(message string, args ...interface{}) (err error) {
 	return client.send(args)
 }
 
+func (client *Client) Close() error {
+	return client.conn.Close()
+}
+
+func (client *Client) Connected() bool {
+	switch client.conn.getState() {
+	case stateNormal, stateUpgrading:
+		return true
+	case stateUnknow, stateClosed:
+		fallthrough
+	default:
+		return false
+	}
+}
+
 func (client *Client) sendConnect() error {
 	packet := packet{
 		Type: _CONNECT,