Damien Neil 8012b444ee internal/fileinit: generate reflect data structures from raw descriptors
This CL takes a significantly different approach to generating support
for protobuf reflection. The previous approach involved generating a
large number of Go literals to represent the reflection information.
While that approach was correct, it resulted in too much binary bloat.

The approach taken here initializes the reflection information from
the raw descriptor proto, which is a relatively dense representation
of the protobuf reflection information. In order to keep initialization
cost low, several measures were taken:
* At program init, the bare minimum is parsed in order to initialize
naming information for enums, messages, extensions, and services declared
in the file. This is done because those top-level declarations are often
relevant for registration.
* Only upon first are most of the other data structures for protobuf
reflection actually initialized.
* Instead of using proto.Unmarshal, a hand-written unmarshaler is used.
This allows us to avoid a dependendency on the descriptor proto and also
because the API for the descriptor proto is fundamentally non-performant
since it requires an allocation for every primitive field.

At a high-level, the new implementation lives in internal/fileinit.

Several changes were made to other parts of the repository:
* cmd/protoc-gen-go:
  * Stop compressing the raw descriptors. While compression does reduce
the size of the descriptors by approximately 2x, it is a pre-mature
optimization since the descriptors themselves are around 1% of the total
binary bloat that is due to generated protobufs.
  * Seeding protobuf reflection from the raw descriptor significantly
simplifies the generator implementation since it is no longer responsible
for constructing a tree of Go literals to represent the same information.
  * We remove the generation of the shadow types and instead call
protoimpl.MessageType.MessageOf. Unfortunately, this incurs an allocation
for every call to ProtoReflect since we need to allocate a tuple that wraps
a pointer to the message value, and a pointer to message type.
* internal/impl:
  * We add a MessageType.GoType field and make it required that it is
set prior to first use. This is done so that we can avoid calling
MessageType.init except for when it is actually needed. The allows code
to call (*FooMessage)(nil).ProtoReflect().Type() without fearing that the
init code will run, possibly triggering a recursive deadlock (where the
init code depends on getting the Type of some dependency which may be
declared within the same file).
* internal/cmd/generate-types:
  * The code to generate reflect/prototype/protofile_list_gen.go was copied
and altered to generated internal/fileinit.desc_list_gen.go.

At a high-level this CL adds significant technical complexity.
However, this is offset by several possible future changes:
* The prototype package can be drastically simplified. We can probably
reimplement internal/legacy to use internal/fileinit instead, allowing us
to drop another dependency on the prototype package. As a result, we can
probably delete most of the constructor types in that package.
* With the prototype package significantly pruned, and the fact that generated
code no longer depend on depends on that package, we can consider merging
what's left of prototype into protodesc.

Change-Id: I6090f023f2e1b6afaf62bd3ae883566242e30715
Reviewed-on: https://go-review.googlesource.com/c/158539
Reviewed-by: Herbie Ong <herbie@google.com>
Reviewed-by: Joe Tsai <thebrokentoaster@gmail.com>
2019-01-30 01:33:46 +00:00

95 lines
3.8 KiB
Go

// Copyright 2018 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package fileinit
// Constants for field numbers of messages declared in descriptor.proto.
const (
// FileDescriptorProto field numbers
fileDesc_Syntax = 12 // optional string
fileDesc_Name = 1 // optional string
fileDesc_Package = 2 // optional string
fileDesc_Imports = 3 // repeated string
fileDesc_PublicImports = 10 // repeated int32
fileDesc_WeakImports = 11 // repeated int32
fileDesc_Enums = 5 // repeated EnumDescriptorProto
fileDesc_Messages = 4 // repeated DescriptorProto
fileDesc_Extensions = 7 // repeated FieldDescriptorProto
fileDesc_Services = 6 // repeated ServiceDescriptorProto
fileDesc_Options = 8 // optional FileOptions
// EnumDescriptorProto field numbers
enumDesc_Name = 1 // optional string
enumDesc_Values = 2 // repeated EnumValueDescriptorProto
enumDesc_ReservedNames = 5 // repeated string
enumDesc_ReservedRanges = 4 // repeated EnumReservedRange
enumDesc_Options = 3 // optional EnumOptions
// EnumReservedRange field numbers
enumReservedRange_Start = 1 // optional int32
enumReservedRange_End = 2 // optional int32
// EnumValueDescriptorProto field numbers
enumValueDesc_Name = 1 // optional string
enumValueDesc_Number = 2 // optional int32
enumValueDesc_Options = 3 // optional EnumValueOptions
// DescriptorProto field numbers
messageDesc_Name = 1 // optional string
messageDesc_Fields = 2 // repeated FieldDescriptorProto
messageDesc_Oneofs = 8 // repeated OneofDescriptorProto
messageDesc_ReservedNames = 10 // repeated string
messageDesc_ReservedRanges = 9 // repeated ReservedRange
messageDesc_ExtensionRanges = 5 // repeated ExtensionRange
messageDesc_Enums = 4 // repeated EnumDescriptorProto
messageDesc_Messages = 3 // repeated DescriptorProto
messageDesc_Extensions = 6 // repeated FieldDescriptorProto
messageDesc_Options = 7 // optional MessageOptions
// ReservedRange field numbers
messageReservedRange_Start = 1 // optional int32
messageReservedRange_End = 2 // optional int32
// ExtensionRange field numbers
messageExtensionRange_Start = 1 // optional int32
messageExtensionRange_End = 2 // optional int32
messageExtensionRange_Options = 3 // optional ExtensionRangeOptions
// MessageOptions field numbers
messageOptions_IsMapEntry = 7 // optional bool
// FieldDescriptorProto field numbers
fieldDesc_Name = 1 // optional string
fieldDesc_Number = 3 // optional int32
fieldDesc_Cardinality = 4 // optional Label
fieldDesc_Kind = 5 // optional Type
fieldDesc_JSONName = 10 // optional string
fieldDesc_Default = 7 // optional string
fieldDesc_OneofIndex = 9 // optional int32
fieldDesc_TypeName = 6 // optional string
fieldDesc_ExtendedType = 2 // optional string
fieldDesc_Options = 8 // optional FieldOptions
// FieldOptions field numbers
fieldOptions_IsPacked = 2 // optional bool
fieldOptions_IsWeak = 10 // optional bool
// OneofDescriptorProto field numbers
oneofDesc_Name = 1 // optional string
oneofDesc_Options = 2 // optional OneofOptions
// ServiceDescriptorProto field numbers
serviceDesc_Name = 1 // optional string
serviceDesc_Methods = 2 // repeated MethodDescriptorProto
serviceDesc_Options = 3 // optional ServiceOptions
// MethodDescriptorProto field numbers
methodDesc_Name = 1 // optional string
methodDesc_InputType = 2 // optional string
methodDesc_OutputType = 3 // optional string
methodDesc_IsStreamingClient = 5 // optional bool
methodDesc_IsStreamingServer = 6 // optional bool
methodDesc_Options = 4 // optional MethodOptions
)