func (p *plugin) hasLoop(field *descriptor.FieldDescriptorProto, visited []*generator.Descriptor, excludes []*generator.Descriptor) *generator.Descriptor { if field.IsMessage() || p.IsGroup(field) { fieldMessage := p.ObjectNamed(field.GetTypeName()).(*generator.Descriptor) fieldTypeName := generator.CamelCaseSlice(fieldMessage.TypeName()) for _, message := range visited { messageTypeName := generator.CamelCaseSlice(message.TypeName()) if fieldTypeName == messageTypeName { for _, e := range excludes { if fieldTypeName == generator.CamelCaseSlice(e.TypeName()) { return nil } } return fieldMessage } } for _, f := range fieldMessage.Field { visited = append(visited, fieldMessage) loopTo := p.hasLoop(f, visited, excludes) if loopTo != nil { return loopTo } } } return nil }
func (p *plugin) checkNameSpace(message *generator.Descriptor) map[string]bool { ccTypeName := generator.CamelCaseSlice(message.TypeName()) names := make(map[string]bool) for _, field := range message.Field { fieldname := generator.CamelCase(*field.Name) if gogoproto.IsEmbed(field) { desc := p.ObjectNamed(field.GetTypeName()) moreNames := p.checkNameSpace(desc.(*generator.Descriptor)) for another := range moreNames { if names[another] { fmt.Fprintf(os.Stderr, "ERROR: duplicate embedded fieldname %v in type %v\n", fieldname, ccTypeName) os.Exit(1) } names[another] = true } } else { if names[fieldname] { fmt.Fprintf(os.Stderr, "ERROR: duplicate embedded fieldname %v in type %v\n", fieldname, ccTypeName) os.Exit(1) } names[fieldname] = true } } return names }
func (p *test) Generate(imports generator.PluginImports, file *generator.FileDescriptor) bool { used := false randPkg := imports.NewImport("math/rand") timePkg := imports.NewImport("time") testingPkg := imports.NewImport("testing") for _, message := range file.Messages() { if !gogoproto.IsUnion(file.FileDescriptorProto, message.DescriptorProto) || !gogoproto.HasTestGen(file.FileDescriptorProto, message.DescriptorProto) { continue } used = true ccTypeName := generator.CamelCaseSlice(message.TypeName()) p.P(`func Test`, ccTypeName, `OnlyOne(t *`, testingPkg.Use(), `.T) {`) p.In() p.P(`popr := `, randPkg.Use(), `.New(`, randPkg.Use(), `.NewSource(`, timePkg.Use(), `.Now().UnixNano()))`) p.P(`p := NewPopulated`, ccTypeName, `(popr, true)`) p.P(`v := p.GetValue()`) p.P(`msg := &`, ccTypeName, `{}`) p.P(`if !msg.SetValue(v) {`) p.In() p.P(`t.Fatalf("OnlyOne: Could not set Value")`) p.Out() p.P(`}`) p.P(`if !p.Equal(msg) {`) p.In() p.P(`t.Fatalf("%#v !OnlyOne Equal %#v", msg, p)`) p.Out() p.P(`}`) p.Out() p.P(`}`) } return used }
func (p *test) Generate(imports generator.PluginImports, file *generator.FileDescriptor) bool { used := false randPkg := imports.NewImport("math/rand") timePkg := imports.NewImport("time") testingPkg := imports.NewImport("testing") fmtPkg := imports.NewImport("fmt") for _, message := range file.Messages() { ccTypeName := generator.CamelCaseSlice(message.TypeName()) if !gogoproto.IsStringer(file.FileDescriptorProto, message.DescriptorProto) { continue } if gogoproto.HasTestGen(file.FileDescriptorProto, message.DescriptorProto) { used = true p.P(`func Test`, ccTypeName, `Stringer(t *`, testingPkg.Use(), `.T) {`) p.In() p.P(`popr := `, randPkg.Use(), `.New(`, randPkg.Use(), `.NewSource(`, timePkg.Use(), `.Now().UnixNano()))`) p.P(`p := NewPopulated`, ccTypeName, `(popr, false)`) p.P(`s1 := p.String()`) p.P(`s2 := `, fmtPkg.Use(), `.Sprintf("%v", p)`) p.P(`if s1 != s2 {`) p.In() p.P(`t.Fatalf("String want %v got %v", s1, s2)`) p.Out() p.P(`}`) p.Out() p.P(`}`) } } return used }
func (p *enumstringer) Generate(file *generator.FileDescriptor) { p.PluginImports = generator.NewPluginImports(p.Generator) p.atleastOne = false p.localName = generator.FileName(file) strconvPkg := p.NewImport("strconv") for _, enum := range file.Enums() { if !gogoproto.IsEnumStringer(file.FileDescriptorProto, enum.EnumDescriptorProto) { continue } if gogoproto.IsGoEnumStringer(file.FileDescriptorProto, enum.EnumDescriptorProto) { panic("old enum string method needs to be disabled, please use gogoproto.old_enum_stringer or gogoproto.old_enum_string_all and set it to false") } p.atleastOne = true ccTypeName := generator.CamelCaseSlice(enum.TypeName()) p.P("func (x ", ccTypeName, ") String() string {") p.In() p.P(`s, ok := `, ccTypeName, `_name[int32(x)]`) p.P(`if ok {`) p.In() p.P(`return s`) p.Out() p.P(`}`) p.P(`return `, strconvPkg.Use(), `.Itoa(int(x))`) p.Out() p.P(`}`) } if !p.atleastOne { return } }
func (p *plugin) checkRepeated(message *generator.Descriptor) { ccTypeName := generator.CamelCaseSlice(message.TypeName()) for _, field := range message.Field { if !gogoproto.IsEmbed(field) { continue } if !field.IsRepeated() { continue } fieldname := generator.CamelCase(*field.Name) fmt.Fprintf(os.Stderr, "ERROR: found repeated embedded field %s in message %s\n", fieldname, ccTypeName) os.Exit(1) } }
func (p *test) Generate(imports generator.PluginImports, file *generator.FileDescriptor) bool { used := false randPkg := imports.NewImport("math/rand") timePkg := imports.NewImport("time") testingPkg := imports.NewImport("testing") protoPkg := imports.NewImport("github.com/dropbox/goprotoc/proto") for _, message := range file.Messages() { ccTypeName := generator.CamelCaseSlice(message.TypeName()) if !gogoproto.HasVerboseEqual(file.FileDescriptorProto, message.DescriptorProto) { continue } if gogoproto.HasTestGen(file.FileDescriptorProto, message.DescriptorProto) { used = true p.P(`func Test`, ccTypeName, `VerboseEqual(t *`, testingPkg.Use(), `.T) {`) p.In() p.P(`popr := `, randPkg.Use(), `.New(`, randPkg.Use(), `.NewSource(`, timePkg.Use(), `.Now().UnixNano()))`) p.P(`p := NewPopulated`, ccTypeName, `(popr, false)`) p.P(`data, err := `, protoPkg.Use(), `.Marshal(p)`) p.P(`if err != nil {`) p.In() p.P(`panic(err)`) p.Out() p.P(`}`) p.P(`msg := &`, ccTypeName, `{}`) p.P(`if err := `, protoPkg.Use(), `.Unmarshal(data, msg); err != nil {`) p.In() p.P(`panic(err)`) p.Out() p.P(`}`) p.P(`if err := p.VerboseEqual(msg); err != nil {`) p.In() p.P(`t.Fatalf("%#v !VerboseEqual %#v, since %v", msg, p, err)`) p.Out() p.P(`}`) p.Out() p.P(`}`) } } return used }
func (p *plugin) checkOverwrite(message *generator.Descriptor, enablers map[string]gogoproto.EnableFunc) { ccTypeName := generator.CamelCaseSlice(message.TypeName()) names := []string{} for name := range enablers { names = append(names, name) } for _, field := range message.Field { if gogoproto.IsEmbed(field) { fieldname := generator.CamelCase(*field.Name) desc := p.ObjectNamed(field.GetTypeName()) msg := desc.(*generator.Descriptor) for errStr, enabled := range enablers { if enabled(msg.File(), msg.DescriptorProto) { fmt.Fprintf(os.Stderr, "WARNING: found non-%v %v with embedded %v %v\n", names, ccTypeName, errStr, fieldname) } } p.checkOverwrite(msg, enablers) } } }
func (p *plugin) Generate(file *generator.FileDescriptor) { p.PluginImports = generator.NewPluginImports(p.Generator) protoPkg := p.NewImport("github.com/dropbox/goprotoc/proto") for _, message := range file.Messages() { if !gogoproto.IsFace(file.FileDescriptorProto, message.DescriptorProto) { continue } if message.DescriptorProto.HasExtension() { panic("face does not support message with extensions") } ccTypeName := generator.CamelCaseSlice(message.TypeName()) p.P(`type `, ccTypeName, `Face interface{`) p.In() p.P(`Proto() `, protoPkg.Use(), `.Message`) for _, field := range message.Field { fieldname := p.GetFieldName(message, field) goTyp, _ := p.GoType(message, field) p.P(`Get`, fieldname, `() `, goTyp) } p.Out() p.P(`}`) } }
func (p *plugin) Generate(file *generator.FileDescriptor) { p.used = false localName := generator.FileName(file) for _, message := range file.Messages() { if !gogoproto.HasDescription(file.FileDescriptorProto, message.DescriptorProto) { continue } p.used = true ccTypeName := generator.CamelCaseSlice(message.TypeName()) p.P(`func (this *`, ccTypeName, `) Description() (desc *google_protobuf.FileDescriptorSet) {`) p.In() p.P(`return `, localName, `Description()`) p.Out() p.P(`}`) } if p.used { p.P(`func `, localName, `Description() (desc *google_protobuf.FileDescriptorSet) {`) p.In() //Don't generate SourceCodeInfo, since it will create too much code. ss := make([]*descriptor.SourceCodeInfo, 0) for _, f := range p.Generator.AllFiles().GetFile() { ss = append(ss, f.SourceCodeInfo) f.SourceCodeInfo = nil } s := fmt.Sprintf("%#v", p.Generator.AllFiles()) for i, f := range p.Generator.AllFiles().GetFile() { f.SourceCodeInfo = ss[i] } p.P(`return `, s) p.Out() p.P(`}`) } }
func (p *union) Generate(file *generator.FileDescriptor) { p.PluginImports = generator.NewPluginImports(p.Generator) for _, message := range file.Messages() { if !gogoproto.IsUnion(file.FileDescriptorProto, message.DescriptorProto) { continue } if message.DescriptorProto.HasExtension() { panic("onlyone does not currently support extensions") } ccTypeName := generator.CamelCaseSlice(message.TypeName()) p.P(`func (this *`, ccTypeName, `) GetValue() interface{} {`) p.In() for _, field := range message.Field { fieldname := p.GetFieldName(message, field) if fieldname == "Value" { panic("cannot have a onlyone message " + ccTypeName + " with a field named Value") } p.P(`if this.`, generator.SetterName(fieldname), ` == true {`) p.In() p.P(`return this.`, fieldname) p.Out() p.P(`}`) } p.P(`return nil`) p.Out() p.P(`}`) p.P(``) p.P(`func (this *`, ccTypeName, `) SetValue(value interface{}) bool {`) p.In() p.P(`switch vt := value.(type) {`) p.In() for _, field := range message.Field { fieldname := p.GetFieldName(message, field) goTyp, _ := p.GoType(message, field) p.P(`case `, goTyp, `:`) p.In() p.P(`this.`, generator.SetterName(fieldname), ` = true`) p.P(`this.`, fieldname, ` = vt`) p.Out() } p.P(`default:`) p.In() for _, field := range message.Field { fieldname := p.GetFieldName(message, field) if field.IsMessage() { goTyp, _ := p.GoType(message, field) obj := p.ObjectNamed(field.GetTypeName()).(*generator.Descriptor) if gogoproto.IsUnion(obj.File(), obj.DescriptorProto) { p.P(`this.`, fieldname, ` = new(`, generator.GoTypeToName(goTyp), `)`) p.P(`if set := this.`, fieldname, `.SetValue(value); set {`) p.In() p.P(`return true`) p.Out() p.P(`}`) p.P(`this.`, fieldname, ` = nil`) } } } p.P(`return false`) p.Out() p.P(`}`) p.P(`return true`) p.Out() p.P(`}`) } }
func (p *stringer) Generate(file *generator.FileDescriptor) { p.PluginImports = generator.NewPluginImports(p.Generator) p.atleastOne = false p.localName = generator.FileName(file) stringsPkg := p.NewImport("strings") for _, message := range file.Messages() { if !gogoproto.IsStringer(file.FileDescriptorProto, message.DescriptorProto) { continue } p.atleastOne = true ccTypeName := generator.CamelCaseSlice(message.TypeName()) p.P(`func (this *`, ccTypeName, `) String() string {`) p.In() p.P(`if this == nil {`) p.In() p.P(`return "nil"`) p.Out() p.P(`}`) p.P("s := ", stringsPkg.Use(), ".Join([]string{`&", ccTypeName, "{`,") for _, field := range message.Field { fieldname := p.GetFieldName(message, field) if field.IsMessage() || p.IsGroup(field) { desc := p.ObjectNamed(field.GetTypeName()) msgname := p.TypeName(desc) msgnames := strings.Split(msgname, ".") typeName := msgnames[len(msgnames)-1] if field.IsRepeated() { p.P("`", fieldname, ":`", ` + `, stringsPkg.Use(), `.Replace(`, p.Pkg["fmt"], `.Sprintf("%v", this.`, fieldname, `[:this.`, generator.SizerName(fieldname), `]), "`, typeName, `","`, msgname, `"`, ", 1) + `,", "`,") } else { fieldValue := "this.Get" + generator.CamelCase(fieldname) + "()" if gogoproto.IsCustomType(field) || gogoproto.IsEmbed(field) { fieldValue = "this." + fieldname } p.P("`", fieldname, ":`", ` + `, stringsPkg.Use(), `.Replace(`, p.Pkg["fmt"], `.Sprintf("%v", `, fieldValue, `), "`, typeName, `","`, msgname, `"`, ", 1) + `,", "`,") } } else if field.IsRepeated() { p.P("`", fieldname, ":`", ` + `, p.Pkg["fmt"], `.Sprintf("%v", this.`, fieldname, "[:this.", generator.SizerName(fieldname), "]) + `,", "`,") } else { fieldValue := "this.Get" + generator.CamelCase(fieldname) + "()" if gogoproto.IsCustomType(field) || gogoproto.IsEmbed(field) { fieldValue = "this." + fieldname } p.P("`", fieldname, ":`", ` + `, p.Pkg["fmt"], `.Sprintf("%v", `, fieldValue, ") + `,", "`,") } } if message.DescriptorProto.HasExtension() { if gogoproto.HasExtensionsMap(file.FileDescriptorProto, message.DescriptorProto) { p.P("`XXX_extensions:` + proto.StringFromExtensionsMap(this.XXX_extensions) + `,`,") } else { p.P("`XXX_extensions:` + proto.StringFromExtensionsBytes(this.XXX_extensions) + `,`,") } } p.P("`XXX_unrecognized:` + ", p.Pkg["fmt"], `.Sprintf("%v", this.XXX_unrecognized) + `, "`,`,") p.P("`}`,") p.P(`}`, `,""`, ")") p.P(`return s`) p.Out() p.P(`}`) } if !p.atleastOne { return } }
func (p *testAPI) Generate(imports generator.PluginImports, file *generator.FileDescriptor) bool { used := false testingPkg := imports.NewImport("testing") randPkg := imports.NewImport("math/rand") timePkg := imports.NewImport("time") for _, message := range file.Messages() { ccTypeName := generator.CamelCaseSlice(message.TypeName()) if !gogoproto.HasTestGen(file.FileDescriptorProto, message.DescriptorProto) { continue } used = true p.P(`func Test`, ccTypeName, `API(t *`, testingPkg.Use(), `.T) {`) p.In() p.P(`popr := `, randPkg.Use(), `.New(`, randPkg.Use(), `.NewSource(`, timePkg.Use(), `.Now().UnixNano()))`) p.P(`p := NewPopulated`, ccTypeName, `(popr, false)`) p.P(`msg := &`, ccTypeName, `{}`) p.P(`if !apiEmpty`, ccTypeName, `(msg, t) {`) p.In() p.P(`t.Fatalf("`, ccTypeName, ` should be empty")`) p.Out() p.P(`}`) p.P(`apiCopy`, ccTypeName, `(msg, p, t)`) if gogoproto.HasVerboseEqual(file.FileDescriptorProto, message.DescriptorProto) { p.P(`if err := p.VerboseEqual(msg); err != nil {`) p.In() p.P(`t.Fatalf("%#v !VerboseEqual %#v, since %v", msg, p, err)`) p.Out() p.P(`}`) } p.P(`if apiEmpty`, ccTypeName, `(p, t) != apiEmpty`, ccTypeName, `(msg, t) {`) p.In() p.P(`t.Fatalf("`, ccTypeName, ` should not be empty")`) p.Out() p.P(`}`) p.P(`if !p.Equal(msg) {`) p.In() p.P(`t.Fatalf("%#v !Proto %#v", msg, p)`) p.Out() p.P(`}`) p.P(`msg.Clear()`) p.P(`if !apiEmpty`, ccTypeName, `(msg, t) {`) p.In() p.P(`t.Fatalf("`, ccTypeName, ` should be empty")`) p.Out() p.P(`}`) p.Out() p.P(`}`) p.P() p.P(`func apiCopy`, ccTypeName, `(dst *`, ccTypeName, `, src *`, ccTypeName, `, t *`, testingPkg.Use(), `.T) {`) p.In() p.P(`if dst == nil || src == nil {`) p.In() p.P(`t.Fatalf("Cannot copy to(%v) or from(%v) nil message", dst, src)`) p.Out() p.P(`}`) for _, field := range message.Field { if gogoproto.IsEmbed(field) { p.P(`t.Skip("Cannot copy embed field")`) break } if gogoproto.IsCustomType(field) { p.P(`t.Skip("Cannot copy costum field")`) break } fieldName := generator.CamelCase(p.GetFieldName(message, field)) fieldType, _ := p.GoType(message, field) fieldTypeName := generator.GoTypeToName(fieldType) if generator.IsRepeated(field) { p.P(`for i := 0; i < src.`, fieldName, `Size(); i++ {`) p.In() if generator.IsMessageType(field) { p.P(`src`, fieldName, `, _ := src.Get`, fieldName, `(i)`) p.P(`dst`, fieldName, `, _ := dst.Add`, fieldName, `()`) p.P(`apiCopy`, fieldTypeName, `(dst`, fieldName, `, src`, fieldName, `, t)`) } else { p.P(`value, _ := src.Get`, fieldName, `(i)`) p.P(`dst.Add`, fieldName, `(value)`) } p.Out() p.P(`}`) } else { p.P(`if src.Has`, fieldName, `() {`) p.In() if generator.IsMessageType(field) { p.P(`src`, fieldName, ` := src.Get`, fieldName, `()`) p.P(`dst`, fieldName, `, _ := dst.Mutate`, fieldName, `()`) p.P(`apiCopy`, fieldTypeName, `(dst`, fieldName, `, src`, fieldName, `, t)`) } else { p.P(`dst.Set`, fieldName, `(src.Get`, fieldName, `())`) } p.Out() p.P(`}`) } } p.P(`src.XXX_unrecognized = dst.XXX_unrecognized`) if len(message.ExtensionRange) > 0 { p.P(`src.XXX_extensions = dst.XXX_extensions`) } p.Out() p.P(`}`) p.P() p.P(`func apiEmpty`, ccTypeName, `(msg *`, ccTypeName, `, t *`, testingPkg.Use(), `.T) bool {`) p.In() p.P(`if msg == nil {`) p.In() p.P(`return true`) p.Out() p.P(`}`) for _, field := range message.Field { if gogoproto.IsEmbed(field) { p.P(`t.Skip("Cannot check embed field")`) break } if gogoproto.IsCustomType(field) { p.P(`t.Skip("Cannot check costum field")`) break } fieldName := generator.CamelCase(p.GetFieldName(message, field)) if generator.IsRepeated(field) { p.P(`if msg.`, fieldName, `Size() != 0 {`) } else { p.P(`if msg.Has`, fieldName, `() {`) } p.In() p.P(`return false`) p.Out() p.P(`}`) } p.P(`return true`) p.Out() p.P(`}`) p.P() } return used }
func (p *testProto) Generate(imports generator.PluginImports, file *generator.FileDescriptor) bool { used := false testingPkg := imports.NewImport("testing") randPkg := imports.NewImport("math/rand") timePkg := imports.NewImport("time") protoPkg := imports.NewImport("github.com/dropbox/goprotoc/proto") for _, message := range file.Messages() { ccTypeName := generator.CamelCaseSlice(message.TypeName()) if gogoproto.HasTestGen(file.FileDescriptorProto, message.DescriptorProto) { used = true p.P(`func Test`, ccTypeName, `Proto(t *`, testingPkg.Use(), `.T) {`) p.In() p.P(`popr := `, randPkg.Use(), `.New(`, randPkg.Use(), `.NewSource(`, timePkg.Use(), `.Now().UnixNano()))`) p.P(`p := NewPopulated`, ccTypeName, `(popr, false)`) p.P(`data, err := `, protoPkg.Use(), `.Marshal(p)`) p.P(`if err != nil {`) p.In() p.P(`panic(err)`) p.Out() p.P(`}`) p.P(`msg := &`, ccTypeName, `{}`) p.P(`if err := `, protoPkg.Use(), `.Unmarshal(data, msg); err != nil {`) p.In() p.P(`panic(err)`) p.Out() p.P(`}`) p.P(`for i := range data {`) p.In() p.P(`data[i] = byte(popr.Intn(256))`) p.Out() p.P(`}`) if gogoproto.HasVerboseEqual(file.FileDescriptorProto, message.DescriptorProto) { p.P(`if err := p.VerboseEqual(msg); err != nil {`) p.In() p.P(`t.Fatalf("%#v !VerboseProto %#v, since %v", msg, p, err)`) p.Out() p.P(`}`) } p.P(`if !p.Equal(msg) {`) p.In() p.P(`t.Fatalf("%#v !Proto %#v", msg, p)`) p.Out() p.P(`}`) p.Out() p.P(`}`) p.P() } if gogoproto.HasTestGen(file.FileDescriptorProto, message.DescriptorProto) { if gogoproto.IsMarshaler(file.FileDescriptorProto, message.DescriptorProto) { p.P(`func Test`, ccTypeName, `MarshalTo(t *`, testingPkg.Use(), `.T) {`) p.In() p.P(`popr := `, randPkg.Use(), `.New(`, randPkg.Use(), `.NewSource(`, timePkg.Use(), `.Now().UnixNano()))`) p.P(`p := NewPopulated`, ccTypeName, `(popr, false)`) p.P(`size := p.Size()`) p.P(`data := make([]byte, size)`) p.P(`for i := range data {`) p.In() p.P(`data[i] = byte(popr.Intn(256))`) p.Out() p.P(`}`) p.P(`_, err := p.MarshalTo(data)`) p.P(`if err != nil {`) p.In() p.P(`panic(err)`) p.Out() p.P(`}`) p.P(`msg := &`, ccTypeName, `{}`) p.P(`if err := `, protoPkg.Use(), `.Unmarshal(data, msg); err != nil {`) p.In() p.P(`panic(err)`) p.Out() p.P(`}`) p.P(`for i := range data {`) p.In() p.P(`data[i] = byte(popr.Intn(256))`) p.Out() p.P(`}`) if gogoproto.HasVerboseEqual(file.FileDescriptorProto, message.DescriptorProto) { p.P(`if err := p.VerboseEqual(msg); err != nil {`) p.In() p.P(`t.Fatalf("%#v !VerboseProto %#v, since %v", msg, p, err)`) p.Out() p.P(`}`) } p.P(`if !p.Equal(msg) {`) p.In() p.P(`t.Fatalf("%#v !Proto %#v", msg, p)`) p.Out() p.P(`}`) p.Out() p.P(`}`) p.P() } } if gogoproto.HasBenchGen(file.FileDescriptorProto, message.DescriptorProto) { used = true p.P(`func Benchmark`, ccTypeName, `ProtoMarshal(b *`, testingPkg.Use(), `.B) {`) p.In() p.P(`popr := `, randPkg.Use(), `.New(`, randPkg.Use(), `.NewSource(616))`) p.P(`total := 0`) p.P(`pops := make([]*`, ccTypeName, `, 10000)`) p.P(`for i := 0; i < 10000; i++ {`) p.In() p.P(`pops[i] = NewPopulated`, ccTypeName, `(popr, false)`) p.Out() p.P(`}`) p.P(`b.ResetTimer()`) p.P(`for i := 0; i < b.N; i++ {`) p.In() p.P(`data, err := `, protoPkg.Use(), `.Marshal(pops[i%10000])`) p.P(`if err != nil {`) p.In() p.P(`panic(err)`) p.Out() p.P(`}`) p.P(`total += len(data)`) p.Out() p.P(`}`) p.P(`b.SetBytes(int64(total / b.N))`) p.Out() p.P(`}`) p.P() p.P(`func Benchmark`, ccTypeName, `ProtoUnmarshal(b *`, testingPkg.Use(), `.B) {`) p.In() p.P(`popr := `, randPkg.Use(), `.New(`, randPkg.Use(), `.NewSource(616))`) p.P(`total := 0`) p.P(`datas := make([][]byte, 10000)`) p.P(`for i := 0; i < 10000; i++ {`) p.In() p.P(`data, err := `, protoPkg.Use(), `.Marshal(NewPopulated`, ccTypeName, `(popr, false))`) p.P(`if err != nil {`) p.In() p.P(`panic(err)`) p.Out() p.P(`}`) p.P(`datas[i] = data`) p.Out() p.P(`}`) p.P(`msg := &`, ccTypeName, `{}`) p.P(`b.ResetTimer()`) p.P(`for i := 0; i < b.N; i++ {`) p.In() p.P(`total += len(datas[i%10000])`) p.P(`if err := `, protoPkg.Use(), `.Unmarshal(datas[i%10000], msg); err != nil {`) p.In() p.P(`panic(err)`) p.Out() p.P(`}`) p.Out() p.P(`}`) p.P(`b.SetBytes(int64(total / b.N))`) p.Out() p.P(`}`) p.P() } } return used }
func (p *plugin) Generate(file *generator.FileDescriptor) { p.atleastOne = false p.PluginImports = generator.NewPluginImports(p.Generator) p.varGen = NewVarGen() p.localName = generator.FileName(file) protoPkg := p.NewImport("github.com/dropbox/goprotoc/proto") for _, message := range file.Messages() { if !gogoproto.HasPopulate(file.FileDescriptorProto, message.DescriptorProto) { continue } p.atleastOne = true ccTypeName := generator.CamelCaseSlice(message.TypeName()) p.P(`func NewPopulated`, ccTypeName, `(r randy`, p.localName, `, easy bool) *`, ccTypeName, ` {`) p.In() p.P(`this := &`, ccTypeName, `{}`) if gogoproto.IsUnion(message.File(), message.DescriptorProto) && len(message.Field) > 0 { loopLevels := make([]int, len(message.Field)) maxLoopLevel := 0 for i, field := range message.Field { loopLevels[i] = p.loops(field, message) if loopLevels[i] > maxLoopLevel { maxLoopLevel = loopLevels[i] } } ran := 0 for i := range loopLevels { ran += int(math.Pow10(maxLoopLevel - loopLevels[i])) } p.P(`fieldNum := r.Intn(`, fmt.Sprintf("%d", ran), `)`) p.P(`switch fieldNum {`) k := 0 for i, field := range message.Field { is := []string{} ran := int(math.Pow10(maxLoopLevel - loopLevels[i])) for j := 0; j < ran; j++ { is = append(is, fmt.Sprintf("%d", j+k)) } k += ran p.P(`case `, strings.Join(is, ","), `:`) p.In() p.GenerateField(message, field) p.Out() } p.P(`}`) } else { var maxFieldNumber int32 for _, field := range message.Field { if field.IsRequired() || !field.IsRepeated() { p.GenerateField(message, field) } else { p.P(`if r.Intn(10) != 0 {`) p.In() p.GenerateField(message, field) p.Out() p.P(`}`) } if field.GetNumber() > maxFieldNumber { maxFieldNumber = field.GetNumber() } } if message.DescriptorProto.HasExtension() { p.P(`if !easy && r.Intn(10) != 0 {`) p.In() p.P(`l := r.Intn(5)`) p.P(`for i := 0; i < l; i++ {`) p.In() if len(message.DescriptorProto.GetExtensionRange()) > 1 { p.P(`eIndex := r.Intn(`, strconv.Itoa(len(message.DescriptorProto.GetExtensionRange())), `)`) p.P(`fieldNumber := 0`) p.P(`switch eIndex {`) for i, e := range message.DescriptorProto.GetExtensionRange() { p.P(`case `, strconv.Itoa(i), `:`) p.In() p.P(`fieldNumber = r.Intn(`, strconv.Itoa(int(e.GetEnd()-e.GetStart())), `) + `, strconv.Itoa(int(e.GetStart()))) p.Out() if e.GetEnd() > maxFieldNumber { maxFieldNumber = e.GetEnd() } } p.P(`}`) } else { e := message.DescriptorProto.GetExtensionRange()[0] p.P(`fieldNumber := r.Intn(`, strconv.Itoa(int(e.GetEnd()-e.GetStart())), `) + `, strconv.Itoa(int(e.GetStart()))) if e.GetEnd() > maxFieldNumber { maxFieldNumber = e.GetEnd() } } p.P(`wire := r.Intn(4)`) p.P(`if wire == 3 { wire = 5 }`) p.P(`data := randField`, p.localName, `(nil, r, fieldNumber, wire)`) p.P(protoPkg.Use(), `.SetRawExtension(this, int32(fieldNumber), data)`) p.Out() p.P(`}`) p.Out() p.P(`}`) } if maxFieldNumber < (1 << 10) { p.P(`if !easy && r.Intn(10) != 0 {`) p.In() p.P(`this.XXX_unrecognized = randUnrecognized`, p.localName, `(r, `, strconv.Itoa(int(maxFieldNumber+1)), `)`) p.Out() p.P(`}`) } } p.P(`return this`) p.Out() p.P(`}`) p.P(``) } if !p.atleastOne { return } p.P(`type randy`, p.localName, ` interface {`) p.In() p.P(`Float32() float32`) p.P(`Float64() float64`) p.P(`Int63() int64`) p.P(`Int31() int32`) p.P(`Uint32() uint32`) p.P(`Intn(n int) int`) p.Out() p.P(`}`) surrogateRange := surrogateMax - surrogateMin maxRand := maxRune - surrogateRange p.P(`func randUTF8Rune`, p.localName, `(r randy`, p.localName, `) rune {`) p.In() p.P(`res := rune(r.Uint32() % `, fmt.Sprintf("%d", maxRand), `)`) p.P(`if `, fmt.Sprintf("%d", surrogateMin), ` <= res {`) p.In() p.P(`res += `, fmt.Sprintf("%d", surrogateRange)) p.Out() p.P(`}`) p.P(`return res`) p.Out() p.P(`}`) p.P(`func randString`, p.localName, `(r randy`, p.localName, `) string {`) p.In() p.P(p.varGen.Next(), ` := r.Intn(100)`) p.P(`tmps := make([]rune, `, p.varGen.Current(), `)`) p.P(`for i := 0; i < `, p.varGen.Current(), `; i++ {`) p.In() p.P(`tmps[i] = randUTF8Rune`, p.localName, `(r)`) p.Out() p.P(`}`) p.P(`return string(tmps)`) p.Out() p.P(`}`) p.P(`func randUnrecognized`, p.localName, `(r randy`, p.localName, `, maxFieldNumber int) (data []byte) {`) p.In() p.P(`l := r.Intn(5)`) p.P(`for i := 0; i < l; i++ {`) p.In() p.P(`wire := r.Intn(4)`) p.P(`if wire == 3 { wire = 5 }`) p.P(`fieldNumber := maxFieldNumber + r.Intn(100)`) p.P(`data = randField`, p.localName, `(data, r, fieldNumber, wire)`) p.Out() p.P(`}`) p.P(`return data`) p.Out() p.P(`}`) p.P(`func randField`, p.localName, `(data []byte, r randy`, p.localName, `, fieldNumber int, wire int) []byte {`) p.In() p.P(`key := uint32(fieldNumber)<<3 | uint32(wire)`) p.P(`switch wire {`) p.P(`case 0:`) p.In() p.P(`data = encodeVarintPopulate`, p.localName, `(data, uint64(key))`) p.P(p.varGen.Next(), ` := r.Int63()`) p.P(`if r.Intn(2) == 0 {`) p.In() p.P(p.varGen.Current(), ` *= -1`) p.Out() p.P(`}`) p.P(`data = encodeVarintPopulate`, p.localName, `(data, uint64(`, p.varGen.Current(), `))`) p.Out() p.P(`case 1:`) p.In() p.P(`data = encodeVarintPopulate`, p.localName, `(data, uint64(key))`) p.P(`data = append(data, byte(r.Intn(256)), byte(r.Intn(256)), byte(r.Intn(256)), byte(r.Intn(256)), byte(r.Intn(256)), byte(r.Intn(256)), byte(r.Intn(256)), byte(r.Intn(256)))`) p.Out() p.P(`case 2:`) p.In() p.P(`data = encodeVarintPopulate`, p.localName, `(data, uint64(key))`) p.P(`ll := r.Intn(100)`) p.P(`data = encodeVarintPopulate`, p.localName, `(data, uint64(ll))`) p.P(`for j := 0; j < ll; j++ {`) p.In() p.P(`data = append(data, byte(r.Intn(256)))`) p.Out() p.P(`}`) p.Out() p.P(`default:`) p.In() p.P(`data = encodeVarintPopulate`, p.localName, `(data, uint64(key))`) p.P(`data = append(data, byte(r.Intn(256)), byte(r.Intn(256)), byte(r.Intn(256)), byte(r.Intn(256)))`) p.Out() p.P(`}`) p.P(`return data`) p.Out() p.P(`}`) p.P(`func encodeVarintPopulate`, p.localName, `(data []byte, v uint64) []byte {`) p.In() p.P(`for v >= 1<<7 {`) p.In() p.P(`data = append(data, uint8(uint64(v)&0x7f|0x80))`) p.P(`v >>= 7`) p.Out() p.P(`}`) p.P(`data = append(data, uint8(v))`) p.P(`return data`) p.Out() p.P(`}`) }
func (p *test) Generate(imports generator.PluginImports, file *generator.FileDescriptor) bool { used := false randPkg := imports.NewImport("math/rand") timePkg := imports.NewImport("time") testingPkg := imports.NewImport("testing") protoPkg := imports.NewImport("github.com/dropbox/goprotoc/proto") for _, message := range file.Messages() { ccTypeName := generator.CamelCaseSlice(message.TypeName()) if !gogoproto.IsSizer(file.FileDescriptorProto, message.DescriptorProto) { continue } if gogoproto.HasTestGen(file.FileDescriptorProto, message.DescriptorProto) { used = true p.P(`func Test`, ccTypeName, `Size(t *`, testingPkg.Use(), `.T) {`) p.In() p.P(`popr := `, randPkg.Use(), `.New(`, randPkg.Use(), `.NewSource(`, timePkg.Use(), `.Now().UnixNano()))`) p.P(`p := NewPopulated`, ccTypeName, `(popr, true)`) p.P(`size2 := `, protoPkg.Use(), `.Size(p)`) p.P(`data, err := `, protoPkg.Use(), `.Marshal(p)`) p.P(`if err != nil {`) p.In() p.P(`panic(err)`) p.Out() p.P(`}`) p.P(`size := p.Size()`) p.P(`if len(data) != size {`) p.In() p.P(`t.Fatalf("size %v != marshalled size %v", size, len(data))`) p.Out() p.P(`}`) p.P(`if size2 != size {`) p.In() p.P(`t.Fatalf("size %v != before marshal proto.Size %v", size, size2)`) p.Out() p.P(`}`) p.P(`size3 := `, protoPkg.Use(), `.Size(p)`) p.P(`if size3 != size {`) p.In() p.P(`t.Fatalf("size %v != after marshal proto.Size %v", size, size3)`) p.Out() p.P(`}`) p.Out() p.P(`}`) p.P() } if gogoproto.HasBenchGen(file.FileDescriptorProto, message.DescriptorProto) { used = true p.P(`func Benchmark`, ccTypeName, `Size(b *`, testingPkg.Use(), `.B) {`) p.In() p.P(`popr := `, randPkg.Use(), `.New(`, randPkg.Use(), `.NewSource(616))`) p.P(`total := 0`) p.P(`pops := make([]*`, ccTypeName, `, 1000)`) p.P(`for i := 0; i < 1000; i++ {`) p.In() p.P(`pops[i] = NewPopulated`, ccTypeName, `(popr, false)`) p.Out() p.P(`}`) p.P(`b.ResetTimer()`) p.P(`for i := 0; i < b.N; i++ {`) p.In() p.P(`total += pops[i%1000].Size()`) p.Out() p.P(`}`) p.P(`b.SetBytes(int64(total / b.N))`) p.Out() p.P(`}`) p.P() } } return used }
func (p *plugin) generateMessage(message *generator.Descriptor, verbose bool, hasExtensionsMap bool) { ccTypeName := generator.CamelCaseSlice(message.TypeName()) if verbose { p.P(`func (this *`, ccTypeName, `) VerboseEqual(that interface{}) error {`) } else { p.P(`func (this *`, ccTypeName, `) Equal(that interface{}) bool {`) } p.In() p.P(`if that == nil {`) p.In() p.P(`if this == nil {`) p.In() if verbose { p.P(`return nil`) } else { p.P(`return true`) } p.Out() p.P(`}`) if verbose { p.P(`return `, p.Pkg["fmt"], `.Errorf("that == nil && this != nil")`) } else { p.P(`return false`) } p.Out() p.P(`}`) p.P(``) p.P(`that1, ok := that.(*`, ccTypeName, `)`) p.P(`if !ok {`) p.In() if verbose { p.P(`return `, p.Pkg["fmt"], `.Errorf("that is not of type *`, ccTypeName, `")`) } else { p.P(`return false`) } p.Out() p.P(`}`) p.P(`if that1 == nil {`) p.In() p.P(`if this == nil {`) p.In() if verbose { p.P(`return nil`) } else { p.P(`return true`) } p.Out() p.P(`}`) if verbose { p.P(`return `, p.Pkg["fmt"], `.Errorf("that is type *`, ccTypeName, ` but is nil && this != nil")`) } else { p.P(`return false`) } p.Out() p.P(`} else if this == nil {`) p.In() if verbose { p.P(`return `, p.Pkg["fmt"], `.Errorf("that is type *`, ccTypeName, `but is not nil && this == nil")`) } else { p.P(`return false`) } p.Out() p.P(`}`) for _, field := range message.Field { fieldname := p.GetFieldName(message, field) repeated := field.IsRepeated() if repeated { p.P(`if this.`, generator.SizerName(fieldname), ` != that1.`, generator.SizerName(fieldname), ` {`) } else { p.P(`if this.`, generator.SetterName(fieldname), ` != that1.`, generator.SetterName(fieldname), ` {`) } p.In() if verbose { p.P(`return `, p.Pkg["fmt"], `.Errorf("that.`, fieldname, ` is not equal to this.`, fieldname, `")`) } else { p.P(`return false`) } p.Out() p.P(`}`) if !repeated { if field.IsMessage() || p.IsGroup(field) { p.P(`if this.`, generator.SetterName(fieldname), ` && !this.`, fieldname, `.Equal(that1.`, fieldname, `) {`) } else if field.IsBytes() { p.P(`if this.`, generator.SetterName(fieldname), ` && !`, p.bytesPkg.Use(), `.Equal(this.`, fieldname, `, that1.`, fieldname, `) {`) } else { p.P(`if this.`, generator.SetterName(fieldname), ` && this.`, fieldname, ` != that1.`, fieldname, `{`) } p.In() if verbose { p.P(`return `, p.Pkg["fmt"], `.Errorf("`, fieldname, ` this(%v) Not Equal that(%v)", this.`, fieldname, `, that1.`, fieldname, `)`) } else { p.P(`return false`) } p.Out() p.P(`}`) } else { p.P(`for i := 0; i < this.`, generator.SizerName(fieldname), `; i++ {`) p.In() if field.IsMessage() || p.IsGroup(field) { p.P(`if !this.`, fieldname, `[i].Equal(that1.`, fieldname, `[i]) {`) } else if field.IsBytes() { p.P(`if !`, p.bytesPkg.Use(), `.Equal(this.`, fieldname, `[i], that1.`, fieldname, `[i]) {`) } else { p.P(`if this.`, fieldname, `[i] != that1.`, fieldname, `[i] {`) } p.In() if verbose { p.P(`return `, p.Pkg["fmt"], `.Errorf("`, fieldname, ` this[%v](%v) Not Equal that[%v](%v)", i, this.`, fieldname, `[i], i, that1.`, fieldname, `[i])`) } else { p.P(`return false`) } p.Out() p.P(`}`) p.Out() p.P(`}`) } } if message.DescriptorProto.HasExtension() { fieldname := "XXX_extensions" if hasExtensionsMap { p.P(`for k, v := range this.`, fieldname, ` {`) p.In() p.P(`if v2, ok := that1.`, fieldname, `[k]; ok {`) p.In() p.P(`if !v.Equal(&v2) {`) p.In() if verbose { p.P(`return `, p.Pkg["fmt"], `.Errorf("`, fieldname, ` this[%v](%v) Not Equal that[%v](%v)", k, this.`, fieldname, `[k], k, that1.`, fieldname, `[k])`) } else { p.P(`return false`) } p.Out() p.P(`}`) p.Out() p.P(`} else {`) p.In() if verbose { p.P(`return `, p.Pkg["fmt"], `.Errorf("`, fieldname, `[%v] Not In that", k)`) } else { p.P(`return false`) } p.Out() p.P(`}`) p.Out() p.P(`}`) p.P(`for k, _ := range that1.`, fieldname, ` {`) p.In() p.P(`if _, ok := this.`, fieldname, `[k]; !ok {`) p.In() if verbose { p.P(`return `, p.Pkg["fmt"], `.Errorf("`, fieldname, `[%v] Not In this", k)`) } else { p.P(`return false`) } p.Out() p.P(`}`) p.Out() p.P(`}`) } else { p.P(`if !`, p.bytesPkg.Use(), `.Equal(this.`, fieldname, `, that1.`, fieldname, `) {`) p.In() if verbose { p.P(`return `, p.Pkg["fmt"], `.Errorf("`, fieldname, ` this(%v) Not Equal that(%v)", this.`, fieldname, `, that1.`, fieldname, `)`) } else { p.P(`return false`) } p.Out() p.P(`}`) } } fieldname := "XXX_unrecognized" p.P(`if !`, p.bytesPkg.Use(), `.Equal(this.`, fieldname, `, that1.`, fieldname, `) {`) p.In() if verbose { p.P(`return `, p.Pkg["fmt"], `.Errorf("`, fieldname, ` this(%v) Not Equal that(%v)", this.`, fieldname, `, that1.`, fieldname, `)`) } else { p.P(`return false`) } p.Out() p.P(`}`) if verbose { p.P(`return nil`) } else { p.P(`return true`) } p.Out() p.P(`}`) }