mirror
/
socketify.py
kopia lustrzana https://github.com/cirospaciari/socketify.py
1
0
Forkuj 0
Kod Zgłoszenia Packages Projekty Wydania Wiki Aktywność

fix some mistakes with ws

pull/39/head
Ciro 2022-11-07 14:49:05 -03:00
rodzic 3872860684
commit 21b262dc2e
2 zmienionych plików z 22 dodań i 14 usunięć
Pokaż statystyki Ściągnij plik aktualizacji Ściągnij plik porównania Expand all files Collapse all files

4
examples/upgrade_async.py
Wyświetl plik

@ -24,7 +24,9 @@ app.ws("/*", {
'idle_timeout': 12, 'idle_timeout': 12,
'open': ws_open, 'open': ws_open,
'message': ws_message, 'message': ws_message,
'upgrade': ws_upgrade 'upgrade': ws_upgrade,
'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.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.listen(3000, lambda config: print("Listening on port http://localhost:%d now\n" % (config.port)))

12
src/socketify/socketify.py
Wyświetl plik

@ -601,29 +601,35 @@ class WebSocket:
def get_user_data_uuid(self): def get_user_data_uuid(self):
if self.got_socket_data: if self.got_socket_data:
return self.socket_data_id return self.socket_data_id
user_data = lib.uws_ws_get_user_data(self.SSL, self._ptr) user_data = lib.uws_ws_get_user_data(self.SSL, self.ws)
if user_data == ffi.NULL: if user_data == ffi.NULL:
return None return None
try:
(data, socket_data_id) = ffi.from_handle(user_data) (data, socket_data_id) = ffi.from_handle(user_data)
self.socket_data_id = socket_data_id self.socket_data_id = socket_data_id
self.socket_data = data self.socket_data = data
self.got_socket_data = True self.got_socket_data = True
return socket_data_id return socket_data_id
except:
return None
def get_user_data(self): def get_user_data(self):
if self.got_socket_data: if self.got_socket_data:
return self.socket_data return self.socket_data
user_data = lib.uws_ws_get_user_data(self.SSL, self._ptr) user_data = lib.uws_ws_get_user_data(self.SSL, self.ws)
if user_data == ffi.NULL: if user_data == ffi.NULL:
return None return None
try:
(data, socket_data_id) = ffi.from_handle(user_data) (data, socket_data_id) = ffi.from_handle(user_data)
self.socket_data_id = socket_data_id self.socket_data_id = socket_data_id
self.socket_data = data self.socket_data = data
self.got_socket_data = True self.got_socket_data = True
return data return data
except:
return None
def get_buffered_amount(self): def get_buffered_amount(self):
return int(lib.uws_ws_get_buffered_amount(self.SSL, self._ptr)) return int(lib.uws_ws_get_buffered_amount(self.SSL, self.ws))
def subscribe(self, topic): def subscribe(self, topic):
try: try: