external/srs
1
0
Fork 0
mirror of https://github.com/ossrs/srs.git synced 2025-02-13 03:41:55 +00:00
Code Issues Releases Wiki Activity

refine code, implements the stat.:

Browse source
This commit is contained in:
winlin 2015-01-05 12:57:48 +08:00
parent 97bea51499
commit ae63af6a4a
2 changed files with 51 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

53
trunk/src/app/srs_app_statistic.cpp
View file

@ -23,7 +23,11 @@ CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
#include <srs_app_statistic.hpp>
#include <sstream>
using namespace std;
#include <srs_protocol_rtmp.hpp>
#include <srs_app_json.hpp>
SrsStatistic* SrsStatistic::_instance = new SrsStatistic();
@ -61,20 +65,63 @@ SrsStatistic* SrsStatistic::instance()
return _instance;
}
int SrsStatistic::on_client(int id, SrsRequest *req)
int SrsStatistic::on_client(int id, SrsRequest* req)
{
int ret = ERROR_SUCCESS;
// create vhost if not exists.
SrsStatisticVhost* vhost = NULL;
if (vhosts.find(req->vhost) == vhosts.end()) {
vhost = new SrsStatisticVhost();
vhost->vhost = req->vhost;
vhosts[req->vhost] = vhost;
} else {
vhost = vhosts[req->vhost];
}
// the url to identify the stream.
std::string url = req->get_stream_url();
// create stream if not exists.
SrsStatisticStream* stream = NULL;
if (streams.find(url) == streams.end()) {
stream = new SrsStatisticStream();
stream->vhost = vhost;
stream->stream = url;
streams[url] = stream;
} else {
stream = streams[url];
}
return ret;
}
int SrsStatistic::dumps_vhosts(std::stringstream& ss)
int SrsStatistic::dumps_vhosts(stringstream& ss)
{
int ret = ERROR_SUCCESS;
std::map<std::string, SrsStatisticVhost*>::iterator it;
for (it = vhosts.begin(); it != vhosts.end(); it++) {
SrsStatisticVhost* vhost = it->second;
ss << __SRS_JOBJECT_START
<< __SRS_JFIELD_STR("name", vhost->vhost)
<< __SRS_JOBJECT_END;
}
return ret;
}
int SrsStatistic::dumps_streams(std::stringstream& ss)
int SrsStatistic::dumps_streams(stringstream& ss)
{
int ret = ERROR_SUCCESS;
std::map<std::string, SrsStatisticStream*>::iterator it;
for (it = streams.begin(); it != streams.end(); it++) {
SrsStatisticStream* stream = it->second;
ss << __SRS_JOBJECT_START
<< __SRS_JFIELD_STR("url", stream->stream)
<< __SRS_JOBJECT_END;
}
return ret;
}

2
trunk/src/app/srs_app_statistic.hpp
View file

@ -77,7 +77,7 @@ public:
* @param id, the client srs id.
* @param req, the client request object.
*/
virtual int on_client(int id, SrsRequest *req);
virtual int on_client(int id, SrsRequest* req);
public:
/**
* dumps the vhosts to sstream in json.