Merge pull request #270 from jwiklund/binary-metadata

base64 decode binary metadata
bojand · May 4, 2021 · ff10171 · ff10171
2 parents 8d555a2 + 57df589
commit ff10171
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/runner/calldata.go b/runner/calldata.go
@@ -2,8 +2,10 @@ package runner
 
 import (
 	"bytes"
+	"encoding/base64"
 	"encoding/json"
 	"math/rand"
+	"strings"
 	"text/template"
 	"text/template/parse"
 	"time"
@@ -181,6 +183,16 @@ func (td *CallData) executeMetadata(metadata string) (map[string]string, error)
 		if err != nil {
 			return nil, err
 		}
+
+		for key, value := range mdMap {
+			if strings.HasSuffix(key, "-bin") {
+				decoded, err := base64.StdEncoding.DecodeString(value)
+				if err != nil {
+					return nil, err
+				}
+				mdMap[key] = string(decoded)
+			}
+		}
 	}
 
 	return mdMap, nil

diff --git a/runner/calldata_test.go b/runner/calldata_test.go
@@ -114,6 +114,11 @@ func TestCallData_ExecuteMetadata(t *testing.T) {
 			map[string]string{"trace_id": "asdf {{.Something}} {{.MethodName}} bob"},
 			false,
 		},
+		{"with binary data",
+			`{"data-bin":"YmFzZTY0"}`,
+			map[string]string{"data-bin": "base64"},
+			false,
+		},
 	}
 
 	for _, tt := range tests {