From fb1efdef55d9e3ddd060410ce1e774bf74e03c8a Mon Sep 17 00:00:00 2001
From: Hang <hang@Hangs-MacBook-Pro.local>
Date: Tue, 26 Apr 2022 16:47:43 -0500
Subject: [PATCH] fast push

---
 mp3/server.py | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/mp3/server.py b/mp3/server.py
index de88ac9..623f8ab 100644
--- a/mp3/server.py
+++ b/mp3/server.py
@@ -640,14 +640,11 @@ def client_handler(conn, SELF_NODE):
                     gprint(f'[server_handler] accounts_map: '
                            f'\n{json.dumps(accounts_map, default=lambda x: x.__dict__)}')
 
-                ret_value = reformat_return_value(ret_value)
+                ret_value = reformat_return_value(ret_value, operation)
                 if 'ABORT' in ret_value:
                     broadcast_msg(format_operation_and_timestamp_json('ABORT', timestamp), connect_info)
                     abort(SELF_NODE, timestamp)
 
-                if operation.startswith('BALANCE'):
-                    account_name = operation.split()[1]
-                    ret_value = f'{account_name} = {ret_value}'
                 send_msg_to_socket(ret_value, conn)
     except Exception as e:
         rprint(f"[ERROR] client_handler error:{e.__class__.__name__}.{e}.{traceback.format_exc()}")
@@ -658,7 +655,7 @@ def client_handler(conn, SELF_NODE):
         # client_send_socket.close()
 
 
-def reformat_return_value(ret_value):
+def reformat_return_value(ret_value, operation):
     if isinstance(ret_value, bool):
         if ret_value:
             return 'True'
@@ -667,6 +664,8 @@ def reformat_return_value(ret_value):
     elif isinstance(ret_value, str):
         return ret_value
     elif isinstance(ret_value, int):
+        account_name = operation.split()[1]
+        ret_value = f'{account_name} = {ret_value}'
         return str(ret_value)
     elif ret_value == 'NOT FOUND, ABORTED':
         return ret_value
-- 
GitLab