/// <summary>
///
/// </summary>
/// <param name="context"></param>
/// <param name="message"></param>
public override void ChannelRead(IChannelHandlerContext context, object message)
{
var msg = (Message)message;
var loginState = ChannelState.GetLoginState(context);
//如果是服务请求消息则进行处理，其它消息则透传
if (msg.Header != null && msg.Header.Type == MessageType.SERVICE_REQ)
{
logger.LogDebug($"Recieve service message. RequestId={msg.Header.RequestId}, Local={context.GetLocalNetString()}, Remote={context.GetRemoteNetString()}");
if (serviceRecieveHandler != null)
{
Task.Run(() =>
{
var handlerTask = serviceRecieveHandler(msg.Body, msg.Header.Attachments, loginState);
if (handlerTask != null)
{
handlerTask.ContinueWith(task =>
{
context.WriteAndFlushAsync(BuildResponse(msg.Header.RequestId, task.Result));
logger.LogDebug($"Service is response. RequestId={msg.Header.RequestId}, Local={context.GetLocalNetString()}, Remote={context.GetRemoteNetString()}");
});
}
});
}
}
else if (msg.Header != null && msg.Header.Type == MessageType.ONE_WAY)
{
logger.LogDebug($"Recieve oneway service message. RequestId={msg.Header.RequestId}, Local={context.GetLocalNetString()}, Remote={context.GetRemoteNetString()}");
if (serviceRecieveHandler != null)
{
Task.Run(() =>
{
serviceRecieveHandler?.Invoke(msg.Body, msg.Header.Attachments, loginState);
});
}
}
else
{
context.FireChannelRead(message);
}
}
/// <summary>
///
/// </summary>
/// <param name="context"></param>
/// <param name="message"></param>
public override void ChannelRead(IChannelHandlerContext context, object message)
{
var msg = (Message)message;
var loginState = ChannelState.GetLoginState(context);
//如果是握手请求消息，处理，其它消息透传
if (msg.Header != null && msg.Header.Type == MessageType.LOGIN_REQ && loginState != null)
{
loginState.IsLoginSuccess = false;
if (loginRecieveHandler != null)
{
Task.Run(() =>
{
loginRecieveHandler(new LoginAuthInfo()
{
Body = msg.Body, Attachments = msg.Header.Attachments, RemoteAddress = context.GetRemoteAddress()
}).ContinueWith(t =>
{
var resData = t.Result;
if (resData != null)
{
loginState.IsLoginSuccess = resData.AuthResult;
loginState.Identity = resData.AuthIdentity;
loginState.RemoteAddress = context.GetRemoteAddress();
if (loginState.IsLoginSuccess)
{
logger.LogInformation($"Login success. Local={context.GetLocalNetString()}, Remote={context.GetRemoteNetString()}");
var loginResp = BuildResponse(resData);
context.WriteAndFlushAsync(loginResp);
}
else
{
logger.LogDebug($"Login failed. MessageType={msg.Header.Type}, Local={context.GetLocalNetString()}, Remote={context.GetRemoteNetString()}, AuthStatusCode={resData.AuthStatusCode}, AuthFailedMessage={resData.AuthFailedMessage}");
var loginResp = BuildResponse(resData);
context.WriteAndFlushAsync(loginResp).ContinueWith(task =>
{
context.CloseAsync();
});
}
}
else
{
logger.LogDebug($"Drop message because LoginResponseData is null. MessageType={msg.Header.Type}, Local={context.GetLocalNetString()}, Remote={context.GetRemoteNetString()}");
var loginResp = BuildResponse((byte)AuthStatusCodes.LoginResponseDataIsNull);
context.WriteAndFlushAsync(loginResp).ContinueWith(task =>
{
context.CloseAsync();
});
}
});
});
}
else
{
logger.LogDebug($"Drop message because LoginRecieveHandler is null. MessageType={msg.Header.Type}, Local={context.GetLocalNetString()}, Remote={context.GetRemoteNetString()}");
var loginResp = BuildResponse((byte)AuthStatusCodes.LoginRecieveHandlerIsNull);
context.WriteAndFlushAsync(loginResp).ContinueWith(task =>
{
context.CloseAsync();
});
}
}
else
{
if (loginState != null && loginState.IsLoginSuccess)
{
context.FireChannelRead(msg);
}
else
{
logger.LogDebug($"Drop message because no login. MessageType={msg.Header.Type}, Local={context.GetLocalNetString()}, Remote={context.GetRemoteNetString()}");
var loginResp = BuildResponse((byte)AuthStatusCodes.NoLogin);
context.WriteAndFlushAsync(loginResp).ContinueWith(task =>
{
context.CloseAsync();
});
}
}
}
