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

add function to convert kafka array to vector

Browse source
This commit is contained in:
winlin 2015-10-22 12:15:30 +08:00
parent 84b3981d47
commit 33a015329d
4 changed files with 60 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

37
trunk/src/protocol/srs_kafka_stack.hpp
View file

@ -666,7 +666,7 @@ public:
*/
class SrsKafkaTopicMetadataResponse : public SrsKafkaResponse
{
private:
public:
SrsKafkaArray<SrsKafkaBroker> brokers;
SrsKafkaArray<SrsKafkaTopicMetadata> metadatas;
public:
@ -741,6 +741,36 @@ public:
* @param pmsg output the received message. user must free it.
*/
virtual int recv_message(SrsKafkaResponse** pmsg);
public:
/**
* expect specified message.
*/
template<typename T>
int expect_message(T** pmsg)
{
int ret = ERROR_SUCCESS;
while (true) {
SrsKafkaResponse* res = NULL;
if ((ret = recv_message(&res)) != ERROR_SUCCESS) {
srs_error("recv response failed. ret=%d", ret);
return ret;
}
// drop not matched.
T* msg = dynamic_cast<T*>(res);
if (!msg) {
srs_info("kafka drop response.");
srs_freep(res);
continue;
}
*pmsg = msg;
break;
}
return ret;
}
};
/**
@ -757,9 +787,12 @@ public:
/**
* fetch the metadata from broker for topic.
*/
virtual int fetch_metadata(std::string topic);
virtual int fetch_metadata(std::string topic, SrsKafkaTopicMetadataResponse** pmsg);
};
// convert kafka array[string] to vector[string]
extern std::vector<std::string> srs_kafka_array2vector(SrsKafkaArray<SrsKafkaString>* arr);
#endif
#endif