Add some error log

ljcui · Oct 29, 2024 · 295baa4 · 295baa4
1 parent d642423
commit 295baa4
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/server/bolt_handler.cpp b/src/server/bolt_handler.cpp
@@ -255,6 +255,18 @@ std::function<void(bolt::BoltConnection &conn, bolt::BoltMsg msg,
             return;
         }
         auto& val = std::any_cast<const std::unordered_map<std::string, std::any>&>(fields[0]);
+        if (!val.count("principal") || !val.count("credentials")
+            || !val.count("user_agent")) {
+            LOG_ERROR() << "Hello msg fields error, "
+                           "'principal' or 'credentials' or 'user_agent' are missing.";
+            bolt::PackStream ps;
+            ps.AppendFailure({{"code", "error"},
+                              {"message", "Hello msg fields error, "
+                               "'principal' or 'credentials' or 'user_agent' are missing."}});
+            conn.Respond(std::move(ps.MutableBuffer()));
+            conn.Close();
+            return;
+        }
         auto& principal = std::any_cast<const std::string&>(val.at("principal"));
         auto& credentials = std::any_cast<const std::string&>(val.at("credentials"));
         auto galaxy = BoltServer::Instance().StateMachine()->GetGalaxy();