4

Change-Id: I84c74a84f5d69fcd9abd5933ca2923dcc8ff2220

Author: callthingsoff

src/encoding/json/encode.go

@@ -329,13 +329,37 @@ func isEmptyValue(v reflect.Value) bool {
 	return false
 }
 
+type isZeroer interface {
+	IsZero() bool
+}
+
+var isZeroerType = reflect.TypeFor[isZeroer]()
+
 func isZeroValue(v reflect.Value) bool {
-	if z, ok := v.Interface().(interface {
-		IsZero() bool
-	}); ok {
-		return z.IsZero()
+	// Provide a function that uses a type's IsZero method.
+	var isZero func(reflect.Value) bool
+
+	switch {
+	case v.Kind() == reflect.Interface && v.Type().Implements(isZeroerType):
+		isZero = func(va reflect.Value) bool {
+			// Avoid panics calling IsZero on a nil interface or
+			// non-nil interface with nil pointer.
+			return va.IsNil() ||
+				(va.Elem().Kind() == reflect.Pointer && va.Elem().IsNil()) ||
+				va.Interface().(isZeroer).IsZero()
+		}
+	case v.Kind() == reflect.Pointer && v.Type().Implements(isZeroerType):
+		isZero = func(va reflect.Value) bool {
+			// Avoid panics calling IsZero on nil pointer.
+			return va.IsNil() || va.Interface().(isZeroer).IsZero()
+		}
+	case v.Type().Implements(isZeroerType):
+		isZero = func(va reflect.Value) bool { return va.Interface().(isZeroer).IsZero() }
+	case reflect.PointerTo(v.Type()).Implements(isZeroerType):
+		isZero = func(va reflect.Value) bool { return va.Addr().Interface().(isZeroer).IsZero() }
 	}
-	return v.IsZero()
+
+	return (isZero == nil && v.IsZero() || (isZero != nil && isZero(v)))
 }
 
 func (e *encodeState) reflectValue(v reflect.Value, opts encOpts) {

src/encoding/json/encode_test.go

@@ -77,6 +77,14 @@ func (nzs NonZeroStruct) IsZero() bool {
 	return false
 }
 
+type NoPanicStruct struct {
+	Int int `json:"int,omitzero"`
+}
+
+func (nps *NoPanicStruct) IsZero() bool {
+	return nps.Int == 0
+}
+
 type OptionalsZero struct {
 	Sr string `json:"sr"`
 	So string `json:"so,omitzero"`
@@ -106,6 +114,12 @@ type OptionalsZero struct {
 
 	Time time.Time     `json:"time,omitzero"`
 	Nzs  NonZeroStruct `json:"nzs,omitzero"`
+
+	IsZeroer interface {
+		IsZero() bool
+	} `json:"iszeroer,omitzero"`
+	NoPanicStruct1 *NoPanicStruct `json:"nps1,omitzero"`
+	NoPanicStruct2 NoPanicStruct  `json:"nps2,omitzero"`
 }
 
 func TestOmitZero(t *testing.T) {