Use easyjson to unmarshal the ProjectionNotification

Author: boekkooi-fresh

driver/sql/projection.go

@@ -5,6 +5,7 @@ import (
 	"database/sql"
 
 	"github.com/hellofresh/goengine"
+	"github.com/mailru/easyjson/jlexer"
 )
 
 type (
@@ -57,3 +58,45 @@ type (
 	// EventStreamLoader loads a event stream based on the provided notification and state
 	EventStreamLoader func(ctx context.Context, conn *sql.Conn, notification *ProjectionNotification, position int64) (goengine.EventStream, error)
 )
+
+// UnmarshalJSON supports json.Unmarshaler interface
+func (p *ProjectionNotification) UnmarshalJSON(data []byte) error {
+	r := jlexer.Lexer{Data: data}
+	p.UnmarshalEasyJSON(&r)
+	return r.Error()
+}
+
+// UnmarshalEasyJSON supports easyjson.Unmarshaler interface
+func (p *ProjectionNotification) UnmarshalEasyJSON(in *jlexer.Lexer) {
+	isTopLevel := in.IsStart()
+	if in.IsNull() {
+		if isTopLevel {
+			in.Consumed()
+		}
+		in.Skip()
+		return
+	}
+	in.Delim('{')
+	for !in.IsDelim('}') {
+		key := in.UnsafeString()
+		in.WantColon()
+		if in.IsNull() {
+			in.Skip()
+			in.WantComma()
+			continue
+		}
+		switch key {
+		case "no":
+			p.No = int64(in.Int64())
+		case "aggregate_id":
+			p.AggregateID = string(in.String())
+		default:
+			in.SkipRecursive()
+		}
+		in.WantComma()
+	}
+	in.Delim('}')
+	if isTopLevel {
+		in.Consumed()
+	}
+}

extension/pq/listener.go

@@ -2,13 +2,13 @@ package pq
 
 import (
 	"context"
-	"encoding/json"
 	"strings"
 	"time"
 
 	"github.com/hellofresh/goengine"
 	"github.com/hellofresh/goengine/driver/sql"
 	"github.com/lib/pq"
+	"github.com/mailru/easyjson"
 )
 
 // Ensure Listener implements sql.Listener
@@ -138,7 +138,7 @@ func (s *Listener) unmarshalNotification(n *pq.Notification) *sql.ProjectionNoti
 	}
 
 	notification := &sql.ProjectionNotification{}
-	if err := json.Unmarshal([]byte(n.Extra), notification); err != nil {
+	if err := easyjson.Unmarshal([]byte(n.Extra), notification); err != nil {
 		logger.WithError(err).Error("received invalid notification data")
 		return nil
 	}