connectionSet.h

/* Copyright (c) 2005-2008, Stefan Eilemann <[email protected]> 
   All rights reserved. */

#ifndef EQNET_CONNECTION_SET_H
#define EQNET_CONNECTION_SET_H

#include <eq/net/connectionListener.h> // base class
#include <eq/net/pipeConnection.h>     // RefPtr member

#include <eq/base/base.h>
#include <eq/base/buffer.h>
#include <eq/base/hash.h>
#include <eq/base/refPtr.h>

#ifndef WIN32
#  include <poll.h>
#endif

namespace eq
{
namespace net
{
    class Message;
    class Network;
    class PipeConnection;

    class EQ_EXPORT ConnectionSet : public ConnectionListener
    {
    public:
        enum Event
        {
            EVENT_NONE = 0,        
            EVENT_CONNECT,         
            EVENT_DISCONNECT,      
            EVENT_DATA,            
            EVENT_TIMEOUT,         
            EVENT_INTERRUPT,       
            EVENT_ERROR,           
            EVENT_SELECT_ERROR,    
            EVENT_INVALID_HANDLE,  
            EVENT_ALL
        };

        ConnectionSet();
        ~ConnectionSet();

        void addConnection( ConnectionPtr connection );
        bool removeConnection( ConnectionPtr connection );
        void clear();
        size_t size()  const { return _connections.size(); }
        bool   empty() const { return _connections.empty(); }

        const ConnectionVector& getConnections() const { return _connections; }

        Event select( const int timeout = -1 );

        void interrupt();

        int           getError()     { return _error; }
        ConnectionPtr getConnection(){ return _connection; }

    private:
        base::SpinLock _mutex;

        ConnectionVector _connections;

        // Note: std::vector had to much overhead here
#ifdef WIN32
        base::Buffer< HANDLE > _fdSet;
#else
        base::Buffer< pollfd > _fdSetCopy; // 'const' set
        base::Buffer< pollfd > _fdSet;     // copy of _fdSetCopy used to poll
#endif
        base::Buffer< Connection* > _fdSetConnections;

        enum SelfCommands
        {
            SELF_INTERRUPT = 42
        };

        base::RefPtr< PipeConnection > _selfConnection;

        // result values
        ConnectionPtr _connection;
        int           _error;

        bool _dirty;

        bool _getEvent( Event& event, Connection::ReadNotifier& fd );
        void _dirtyFDSet();
        bool _setupFDSet();
        bool _buildFDSet();
        virtual void notifyStateChanged( Connection* ) { _dirty = true; }

        Event _handleSelfCommand();
        Event _getSelectResult( const uint32_t index );
    };

    EQ_EXPORT std::ostream& operator << ( std::ostream& os, 
                                          const ConnectionSet* set );
    EQ_EXPORT std::ostream& operator << ( std::ostream& os, 
                                          const ConnectionSet::Event event );
}
}

#endif // EQNET_CONNECTION_SET_H