convert metadata to partitions

trunk/src/app/srs_app_kafka.cpp

@@ -41,6 +41,83 @@ using namespace std;
 
 #define SRS_KAKFA_CYCLE_INTERVAL_MS 3000
 
+std::string srs_kafka_metadata_summary(SrsKafkaTopicMetadataResponse* metadata)
+{
+    vector<string> bs;
+    for (int i = 0; i < metadata->brokers.size(); i++) {
+        SrsKafkaBroker* broker = metadata->brokers.at(i);
+        
+        string hostport = srs_int2str(broker->node_id) + "/" + broker->host.to_str();
+        if (broker->port > 0) {
+            hostport += ":" + srs_int2str(broker->port);
+        }
+        
+        bs.push_back(hostport);
+    }
+    
+    vector<string> ps;
+    for (int i = 0; i < metadata->metadatas.size(); i++) {
+        SrsKafkaTopicMetadata* topic = metadata->metadatas.at(i);
+        
+        for (int j = 0; j < topic->metadatas.size(); j++) {
+            string desc = "topic=" + topic->name.to_str();
+            
+            SrsKafkaPartitionMetadata* partition = topic->metadatas.at(j);
+            
+            desc += "?partition=" + srs_int2str(partition->partition_id);
+            desc += "&leader=" + srs_int2str(partition->leader);
+            
+            vector<string> replicas = srs_kafka_array2vector(&partition->replicas);
+            desc += "&replicas=" + srs_join_vector_string(replicas, ",");
+            
+            ps.push_back(desc);
+        }
+    }
+    
+    std::stringstream ss;
+    ss << "brokers=" << srs_join_vector_string(bs, ",");
+    ss << ", " << srs_join_vector_string(ps, ", ");
+    
+    return ss.str();
+}
+
+std::string srs_kafka_summary_partitions(const vector<SrsKafkaPartition*>& partitions)
+{
+    vector<string> ret;
+    
+    vector<SrsKafkaPartition*>::const_iterator it;
+    for (it = partitions.begin(); it != partitions.end(); ++it) {
+        SrsKafkaPartition* partition = *it;
+        
+        string desc = "tcp://";
+        desc += partition->host + ":" + srs_int2str(partition->port);
+        desc += "?broker=" + srs_int2str(partition->broker);
+        desc += "&partition=" + srs_int2str(partition->id);
+        ret.push_back(desc);
+    }
+    
+    return srs_join_vector_string(ret, ", ");
+}
+
+SrsKafkaPartition::SrsKafkaPartition()
+{
+    id = broker = 0;
+    port = SRS_CONSTS_KAFKA_DEFAULT_PORT;
+}
+
+SrsKafkaPartition::~SrsKafkaPartition()
+{
+}
+
+string SrsKafkaPartition::hostport()
+{
+    if (ep.empty()) {
+        ep = host + ":" + srs_int2str(port);
+    }
+    
+    return ep;
+}
+
 SrsKafkaProducer::SrsKafkaProducer()
 {
     meatadata_ok = false;
@@ -57,6 +134,13 @@ SrsKafkaProducer::SrsKafkaProducer()
 
 SrsKafkaProducer::~SrsKafkaProducer()
 {
+    vector<SrsKafkaPartition*>::iterator it;
+    for (it = partitions.begin(); it != partitions.end(); ++it) {
+        SrsKafkaPartition* partition = *it;
+        srs_freep(partition);
+    }
+    partitions.clear();
+    
     srs_freep(lb);
     srs_freep(kafka);
     srs_freep(transport);
@@ -203,46 +287,33 @@ int SrsKafkaProducer::request_metadata()
     SrsAutoFree(SrsKafkaTopicMetadataResponse, metadata);
     
     // show kafka metadata.
-    string summary;
+    string summary = srs_kafka_metadata_summary(metadata);
-    if (true) {
+    srs_trace("kafka metadata: %s", summary.c_str());
-        vector<string> bs;
+    
-        for (int i = 0; i < metadata->brokers.size(); i++) {
+    // generate the partition info.
-            SrsKafkaBroker* broker = metadata->brokers.at(i);
+    for (int i = 0; i < metadata->metadatas.size(); i++) {
-            
+        SrsKafkaTopicMetadata* topic = metadata->metadatas.at(i);
-            string hostport = srs_int2str(broker->node_id) + "/" + broker->host.to_str();
-            if (broker->port > 0) {
-                hostport += ":" + srs_int2str(broker->port);
-            }
-            
-            bs.push_back(hostport);
-        }
         
-        vector<string> ps;
+        for (int j = 0; j < topic->metadatas.size(); j++) {
-        for (int i = 0; i < metadata->metadatas.size(); i++) {
+            SrsKafkaPartitionMetadata* partition = topic->metadatas.at(j);
-            SrsKafkaTopicMetadata* topic = metadata->metadatas.at(i);
             
-            string desc = "topic=" + topic->name.to_str();
+            SrsKafkaPartition* p = new SrsKafkaPartition();
+            p->id = partition->partition_id;
+            p->broker = partition->leader;
             
-            for (int j = 0; j < topic->metadatas.size(); j++) {
+            for (int i = 0; i < metadata->brokers.size(); i++) {
-                SrsKafkaPartitionMetadata* partition = topic->metadatas.at(j);
+                SrsKafkaBroker* broker = metadata->brokers.at(i);
-                
+                if (broker->node_id == p->broker) {
-                desc += ", partition" + srs_int2str(partition->partition_id) +"=";
+                    p->host = broker->host.to_str();
-                desc += srs_int2str(partition->leader) + "/";
+                    p->port = broker->port;
-                
+                    break;
-                vector<string> replicas = srs_kafka_array2vector(&partition->replicas);
+                }
-                desc += srs_join_vector_string(replicas, ",");
             }
             
-            ps.push_back(desc);
+            partitions.push_back(p);
         }
-        
-        std::stringstream ss;
-        ss << "brokers=" << srs_join_vector_string(bs, ",");
-        ss << ", " << srs_join_vector_string(ps, ",");
-    
-        summary = ss.str();
     }
-    srs_trace("kafka metadata: %s", summary.c_str());
+    srs_trace("kafka connector: %s", srs_kafka_summary_partitions(partitions).c_str());
     
     meatadata_ok = true;
     

trunk/src/app/srs_app_kafka.hpp

@@ -29,6 +29,8 @@
 */
 #include <srs_core.hpp>
 
+#include <vector>
+
 class SrsLbRoundRobin;
 class SrsAsyncCallWorker;
 class SrsTcpClient;
@@ -38,6 +40,26 @@ class SrsKafkaClient;
 
 #ifdef SRS_AUTO_KAFKA
 
+/**
+ * the kafka partition info.
+ */
+struct SrsKafkaPartition
+{
+private:
+    std::string ep;
+public:
+    int id;
+    // leader.
+    int broker;
+    std::string host;
+    int port;
+public:
+    SrsKafkaPartition();
+    virtual ~SrsKafkaPartition();
+public:
+    virtual std::string hostport();
+};
+
 /**
  * the kafka producer used to save log to kafka cluster.
  */
@@ -49,6 +71,8 @@ private:
 private:
     bool meatadata_ok;
     st_cond_t metadata_expired;
+public:
+    std::vector<SrsKafkaPartition*> partitions;
 private:
     SrsLbRoundRobin* lb;
     SrsAsyncCallWorker* worker;