compression: Process zstd dispatcher

Signed-off-by: Hiroshi Hatake <[email protected]>

Author: cosmo0920

include/fluent-bit/flb_compression.h

@@ -26,6 +26,7 @@
 
 #define FLB_COMPRESSION_ALGORITHM_NONE                    0
 #define FLB_COMPRESSION_ALGORITHM_GZIP                    1
+#define FLB_COMPRESSION_ALGORITHM_ZSTD                    2
 
 #define FLB_DECOMPRESSOR_STATE_FAILED                    -1
 #define FLB_DECOMPRESSOR_STATE_EXPECTING_HEADER           0

src/flb_compression.c

@@ -21,6 +21,7 @@
 #include <fluent-bit/flb_mem.h>
 #include <fluent-bit/flb_log.h>
 #include <fluent-bit/flb_gzip.h>
+#include <fluent-bit/flb_zstd.h>
 #include <fluent-bit/flb_compression.h>
 
 static size_t flb_decompression_context_get_read_buffer_offset(
@@ -131,7 +132,12 @@ void flb_decompression_context_destroy(struct flb_decompression_context *context
         }
 
         if (context->inner_context != NULL) {
-            flb_gzip_decompression_context_destroy(context->inner_context);
+            if (context->algorithm == FLB_COMPRESSION_ALGORITHM_GZIP) {
+                flb_gzip_decompression_context_destroy(context->inner_context);
+            }
+            else if (context->algorithm == FLB_COMPRESSION_ALGORITHM_ZSTD) {
+                flb_zstd_decompression_context_destroy(context->inner_context);
+            }
 
             context->inner_context = NULL;
         }
@@ -178,6 +184,9 @@ struct flb_decompression_context *flb_decompression_context_create(int algorithm
     if (algorithm == FLB_COMPRESSION_ALGORITHM_GZIP) {
         context->inner_context = flb_gzip_decompression_context_create();
     }
+    else if (algorithm == FLB_COMPRESSION_ALGORITHM_ZSTD) {
+        context->inner_context = flb_zstd_decompression_context_create();
+    }
     else {
         flb_error("invalid compression algorithm : %d", algorithm);
 
@@ -199,7 +208,12 @@ struct flb_decompression_context *flb_decompression_context_create(int algorithm
     context->input_buffer_size = input_buffer_size;
     context->read_buffer = context->read_buffer;
     context->algorithm = algorithm;
-    context->state = FLB_DECOMPRESSOR_STATE_EXPECTING_HEADER;
+    if (algorithm == FLB_COMPRESSION_ALGORITHM_GZIP) {
+        context->state = FLB_DECOMPRESSOR_STATE_EXPECTING_HEADER;
+    }
+    else if (algorithm == FLB_COMPRESSION_ALGORITHM_ZSTD) {
+        context->state = FLB_DECOMPRESSOR_STATE_EXPECTING_BODY;
+    }
 
     return context;
 }
@@ -215,6 +229,12 @@ int flb_decompress(struct flb_decompression_context *context,
                                                   output_length);
 
         }
+        else if (context->algorithm == FLB_COMPRESSION_ALGORITHM_ZSTD) {
+            return flb_zstd_decompressor_dispatch(context,
+                                                  output_buffer,
+                                                  output_length);
+
+        }
     }
 
     return FLB_DECOMPRESSOR_FAILURE;