channelSearchManager.h

Go to the documentation of this file.

#ifndef CHANNELSEARCHMANAGER_H
#define CHANNELSEARCHMANAGER_H

#ifdef epicsExportSharedSymbols
#   define channelSearchManagerEpicsExportSharedSymbols
#   undef epicsExportSharedSymbols
#endif

#include <osiSock.h>

#ifdef channelSearchManagerEpicsExportSharedSymbols
#   define epicsExportSharedSymbols
#       undef channelSearchManagerEpicsExportSharedSymbols
#endif

#include <pv/pvaDefs.h>
#include <pv/remote.h>

namespace epics {
namespace pvAccess {

class SearchInstance {
public:
    POINTER_DEFINITIONS(SearchInstance);

    virtual ~SearchInstance() {}

    virtual pvAccessID getSearchInstanceID() = 0;

    virtual const std::string& getSearchInstanceName() = 0;

    virtual int32_t& getUserValue() = 0;

    // TODO make serverAddress an URI or similar
    virtual void searchResponse(const ServerGUID & guid, int8_t minorRevision, osiSockAddr* serverAddress) = 0;
};


class ChannelSearchManager :
        public epics::pvData::TimerCallback,
        public std::tr1::enable_shared_from_this<ChannelSearchManager>
{
public:
    POINTER_DEFINITIONS(ChannelSearchManager);

    virtual ~ChannelSearchManager();
    void cancel();
    int32_t registeredCount();
    void registerSearchInstance(SearchInstance::shared_pointer const & channel, bool penalize = false);
    void unregisterSearchInstance(SearchInstance::shared_pointer const & channel);
    void searchResponse(const ServerGUID & guid, pvAccessID cid, int32_t seqNo, int8_t minorRevision, osiSockAddr* serverAddress);
    void newServerDetected();

    virtual void callback() OVERRIDE FINAL;

    virtual void timerStopped() OVERRIDE FINAL;

    ChannelSearchManager(Context::shared_pointer const & context);
    void activate();

private:

    bool generateSearchRequestMessage(SearchInstance::shared_pointer const & channel, bool allowNewFrame, bool flush);

    static bool generateSearchRequestMessage(SearchInstance::shared_pointer const & channel,
            epics::pvData::ByteBuffer* byteBuffer, TransportSendControl* control);

    void boost();

    void initializeSendBuffer();
    void flushSendBuffer();

    static bool isPowerOfTwo(int32_t x);

    Context::weak_pointer m_context;

    osiSockAddr m_responseAddress;

    AtomicBoolean m_canceled;

    int32_t m_sequenceNumber;

    epics::pvData::ByteBuffer m_sendBuffer;

    typedef std::map<pvAccessID,SearchInstance::weak_pointer> m_channels_t;
    m_channels_t m_channels;

    int64_t m_lastTimeSent;

    epics::pvData::Mutex m_channelMutex;

    epics::pvData::Mutex m_userValueMutex;

    epics::pvData::Mutex m_mutex;
};

}
}

#endif