From e2b9c8a969cfa1c5947c7ffbf4c6ac84945e3db6 Mon Sep 17 00:00:00 2001
From: Ciro <ciro.spaciari@gmail.com>
Date: Thu, 1 Dec 2022 07:57:14 -0300
Subject: [PATCH] fix ws.close #41 thanks @seanr3

---
 examples/ws_close_connection.py | 38 +++++++++++++++++++++++++++++++++
 src/socketify/socketify.py      |  2 +-
 2 files changed, 39 insertions(+), 1 deletion(-)
 create mode 100644 examples/ws_close_connection.py

diff --git a/examples/ws_close_connection.py b/examples/ws_close_connection.py
new file mode 100644
index 0000000..29d25c5
--- /dev/null
+++ b/examples/ws_close_connection.py
@@ -0,0 +1,38 @@
+from socketify import App, AppOptions, OpCode, CompressOptions
+
+
+def ws_open(ws):
+    print("A WebSocket got connected!")
+    ws.send("Hello World!", OpCode.TEXT)
+
+
+def ws_message(ws, message, opcode):
+    print(message, opcode)
+    if message == 'close':
+        ws.close()
+    else:
+        # Ok is false if backpressure was built up, wait for drain
+        ok = ws.send(message, opcode)
+
+
+app = App()
+app.ws(
+    "/*",
+    {
+        "compression": CompressOptions.SHARED_COMPRESSOR,
+        "max_payload_length": 16 * 1024 * 1024,
+        "idle_timeout": 12,
+        "open": ws_open,
+        "message": ws_message,
+        "drain": lambda ws: print(
+            "WebSocket backpressure: %s", ws.get_buffered_amount()
+        ),
+        "close": lambda ws, code, message: print("WebSocket closed"),
+    },
+)
+app.any("/", lambda res, req: res.end("Nothing to see here!'"))
+app.listen(
+    3000,
+    lambda config: print("Listening on port http://localhost:%d now\n" % (config.port)),
+)
+app.run()
diff --git a/src/socketify/socketify.py b/src/socketify/socketify.py
index 8f20b33..23e71e6 100644
--- a/src/socketify/socketify.py
+++ b/src/socketify/socketify.py
@@ -1189,7 +1189,7 @@ class WebSocket:
             return self
 
     def close(self):
-        lib.uws_ws_close(self.SSL, self._ptr)
+        lib.uws_ws_close(self.SSL, self.ws)
         return self
 
     def cork(self, callback):