slist: on split_tokens add suppor for 'max_split' option

Signed-off-by: Eduardo Silva <[email protected]>

Author: edsiper

include/fluent-bit/flb_slist.h

@@ -38,7 +38,7 @@ int flb_slist_add_sds(struct mk_list *head, flb_sds_t str);
 void flb_slist_destroy(struct mk_list *list);
 int flb_slist_split_string(struct mk_list *list, const char *str,
                            int separator, int max_split);
-int flb_slist_split_tokens(struct mk_list *list, const char *str);
+int flb_slist_split_tokens(struct mk_list *list, const char *str, int max_split);
 
 void flb_slist_dump(struct mk_list *list);
 struct flb_slist_entry *flb_slist_entry_get(struct mk_list *list, int n);

src/flb_slist.c

@@ -176,8 +176,10 @@ static flb_sds_t token_retrieve(char **str)
     return out;
 }
 
-int flb_slist_split_tokens(struct mk_list *list, const char *str)
+int flb_slist_split_tokens(struct mk_list *list, const char *str, int max_split)
 {
+    int count = 0;
+    char *p;
     char *buf;
     flb_sds_t tmp = NULL;
 
@@ -187,6 +189,20 @@ int flb_slist_split_tokens(struct mk_list *list, const char *str)
         if (!buf) {
             break;
         }
+        count++;
+
+        /* Append remaining string if we use a maximum number of tokens */
+        if (count >= max_split && max_split > 0) {
+            p = buf;
+            while (*p == ' ') {
+                p++;
+            }
+
+            if (*p) {
+                flb_slist_add(list, p);
+            }
+            break;
+        }
     }
 
     return 0;