/// <summary>
/// 增对不同的客户端和消息类型进行处理
/// </summary>
/// <param name="regClient"></param>
/// <param name="regMessage"></param>
public void Process(RegistryClient regClient, CometMessage regMessage)
{
if (regMessage.Action == CometMessageAction.Hello)
{
ConsumerConfig consumerConfig = XmlUtil.LoadObjFromXML<ConsumerConfig>(regMessage.MessageBody);
regClient.ConsumerConfig = consumerConfig;
regClient.ProcessorID = regMessage.ProcessorID;
regClient.DotNetFramworkVersion = regMessage.DotNetFramworkVersion;
regClient.OSVersion = regMessage.OSVersion;
regClient.ClientApplicationName = consumerConfig.ApplicationName;
ESBConfig esbConfig = GetESBConfig(regClient);
m_RegistryCenter.SendData(regClient, CometMessageAction.ServiceConfig, esbConfig.ToXml(), regMessage.IsAsync);
}
else if (regMessage.Action == CometMessageAction.ListRegistryClient)
{
String message = XmlUtil.SaveXmlFromObj<List<RegistryClient>>(m_RegistryCenter.RegistryClients);
m_RegistryCenter.SendData(regClient, CometMessageAction.ListRegistryClient, message, regMessage.IsAsync);
}
else if (regMessage.Action == CometMessageAction.ResendConfig)//--管理中心向每个客户端发送配置文件
{
foreach (var item in m_RegistryCenter.RegistryClients)
{
if (item.RegistryClientType == CometClientType.Consumer
|| item.RegistryClientType == CometClientType.CallCenter
|| item.RegistryClientType == CometClientType.Monitor
|| item.RegistryClientType == CometClientType.QueueCenter
|| item.RegistryClientType == CometClientType.Portal)
{
ESBConfig esbConfig = GetESBConfig(item);
m_RegistryCenter.SendData(item, CometMessageAction.ServiceConfig, esbConfig.ToXml());
}
}
}
}
/// <summary>
/// 接收请求
/// </summary>
/// <param name="ar"></param>
private void AcceptCallback(IAsyncResult ar)
{
try
{
TcpListener listener = ar.AsyncState as TcpListener;
Socket socket = listener.EndAcceptSocket(ar);
listener.BeginAcceptSocket(new AsyncCallback(AcceptCallback), listener);
RegistryClient registryClient = new RegistryClient(socket);
registryClient.ID = Guid.NewGuid().ToString();
registryClient.ReceiveDateTime = DateTime.Now;
registryClient.RegistryClientType = CometClientType.AnKnown;
registryClient.ClientIP = registryClient.Socket.RemoteEndPoint.ToString();
registryClient.ClearBuffer();
registryClient.Socket.BeginReceive(registryClient.ReceiveBuffer, 0, registryClient.ReceiveBuffer.Length
, SocketFlags.None, new AsyncCallback(ReceiveCallback), registryClient);
Console.WriteLine("接收客户端：{0}", registryClient.Socket.RemoteEndPoint.ToString());
lock (m_RegistryClients)
{
m_RegistryClients.Add(registryClient);
}
}
catch (Exception ex)
{
XTrace.WriteLine(ex.ToString());
}
}
/// <summary>
/// 增对不同的客户端和消息类型进行处理
/// </summary>
/// <param name="regClient"></param>
/// <param name="regMessage"></param>
public void Process(RegistryClient regClient, RegistryMessage regMessage)
{
if (regMessage.ClientType == RegistryClientType.Consumer && regMessage.Action == RegistryMessageAction.Hello)
{
ConsumerConfig consumerConfig = XmlUtil.LoadObjFromXML<ConsumerConfig>(regMessage.MessageBody);
ESBConfig esbConfig = GetESBConfig(consumerConfig);
m_RegistryCenter.SendData(regClient, RegistryMessageAction.ServiceConfig, esbConfig.ToXml());
}
}
/// <summary>
/// 获取到ESBConfig
/// </summary>
/// <returns></returns>
private ESBConfig GetESBConfig(RegistryClient regClient)
{
ConsumerConfig consumerConfig = regClient.ConsumerConfig;
ESBConfig esbConfig = new ESBConfig();
//--处理中心地址
foreach (SettingUri uri in SettingUri.FindAll())
{
switch (uri.UriType)
{
case 0: //--注册中心
esbConfig.Registry.Add(new RegistryItem() { Uri = String.Format("{0}:{1}", uri.Uri, uri.Port), Load = 1 });
break;
case 1: //--监控中心
esbConfig.Monitor.Add(new MonitorItem() { Uri = String.Format("{0}:{1}", uri.Uri, uri.Port), Load = 1 });
break;
case 2: //--调用中心
esbConfig.CallCenter.Add(new CallCenterItem() { Uri = uri.Uri, Load = 1 });
break;
case 3: //--队列中心
esbConfig.MessageQueue.Add(new MessageQueueItem() { Uri = String.Format("{0}:{1}:{2}:{3}", uri.Uri, uri.Port, uri.UserName, uri.PassWord), Load = 1 });
break;
case 4: //--缓存中心
esbConfig.Cache.Add(new CacheItem() { Uri = String.Format("{0}:{1}", uri.Uri, uri.Port), Load = 1 });
break;
default:
break;
}
}
//--处理服务
if (regClient.RegistryClientType == CometClientType.Consumer
|| regClient.RegistryClientType == CometClientType.Portal
|| regClient.RegistryClientType == CometClientType.Monitor)
{
foreach (var refService in consumerConfig.Reference)
{
List<EsbView_ServiceVersion> lstBS = EsbView_ServiceVersion.FindAllPublish();
foreach(EsbView_ServiceVersion bs in lstBS.Where(x=>x.ServiceName == refService.ServiceName))
{
ServiceItem si = new ServiceItem();
si.ServiceName = bs.ServiceName;
si.Version = bs.BigVer;
si.Binding = bs.Binding;
si.IsDefault = (bs.BigVer == bs.DefaultVersion);
esbConfig.Service.Add(si);
}
}
}
else if(regClient.RegistryClientType == CometClientType.CallCenter
|| regClient.RegistryClientType == CometClientType.QueueCenter)
{
EntityList<EsbView_ServiceVersion> lstBS = EsbView_ServiceVersion.FindAllPublish();
foreach (var bs in lstBS)
{
ServiceItem si = new ServiceItem();
si.ServiceName = bs.ServiceName;
si.Version = bs.BigVer;
si.Binding = bs.Binding;
si.IsDefault = (bs.BigVer == bs.DefaultVersion);
esbConfig.Service.Add(si);
}
}
//--处理服务配置
if (regClient.RegistryClientType == CometClientType.Consumer
|| regClient.RegistryClientType == CometClientType.Portal
|| regClient.RegistryClientType == CometClientType.Monitor)
{
List<EsbView_ServiceConfig> lstSC = EsbView_ServiceConfig.FindAll();
foreach (var refService in consumerConfig.Reference)
{
List<EsbView_ServiceConfig> scs = lstSC.FindAll(x => x.ServiceName == refService.ServiceName);
foreach (EsbView_ServiceConfig sc in scs)
{
esbConfig.ServiceConfig.Add(sc);
}
}
}
else if (regClient.RegistryClientType == CometClientType.CallCenter
|| regClient.RegistryClientType == CometClientType.QueueCenter)
{
esbConfig.ServiceConfig = EsbView_ServiceConfig.FindAll();
}
return esbConfig;
}
/// <summary>
/// 向单个客户端发送数据
/// </summary>
/// <param name="rsClient"></param>
/// <param name="data"></param>
/// <param name="isAsync">默认为异步调用</param>
public void SendData(RegistryClient registryClient, CometMessageAction action, String data, Boolean isAsync = true)
{
try
{
CometMessage rm = new CometMessage()
{
Action = action,
MessageBody = data,
IsAsync = isAsync
};
String messageData = XmlUtil.SaveXmlFromObj<CometMessage>(rm);
Console.WriteLine("发送数据：{0}", messageData);
Byte[] msg = Encoding.UTF8.GetBytes(messageData);
registryClient.Socket.BeginSend(msg, 0, msg.Length, SocketFlags.None, new AsyncCallback(SendCallback), registryClient);
}
catch (Exception ex)
{
XTrace.WriteLine("发送数据时发生异常：" + ex.ToString());
lock (m_RegistryClients)
{
m_RegistryClients.Remove(registryClient);
registryClient.Dispose();
}
}
}