Simple implementation for web socket (alibaba#1109)

Simple implementation for web socket

Author: Coldwings

CMakeLists.txt

@@ -220,6 +220,7 @@ file(GLOB PHOTON_SRC RELATIVE "${PROJECT_SOURCE_DIR}"
         io/reset_handle.cpp
         net/*.cpp
         net/http/*.cpp
+        net/http/websocket.cpp
         net/security-context/tls-stream.cpp
         net/vdma/*.cpp
         rpc/*.cpp

include/photon/net/http/websocket.h

@@ -0,0 +1 @@
+../../../../net/http/websocket.h

net/http/client.h

@@ -32,6 +32,8 @@ namespace net {
 class TLSContext;
 namespace http {
 
+class IWebSocketStream;  // Forward declaration for websocket_connect
+
 class ICookieJar : public Object {
 public:
     virtual int get_cookies_from_headers(std::string_view host, Message* message) = 0;
@@ -166,6 +168,16 @@ class Client : public Object {
 
     virtual ISocketStream* native_connect(std::string_view host, uint16_t port,
                                           bool secure = false, uint64_t timeout = -1UL) = 0;
+
+    /**
+     * @brief Connect to a WebSocket server
+     * 
+     * @param url Full URL for the WebSocket endpoint (e.g., "http://example.com/ws")
+     * @param timeout Timeout in microseconds (-1 for infinite)
+     * @return Pointer to IWebSocketStream on success, nullptr on failure
+     */
+    IWebSocketStream* websocket_connect(std::string_view url, uint64_t timeout = -1UL);
+
 protected:
     StoredURL m_proxy_url;
     std::string m_user_agent;

net/http/message.h

@@ -92,6 +92,15 @@ class Message : public IStream {
     ssize_t write_stream(IStream *stream, size_t size_limit = -1);
     int close() override { return 0; }
 
+    // Release ownership of socket stream and return it (like unique_ptr::release)
+    // After this call, the Message no longer owns or references the socket
+    net::ISocketStream* steal_socket_stream() {
+        auto* s = m_stream;
+        m_stream = nullptr;
+        m_stream_ownership = false;
+        return s;
+    }
+
     // size of body: infer from Content-Range/Content-Length in response header
     size_t body_size() const;
     // size of origin resource: infer from Content-Range/Content-Length in response header

net/http/test/CMakeLists.txt

@@ -18,3 +18,7 @@ add_executable(client_tls_test client_tls_test.cpp)
 target_include_directories(client_tls_test PRIVATE ${OPENSSL_INCLUDE_DIRS})
 target_link_libraries(client_tls_test PRIVATE photon_shared ${testing_libs})
 add_test(NAME client_tls_test COMMAND $<TARGET_FILE:client_tls_test>)
+
+add_executable(websocket_test websocket_test.cpp)
+target_link_libraries(websocket_test PRIVATE photon_shared ${testing_libs})
+add_test(NAME websocket_test COMMAND $<TARGET_FILE:websocket_test>)