YiSignalHandler.h

Go to the documentation of this file.

// © You i Labs Inc. 2000-2017. All rights reserved.
#ifndef _YI_SIGNAL_HANDLER_H_
#define _YI_SIGNAL_HANDLER_H_

#include "thread/YiThread.h"
#include "thread/YiRecursiveMutex.h"
#include "utility/YiLazy.h"

#define YI_RANDOM_SLEEP(X) YI_SLEEP(((rand() % 5) + 1) * (X))

class CYISignalBase;

class CYISignalHandler : public CYIThread::Listener
{
    friend class CYISignalBase;

public:
    CYISignalHandler();
    CYISignalHandler(const CYISignalHandler &rSignalHandler);
    virtual ~CYISignalHandler();
    CYISignalHandler &operator=(const CYISignalHandler &rSignalHandler);

    void MoveToThread(CYIThread *pThread);

    CYIThreadHandle GetThreadAffinity() const;

    void SetThreadAffinity(const CYIThreadHandle &rThreadAffinity);

    virtual bool IsConnected() const;

    virtual bool IsConnected(const CYISignalBase &rSignal) const;

    void Disconnect(CYISignalBase &rSignal);

    void DisconnectFromAllSignals();

private:
    enum YI_NOTIFY_FLAG
    {
        YI_DO_NOT_NOTIFY = 0x0,
        YI_NOTIFY        = 0x1,
    };

    struct HandlerObjects
    {
        HandlerObjects();

        mutable CYIRecursiveMutex m_handlerMutex;
        std::vector<CYISignalBase *> m_signalList;
        CYIThread *m_pThreadOverride;
    };

    void AddSignal(CYISignalBase &rSignal) const;
    void RemoveSignal(CYISignalBase &rSignal, YI_NOTIFY_FLAG notifySignal);
    void RemoveAllSignals(YI_NOTIFY_FLAG notifySignal);
    bool HasSignal(const CYISignalBase &rSignal) const;
    void CopySignalHandler(const CYISignalHandler &rSignalHandler);

    void ExclusiveLock(CYIRecursiveMutex &rHandlerMutex) const;
    void ExclusiveLock(const CYISignalBase &rSignal, CYIRecursiveMutex &rHandlerMutex) const;
    void ExclusiveUnlock(CYIRecursiveMutex &rHandlerMutex) const;
    void ExclusiveUnlock(const CYISignalBase &rSignal, CYIRecursiveMutex &rHandlerMutex) const;

    virtual void OnThreadDestroyed(CYIThread *pThread) override;

    CYILazy<HandlerObjects> m_handlerObjects;
    CYIThreadHandle m_threadAffinity;
};

#endif /* _YI_SIGNAL_HANDLER_H_ */