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

kafka refine code

Browse source
This commit is contained in:
winlin 2015-10-22 17:43:15 +08:00
parent f9f5b56951
commit 7a0aaf5900
1 changed files with 10 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -281,28 +281,32 @@ int SrsKafkaCache::flush(SrsKafkaPartition* partition, int key, SrsKafkaPartitio
// ensure the key exists. // ensure the key exists.
srs_assert (cache.find(key) != cache.end()); srs_assert (cache.find(key) != cache.end());
// the cache is vector, which is continous store.
// we remember the messages we have written and clear it when completed.
int nb_msgs = (int)pc->size();
if (pc->empty()) {
return ret;
}
// connect transport. // connect transport.
if ((ret = partition->connect()) != ERROR_SUCCESS) { if ((ret = partition->connect()) != ERROR_SUCCESS) {
srs_error("connect to partition failed. ret=%d", ret); srs_error("connect to partition failed. ret=%d", ret);
return ret; return ret;
} }
// copy the messages to a temp cache.
SrsKafkaPartitionCache tpc(*pc);
// TODO: FIXME: implements it. // TODO: FIXME: implements it.
// free all wrote messages. // free all wrote messages.
for (vector<SrsJsonObject*>::iterator it = tpc.begin(); it != tpc.end(); ++it) { for (vector<SrsJsonObject*>::iterator it = pc->begin(); it != pc->end(); ++it) {
SrsJsonObject* obj = *it; SrsJsonObject* obj = *it;
srs_freep(obj); srs_freep(obj);
} }
// remove the messages from cache. // remove the messages from cache.
if (pc->size() == tpc.size()) { if (pc->size() == nb_msgs) {
pc->clear(); pc->clear();
} else { } else {
pc->erase(pc->begin(), pc->begin() + tpc.size()); pc->erase(pc->begin(), pc->begin() + nb_msgs);
} }
return ret; return ret;