From cbda08a525662e9f2416ad7cca645a0b49207297 Mon Sep 17 00:00:00 2001
From: Masaaki Goshima <goccy54@gmail.com>
Date: Sat, 5 Jun 2021 13:00:26 +0900
Subject: [PATCH] Add DecodeFieldPriorityFirstWin option

---
 benchmarks/decode_test.go  | 28 +++++++++++++++
 internal/decoder/struct.go | 74 ++++++++++++++++++++++++++++++--------
 2 files changed, 88 insertions(+), 14 deletions(-)

diff --git a/benchmarks/decode_test.go b/benchmarks/decode_test.go
index 805645c..ae3439a 100644
--- a/benchmarks/decode_test.go
+++ b/benchmarks/decode_test.go
@@ -375,6 +375,34 @@ func Benchmark_Decode_LargeStruct_Unmarshal_GoJsonNoEscape(b *testing.B) {
 	}
 }
 
+func Benchmark_Decode_LargeStruct_Unmarshal_GoJsonFirstWinMode(b *testing.B) {
+	b.ReportAllocs()
+	for i := 0; i < b.N; i++ {
+		result := LargePayload{}
+		if err := gojson.UnmarshalWithOption(
+			LargeFixture,
+			&result,
+			gojson.DecodeFieldPriorityFirstWin(),
+		); err != nil {
+			b.Fatal(err)
+		}
+	}
+}
+
+func Benchmark_Decode_LargeStruct_Unmarshal_GoJsonNoEscapeFirstWinMode(b *testing.B) {
+	b.ReportAllocs()
+	for i := 0; i < b.N; i++ {
+		result := LargePayload{}
+		if err := gojson.UnmarshalNoEscape(
+			LargeFixture,
+			&result,
+			gojson.DecodeFieldPriorityFirstWin(),
+		); err != nil {
+			b.Fatal(err)
+		}
+	}
+}
+
 func Benchmark_Decode_LargeStruct_Stream_EncodingJson(b *testing.B) {
 	b.ReportAllocs()
 	reader := bytes.NewReader(LargeFixture)
diff --git a/internal/decoder/struct.go b/internal/decoder/struct.go
index c5de021..ae33bdd 100644
--- a/internal/decoder/struct.go
+++ b/internal/decoder/struct.go
@@ -17,22 +17,24 @@ type structFieldSet struct {
 	dec         Decoder
 	offset      uintptr
 	isTaggedKey bool
+	fieldIdx    int
 	key         string
 	keyLen      int64
 	err         error
 }
 
 type structDecoder struct {
-	fieldMap         map[string]*structFieldSet
-	stringDecoder    *stringDecoder
-	structName       string
-	fieldName        string
-	isTriedOptimize  bool
-	keyBitmapUint8   [][256]uint8
-	keyBitmapUint16  [][256]uint16
-	sortedFieldSets  []*structFieldSet
-	keyDecoder       func(*structDecoder, []byte, int64) (int64, *structFieldSet, error)
-	keyStreamDecoder func(*structDecoder, *Stream) (*structFieldSet, string, error)
+	fieldMap           map[string]*structFieldSet
+	fieldUniqueNameNum int
+	stringDecoder      *stringDecoder
+	structName         string
+	fieldName          string
+	isTriedOptimize    bool
+	keyBitmapUint8     [][256]uint8
+	keyBitmapUint16    [][256]uint16
+	sortedFieldSets    []*structFieldSet
+	keyDecoder         func(*structDecoder, []byte, int64) (int64, *structFieldSet, error)
+	keyStreamDecoder   func(*structDecoder, *Stream) (*structFieldSet, string, error)
 }
 
 var (
@@ -66,6 +68,21 @@ const (
 )
 
 func (d *structDecoder) tryOptimize() {
+	fieldUniqueNameMap := map[string]int{}
+	fieldIdx := -1
+	for k, v := range d.fieldMap {
+		lower := strings.ToLower(k)
+		idx, exists := fieldUniqueNameMap[lower]
+		if exists {
+			v.fieldIdx = idx
+		} else {
+			fieldIdx++
+			v.fieldIdx = fieldIdx
+		}
+		fieldUniqueNameMap[lower] = fieldIdx
+	}
+	d.fieldUniqueNameNum = len(fieldUniqueNameMap)
+
 	if d.isTriedOptimize {
 		return
 	}
@@ -706,6 +723,14 @@ func (d *structDecoder) Decode(ctx *RuntimeContext, cursor, depth int64, p unsaf
 		cursor++
 		return cursor, nil
 	}
+	var (
+		seenFields   map[int]struct{}
+		seenFieldNum int
+	)
+	firstWin := ctx.Option.FirstWin
+	if firstWin {
+		seenFields = make(map[int]struct{}, d.fieldUniqueNameNum)
+	}
 	for {
 		c, field, err := d.keyDecoder(d, buf, cursor)
 		if err != nil {
@@ -723,11 +748,32 @@ func (d *structDecoder) Decode(ctx *RuntimeContext, cursor, depth int64, p unsaf
 			if field.err != nil {
 				return 0, field.err
 			}
-			c, err := field.dec.Decode(ctx, cursor, depth, unsafe.Pointer(uintptr(p)+field.offset))
-			if err != nil {
-				return 0, err
+			if firstWin {
+				if _, exists := seenFields[field.fieldIdx]; exists {
+					c, err := skipValue(buf, cursor, depth)
+					if err != nil {
+						return 0, err
+					}
+					cursor = c
+				} else {
+					c, err := field.dec.Decode(ctx, cursor, depth, unsafe.Pointer(uintptr(p)+field.offset))
+					if err != nil {
+						return 0, err
+					}
+					cursor = c
+					seenFieldNum++
+					if d.fieldUniqueNameNum <= seenFieldNum {
+						return skipObject(buf, cursor, depth)
+					}
+					seenFields[field.fieldIdx] = struct{}{}
+				}
+			} else {
+				c, err := field.dec.Decode(ctx, cursor, depth, unsafe.Pointer(uintptr(p)+field.offset))
+				if err != nil {
+					return 0, err
+				}
+				cursor = c
 			}
-			cursor = c
 		} else {
 			c, err := skipValue(buf, cursor, depth)
 			if err != nil {