mcs/class/RabbitMQ.Client/src/client/impl/MainSession.cs

   1 // This source code is dual-licensed under the Apache License, version
   2 // 2.0, and the Mozilla Public License, version 1.1.
   3 //
   4 // The APL v2.0:
   5 //
   6 //---------------------------------------------------------------------------
   7 //   Copyright (C) 2007, 2008 LShift Ltd., Cohesive Financial
   8 //   Technologies LLC., and Rabbit Technologies Ltd.
   9 //
  10 //   Licensed under the Apache License, Version 2.0 (the "License");
  11 //   you may not use this file except in compliance with the License.
  12 //   You may obtain a copy of the License at
  13 //
  14 //       http://www.apache.org/licenses/LICENSE-2.0
  15 //
  16 //   Unless required by applicable law or agreed to in writing, software
  17 //   distributed under the License is distributed on an "AS IS" BASIS,
  18 //   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  19 //   See the License for the specific language governing permissions and
  20 //   limitations under the License.
  21 //---------------------------------------------------------------------------
  22 //
  23 // The MPL v1.1:
  24 //
  25 //---------------------------------------------------------------------------
  26 //   The contents of this file are subject to the Mozilla Public License
  27 //   Version 1.1 (the "License"); you may not use this file except in
  28 //   compliance with the License. You may obtain a copy of the License at
  29 //   http://www.rabbitmq.com/mpl.html
  30 //
  31 //   Software distributed under the License is distributed on an "AS IS"
  32 //   basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See the
  33 //   License for the specific language governing rights and limitations
  34 //   under the License.
  35 //
  36 //   The Original Code is The RabbitMQ .NET Client.
  37 //
  38 //   The Initial Developers of the Original Code are LShift Ltd.,
  39 //   Cohesive Financial Technologies LLC., and Rabbit Technologies Ltd.
  40 //
  41 //   Portions created by LShift Ltd., Cohesive Financial Technologies
  42 //   LLC., and Rabbit Technologies Ltd. are Copyright (C) 2007, 2008
  43 //   LShift Ltd., Cohesive Financial Technologies LLC., and Rabbit
  44 //   Technologies Ltd.;
  45 //
  46 //   All Rights Reserved.
  47 //
  48 //   Contributor(s): ______________________________________.
  49 //
  50 //---------------------------------------------------------------------------
  51 using System;
  52
  53 using RabbitMQ.Client;
  54 using RabbitMQ.Client.Events;
  55 using RabbitMQ.Client.Exceptions;
  56
  57 // We use spec version 0-9 for common constants such as frame types,
  58 // error codes, and the frame end byte, since they don't vary *within
  59 // the versions we support*. Obviously we may need to revisit this if
  60 // that ever changes.
  61 using CommonFraming = RabbitMQ.Client.Framing.v0_9;
  62
  63 namespace RabbitMQ.Client.Impl
  64 {
  65     ///<summary>Small ISession implementation used only for channel 0.</summary>
  66     public class MainSession: Session
  67     {
  68         public bool m_closing = false;
  69         public int m_closeClassId;
  70         public int m_closeMethodId;
  71         public int m_closeOkClassId;
  72         public int m_closeOkMethodId;
  73
  74         public bool m_closeServerInitiated;
  75
  76         private readonly object m_closingLock = new object();
  77         public delegate void SessionCloseDelegate();
  78         public SessionCloseDelegate handler;
  79
  80         public MainSession(ConnectionBase connection)
  81             : base(connection, 0)
  82         {
  83             Command request;
  84             connection.Protocol.CreateConnectionClose(0,"",
  85                                                       out request,
  86                                                       out m_closeOkClassId,
  87                                                       out m_closeOkMethodId);
  88             m_closeClassId = request.Method.ProtocolClassId;
  89             m_closeMethodId = request.Method.ProtocolMethodId;
  90         }
  91
  92         ///<summary> Set channel 0 as quiescing </summary>
  93         ///<remarks>
  94         /// Method should be idempotent. Cannot use base.Close
  95         /// method call because that would prevent us from
  96         /// sending/receiving Close/CloseOk commands
  97         ///</remarks>
  98         public void SetSessionClosing(bool closeServerInitiated)
  99         {
 100             lock(m_closingLock)
 101             {
 102                 if (!m_closing)
 103                 {
 104                     m_closing = true;
 105                     m_closeServerInitiated = closeServerInitiated;
 106                 }
 107             }
 108         }
 109
 110         public SessionCloseDelegate Handler
 111         {
 112             get { return handler; }
 113             set { handler = new SessionCloseDelegate(value); }
 114         }
 115
 116         public override void HandleFrame(Frame frame)
 117         {
 118
 119             lock(m_closingLock)
 120             {
 121                 if (!m_closing)
 122                 {
 123                     base.HandleFrame(frame);
 124                     return;
 125                 }
 126             }
 127
 128             if (!m_closeServerInitiated
 129                 && (frame.Type == CommonFraming.Constants.FrameMethod))
 130             {
 131                 MethodBase method = Connection.Protocol.DecodeMethodFrom(frame.GetReader());
 132                 if ((method.ProtocolClassId == m_closeClassId)
 133                     && (method.ProtocolMethodId == m_closeMethodId))
 134                 {
 135                     base.HandleFrame(frame);
 136                     return;
 137                 }
 138
 139                 if ((method.ProtocolClassId == m_closeOkClassId)
 140                     && (method.ProtocolMethodId == m_closeOkMethodId))
 141                 {
 142                     // This is the reply (CloseOk) we were looking for
 143                     // Call any listener attached to this session
 144                     this.Handler();
 145                     return;
 146                 }
 147             }
 148
 149             // Either a non-method frame, or not what we were looking
 150             // for. Ignore it - we're quiescing.
 151         }
 152
 153
 154         public override void Transmit(Command cmd)
 155         {
 156             lock(m_closingLock)
 157             {
 158                 if (!m_closing)
 159                 {
 160                     base.Transmit(cmd);
 161                     return;
 162                 }
 163             }
 164
 165             // Allow always for sending close ok
 166             // Or if application initiated, allow also for sending close
 167             MethodBase method = cmd.m_method;
 168             if ( ((method.ProtocolClassId == m_closeOkClassId)
 169                   && (method.ProtocolMethodId == m_closeOkMethodId))
 170                   || (!m_closeServerInitiated && (
 171                       (method.ProtocolClassId == m_closeClassId) &&
 172                       (method.ProtocolMethodId == m_closeMethodId))
 173                       ))
 174             {
 175                 base.Transmit(cmd);
 176                 return;
 177             }
 178         }
 179     }
 180 }