TSG-17862 Proxy支持Internal IP address和Exteral IP Address的扫描, 支持以Topic方式上传HTTP请求体/应答体v4.8.51-20231215

1 files changed, 20 insertions, 12 deletions

diff --git a/common/include/tfe_kafka_logger.h b/common/include/tfe_kafka_logger.h
index de182ba..2bed406 100644
--- a/common/include/tfe_kafka_logger.h
+++ b/common/include/tfe_kafka_logger.h
@@ -9,26 +9,34 @@ extern "C"
 #include <tfe_utils.h>
 #include <librdkafka/rdkafka.h>
 
-    typedef struct tfe_kafka_logger_s
-    {
-        int enable;
-        int t_vsys_id;
+enum kafka_topic_type
+{   
+    TOPIC_LOGGER,
+    TOPIC_BUCKET,
+    TOPIC_MAX
+};
+
+typedef struct tfe_kafka_logger_s
+{
+    int enable;
+    int t_vsys_id;
 
-        unsigned int local_ip_num;
-        char local_ip_str[TFE_SYMBOL_MAX];
+    unsigned int local_ip_num;
+    char local_ip_str[TFE_SYMBOL_MAX];
 
-        char topic_name[TFE_STRING_MAX];
-        char broker_list[TFE_STRING_MAX];
+    char topic_name[TOPIC_MAX][TFE_STRING_MAX];
+    char broker_list[TFE_STRING_MAX];
 
-        rd_kafka_t *kafka_handle;
-        rd_kafka_topic_t *kafka_topic;
-    } tfe_kafka_logger_t;
+    rd_kafka_t *kafka_handle;
+    rd_kafka_topic_t *kafka_topic[TOPIC_MAX];
+} tfe_kafka_logger_t;
 
 tfe_kafka_logger_t *tfe_kafka_logger_create(int enable, const char *nic_name, const char *brokerlist, const char *topic_name,
                                        const char *sasl_username, const char *sasl_passwd, void *local_logger);
 
+int tfe_kafka_logger_topic_new(tfe_kafka_logger_t *logger, const char *topic_name, void *local_logger);
 void tfe_kafka_logger_destroy(tfe_kafka_logger_t *logger);
-int tfe_kafka_logger_send(tfe_kafka_logger_t *logger, const char *data, int len);
+int tfe_kafka_logger_send(tfe_kafka_logger_t *logger, int topic_id, const char *data, int len);
 
 #ifdef __cpluscplus
 }