Browse Source

修改代码

denny 2 years ago
parent
commit
6a0b5df3b9
1 changed files with 7 additions and 5 deletions
  1. 7 5
      src/InteractionCenter/FastestServerInteractionCenterService.cs

+ 7 - 5
src/InteractionCenter/FastestServerInteractionCenterService.cs

@@ -176,7 +176,7 @@ namespace WingCloudServer.InteractionCenter
                     }
                     else if (absPath.Contains("LoginSuccess"))
                     {
-                        LoginSuccess(context);
+                        await LoginSuccess(context);
                         return;
                     }
                     else if (absPath.Contains("GetServerList"))
@@ -205,19 +205,19 @@ namespace WingCloudServer.InteractionCenter
         /// 登录成功Get请求
         /// </summary>
         /// <param name="context">监听到的上下问对象</param>
-        private void LoginSuccess(HttpListenerContext context) 
+        private async Task<bool> LoginSuccess(HttpListenerContext context) 
         {
             var requestToken = "";
             if (context.Request.Headers.Count > 0) {
                 try 
                 {
-                    requestToken = context.Request.Headers?.GetValues("Token")?.ToList()?.FirstOrDefault();
+                    requestToken = context.Request.Headers?.GetValues("Token")?.ToList()?.FirstOrDefault() ?? string.Empty;
                 }
                 catch (Exception ex) 
                 {
                     Logger.WriteLineError($"GetHeadersError:{ex}");
                     ResponseContent(context, false, JsonConvert.SerializeObject(new FastestServerBaseDTO() { Code = 0 }));
-                    return;
+                    return true;
                 }
             }
             string requestIp = null;
@@ -234,9 +234,11 @@ namespace WingCloudServer.InteractionCenter
                 {
                     _fastestServerManager.Remove(tokenInfo.Code);
                     ResponseContent(context, true, JsonConvert.SerializeObject(new FastestServerBaseDTO() { Code = 1 }));
-                    return;
+                    return true;
                 }
             }
+            ResponseContent(context, false, JsonConvert.SerializeObject(new FastestServerBaseDTO() { Code = 0 }));
+            return false;
         }
 
         /// <summary>