blame | 历史 | 原始文档
mrDarker
2025-07-08 ed84512b71ff17024524376074fe3cdf5fca5b67 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50

#ifndef WEBSOCKETCLIENTWITHPOOL_H


#define WEBSOCKETCLIENTWITHPOOL_H


 


#include "WebSocketClient.h"


#include "ThreadPool.h"


 


class WebSocketClientPool {


public:


    explicit WebSocketClientPool(size_t thread_count);


    ~WebSocketClientPool();


 


    // ³õʼ»¯¿Í»§¶Ë


    void init_clients(const std::vector<std::string>& uris);


 


    // Í˳öËùÓпͻ§¶Ë


    void exit_clients();


 


    // Ê¹ÓÃÏ̳߳ط¢ËÍÏûÏ¢


    void post_to_all(const std::string& message);


    void post_binary_to_all(const std::vector<char>& binary_data);


 


    // ´Ó³ØÖÐÑ¡ÔñÒ»¸ö¿ÕÏеĿͻ§¶Ë½øÐз¢ËÍ£¬²»ÐèÒª·þÎñÆ÷ÏìÓ¦


    bool post_to_idle_client(const std::string& message);


    bool post_binary_to_idle_client(const std::vector<char>& binary_data);


 


    // ´Ó³ØÖÐÑ¡ÔñÒ»¸ö¿ÕÏеĿͻ§¶Ë½øÐз¢ËÍ£¬²¢ÐèÒª·þÎñÆ÷ÏìÓ¦


    bool send_to_idle_client(const std::string& message, int timeout_ms);


    bool send_binary_to_idle_client(const std::vector<char>& binary_data, int timeout_ms);


 


    // ´Ó³ØÖÐÑ¡ÔñÒ»¸ö¿ÕÏеĿͻ§¶Ë½øÐз¢ËÍ£¬²¢µÈ´ý·þÎñÆ÷ÏìÓ¦


    bool send_to_idle_client_and_wait(const std::string& message, int timeout_ms);


    bool send_binary_to_idle_client_and_wait(const std::vector<char>& binary_data, int timeout_ms);


 


    // ´Ó³ØÖÐÑ¡ÔñÒ»¸ö¿ÕÏеĿͻ§¶Ë½øÐз¢ËÍ£¬²¢µÈ´ý·þÎñÆ÷ÏìÓ¦£¬Èç¹û³¬Ê±ÔòÖØÊÔ


    bool send_to_idle_client_with_retry(const std::string& message, int total_timeout_ms, int retry_interval_ms);


    bool send_binary_to_idle_client_with_retry(std::vector<char>&& binary_data, int total_timeout_ms, int retry_interval_ms);


 


    // ÉèÖàWebSocket ¿Í»§¶Ëʼþ´¦Àí³ÌÐò


    void set_open_handler(WebSocketClient::OpenHandler handler);


    void set_message_handler(WebSocketClient::MessageHandler handler);


    void set_close_handler(WebSocketClient::CloseHandler handler);


    void set_fail_handler(WebSocketClient::FailHandler handler);


    void set_pong_handler(WebSocketClient::PongHandler handler);


 


private:


    std::vector<std::shared_ptr<WebSocketClient>> m_clients;


    ThreadPool m_thread_pool;


};


 


#endif // WEBSOCKETCLIENTWITHPOOL_H