mirror of
https://gitea.com/Lydanne/buildx.git
synced 2025-05-28 16:37:43 +08:00
64c5139ab6
Integrates vtproto into buildx. The generated files dockerfile has been modified to copy the buildkit equivalent file to ensure files are laid out in the appropriate way for imports. An import has also been included to change the grpc codec to the version in buildkit that supports vtproto. This will allow buildx to utilize the speed and memory improvements from that. Also updates the gc control options for prune. Signed-off-by: Jonathan A. Sternberg <jonathan.sternberg@docker.com>
106 lines
2.5 KiB
Go
106 lines
2.5 KiB
Go
package pool
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"google.golang.org/protobuf/compiler/protogen"
|
|
"google.golang.org/protobuf/reflect/protoreflect"
|
|
|
|
"github.com/planetscale/vtprotobuf/generator"
|
|
)
|
|
|
|
func init() {
|
|
generator.RegisterFeature("pool", func(gen *generator.GeneratedFile) generator.FeatureGenerator {
|
|
return &pool{GeneratedFile: gen}
|
|
})
|
|
}
|
|
|
|
type pool struct {
|
|
*generator.GeneratedFile
|
|
once bool
|
|
}
|
|
|
|
var _ generator.FeatureGenerator = (*pool)(nil)
|
|
|
|
func (p *pool) GenerateFile(file *protogen.File) bool {
|
|
for _, message := range file.Messages {
|
|
p.message(message)
|
|
}
|
|
return p.once
|
|
}
|
|
|
|
func (p *pool) message(message *protogen.Message) {
|
|
for _, nested := range message.Messages {
|
|
p.message(nested)
|
|
}
|
|
|
|
if message.Desc.IsMapEntry() || !p.ShouldPool(message) {
|
|
return
|
|
}
|
|
|
|
p.once = true
|
|
ccTypeName := message.GoIdent
|
|
|
|
p.P(`var vtprotoPool_`, ccTypeName, ` = `, p.Ident("sync", "Pool"), `{`)
|
|
p.P(`New: func() interface{} {`)
|
|
p.P(`return &`, ccTypeName, `{}`)
|
|
p.P(`},`)
|
|
p.P(`}`)
|
|
|
|
p.P(`func (m *`, ccTypeName, `) ResetVT() {`)
|
|
p.P(`if m != nil {`)
|
|
var saved []*protogen.Field
|
|
for _, field := range message.Fields {
|
|
fieldName := field.GoName
|
|
|
|
if field.Desc.IsList() {
|
|
switch field.Desc.Kind() {
|
|
case protoreflect.MessageKind, protoreflect.GroupKind:
|
|
p.P(`for _, mm := range m.`, fieldName, `{`)
|
|
if p.ShouldPool(field.Message) {
|
|
p.P(`mm.ResetVT()`)
|
|
} else {
|
|
p.P(`mm.Reset()`)
|
|
}
|
|
p.P(`}`)
|
|
}
|
|
p.P(fmt.Sprintf("f%d", len(saved)), ` := m.`, fieldName, `[:0]`)
|
|
saved = append(saved, field)
|
|
} else if field.Oneof != nil && !field.Oneof.Desc.IsSynthetic() {
|
|
if p.ShouldPool(field.Message) {
|
|
p.P(`if oneof, ok := m.`, field.Oneof.GoName, `.(*`, field.GoIdent, `); ok {`)
|
|
p.P(`oneof.`, fieldName, `.ReturnToVTPool()`)
|
|
p.P(`}`)
|
|
}
|
|
} else {
|
|
switch field.Desc.Kind() {
|
|
case protoreflect.MessageKind, protoreflect.GroupKind:
|
|
if !field.Desc.IsMap() && p.ShouldPool(field.Message) {
|
|
p.P(`m.`, fieldName, `.ReturnToVTPool()`)
|
|
}
|
|
case protoreflect.BytesKind:
|
|
p.P(fmt.Sprintf("f%d", len(saved)), ` := m.`, fieldName, `[:0]`)
|
|
saved = append(saved, field)
|
|
}
|
|
}
|
|
}
|
|
|
|
p.P(`m.Reset()`)
|
|
for i, field := range saved {
|
|
p.P(`m.`, field.GoName, ` = `, fmt.Sprintf("f%d", i))
|
|
}
|
|
p.P(`}`)
|
|
p.P(`}`)
|
|
|
|
p.P(`func (m *`, ccTypeName, `) ReturnToVTPool() {`)
|
|
p.P(`if m != nil {`)
|
|
p.P(`m.ResetVT()`)
|
|
p.P(`vtprotoPool_`, ccTypeName, `.Put(m)`)
|
|
p.P(`}`)
|
|
p.P(`}`)
|
|
|
|
p.P(`func `, ccTypeName, `FromVTPool() *`, ccTypeName, `{`)
|
|
p.P(`return vtprotoPool_`, ccTypeName, `.Get().(*`, ccTypeName, `)`)
|
|
p.P(`}`)
|
|
}
|