public MqClient(string address) : base(address)
{
this.heartbeatMessage = new Message();
this.heartbeatMessage.Headers[Protocol.CMD] = Protocol.PING;
OnMessage += async(msg) => {
string mq = (string)msg.Headers[Protocol.MQ];
string channel = (string)msg.Headers[Protocol.CHANNEL];
if (mq == null || channel == null)
{
logger.Warn("Missing mq or channel in response: " + JsonKit.SerializeObject(msg));
return;
}
MqHandler mqHandler = null;
foreach (var e in this.handlers)
{
if (e.Mq == mq && e.Channel == channel)
{
mqHandler = e;
break;
}
}
if (mqHandler == null)
{
logger.Warn(string.Format("Missing handler for mq={}, channel={}", mq, channel));
return;
}
mqHandler.Handler(msg);
string windowStr = (string)msg.Headers[Protocol.WINDOW];
int? window = null;
if (windowStr != null)
{
window = int.Parse(windowStr);
}
if (window != null && window <= mqHandler.Window / 2)
{
var sub = new Message();
sub.Headers[Protocol.CMD] = Protocol.SUB;
sub.Headers[Protocol.MQ] = mq;
sub.Headers[Protocol.CHANNEL] = channel;
sub.Headers[Protocol.ACK] = false;
sub.Headers[Protocol.WINDOW] = mqHandler.Window.ToString();
await this.SendAsync(sub);
}
};
}
protected async Task SendUnsafeAsync(Message req, Action <Message> beforeSend = null, CancellationToken?token = null)
{
if (token == null)
{
token = CancellationToken.None;
}
if (beforeSend == null)
{
beforeSend = BeforeSend;
}
if (beforeSend != null)
{
beforeSend(req);
}
if (AuthEnabled)
{
Auth.Sign(ApiKey, SecretKey, req);
}
string msg = JsonKit.SerializeObject(req);
UTF8Encoding encoder = new UTF8Encoding();
await socket.SendAsync(new ArraySegment <byte>(encoder.GetBytes(msg)),
WebSocketMessageType.Text, true, token.Value);
}
public void Start()
{
if (Mq == null)
{
throw new MissingFieldException("missing mq field");
}
if (Channel == null)
{
Channel = Mq;
}
processor.MountDoc();
MqClient client = new MqClient(MqServerAddress);
if (AuthEnabled)
{
client.AuthEnabled = AuthEnabled;
client.ApiKey = ApiKey;
client.SecretKey = SecretKey;
}
client.OnOpen += async(cli) =>
{
Message msg = new Message();
msg.Headers[Protocol.CMD] = Protocol.CREATE;
msg.Headers[Protocol.MQ] = Mq;
msg.Headers[Protocol.MQ_TYPE] = MqType;
msg.Headers[Protocol.CHANNEL] = Channel;
var res = await client.InvokeAsync(msg);
logger.Info(JsonKit.SerializeObject(res));
msg = new Message();
msg.Headers[Protocol.CMD] = Protocol.SUB;
msg.Headers[Protocol.MQ] = Mq;
msg.Headers[Protocol.CHANNEL] = Channel;
res = await client.InvokeAsync(msg);
logger.Info(JsonKit.SerializeObject(res));
msg = new Message();
msg.Headers[Protocol.CMD] = Protocol.BIND;
msg.Headers[Protocol.MQ] = Mq;
msg.Body = processor.UrlEntryList(Mq);
res = await client.InvokeAsync(msg);
logger.Info(JsonKit.SerializeObject(res));
};
client.AddMqHandler(Mq, Channel, async(request) =>
{
string prefix = processor.UrlPrefix;
string url = request.Url;
if (url != null && url.StartsWith(prefix))
{
url = url.Substring(prefix.Length);
request.Url = PathKit.Join(url);
}
string id = (string)request.Headers[Protocol.ID];
string source = (string)request.Headers[Protocol.SOURCE];
Message response = new Message();
try
{
await processor.ProcessAsync(request, response);
}
catch (Exception e)
{
while (e.InnerException != null)
{
e = e.InnerException;
}
response.Status = 500;
response.Headers["content-type"] = "text/plain; charset=utf8;";
response.Body = e.Message;
}
response.Headers[Protocol.CMD] = Protocol.ROUTE;
response.Headers[Protocol.ID] = id;
response.Headers[Protocol.TARGET] = source;
await client.SendAsync(response);
});
client.ConnectAsync().Wait();
}
private bool checkParams(Message req, Message res, MethodInfo method, object[] args, object[] invokeArgs)
{
ParameterInfo[] pinfo = method.GetParameters();
int count = 0;
foreach (ParameterInfo info in pinfo)
{
if (typeof(Message).IsAssignableFrom(info.ParameterType))
{
continue;
}
count++;
}
if (count != args.Length)
{
reply(res, 400, string.Format("Request(Url={0}, Method={1}, Params={2}) Bad Format", req.Url, method.Name, JsonKit.SerializeObject(args)));
return(false);
}
int j = 0;
for (int i = 0; i < pinfo.Length; i++)
{
if (typeof(Message).IsAssignableFrom(pinfo[i].ParameterType))
{
invokeArgs[i] = req;
continue;
}
invokeArgs[i] = JsonKit.Convert(args[j++], pinfo[i].ParameterType);
}
return(true);
}