proto: inline the implementation of protoimpl.X.ExtensionDescFromType (…

…#1056)
golang · Mar 13, 2020 · b860323 · b860323
1 parent 7bd8073
commit b860323
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/proto/extensions.go b/proto/extensions.go
@@ -286,8 +286,10 @@ func ExtensionDescs(m Message) ([]*ExtensionDesc, error) {
 	extDescs := make(map[protoreflect.FieldNumber]*ExtensionDesc)
 	mr.Range(func(fd protoreflect.FieldDescriptor, v protoreflect.Value) bool {
 		if fd.IsExtension() {
-			xts := fd.(protoreflect.ExtensionTypeDescriptor)
-			extDescs[fd.Number()] = protoimpl.X.ExtensionDescFromType(xts.Type())
+			xt := fd.(protoreflect.ExtensionTypeDescriptor)
+			if xd, ok := xt.Type().(*ExtensionDesc); ok {
+				extDescs[fd.Number()] = xd
+			}
 		}
 		return true
 	})