From 548150779dbe40d7a1a4285fb45c7231eb33cc35 Mon Sep 17 00:00:00 2001
From: aozhiwei <aozhiwei@kingsome.cn>
Date: Sun, 28 May 2023 09:13:38 +0800
Subject: [PATCH] 1

---
 a8/asiotcpclient.cc   |  2 +-
 a8/asiotcpclient.h    |  2 +-
 a8/websocketclient.cc | 25 +++++++++++++++++++++++++
 a8/websocketclient.h  | 28 ++++++++++++++++++++++++++++
 4 files changed, 55 insertions(+), 2 deletions(-)
 create mode 100644 a8/websocketclient.cc
 create mode 100644 a8/websocketclient.h

diff --git a/a8/asiotcpclient.cc b/a8/asiotcpclient.cc
index ed5e253..db32390 100644
--- a/a8/asiotcpclient.cc
+++ b/a8/asiotcpclient.cc
@@ -8,7 +8,7 @@
 #include <a8/a8.h>
 #include <a8/asiotcpclient.h>
 
-#ifdef USE_ASIO
+#ifdef USE_BOOST
 
 const int MAX_RECV_BUFFERSIZE = 1024 * 64;
 
diff --git a/a8/asiotcpclient.h b/a8/asiotcpclient.h
index a5b6b11..ca59cb9 100644
--- a/a8/asiotcpclient.h
+++ b/a8/asiotcpclient.h
@@ -1,6 +1,6 @@
 #pragma once
 
-#ifdef USE_ASIO
+#ifdef USE_BOOST
 
 #include <asio.hpp>
 
diff --git a/a8/websocketclient.cc b/a8/websocketclient.cc
new file mode 100644
index 0000000..f773e16
--- /dev/null
+++ b/a8/websocketclient.cc
@@ -0,0 +1,25 @@
+#include <a8/a8.h>
+
+#include <a8/websocketclient.h>
+
+#ifdef USE_BOOST
+
+namespace a8
+{
+
+    WebSocketClient::WebSocketClient(asio::io_context& io_context, const std::string& remote_ip, int remote_port): AsioTcpClient(io_context, remote_ip, remote_port)
+    {
+        decoded_buff_ = (char *)malloc(1024 * 64 + 1);
+        decoded_bufflen_ = 0;
+    }
+
+    WebSocketClient::~WebSocketClient()
+    {
+        free(decoded_buff_);
+        decoded_buff_ = nullptr;
+        decoded_bufflen_ = 0;
+    }
+
+}
+
+#endif
diff --git a/a8/websocketclient.h b/a8/websocketclient.h
new file mode 100644
index 0000000..2d74705
--- /dev/null
+++ b/a8/websocketclient.h
@@ -0,0 +1,28 @@
+#pragma once
+
+#include <a8/asiotcpclient.h>
+
+#ifdef USE_BOOST
+
+#include <asio.hpp>
+
+using asio::ip::tcp;
+
+namespace a8
+{
+
+    class WebSocketClient : public AsioTcpClient
+    {
+    public:
+        WebSocketClient(asio::io_context& io_context, const std::string& remote_ip, int remote_port);
+        virtual ~WebSocketClient() override;
+
+    private:
+        char *decoded_buff_ = nullptr;
+        int decoded_bufflen_ = 0;
+        bool handshook_ = false;
+    };
+
+}
+
+#endif