From 20580a4c918a1f54d137a28bb34fa57a96771fc2 Mon Sep 17 00:00:00 2001
From: taynpg <taynpg@163.com>
Date: Tue, 24 Dec 2024 07:59:56 +0800
Subject: [PATCH] =?UTF-8?q?change=EF=BC=9Aupdate=E8=AF=BB=E5=85=A5?=
 =?UTF-8?q?=E6=96=87=E4=BB=B6=E6=94=B9=E4=B8=BA=E8=BE=93=E5=85=A5=E8=80=8C?=
 =?UTF-8?q?=E9=9D=9E=E9=BB=98=E8=AE=A4=E3=80=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 client/client.cpp | 35 +++++++++++++++++------------------
 1 file changed, 17 insertions(+), 18 deletions(-)

diff --git a/client/client.cpp b/client/client.cpp
index 7c6f323..2ff6ceb 100644
--- a/client/client.cpp
+++ b/client/client.cpp
@@ -50,7 +50,7 @@ void CClient::run(const std::string& ip, const std::string& port)
     client_->async_recv();
     hearts_ = std::thread([&]() { hearts(); });
     std::thread thread([&]() { io_context_.run(); });
-    logger_->warn("SupportCmd:Get|Up|Down|Cancel|Update(list.txt)");
+    logger_->warn("SupportCmd:Get|Up|Down|Cancel|Update");
     char line[512]{};
     while (std::cin.getline(line, 512)) {
 
@@ -73,26 +73,25 @@ void CClient::run(const std::string& ip, const std::string& port)
             continue;
         }
         auto vec = COfStr::split(cmd_input, " ");
-        if (vec.size() < 2) {
+        if (vec.size() == 3) {
+            if (vec[0] == "Update") {
+                request_update_list(vec[2], std::stoi(vec[1]));
+                continue;
+            }
             logger_->error("input's invalid format.");
             continue;
+        } else if (vec.size() == 2) {
+            if (vec[0] == "Down") {
+                down_task(vec[1]);
+                continue;
+            }
+            if (vec[0] == "Up") {
+                up_task(cmd_input);
+                continue;
+            }
+        } else {
+            logger_->error("No matched cmd.");
         }
-        std::string cmd = vec[0];
-        std::string param = vec[1];
-
-        if (cmd == "Down") {
-            down_task(vec[1]);
-            continue;
-        }
-        if (cmd == "Up") {
-            up_task(cmd_input);
-            continue;
-        }
-        if (cmd == "Update") {
-            request_update_list("list.txt", std::stoi(param));
-            continue;
-        }
-        logger_->error("No matched cmd.");
     }
     client_->disconnect();
     thread.join();