Skip to content

Commit 61e5afd

Browse files
pwhelanedsiper
authored andcommitted
input: add support for upstream configuration for input client plugins.
Signed-off-by: Phillip Whelan <[email protected]>
1 parent bb4611b commit 61e5afd

File tree

1 file changed

+15
-5
lines changed

1 file changed

+15
-5
lines changed

src/flb_input.c

Lines changed: 15 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,7 @@
3535
#include <fluent-bit/flb_metrics.h>
3636
#include <fluent-bit/flb_storage.h>
3737
#include <fluent-bit/flb_downstream.h>
38+
#include <fluent-bit/flb_upstream.h>
3839
#include <fluent-bit/flb_plugin.h>
3940
#include <fluent-bit/flb_kv.h>
4041
#include <fluent-bit/flb_hash_table.h>
@@ -888,11 +889,20 @@ int flb_input_net_property_check(struct flb_input_instance *ins,
888889
{
889890
int ret = 0;
890891

891-
/* Get Downstream net_setup configmap */
892-
ins->net_config_map = flb_downstream_get_config_map(config);
893-
if (!ins->net_config_map) {
894-
flb_input_instance_destroy(ins);
895-
return -1;
892+
/* Get Downstream or Upstream net_setup configmap */
893+
if (ins->p->flags & FLB_INPUT_NET_SERVER) {
894+
ins->net_config_map = flb_downstream_get_config_map(config);
895+
if (!ins->net_config_map) {
896+
flb_input_instance_destroy(ins);
897+
return -1;
898+
}
899+
}
900+
else if (ins->p->flags & FLB_INPUT_NET) {
901+
ins->net_config_map = flb_upstream_get_config_map(config);
902+
if (!ins->net_config_map) {
903+
flb_input_instance_destroy(ins);
904+
return -1;
905+
}
896906
}
897907

898908
/*

0 commit comments

Comments
 (0)