Exemplo n.º 1
0
 /// <summary>
 /// 客户端取消订阅
 /// </summary>
 /// <param name="unSubScribeFrame"></param>
 /// <param name="stompBehavior"></param>
 public void UnSubScribe(StompFrame unSubScribeFrame, StompBehavior stompBehavior)
 {
     lock (stompSubscribeDictionaryLock)
     {
         if (unSubScribeFrame.Headers.ContainsKey(StompHeaders.Id) && string.IsNullOrEmpty(unSubScribeFrame.Headers[StompHeaders.Id]))
         {
             string destinationOfsubid = stompBehavior.topicPathSubidDictionary.FirstOrDefault(d => d.Value == unSubScribeFrame.Headers[StompHeaders.Id]).Key;
             if (stompSubscribeDictionary.ContainsKey(destinationOfsubid))
             {
                 if (stompSubscribeDictionary[destinationOfsubid].ContainsKey(stompBehavior.ID))
                 {
                     stompSubscribeDictionary[destinationOfsubid].Remove(stompBehavior.ID);
                 }
             }
         }
     }
 }
Exemplo n.º 2
0
        /// <summary>
        /// when stompBehavior closing,need remove subscribe
        /// </summary>
        /// <param name="stompBehavior"></param>
        public void StompBehaviorOnClose(StompBehavior stompBehavior)
        {
            lock (stompSubscribeDictionaryLock)
            {
                foreach (var destination in stompBehavior.topicPathSubidDictionary.Keys)
                {
                    if (stompSubscribeDictionary.ContainsKey(destination))
                    {
                        if (stompSubscribeDictionary[destination].ContainsKey(stompBehavior.ID))
                        {
                            stompSubscribeDictionary[destination].Remove(stompBehavior.ID);

                            if (stompSubscribeDictionary[destination].Count == 0)
                            {
                                stompSubscribeDictionary.Remove(destination);
                            }
                        }
                    }
                }
            }
        }
Exemplo n.º 3
0
        /// <summary>
        /// 添加一个新的订阅地址
        /// </summary>
        /// <param name="subScribeFrame">客户端的订阅数据包</param>
        /// <param name="stompBehavior">客户端连接对象</param>
        /// <returns>null:Success,not NULL:ERROR</returns>
        public StompFrame SubScribe(StompFrame subScribeFrame, StompBehavior stompBehavior)
        {
            //* If the server cannot successfully create the subscription, the server MUST send the client an ERROR frame and disconnect the client.
            lock (stompSubscribeDictionaryLock)
            {
                if (!subScribeFrame.Headers.ContainsKey(StompHeaders.Destination))
                {
                    return(new ErrorFrame("Headers missing:'destination'", "SUBSCRIBE 包必须携带'destination'头,而且值不能为空.\nSUBSCRIBE Frame's headers MUST Contain 'destination',and the value MUST NOT null/empty", subScribeFrame));
                }
                else if (string.IsNullOrEmpty(subScribeFrame.Headers[StompHeaders.Destination]))
                {
                    return(new ErrorFrame("Header['destination'] MUST NOT be null or empty", "SUBSCRIBE 包规定header['destination']的值不能为空.\nSUBSCRIBE Frame's header['destination']: MUST NOT be null/empty", subScribeFrame));
                }
                else if (!subscribeSupportTopicList.Contains(subScribeFrame.Headers[StompHeaders.Destination]))
                {
                    return(new ErrorFrame("Stomp Server not support this destination", "Stomp Server not support this destination : '" + subScribeFrame.Headers[StompHeaders.Destination] + "'\n", subScribeFrame));
                }
                else if (!subScribeFrame.Headers.ContainsKey(StompHeaders.Id))//subid
                {
                    return(new ErrorFrame("Headers missing: 'id'", "SUBSCRIBE 包必须携带'id'头,而且值不能为空.\nSUBSCRIBE Frame's headers MUST Contain 'id',and the value MUST uniquely" + "\nSince a single connection can have multiple open subscriptions with a server, an id header MUST be included in the frame to uniquely identify the subscription.", subScribeFrame));
                }
                else if (string.IsNullOrEmpty(subScribeFrame.Headers[StompHeaders.Id]))
                {
                    return(new ErrorFrame("Header['id'] MUST NOT be null or empty,and MUST uniquely.", "SUBSCRIBE 包必须携带'id'头,而且值必须唯一.\nSUBSCRIBE Frame's header['id'] MUST NOT null/empty,and MUST uniquely.", subScribeFrame));
                }
                if (stompBehavior.topicPathSubidDictionary.Values.Contains(subScribeFrame.Headers[StompHeaders.Id]))// check id
                {
                    if (stompBehavior.topicPathSubidDictionary.Keys.Contains(subScribeFrame.Headers[StompHeaders.Destination]) && stompBehavior.topicPathSubidDictionary[subScribeFrame.Headers[StompHeaders.Destination]] == subScribeFrame.Headers[StompHeaders.Id])
                    {
                        return(null);
                    }
                    else
                    {
                        // subid 已经用在了别的地址上
                        string destinationOfsubid = stompBehavior.topicPathSubidDictionary.FirstOrDefault(d => d.Value == subScribeFrame.Headers[StompHeaders.Id]).Key;
                        return(new ErrorFrame("SUBSCRIBE: ['id'] MUST uniquely", "this sub-id is alread been used with this destination:" + destinationOfsubid + ".\n sub-id MUST uniquely,please use another id", subScribeFrame));

                        //stompBehavior.topicPathSubidDictionary[subScribeFrame.Headers[StompHeaders.Destination]] = subScribeFrame.Headers[StompHeaders.Id];//update sub-id
                    }
                }

                if (stompBehavior.topicPathSubidDictionary.ContainsKey(subScribeFrame.Headers[StompHeaders.Destination]))//check destination
                {
                    // 已经订阅过,id都匹配
                    if (stompBehavior.topicPathSubidDictionary[subScribeFrame.Headers[StompHeaders.Destination]] == subScribeFrame.Headers[StompHeaders.Id])
                    {
                        return(null);
                    }
                    else //换了一个sub-id
                    {
                        // destination已经匹配了一个sub-id
                        //return new ErrorFrame("SUBSCRIBE: ['destination'] already linked a sub-id", "this destination is alread linked a sub-id:" + stompBehavior.topicPathSubidDictionary[subScribeFrame.Headers[StompHeaders.Destination]] + ".\n Stomp Server Don't want client Subscribe one destination by many times.", subScribeFrame);


                        // update sub-id
                        stompBehavior.topicPathSubidDictionary[subScribeFrame.Headers[StompHeaders.Destination]] = subScribeFrame.Headers[StompHeaders.Id];
                    }
                }

                //
                // ack ... Not support!
                //


                //
                // Stomp1.0
                // The body of the SUBSCRIBE command is ignored.
                //



                if (!stompBehavior.topicPathSubidDictionary.ContainsKey(subScribeFrame.Headers[StompHeaders.Destination]))
                {
                    stompBehavior.topicPathSubidDictionary.Add(subScribeFrame.Headers[StompHeaders.Destination], subScribeFrame.Headers[StompHeaders.Id]);//stompBehavior<destination:id> Add
                }

                //SubscribeFlowDirction subscribeFlowDirction = GetSubscribeFlowDirction(subScribeFrame.Headers[StompHeaders.Destination]);
                //if (subscribeFlowDirction == SubscribeFlowDirction.Application || subscribeFlowDirction == SubscribeFlowDirction.Unkonw)//Unkonw  is Error ?
                //{
                //    return null;
                //}


                if (stompSubscribeDictionary.ContainsKey(subScribeFrame.Headers[StompHeaders.Destination]))
                {
                    var idStompBehaviorKV = stompSubscribeDictionary[subScribeFrame.Headers[StompHeaders.Destination]];
                    if (!idStompBehaviorKV.ContainsKey(stompBehavior.ID))
                    {
                        stompSubscribeDictionary[subScribeFrame.Headers[StompHeaders.Destination]].Add(stompBehavior.ID, stompBehavior);
                    }
                }
                else
                {
                    stompSubscribeDictionary.Add(subScribeFrame.Headers[StompHeaders.Destination], new Dictionary <string, StompBehavior>()
                    {
                        { stompBehavior.ID, stompBehavior }
                    });
                }
            }
            return(null);
        }