69 lines
2.8 KiB
C++
69 lines
2.8 KiB
C++
#pragma once
|
|
#include <string>
|
|
#include "../util/tinythread.h"
|
|
#include "../util/json.h"
|
|
#include "user.h"
|
|
|
|
namespace Buffer{
|
|
class Stream{
|
|
public:
|
|
/// Get a reference to this Stream object.
|
|
static Stream * get();
|
|
/// Get the current statistics in JSON format.
|
|
std::string getStats();
|
|
/// Get a new DTSC::Ring object for a user.
|
|
DTSC::Ring * getRing();
|
|
/// Drop a DTSC::Ring object.
|
|
void dropRing(DTSC::Ring * ring);
|
|
/// Get the (constant) header data of this stream.
|
|
std::string & getHeader();
|
|
/// Set the IP address to accept push data from.
|
|
void setWaitingIP(std::string ip);
|
|
/// Check if this is the IP address to accept push data from.
|
|
bool checkWaitingIP(std::string ip);
|
|
/// Sets the current socket for push data.
|
|
bool setInput(Socket::Connection S);
|
|
/// Gets the current socket for push data.
|
|
Socket::Connection & getIPInput();
|
|
/// Stores intermediate statistics.
|
|
void saveStats(std::string username, Stats & stats);
|
|
/// Stores final statistics.
|
|
void clearStats(std::string username, Stats & stats, std::string reason);
|
|
/// Cleans up broken connections
|
|
void cleanUsers();
|
|
/// Blocks until writing is safe.
|
|
void getWriteLock();
|
|
/// Drops a previously gotten write lock.
|
|
void dropWriteLock(bool newpackets_available);
|
|
/// Blocks until reading is safe.
|
|
void getReadLock();
|
|
/// Drops a previously gotten read lock.
|
|
void dropReadLock();
|
|
/// Retrieves a reference to the DTSC::Stream
|
|
DTSC::Stream * getStream();
|
|
/// Sets the buffer name.
|
|
void setName(std::string n);
|
|
/// Add a user to the userlist.
|
|
void addUser(user * new_user);
|
|
/// Blocks the thread until new data is available.
|
|
void waitForData();
|
|
/// Cleanup function
|
|
~Stream();
|
|
private:
|
|
volatile int readers;///< Current count of active readers;
|
|
volatile int writers;///< Current count of waiting/active writers.
|
|
tthread::mutex rw_mutex; ///< Mutex for read/write locking.
|
|
tthread::condition_variable rw_change; ///< Triggered when reader/writer count changes.
|
|
static Stream * ref;
|
|
Stream();
|
|
JSON::Value Storage; ///< Global storage of data.
|
|
DTSC::Stream * Strm;
|
|
std::string waiting_ip; ///< IP address for media push.
|
|
Socket::Connection ip_input; ///< Connection used for media push.
|
|
tthread::mutex stats_mutex; ///< Mutex for stats/users modifications.
|
|
std::vector<user*> users; ///< All connected users.
|
|
std::vector<user*>::iterator usersIt; ///< Iterator for all connected users.
|
|
std::string name; ///< Name for this buffer.
|
|
tthread::condition_variable moreData; ///< Triggered when more data becomes available.
|
|
};
|
|
};
|