execute processor condition template

go.mod

@@ -5,6 +5,7 @@ go 1.21.1
 require (
 	buf.build/gen/go/grpc-ecosystem/grpc-gateway/protocolbuffers/go v1.32.0-20231027202514-3f42134f4c56.1
 	github.com/Masterminds/semver/v3 v3.2.1
+	github.com/Masterminds/sprig/v3 v3.2.3
 	github.com/NYTimes/gziphandler v1.1.1
 	github.com/antchfx/jsonquery v1.3.3
 	github.com/bufbuild/buf v1.29.0
@@ -71,7 +72,6 @@ require (
 	github.com/GaijinEntertainment/go-exhaustruct/v3 v3.1.0 // indirect
 	github.com/Masterminds/goutils v1.1.1 // indirect
 	github.com/Masterminds/semver v1.5.0 // indirect
-	github.com/Masterminds/sprig/v3 v3.2.3 // indirect
 	github.com/Masterminds/squirrel v1.5.4 // indirect
 	github.com/Microsoft/go-winio v0.6.1 // indirect
 	github.com/OpenPeeDeeP/depguard/v2 v2.1.0 // indirect

pkg/processor/template_utils.go

@@ -0,0 +1,67 @@
+// Copyright © 2024 Meroxa, Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package processor
+
+import (
+	"bytes"
+	"fmt"
+	"strconv"
+	"text/template"
+
+	"github.com/Masterminds/sprig/v3"
+	"github.com/conduitio/conduit/pkg/record"
+)
+
+// TemplateUtils provides methods to parse, execute go templates for provided records, and return the boolean value of the output.
+type TemplateUtils struct {
+	condition string
+	rec       record.Record
+	tmpl      *template.Template
+}
+
+func NewTemplateUtils(condition string, rec record.Record) *TemplateUtils {
+	return &TemplateUtils{
+		condition: condition,
+		rec:       rec,
+	}
+}
+
+// parse parses the provided template with injecting `sprig` functions, returns an error if parsing failed.
+func (t *TemplateUtils) parse() error {
+	tmpl, err := template.New("").Funcs(sprig.FuncMap()).Parse(t.condition)
+	if err != nil {
+		return err
+	}
+	t.tmpl = tmpl
+	return nil
+}
+
+// execute executes the template for the provided record, and parses the template output into a boolean, returns an error
+// if output is not a boolean, or if template is nil (should be called after TemplateUtils.parse).
+func (t *TemplateUtils) execute() (bool, error) {
+	if t.tmpl == nil {
+		return false, fmt.Errorf("template is nil, make sure to parse it first by calling TemplateUtils.parse()")
+	}
+	var b bytes.Buffer
+	err := t.tmpl.Execute(&b, t.rec)
+	if err != nil {
+		return false, err
+	}
+	output, err := strconv.ParseBool(b.String())
+	if err != nil {
+		return false, fmt.Errorf("error converting the condition go-template output to boolean, %w", err)
+	}
+	return output, nil
+}

pkg/processor/template_utils_test.go

@@ -0,0 +1,64 @@
+// Copyright © 2024 Meroxa, Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package processor
+
+import (
+	"testing"
+
+	"github.com/conduitio/conduit/pkg/record"
+	"github.com/matryer/is"
+)
+
+func Test_TemplateUtils_ExecuteTrue(t *testing.T) {
+	is := is.New(t)
+	condition := "{{ eq .Metadata.key \"val\" }}"
+	rec := record.Record{
+		Position: record.Position("position-out"),
+		Metadata: record.Metadata{"key": "val"},
+	}
+	tmpl := NewTemplateUtils(condition, rec)
+	is.NoErr(tmpl.parse())
+	res, err := tmpl.execute()
+	is.NoErr(err)
+	is.True(res)
+}
+
+func Test_TemplateUtils_ExecuteFalse(t *testing.T) {
+	is := is.New(t)
+	condition := "{{ eq .Metadata.key \"wrongVal\" }}"
+	rec := record.Record{
+		Position: record.Position("position-out"),
+		Metadata: record.Metadata{"key": "val"},
+	}
+	tmpl := NewTemplateUtils(condition, rec)
+	is.NoErr(tmpl.parse())
+	res, err := tmpl.execute()
+	is.NoErr(err)
+	is.True(res == false)
+}
+
+func Test_TemplateUtils_NonBooleanValue(t *testing.T) {
+	is := is.New(t)
+	condition := "{{ printf \"hi\" }}"
+	rec := record.Record{
+		Position: record.Position("position-out"),
+		Metadata: record.Metadata{"key": "val"},
+	}
+	tmpl := NewTemplateUtils(condition, rec)
+	is.NoErr(tmpl.parse())
+	res, err := tmpl.execute()
+	is.True(err != nil)
+	is.True(res == false)
+}