streamstats.h

Go to the documentation of this file.

/********************************************************************************************
* Twitch Broadcasting SDK
*
* This software is supplied under the terms of a license agreement with Twitch Interactive, Inc. and
* may not be copied or used except in accordance with the terms of that agreement
* Copyright (c) 2012-2016 Twitch Interactive, Inc.
*********************************************************************************************/

#pragma once

#include "twitchsdk/core/concurrentqueue.h"
#include "twitchsdk/core/eventsource.h"
#include "twitchsdk/broadcast/broadcasttypes.h"


namespace ttv
{
    namespace broadcast
    {
        class StreamStats;
        class StreamStatsListenerProxy;
    }
}


class ttv::broadcast::StreamStats
{
public:
    enum class StatType
    {
        RtmpState, 
        RtmpTotalBytesSent, 
        TotalVideoPacketsSent 
    };

    class IListener
    {
    public:
        virtual void OnStatReceived(StreamStats* source, StatType type, uint64_t data) = 0;
    };

public:
    void AddListener(const std::shared_ptr<IListener>& listener);
    void RemoveListener(const std::shared_ptr<IListener>& listener);
    void Flush();
    void Add(StatType type, uint64_t data);

private:
    typedef std::pair<StatType, uint64_t> value_t;
    ConcurrentQueue< value_t > mQueue;
    EventSource<IListener> mListeners;
};


class ttv::broadcast::StreamStatsListenerProxy : public StreamStats::IListener
{
public:
    typedef std::function<void(StreamStats* source, StreamStats::StatType type, uint64_t data)> OnStatReceivedFunc;

public:
    // StreamStats::Listener implementation
    virtual void OnStatReceived(StreamStats* source, StreamStats::StatType type, uint64_t data) override
    {
        if (mOnStatReceivedFunc != nullptr)
        {
            mOnStatReceivedFunc(source, type, data);
        }
    }

public:
    OnStatReceivedFunc mOnStatReceivedFunc;
};