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

refine kafka

Browse source
This commit is contained in:
winlin 2016-12-08 11:44:49 +08:00
parent 5a76f15e4e
commit 9a3674c624
8 changed files with 101 additions and 53 deletions
Download patch file Download diff file Expand all files Collapse all files

37
trunk/src/app/srs_app_kafka.cpp
View file

@ -321,6 +321,35 @@ ISrsKafkaCluster::~ISrsKafkaCluster()
{
}
// @global kafka event producer, user must use srs_initialize_kafka to initialize it.
ISrsKafkaCluster* _srs_kafka = NULL;
int srs_initialize_kafka()
{
int ret = ERROR_SUCCESS;
SrsKafkaProducer* kafka = new SrsKafkaProducer();
_srs_kafka = kafka;
if ((ret = kafka->initialize()) != ERROR_SUCCESS) {
srs_error("initialize the kafka producer failed. ret=%d", ret);
return ret;
}
if ((ret = kafka->start()) != ERROR_SUCCESS) {
srs_error("start kafka failed. ret=%d", ret);
return ret;
}
return ret;
}
void srs_dispose_kafka()
{
SrsKafkaProducer* kafka = dynamic_cast<SrsKafkaProducer*>(_srs_kafka);
kafka->stop();
}
SrsKafkaProducer::SrsKafkaProducer()
{
metadata_ok = false;
@ -362,6 +391,10 @@ int SrsKafkaProducer::start()
{
int ret = ERROR_SUCCESS;
if (!enabled) {
return ret;
}
if ((ret = worker->start()) != ERROR_SUCCESS) {
srs_error("start kafka worker failed. ret=%d", ret);
return ret;
@ -378,6 +411,10 @@ int SrsKafkaProducer::start()
void SrsKafkaProducer::stop()
{
if (!enabled) {
return;
}
pthread->stop();
worker->stop();
}