fms-m/client_session.cpp at main · damirn/fms-m · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
#include "pch.h"
#include "client_session.h"
#include "app_host.h"
#include "rtmp_application.h"

namespace fms
{
	client_session::client_session(std::uint32_t id, app_host *app_mngr)
		: m_id(id)
		, m_app_manager(app_mngr)
		, m_time(std::chrono::system_clock::now())
		, m_bytes_read(0)
		, m_bytes_written(0)
		, m_messages_read(0)
		, m_messages_written(0)
	{}

	void client_session::close()
	{
		m_app_manager->delete_connection(m_id);
	}

	std::uint32_t client_session::get_timestamp()
	{
		std::chrono::system_clock::time_point const now(std::chrono::system_clock::now());
		std::chrono::system_clock::duration const delta = now - m_time;
		return static_cast<std::uint32_t>(std::chrono::duration_cast<std::chrono::milliseconds>(delta).count());
	}

	void client_session::handle_bytes_read(std::size_t bytes_transferred)
	{
		m_bytes_read += static_cast<std::uint32_t>(bytes_transferred);
		if (rtmp_application *const app = get_app())
			app->update_stats(true, true, static_cast<std::uint32_t>(bytes_transferred));
	}

	void client_session::handle_bytes_written(std::size_t bytes_written)
	{
		m_bytes_written += static_cast<std::uint32_t>(bytes_written);
		if (rtmp_application *const app = get_app())
			app->update_stats(false, true, static_cast<std::uint32_t>(bytes_written));
	}

	std::uint32_t client_session::reserve_stream_id()
	{
		std::uint32_t stream_id = 1;
		while (true)
		{
			if (!m_stream_ids.contains(stream_id))
			{
				m_stream_ids.insert(stream_id);
				break;
			}
			++stream_id;
		}
		return stream_id;
	}

	void client_session::unreserve_stream_id(std::uint32_t stream_id)
	{
		m_stream_ids.erase(stream_id);
	}
}