Use easyjson.Marshaler, json.Marshaler or json.Marshal

In order to allow for faster JSON marshalling use `internal.MarshalJSON` which will pick the best possible json marshalling action.

Author: boekkooi-fresh

metadata/metadata_test.go

@@ -10,6 +10,7 @@ import (
 	"unicode"
 
 	"github.com/hellofresh/goengine/metadata"
+	"github.com/mailru/easyjson"
 	"github.com/stretchr/testify/assert"
 )
 
@@ -253,7 +254,7 @@ func BenchmarkMarshalJSON(b *testing.B) {
 
 	b.ResetTimer()
 	for i := 0; i < b.N; i++ {
-		_, err := json.Marshal(m)
+		_, err := easyjson.Marshal(m.(easyjson.Marshaler))
 		if err != nil {
 			b.Fail()
 		}

strategy/json/internal/json.go

@@ -0,0 +1,24 @@
+package internal
+
+import (
+	"encoding/json"
+
+	"github.com/mailru/easyjson"
+	"github.com/mailru/easyjson/jwriter"
+)
+
+// MarshalJSON returns the JSON encoding of v.
+// This is done using `easyjson.Marshaler`, `json.Marshaler` or `json.Marshal`
+func MarshalJSON(v interface{}) ([]byte, error) {
+	if vm, ok := v.(easyjson.Marshaler); ok {
+		w := &jwriter.Writer{}
+		vm.MarshalEasyJSON(w)
+		return w.Buffer.BuildBytes(), w.Error
+	}
+
+	if vm, ok := v.(json.Marshaler); ok {
+		return vm.MarshalJSON()
+	}
+
+	return json.Marshal(v)
+}

strategy/json/payload_transformer.go

@@ -8,6 +8,7 @@ import (
 
 	"github.com/hellofresh/goengine"
 	reflectUtil "github.com/hellofresh/goengine/internal/reflect"
+	"github.com/hellofresh/goengine/strategy/json/internal"
 )
 
 var (
@@ -66,7 +67,7 @@ func (p *PayloadTransformer) ConvertPayload(payload interface{}) (string, []byte
 		return "", nil, err
 	}
 
-	data, err := json.Marshal(payload)
+	data, err := internal.MarshalJSON(payload)
 	if err != nil {
 		return "", nil, ErrPayloadCannotBeSerialized
 	}

strategy/json/sql/message_factory_aggregate_test.go

@@ -3,18 +3,17 @@
 package sql_test
 
 import (
-	"encoding/json"
 	"errors"
 	"testing"
 	"time"
 
-	"github.com/golang/mock/gomock"
-
 	sqlmock "github.com/DATA-DOG/go-sqlmock"
+	"github.com/golang/mock/gomock"
 	"github.com/hellofresh/goengine"
 	"github.com/hellofresh/goengine/aggregate"
 	"github.com/hellofresh/goengine/metadata"
 	"github.com/hellofresh/goengine/mocks"
+	"github.com/hellofresh/goengine/strategy/json/internal"
 	"github.com/hellofresh/goengine/strategy/json/sql"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
@@ -70,10 +69,10 @@ func TestAggregateChangedFactory_CreateFromRows(t *testing.T) {
 				payloadFactory := mocks.NewMessagePayloadFactory(ctrl)
 				mockRows := sqlmock.NewRows(rowColumns)
 				for i, msg := range expectedMessages {
-					rowPayload, err := json.Marshal(msg.Payload())
+					rowPayload, err := internal.MarshalJSON(msg.Payload())
 					require.NoError(t, err)
 
-					rowMetadata, err := json.Marshal(msg.Metadata())
+					rowMetadata, err := internal.MarshalJSON(msg.Metadata())
 					require.NoError(t, err)
 
 					msgNr := i + 1

strategy/json/sql/postgres/single_stream_strategy.go

@@ -1,15 +1,14 @@
 package postgres
 
 import (
-	"encoding/json"
 	"fmt"
 	"regexp"
 	"strings"
 
-	"github.com/hellofresh/goengine/driver/sql/postgres"
-
 	"github.com/hellofresh/goengine"
 	"github.com/hellofresh/goengine/driver/sql"
+	"github.com/hellofresh/goengine/driver/sql/postgres"
+	"github.com/hellofresh/goengine/strategy/json/internal"
 )
 
 var (
@@ -81,7 +80,7 @@ func (s *SingleStreamStrategy) PrepareData(messages []goengine.Message) ([]inter
 			return nil, err
 		}
 
-		meta, err := json.Marshal(msg.Metadata())
+		meta, err := internal.MarshalJSON(msg.Metadata())
 		if err != nil {
 			return nil, err
 		}

strategy/json/sql/postgres/single_stream_strategy_test.go

@@ -3,7 +3,6 @@
 package postgres_test
 
 import (
-	"encoding/json"
 	"errors"
 	"fmt"
 	"testing"
@@ -13,6 +12,7 @@ import (
 	"github.com/hellofresh/goengine"
 	"github.com/hellofresh/goengine/metadata"
 	"github.com/hellofresh/goengine/mocks"
+	"github.com/hellofresh/goengine/strategy/json/internal"
 	"github.com/hellofresh/goengine/strategy/json/sql/postgres"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
@@ -181,7 +181,7 @@ func TestPrepareData(t *testing.T) {
 
 			pc.EXPECT().ConvertPayload(payload).Return(payloadType, payload, nil).AnyTimes()
 
-			metaJSON, err := json.Marshal(meta)
+			metaJSON, err := internal.MarshalJSON(meta)
 			require.NoError(t, err)
 
 			expectedColumns = append(