1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89
|
/*
* Copyright (c) 2014 Pierre-Yves Ritschard <pyr@spootnik.org>
* Copyright (c) 2013-2019 Balabit
* Copyright (c) 2019 Balazs Scheidler
*
* This program is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License version 2 as published
* by the Free Software Foundation, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
*
* As an additional exemption you are allowed to compile & link against the
* OpenSSL libraries as published by the OpenSSL project. See the file
* COPYING for details.
*
*/
#ifndef KAFKA_H_INCLUDED
#define KAFKA_H_INCLUDED
#include "logthrdest/logthrdestdrv.h"
#pragma GCC diagnostic push "-Wignored-qualifiers"
#pragma GCC diagnostic ignored "-Wignored-qualifiers"
#include <librdkafka/rdkafka.h>
#pragma GCC diagnostic pop "-Wignored-qualifiers"
typedef struct
{
LogThreadedDestDriver super;
LogTemplateOptions template_options;
LogTemplate *key;
LogTemplate *message;
LogTemplate *topic_name;
GHashTable *topics;
GMutex topics_lock;
gboolean transaction_commit;
GList *config;
gchar *bootstrap_servers;
gchar *fallback_topic_name;
rd_kafka_topic_t *topic;
rd_kafka_t *kafka;
gint flush_timeout_on_shutdown;
gint flush_timeout_on_reload;
gint poll_timeout;
gboolean transaction_inited;
} KafkaDestDriver;
#define TOPIC_NAME_ERROR topic_name_error_quark()
GQuark topic_name_error_quark(void);
enum KafkaTopicError
{
TOPIC_LENGTH_ZERO,
TOPIC_DOT_TWO_DOTS,
TOPIC_EXCEEDS_MAX_LENGTH,
TOPIC_INVALID_PATTERN,
};
void kafka_dd_set_topic(LogDriver *d, LogTemplate *topic);
gboolean kafka_dd_reopen(LogDriver *d);
void kafka_dd_set_fallback_topic(LogDriver *d, const gchar *fallback_topic);
void kafka_dd_merge_config(LogDriver *d, GList *props);
void kafka_dd_set_bootstrap_servers(LogDriver *d, const gchar *bootstrap_servers);
void kafka_dd_set_key_ref(LogDriver *d, LogTemplate *key);
void kafka_dd_set_message_ref(LogDriver *d, LogTemplate *message);
void kafka_dd_shutdown(LogThreadedDestDriver *s);
void kafka_dd_set_flush_timeout_on_shutdown(LogDriver *d, gint shutdown_timeout);
void kafka_dd_set_flush_timeout_on_reload(LogDriver *d, gint reload_timeout);
void kafka_dd_set_poll_timeout(LogDriver *d, gint poll_timeout);
void kafka_dd_set_transaction_commit(LogDriver *d, gboolean transaction_commit);
gboolean kafka_dd_validate_topic_name(const gchar *name, GError **error);
gboolean kafka_dd_is_topic_name_a_template(KafkaDestDriver *self);
rd_kafka_topic_t *kafka_dd_query_insert_topic(KafkaDestDriver *self, const gchar *name);
LogTemplateOptions *kafka_dd_get_template_options(LogDriver *d);
LogDriver *kafka_dd_new(GlobalConfig *cfg);
#endif
|