Add Close method

invliD · Feb 18, 2024 · b29a82c · b29a82c
1 parent 9ac3f62
commit b29a82c
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -21,6 +21,7 @@ import (
 
 func main() {
 	client := palworldrcon.NewClient("127.0.0.1:25575", "password")
+	defer client.Close()
 
 	info, err := client.Info()
 	if err != nil {

diff --git a/rcon.go b/rcon.go
@@ -31,6 +31,17 @@ func NewClient(address string, password string) *Client {
 	return client
 }
 
+// Close closes the connection to the RCON server. Calling any other method after closing will reopen the connection.
+func (r *Client) Close() error {
+	if r.conn == nil {
+		return nil
+	} else {
+		err := r.conn.Close()
+		r.conn = nil
+		return err
+	}
+}
+
 func (r *Client) connect() error {
 	if r.conn != nil {
 		r.conn.Close()