ws获取事件函数

Author: cyanray

include/mirai_bot.hpp

@@ -47,8 +47,18 @@ namespace Cyan
 	class EXPORTED MiraiBot
 	{
 	public:
-		MiraiBot() :qq_(0), pool_(4), http_client_("localhost", 8080) {}
-		MiraiBot(const string& host, int port) : qq_(0), pool_(4), http_client_(host, port) {}
+		MiraiBot() :
+			qq_(0), 
+			pool_(4), 
+			http_client_("localhost", 8080),
+			host_("localhost"),
+			port_(8080) {}
+		MiraiBot(const string& host, int port) : 
+			qq_(0), 
+			pool_(4), 
+			http_client_(host, port), 
+			host_(host),
+			port_(port) {}
 		~MiraiBot()
 		{
 			Release();
@@ -117,7 +127,8 @@ namespace Cyan
 	private:
 		bool SessionVerify();
 		bool SessionRelease();
-		unsigned int FetchEvents(unsigned int count = 10);
+		unsigned int FetchEvents_HTTP(unsigned int count = 10);
+		void FetchEvents_WS();
 		void ProcessEvents(const nlohmann::json& ele);
 		template<typename T>
 		inline WeakEvent MakeWeakEvent(const json& json_)
@@ -157,6 +168,8 @@ namespace Cyan
 		string authKey_;
 		QQ_t qq_;
 		string sessionKey_;
+		string host_;
+		int port_;
 		httplib::Client http_client_;
 		unordered_map<MiraiEvent, function<void(WeakEvent)> > processors_;
 		ThreadPool pool_;

src/mirai_bot.cpp

@@ -1,4 +1,7 @@
 #include "mirai_bot.hpp"
+#include "easywsclient.hpp"
+#define _SSIZE_T_DEFINED
+#include "easywsclient.cpp"
 
 namespace Cyan
 {
@@ -571,7 +574,8 @@ namespace Cyan
 			unsigned count = 0;
 			try
 			{
-				count = FetchEvents(count_per_loop);
+				FetchEvents_WS();
+				count = FetchEvents_HTTP(count_per_loop);
 			}
 			catch (const std::exception& ex)
 			{
@@ -656,7 +660,7 @@ namespace Cyan
 	}
 
 
-	unsigned int MiraiBot::FetchEvents(unsigned int count)
+	unsigned int MiraiBot::FetchEvents_HTTP(unsigned int count)
 	{
 		int received_count = 0;
 		stringstream api_url;
@@ -699,6 +703,29 @@ namespace Cyan
 
 	}
 
+	void MiraiBot::FetchEvents_WS()
+	{
+		using namespace easywsclient;
+		stringstream url;
+		url << "ws://" << host_ << ":" << port_ << "/all?sessionKey=" << sessionKey_;
+		std::shared_ptr<WebSocket> ws(WebSocket::from_url(url.str()));
+		if (!ws)
+		{
+			throw std::runtime_error("无法建立 WebSocket 连接!");
+		}
+		while (ws->getReadyState() != WebSocket::CLOSED)
+		{
+			ws->poll();
+			string eventJsonStr;
+			ws->dispatch([&](const std::string& message)
+				{
+					eventJsonStr = message;
+				});
+			json j = json::parse(eventJsonStr);
+			ProcessEvents(j);
+		}
+	}
+
 	void MiraiBot::ProcessEvents(const nlohmann::json& ele)
 	{
 		string event_name = ele["type"].get<string>();