From 289c19b307e492717b1a2a60dc90aba638e93d87 Mon Sep 17 00:00:00 2001
From: aozhiwei <aozhiwei@kingsome.cn>
Date: Sun, 26 Nov 2023 13:00:22 +0800
Subject: [PATCH] 1

---
 server/robotserver/android_agent.cc | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/server/robotserver/android_agent.cc b/server/robotserver/android_agent.cc
index 892ae2c6..86734425 100644
--- a/server/robotserver/android_agent.cc
+++ b/server/robotserver/android_agent.cc
@@ -113,6 +113,16 @@ behaviac::EBTStatus AndroidAgent::CoLogin()
 
 behaviac::EBTStatus AndroidAgent::CoConnectBattleServer()
 {
+    PRE_ENTER_COROUTINE();
+    auto context = MAKE_BTCONTEXT
+        (
+         std::function<void (a8::WebSocketClient*, int)> old_on_error;
+         std::function<void (a8::WebSocketClient*)> old_on_connect;
+         std::function<void (a8::WebSocketClient*)> old_on_disconnect;
+         );
+    context->old_on_error = owner_->GetWebSocket()->on_error;
+    context->old_on_connect = owner_->GetWebSocket()->on_connect;
+    context->old_on_disconnect = owner_->GetWebSocket()->on_disconnect;
     owner_->GetWebSocket()->on_error =
         [this] (a8::WebSocketClient*,int err_code)
         {