Wrap with pipelines should return pipe type

redpanda-data · Jan 18, 2018 · 7edbd9e · 7edbd9e
1 parent fe7ab2f
commit 7edbd9e
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/lib/input/wrap_with_pipeline.go b/lib/input/wrap_with_pipeline.go
@@ -55,14 +55,19 @@ func WrapWithPipeline(in Type, pipeConstructor pipeline.ConstructorFunc) (*WithP
 }
 
 // WrapWithPipelines wraps an input with a variadic number of pipelines.
-func WrapWithPipelines(in Type, pipeConstructors ...pipeline.ConstructorFunc) (Type, error) {
+func WrapWithPipelines(in Type, pipeConstructors ...pipeline.ConstructorFunc) (*WithPipeline, error) {
 	var err error
-	for _, ctor := range pipeConstructors {
-		if in, err = WrapWithPipeline(in, ctor); err != nil {
+	var pipe *WithPipeline
+	for i, ctor := range pipeConstructors {
+		if i == 0 {
+			if pipe, err = WrapWithPipeline(in, ctor); err != nil {
+				return nil, err
+			}
+		} else if pipe, err = WrapWithPipeline(pipe, ctor); err != nil {
 			return nil, err
 		}
 	}
-	return in, nil
+	return pipe, nil
 }
 
 //------------------------------------------------------------------------------