封装一个websocket 的类
时间: 2023-08-04 18:20:07 浏览: 36
好的,这里提供一个WebSocket类的封装示例,使用Python的websocket库:
```python
import websocket
class WebSocketClient:
def __init__(self, url):
self.url = url
self.ws = None
def on_open(self):
print("WebSocket connection has been opened.")
def on_message(self, message):
print(f"Received message: {message}")
def on_error(self, error):
print(f"Error: {error}")
def on_close(self):
print("WebSocket connection has been closed.")
def connect(self):
self.ws = websocket.WebSocketApp(self.url,
on_open=self.on_open,
on_message=self.on_message,
on_error=self.on_error,
on_close=self.on_close)
self.ws.run_forever()
def send(self, message):
self.ws.send(message)
```
这个类实现了WebSocket的基本功能,可以通过`connect()`方法连接到指定URL,并通过`send()`方法发送消息。`on_open()`、`on_message()`、`on_error()`、`on_close()`是WebSocket的事件回调函数,可以根据需要进行自定义。