XrdClStream.hh

Go to the documentation of this file.

//------------------------------------------------------------------------------
// Copyright (c) 2011-2012 by European Organization for Nuclear Research (CERN)
// Author: Lukasz Janyst <ljanyst@cern.ch>
//------------------------------------------------------------------------------
// XRootD is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// XRootD is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with XRootD.  If not, see <http://www.gnu.org/licenses/>.
//------------------------------------------------------------------------------
 
#ifndef __XRD_CL_STREAM_HH__
#define __XRD_CL_STREAM_HH__
 
#include "XrdCl/XrdClPoller.hh"
#include "XrdCl/XrdClStatus.hh"
#include "XrdCl/XrdClURL.hh"
#include "XrdCl/XrdClPostMasterInterfaces.hh"
#include "XrdCl/XrdClChannelHandlerList.hh"
#include "XrdCl/XrdClJobManager.hh"
#include "XrdCl/XrdClInQueue.hh"
#include "XrdCl/XrdClUtils.hh"
 
#include "XrdSys/XrdSysPthread.hh"
#include "XrdNet/XrdNetAddr.hh"
#include <list>
#include <vector>
#include <functional>
 
namespace XrdCl
{
  class  Message;
  class  Channel;
  class  TransportHandler;
  class  TaskManager;
  struct SubStreamData;
 
  //----------------------------------------------------------------------------
  //----------------------------------------------------------------------------
  class Stream
  {
    public:
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      enum StreamStatus
      {
        Disconnected    = 0,    
        Connected       = 1,    
        Connecting      = 2,    
        Error           = 3     
      };
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      Stream( const URL *url, uint16_t streamNum );
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      ~Stream();
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      Status Initialize();
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      Status Send( Message              *msg,
                   OutgoingMsgHandler   *handler,
                   bool                  stateful,
                   time_t                expires );
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      void SetTransport( TransportHandler *transport )
      {
        pTransport = transport;
      }
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      void SetPoller( Poller *poller )
      {
        pPoller = poller;
      }
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      void SetIncomingQueue( InQueue *incomingQueue )
      {
        pIncomingQueue = incomingQueue;
        delete pQueueIncMsgJob;
        pQueueIncMsgJob = new QueueIncMsgJob( incomingQueue );
      }
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      void SetChannelData( AnyObject *channelData )
      {
        pChannelData = channelData;
      }
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      void SetTaskManager( TaskManager *taskManager )
      {
        pTaskManager = taskManager;
      }
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      void SetJobManager( JobManager *jobManager )
      {
        pJobManager = jobManager;
      }
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      Status EnableLink( PathID &path );
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      void Disconnect( bool force = false );
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      void Tick( time_t now );
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      const URL *GetURL() const
      {
        return pUrl;
      }
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      uint16_t GetStreamNumber() const
      {
        return pStreamNum;
      }
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      void ForceConnect();
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      const std::string &GetName() const
      {
        return pStreamName;
      }
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      void DisableIfEmpty( uint16_t subStream );
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      void OnIncoming( uint16_t  subStream,
                       Message  *msg,
                       uint32_t  bytesReceived );
 
      //------------------------------------------------------------------------
      // Call when one of the sockets is ready to accept a new message
      //------------------------------------------------------------------------
      std::pair<Message *, OutgoingMsgHandler *>
        OnReadyToWrite( uint16_t subStream );
 
      //------------------------------------------------------------------------
      // Call when a message is written to the socket
      //------------------------------------------------------------------------
      void OnMessageSent( uint16_t  subStream,
                          Message  *msg,
                          uint32_t  bytesSent );
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      void OnConnect( uint16_t subStream );
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      void OnConnectError( uint16_t subStream, Status status );
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      void OnError( uint16_t subStream, Status status );
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      void ForceError( Status status );
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      void OnReadTimeout( uint16_t subStream, bool &isBroken );
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      void OnWriteTimeout( uint16_t subStream );
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      void RegisterEventHandler( ChannelEventHandler *handler );
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      void RemoveEventHandler( ChannelEventHandler *handler );
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      std::pair<IncomingMsgHandler *, bool>
        InstallIncHandler( Message *msg, uint16_t stream );
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      void SetOnConnectHandler( Job *onConnJob )
      {
        delete pOnConnJob;
        pOnConnJob = onConnJob;
      }
 
    private:
 
      //------------------------------------------------------------------------
      // Job queuing the incoming messages
      //------------------------------------------------------------------------
      class QueueIncMsgJob: public Job
      {
        public:
          QueueIncMsgJob( InQueue *queue ): pQueue( queue ) {};
          virtual ~QueueIncMsgJob() {};
          virtual void Run( void *arg )
          {
            Message *msg = (Message *)arg;
            pQueue->AddMessage( msg );
          }
        private:
          InQueue *pQueue;
      };
 
      //------------------------------------------------------------------------
      // Job handling the incoming messages
      //------------------------------------------------------------------------
      class HandleIncMsgJob: public Job
      {
        public:
          HandleIncMsgJob( IncomingMsgHandler *handler ): pHandler( handler ) {};
          virtual ~HandleIncMsgJob() {};
          virtual void Run( void *arg )
          {
            Message *msg = (Message *)arg;
            pHandler->Process( msg );
            delete this;
          }
        private:
          IncomingMsgHandler *pHandler;
      };
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      void OnFatalError( uint16_t           subStream,
                         Status             status,
                         XrdSysMutexHelper &lock );
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      void MonitorDisconnection( Status status );
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      Status RequestClose( Message  *resp );
 
      typedef std::vector<SubStreamData*> SubStreamList;
 
      //------------------------------------------------------------------------
      // Data members
      //------------------------------------------------------------------------
      const URL                     *pUrl;
      uint16_t                       pStreamNum;
      std::string                    pStreamName;
      TransportHandler              *pTransport;
      Poller                        *pPoller;
      TaskManager                   *pTaskManager;
      JobManager                    *pJobManager;
      XrdSysRecMutex                 pMutex;
      InQueue                       *pIncomingQueue;
      AnyObject                     *pChannelData;
      uint32_t                       pLastStreamError;
      Status                         pLastFatalError;
      uint16_t                       pStreamErrorWindow;
      uint16_t                       pConnectionCount;
      uint16_t                       pConnectionRetry;
      time_t                         pConnectionInitTime;
      uint16_t                       pConnectionWindow;
      SubStreamList                  pSubStreams;
      std::vector<XrdNetAddr>        pAddresses;
      Utils::AddressType             pAddressType;
      ChannelHandlerList             pChannelEvHandlers;
      uint64_t                       pSessionId;
 
      //------------------------------------------------------------------------
      // Jobs
      //------------------------------------------------------------------------
      QueueIncMsgJob                *pQueueIncMsgJob;
 
      //------------------------------------------------------------------------
      // Monitoring info
      //------------------------------------------------------------------------
      timeval                        pConnectionStarted;
      timeval                        pConnectionDone;
      uint64_t                       pBytesSent;
      uint64_t                       pBytesReceived;
 
      //------------------------------------------------------------------------
      // Data stream on-connect handler
      //------------------------------------------------------------------------
      Job                           *pOnConnJob;
  };
}
 
#endif // __XRD_CL_STREAM_HH__