public void Start()
{
List <TopCometStreamRequest> cometRequests = conf.GetConnectReqParam();
msgConsumeFactory = new StreamMsgConsumeFactory(conf.GetMinThreads(), conf.GetMaxThreads(), conf.GetQueueSize());
for (int i = 0; i < cometRequests.Count; i++)
{
try
{
TopCometStreamRequest cometRequest = cometRequests[i];
if (cometRequest.GetConnectListener() == null)
{
cometRequest.SetConnectListener(connectionListener);
}
if (cometRequest.GetMsgListener() == null)
{
cometRequest.SetMsgListener(cometMessageListener);
}
Thread controlThread = new Thread(delegate()
{
ControlThread(msgConsumeFactory, ref bstop, conf, cometRequest);
});
controlThread.Name = "stream-control-thread-connectid-" + cometRequest.GetConnectId();
controlThread.Start();
controlThreads.Add(controlThread);
}
catch (Exception e)
{
continue;
}
}
}
/**
* 1，userid是用户的id，是大于0的整数。
* 如果不需要传入userid，则可以传入小于等于0的数值。
* </br>
* 2，如果不指定userid，则所有与此appkey关联的所有用户的消息都会发送到此连接上</br>
* 3，connectId为连接标识（用于多连接）。可以为空。对于多连接的详细描述请参考文档。
* @param appkey
* @param secret
* @param userid 传入用户的userid
* @param connectId 连接标识
*/
public Configuration(string appkey, string secret, string userid, string connectId)
{
TopCometStreamRequest cometReq = new TopCometStreamRequest(appkey, secret, userid, connectId);
connectReqParam = new List <TopCometStreamRequest>(1);
connectReqParam.Add(cometReq);
}
public IStreamImplementation GetMsgStreamImpl(TopCometStreamRequest cometReq)
{
if (cometReq != null)
{
cometReq.GetConnectListener().OnBeforeConnect();
}
TopDictionary param = new TopDictionary();
param.Add(StreamConstants.PARAM_APPKEY, cometReq.GetAppkey());
if (!String.IsNullOrEmpty(cometReq.GetUserId()))
{
param.Add(StreamConstants.PARAM_USERID, cometReq.GetUserId());
}
if (!String.IsNullOrEmpty(cometReq.GetConnectId()))
{
param.Add(StreamConstants.PARAM_CONNECT_ID, cometReq.GetConnectId());
}
param.Add(StreamConstants.PARAM_TIMESTAMP, DateTime.Now.Ticks);
IDictionary <string, string> otherParam = cometReq.GetOtherParam();
if (otherParam != null && otherParam.Count > 0)
{
IEnumerator <KeyValuePair <string, string> > kvps = otherParam.GetEnumerator();
while (kvps.MoveNext())
{
param.Add(kvps.Current.Key, kvps.Current.Value);
}
}
string sign = null;
try
{
sign = TopUtils.SignTopRequest(param, cometReq.GetSecret(), true);
if (String.IsNullOrEmpty(sign))
{
throw new Exception("Get sign error");
}
}
catch (Exception e)
{
throw e;
}
param.Add(StreamConstants.PARAM_SIGN, sign);
HttpClient httpClient = new HttpClient(conf, param);
HttpResponse response;
response = httpClient.Post();
return(currentStreamImpl = new MessageStreamImpl(msgConsumeFactory, response, cometMessageListener, this));
}
public IStreamImplementation GetMsgStreamImpl(TopCometStreamRequest cometReq)
{
if (cometReq != null)
{
cometReq.GetConnectListener().OnBeforeConnect();
}
TopDictionary param = new TopDictionary();
param.Add(StreamConstants.PARAM_APPKEY, cometReq.GetAppkey());
if (!String.IsNullOrEmpty(cometReq.GetUserId()))
{
param.Add(StreamConstants.PARAM_USERID, cometReq.GetUserId());
}
if (!String.IsNullOrEmpty(cometReq.GetConnectId()))
{
param.Add(StreamConstants.PARAM_CONNECT_ID, cometReq.GetConnectId());
}
param.Add(StreamConstants.PARAM_TIMESTAMP, DateTime.Now.Ticks);
IDictionary<string, string> otherParam = cometReq.GetOtherParam();
if (otherParam != null && otherParam.Count > 0)
{
IEnumerator<KeyValuePair<string, string>> kvps = otherParam.GetEnumerator();
while (kvps.MoveNext())
{
param.Add(kvps.Current.Key, kvps.Current.Value);
}
}
string sign = null;
try
{
sign = TopUtils.SignTopRequest(param, cometReq.GetSecret(), true);
if (String.IsNullOrEmpty(sign))
{
throw new Exception("Get sign error");
}
}
catch (Exception e)
{
throw e;
}
param.Add(StreamConstants.PARAM_SIGN, sign);
HttpClient httpClient = new HttpClient(conf, param);
HttpResponse response;
response = httpClient.Post();
return currentStreamImpl = new MessageStreamImpl(msgConsumeFactory, response, cometMessageListener, this);
}
private void StartConsumeThread(TopCometStreamRequest cometReq)
{
IStreamImplementation stream = null;
try
{
stream = GetMsgStreamImpl(cometReq);
if (cometReq.GetConnectListener() != null)
{
cometReq.GetConnectListener().OnConnect();
}
}
catch (TopCometSysErrorException e)
{
bstop = true;
logger.Error(e.Message);
if (cometReq.GetConnectListener() != null)
{
cometReq.GetConnectListener().OnSysErrorException(e);
}
}
catch (Exception ex)
{
bstop = true;
logger.Error(ex.Message);
if (cometReq.GetConnectListener() != null)
{
cometReq.GetConnectListener().OnConnectError(ex);
}
}
lastStartConsumeThread = DateTime.Now.Ticks;
Thread TopCometStreamConsumeThread = new Thread(delegate()
{
TopCometStreamConsume(lastStartConsumeThread, ref bstop, stream, cometReq.GetConnectListener());
});
TopCometStreamConsumeThread.Name = "top-stream-consume-thread" + cometReq.GetConnectId();
TopCometStreamConsumeThread.Start();
}
/**
* 1，userid是用户的id，是大于0的整数。
* 如果不需要传入userid，则可以传入小于等于0的数值。
* </br>
* 2，如果不指定userid，则所有与此appkey关联的所有用户的消息都会发送到此连接上</br>
* 3，connectId为连接标识（用于多连接）。可以为空。对于多连接的详细描述请参考文档。
* @param appkey
* @param secret
* @param userid 传入用户的userid
* @param connectId 连接标识
*/
public Configuration(string appkey, string secret, string userid, string connectId)
{
TopCometStreamRequest cometReq = new TopCometStreamRequest(appkey, secret, userid, connectId);
connectReqParam = new List<TopCometStreamRequest>(1);
connectReqParam.Add(cometReq);
}
private void StartConsumeThread(TopCometStreamRequest cometReq)
{
IStreamImplementation stream = null;
try
{
stream = GetMsgStreamImpl(cometReq);
if (cometReq.GetConnectListener() != null)
{
cometReq.GetConnectListener().OnConnect();
}
}
catch (TopCometSysErrorException e)
{
bstop = true;
logger.Error(e.Message);
if (cometReq.GetConnectListener() != null)
{
cometReq.GetConnectListener().OnSysErrorException(e);
}
}
catch (Exception ex)
{
bstop = true;
logger.Error(ex.Message);
if (cometReq.GetConnectListener() != null)
{
cometReq.GetConnectListener().OnConnectError(ex);
}
}
lastStartConsumeThread = DateTime.Now.Ticks;
Thread TopCometStreamConsumeThread = new Thread(delegate()
{
TopCometStreamConsume(lastStartConsumeThread, ref bstop, stream, cometReq.GetConnectListener());
});
TopCometStreamConsumeThread.Name = "top-stream-consume-thread" + cometReq.GetConnectId();
TopCometStreamConsumeThread.Start();
}
void ControlThread(StreamMsgConsumeFactory msgConsumeFactory, ref bool bstop, Configuration conf, TopCometStreamRequest cometReq)
{
long lastSleepTime = 0;
while (!bstop)
{
if (allStop)
{
break;
}
try
{
if (StreamConstants.SERVER_DEPLOY.Equals(serverRespCode))
{ // 服务端在发布
logger.Info("Server is upgrade sleep " + conf.GetSleepTimeOfServerInUpgrade() + " seconds");
Thread.Sleep(conf.GetSleepTimeOfServerInUpgrade() * 1000);
StartConsumeThread(cometReq);
}
else if (/*客户端第一次发起连接请求*/
StreamConstants.CLIENT_FIRST_CONNECT.Equals(serverRespCode) ||
/*服务端主动断开了所有的连接*/
StreamConstants.SERVER_REHASH.Equals(serverRespCode) ||
/*连接到达最大时间*/
StreamConstants.CONNECT_REACH_MAX_TIME.Equals(serverRespCode) ||
/*在一些异常情况下需要重连*/
StreamConstants.RECONNECT.Equals(serverRespCode))
{
StartConsumeThread(cometReq);
}
else if (/*客户端自己把自己踢开*/
StreamConstants.CLIENT_KICKOFF.Equals(serverRespCode) ||
/*服务端把客户端踢开*/
StreamConstants.SERVER_KICKOFF.Equals(serverRespCode))
{
if ((StreamConstants.CLIENT_KICKOFF.Equals(serverRespCode) && !isReconnect) ||
StreamConstants.SERVER_KICKOFF.Equals(serverRespCode))
{
break;// 终止掉当前线程
}
}
else
{ //错误码设置出错，停止线程
bstop = true;
break;
}
//连接成功，开始休眠
try
{
Monitor.Enter(objLock);
{
lastSleepTime = DateTime.Now.Ticks;
Monitor.Wait(objLock, conf.GetHttpReconnectInterval() * 1000);
if (DateTime.Now.Ticks - lastSleepTime >= (conf.GetHttpReconnectInterval()) * 1000 * 10000)
{
/*
* 快要到达连接的最大时间了，需要重新发起连接
*/
serverRespCode = StreamConstants.RECONNECT;
isReconnect = true;
}//否则，是由于某种原因被唤醒的
}
}
catch (Exception e)
{
logger.Error(e.Message);
}
finally
{
Monitor.Exit(objLock);
}
}
catch (Exception e)
{
logger.Error("Occur some error,stop the stream consume" + e.Message);
bstop = true;
try
{
Monitor.Enter(objLock);
Monitor.PulseAll(objLock);
}
finally
{
Monitor.Exit(objLock);
}
}
}
if (currentStreamImpl != null)
{
try
{
currentStreamImpl.Close();
}
catch (Exception e)
{
// ignore
}
}
logger.Info("Stop stream consume");
}
void ControlThread(StreamMsgConsumeFactory msgConsumeFactory, ref bool bstop, Configuration conf, TopCometStreamRequest cometReq)
{
long lastSleepTime = 0;
while (!bstop)
{
if (allStop)
{
break;
}
try
{
if (StreamConstants.SERVER_DEPLOY.Equals(serverRespCode))
{ // 服务端在发布
logger.Info("Server is upgrade sleep " + conf.GetSleepTimeOfServerInUpgrade() + " seconds");
Thread.Sleep(conf.GetSleepTimeOfServerInUpgrade() * 1000);
StartConsumeThread(cometReq);
}
else if (/*客户端第一次发起连接请求*/
StreamConstants.CLIENT_FIRST_CONNECT.Equals(serverRespCode) ||
/*服务端主动断开了所有的连接*/
StreamConstants.SERVER_REHASH.Equals(serverRespCode) ||
/*连接到达最大时间*/
StreamConstants.CONNECT_REACH_MAX_TIME.Equals(serverRespCode) ||
/*在一些异常情况下需要重连*/
StreamConstants.RECONNECT.Equals(serverRespCode))
{
StartConsumeThread(cometReq);
}
else if (/*客户端自己把自己踢开*/
StreamConstants.CLIENT_KICKOFF.Equals(serverRespCode) ||
/*服务端把客户端踢开*/
StreamConstants.SERVER_KICKOFF.Equals(serverRespCode))
{
if ((StreamConstants.CLIENT_KICKOFF.Equals(serverRespCode) && !isReconnect) ||
StreamConstants.SERVER_KICKOFF.Equals(serverRespCode))
{
break;// 终止掉当前线程
}
}
else
{ //错误码设置出错，停止线程
bstop = true;
break;
}
//连接成功，开始休眠
try
{
Monitor.Enter(objLock);
{
lastSleepTime = DateTime.Now.Ticks;
Monitor.Wait(objLock, conf.GetHttpReconnectInterval() * 1000);
if (DateTime.Now.Ticks - lastSleepTime >= (conf.GetHttpReconnectInterval()) * 1000 * 10000)
{
/*
* 快要到达连接的最大时间了，需要重新发起连接
*/
serverRespCode = StreamConstants.RECONNECT;
isReconnect = true;
}//否则，是由于某种原因被唤醒的
}
}
catch (Exception e)
{
logger.Error(e.Message);
}
finally
{
Monitor.Exit(objLock);
}
}
catch (Exception e)
{
logger.Error("Occur some error,stop the stream consume" + e.Message);
bstop = true;
try
{
Monitor.Enter(objLock);
Monitor.PulseAll(objLock);
}
finally
{
Monitor.Exit(objLock);
}
}
}
if (currentStreamImpl != null)
{
try
{
currentStreamImpl.Close();
}
catch (Exception e)
{
// ignore
}
}
logger.Info("Stop stream consume");
}
/// <summary>
/// 三方集成用
/// </summary>
/// <param name="appkey"></param>
/// <param name="secret"></param>
/// <param name="connectId"></param>
/// <param name="topics"></param>
/// <param name="otherParam"></param>
public Configuration(string appkey, string secret, string connectId, List<string> topics, IDictionary<string, string> otherParam)
{
TopCometStreamRequest cometReq = new TopCometStreamRequest(appkey, secret, null, connectId);
if (otherParam == null)
otherParam = new Dictionary<string, string>(1);
if (topics != null && topics.Count > 0)
{
StringBuilder topic = new StringBuilder();
string strTopic = null;
foreach (string t in topics)
{
topic.Append(t).Append(";");
strTopic = topic.ToString();
}
otherParam.Add("topic", strTopic.Substring(0, strTopic.Length - 1));
}
cometReq.SetOtherParam(otherParam);
connectReqParam = new List<TopCometStreamRequest>(1);
connectReqParam.Add(cometReq);
}
