mirror of
https://gitea.com/Lydanne/buildx.git
synced 2025-05-18 09:17:49 +08:00
b35a0f4718
Removes gogo/protobuf from buildx and updates to a version of moby/buildkit where gogo is removed. This also changes how the proto files are generated. This is because newer versions of protobuf are more strict about name conflicts. If two files have the same name (even if they are relative paths) and are used in different protoc commands, they'll conflict in the registry. Since protobuf file generation doesn't work very well with `paths=source_relative`, this removes the `go:generate` expression and just relies on the dockerfile to perform the generation. Signed-off-by: Jonathan A. Sternberg <jonathan.sternberg@docker.com>
104 lines
3.1 KiB
Go
104 lines
3.1 KiB
Go
// Copyright 2019 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 proto
|
|
|
|
import (
|
|
"google.golang.org/protobuf/encoding/protowire"
|
|
"google.golang.org/protobuf/internal/encoding/messageset"
|
|
"google.golang.org/protobuf/reflect/protoreflect"
|
|
"google.golang.org/protobuf/runtime/protoiface"
|
|
)
|
|
|
|
// Size returns the size in bytes of the wire-format encoding of m.
|
|
func Size(m Message) int {
|
|
return MarshalOptions{}.Size(m)
|
|
}
|
|
|
|
// Size returns the size in bytes of the wire-format encoding of m.
|
|
func (o MarshalOptions) Size(m Message) int {
|
|
// Treat a nil message interface as an empty message; nothing to output.
|
|
if m == nil {
|
|
return 0
|
|
}
|
|
|
|
return o.size(m.ProtoReflect())
|
|
}
|
|
|
|
// size is a centralized function that all size operations go through.
|
|
// For profiling purposes, avoid changing the name of this function or
|
|
// introducing other code paths for size that do not go through this.
|
|
func (o MarshalOptions) size(m protoreflect.Message) (size int) {
|
|
methods := protoMethods(m)
|
|
if methods != nil && methods.Size != nil {
|
|
out := methods.Size(protoiface.SizeInput{
|
|
Message: m,
|
|
Flags: o.flags(),
|
|
})
|
|
return out.Size
|
|
}
|
|
if methods != nil && methods.Marshal != nil {
|
|
// This is not efficient, but we don't have any choice.
|
|
// This case is mainly used for legacy types with a Marshal method.
|
|
out, _ := methods.Marshal(protoiface.MarshalInput{
|
|
Message: m,
|
|
Flags: o.flags(),
|
|
})
|
|
return len(out.Buf)
|
|
}
|
|
return o.sizeMessageSlow(m)
|
|
}
|
|
|
|
func (o MarshalOptions) sizeMessageSlow(m protoreflect.Message) (size int) {
|
|
if messageset.IsMessageSet(m.Descriptor()) {
|
|
return o.sizeMessageSet(m)
|
|
}
|
|
m.Range(func(fd protoreflect.FieldDescriptor, v protoreflect.Value) bool {
|
|
size += o.sizeField(fd, v)
|
|
return true
|
|
})
|
|
size += len(m.GetUnknown())
|
|
return size
|
|
}
|
|
|
|
func (o MarshalOptions) sizeField(fd protoreflect.FieldDescriptor, value protoreflect.Value) (size int) {
|
|
num := fd.Number()
|
|
switch {
|
|
case fd.IsList():
|
|
return o.sizeList(num, fd, value.List())
|
|
case fd.IsMap():
|
|
return o.sizeMap(num, fd, value.Map())
|
|
default:
|
|
return protowire.SizeTag(num) + o.sizeSingular(num, fd.Kind(), value)
|
|
}
|
|
}
|
|
|
|
func (o MarshalOptions) sizeList(num protowire.Number, fd protoreflect.FieldDescriptor, list protoreflect.List) (size int) {
|
|
sizeTag := protowire.SizeTag(num)
|
|
|
|
if fd.IsPacked() && list.Len() > 0 {
|
|
content := 0
|
|
for i, llen := 0, list.Len(); i < llen; i++ {
|
|
content += o.sizeSingular(num, fd.Kind(), list.Get(i))
|
|
}
|
|
return sizeTag + protowire.SizeBytes(content)
|
|
}
|
|
|
|
for i, llen := 0, list.Len(); i < llen; i++ {
|
|
size += sizeTag + o.sizeSingular(num, fd.Kind(), list.Get(i))
|
|
}
|
|
return size
|
|
}
|
|
|
|
func (o MarshalOptions) sizeMap(num protowire.Number, fd protoreflect.FieldDescriptor, mapv protoreflect.Map) (size int) {
|
|
sizeTag := protowire.SizeTag(num)
|
|
|
|
mapv.Range(func(key protoreflect.MapKey, value protoreflect.Value) bool {
|
|
size += sizeTag
|
|
size += protowire.SizeBytes(o.sizeField(fd.MapKey(), key.Value()) + o.sizeField(fd.MapValue(), value))
|
|
return true
|
|
})
|
|
return size
|
|
}
|