From b5ccc35340785f02e441d20a93d0dd91c70e365b Mon Sep 17 00:00:00 2001
From: winlin <winlin@vip.126.com>
Date: Tue, 22 Sep 2015 15:10:43 +0800
Subject: [PATCH] add kafka config

---
 trunk/conf/full.conf             | 12 ++++++++++++
 trunk/src/app/srs_app_config.cpp | 30 +++++++++++++++++++++++++++++-
 trunk/src/app/srs_app_config.hpp |  6 ++++++
 3 files changed, 47 insertions(+), 1 deletion(-)

diff --git a/trunk/conf/full.conf b/trunk/conf/full.conf
index b6a0d4b8c..0ffd986c0 100644
--- a/trunk/conf/full.conf
+++ b/trunk/conf/full.conf
@@ -230,6 +230,18 @@ stream_caster {
     listen          8936;
 }
 
+#############################################################################################
+# Kafka sections
+#############################################################################################
+# Apache Kafka is a high-throughput distributed messaging system.
+# SRS is a Kafka producer to send message to kafka.
+# @see https://kafka.apache.org/documentation.html#introduction
+kafka {
+    # whether enabled kafka.
+    # default: off
+    enabled         off;
+}
+
 #############################################################################################
 # RTMP/HTTP VHOST sections
 #############################################################################################
diff --git a/trunk/src/app/srs_app_config.cpp b/trunk/src/app/srs_app_config.cpp
index 2d70bce0b..cf4071fdf 100644
--- a/trunk/src/app/srs_app_config.cpp
+++ b/trunk/src/app/srs_app_config.cpp
@@ -3488,7 +3488,7 @@ int SrsConfig::check_config()
             && n != "srs_log_tank" && n != "srs_log_level" && n != "srs_log_file"
             && n != "max_connections" && n != "daemon" && n != "heartbeat"
             && n != "http_api" && n != "stats" && n != "vhost" && n != "pithy_print_ms"
-            && n != "http_server" && n != "stream_caster"
+            && n != "http_server" && n != "stream_caster" && n != "kafka"
             && n != "utc_time"
         ) {
             ret = ERROR_SYSTEM_CONFIG_INVALID;
@@ -3530,6 +3530,17 @@ int SrsConfig::check_config()
             }
         }
     }
+    if (true) {
+        SrsConfDirective* conf = root->get("kafka");
+        for (int i = 0; conf && i < (int)conf->directives.size(); i++) {
+            string n = conf->at(i)->name;
+            if (n != "enabled") {
+                ret = ERROR_SYSTEM_CONFIG_INVALID;
+                srs_error("unsupported kafka directive %s, ret=%d", n.c_str(), ret);
+                return ret;
+            }
+        }
+    }
     if (true) {
         SrsConfDirective* conf = get_heartbeart();
         for (int i = 0; conf && i < (int)conf->directives.size(); i++) {
@@ -4243,6 +4254,23 @@ int SrsConfig::get_stream_caster_rtp_port_max(SrsConfDirective* conf)
     return ::atoi(conf->arg0().c_str());
 }
 
+bool SrsConfig::get_kafka_enabled()
+{
+    static bool DEFAULT = false;
+    
+    SrsConfDirective* conf = root->get("kafka");
+    if (!conf) {
+        return DEFAULT;
+    }
+    
+    conf = conf->get("enabled");
+    if (!conf || conf->arg0().empty()) {
+        return DEFAULT;
+    }
+    
+    return SRS_CONF_PERFER_FALSE(conf->arg0());
+}
+
 SrsConfDirective* SrsConfig::get_vhost(string vhost, bool try_default_vhost)
 {
     srs_assert(root);
diff --git a/trunk/src/app/srs_app_config.hpp b/trunk/src/app/srs_app_config.hpp
index e99a9ad68..20dcb8749 100644
--- a/trunk/src/app/srs_app_config.hpp
+++ b/trunk/src/app/srs_app_config.hpp
@@ -628,6 +628,12 @@ public:
     * get the max udp port for rtp of stream caster rtsp.
     */
     virtual int                 get_stream_caster_rtp_port_max(SrsConfDirective* conf);
+// kafka section.
+public:
+    /**
+     * whether the kafka enabled.
+     */
+    virtual bool                get_kafka_enabled();
 // vhost specified section
 public:
     /**