external/srs
1
0
Fork 0
mirror of https://github.com/ossrs/srs.git synced 2025-03-09 15:49:59 +00:00
Code Issues Releases Wiki Activity

Kernel: Extract SrsPps to kernel

Browse source
This commit is contained in:
winlin 2021-02-25 13:46:52 +08:00
parent df6e47ebfc
commit d8ba72fdc5
7 changed files with 236 additions and 161 deletions
Download patch file Download diff file Expand all files Collapse all files

96
trunk/src/protocol/srs_protocol_kbps.cpp
View file

@ -25,87 +25,6 @@
#include <srs_kernel_utility.hpp>
SrsRateSample::SrsRateSample()
{
total = time = -1;
rate = 0;
}
SrsRateSample::~SrsRateSample()
{
}
SrsRateSample* SrsRateSample::update(int64_t nn, srs_utime_t t, int k)
{
total = nn;
time = t;
rate = k;
return this;
}
void srs_pps_init(SrsRateSample& sample, int64_t nn, srs_utime_t now)
{
if (sample.time < 0 || nn < sample.total) {
sample.update(nn, now, 0);
}
}
void srs_pps_update(SrsRateSample& sample, int64_t nn, srs_utime_t now)
{
int pps = (int)((nn - sample.total) * 1000 / srsu2ms(now - sample.time));
if (pps == 0 && nn > sample.total) {
pps = 1; // For pps in (0, 1), we set to 1.
}
sample.update(nn, now, pps);
}
SrsPps::SrsPps(SrsWallClock* c)
{
clk_ = c;
sugar = 0;
}
SrsPps::~SrsPps()
{
}
void SrsPps::update()
{
update(sugar);
}
void SrsPps::update(int64_t nn)
{
srs_utime_t now = clk_->now();
srs_pps_init(sample_10s_, nn, now);
srs_pps_init(sample_30s_, nn, now);
srs_pps_init(sample_1m_, nn, now);
srs_pps_init(sample_5m_, nn, now);
srs_pps_init(sample_60m_, nn, now);
if (now - sample_10s_.time >= 10 * SRS_UTIME_SECONDS) {
srs_pps_update(sample_10s_, nn, now);
}
if (now - sample_30s_.time >= 30 * SRS_UTIME_SECONDS) {
srs_pps_update(sample_30s_, nn, now);
}
if (now - sample_1m_.time >= 60 * SRS_UTIME_SECONDS) {
srs_pps_update(sample_1m_, nn, now);
}
if (now - sample_5m_.time >= 300 * SRS_UTIME_SECONDS) {
srs_pps_update(sample_5m_, nn, now);
}
if (now - sample_60m_.time >= 3600 * SRS_UTIME_SECONDS) {
srs_pps_update(sample_60m_, nn, now);
}
}
int SrsPps::r10s()
{
return sample_10s_.rate;
}
SrsKbpsSlice::SrsKbpsSlice(SrsWallClock* c)
{
clk = c;
@ -166,19 +85,6 @@ ISrsKbpsDelta::~ISrsKbpsDelta()
{
}
SrsWallClock::SrsWallClock()
{
}
SrsWallClock::~SrsWallClock()
{
}
srs_utime_t SrsWallClock::now()
{
return srs_get_system_time();
}
SrsKbps::SrsKbps(SrsWallClock* c) : is(c), os(c)
{
clk = c;
@ -360,5 +266,3 @@ int SrsKbps::size_memory()
return sizeof(SrsKbps);
}
SrsWallClock* _srs_clock = new SrsWallClock();

63
trunk/src/protocol/srs_protocol_kbps.hpp
View file

@ -27,50 +27,7 @@
#include <srs_core.hpp>
#include <srs_protocol_io.hpp>
class SrsWallClock;
// A sample for rate-based stat, such as kbps or kps.
class SrsRateSample
{
public:
int64_t total;
srs_utime_t time;
// kbps or kps
int rate;
public:
SrsRateSample();
virtual ~SrsRateSample();
public:
virtual SrsRateSample* update(int64_t nn, srs_utime_t t, int k);
};
// A pps manager every some duration.
class SrsPps
{
private:
SrsWallClock* clk_;
private:
// samples
SrsRateSample sample_10s_;
SrsRateSample sample_30s_;
SrsRateSample sample_1m_;
SrsRateSample sample_5m_;
SrsRateSample sample_60m_;
public:
// Sugar for target to stat.
int64_t sugar;
public:
SrsPps(SrsWallClock* clk);
virtual ~SrsPps();
public:
// Update with the nn which is target.
void update();
// Update with the nn.
void update(int64_t nn);
// Get the 10s average stat.
int r10s();
};
#include <srs_kernel_kbps.hpp>
/**
* a slice of kbps statistic, for input or output.
@ -144,21 +101,6 @@ public:
virtual void remark(int64_t* in, int64_t* out) = 0;
};
/**
* A time source to provide wall clock.
*/
class SrsWallClock
{
public:
SrsWallClock();
virtual ~SrsWallClock();
public:
/**
* Current time in srs_utime_t.
*/
virtual srs_utime_t now();
};
/**
* to statistic the kbps of io.
* itself can be a statistic source, for example, used for SRS bytes stat.
@ -258,7 +200,4 @@ public:
virtual int size_memory();
};
// The global clock.
extern SrsWallClock* _srs_clock;
#endif