func (p *plugin) GenerateField(message *generator.Descriptor, field *descriptor.FieldDescriptorProto) { goTyp, _ := p.GoType(message, field) fieldname := p.GetFieldName(message, field) goTypName := generator.GoTypeToName(goTyp) ctype := "" if gogoproto.IsCustomType(field) { _, typ, err := generator.GetCustomType(field) if err != nil { panic(err) } ctype = typ } if field.IsMessage() || p.IsGroup(field) { funcName := "NewPopulated" + goTypName goTypNames := strings.Split(goTypName, ".") if len(goTypNames) == 2 { funcName = goTypNames[0] + ".NewPopulated" + goTypNames[1] } else if len(goTypNames) != 1 { panic(fmt.Errorf("unreachable: too many dots in %v", goTypName)) } funcCall := funcName + "(r, easy)" if field.IsRepeated() { p.P(p.varGen.Next(), ` := r.Intn(10)`) p.P(`this.`, fieldname, ` = make(`, goTyp, `, `, p.varGen.Current(), `)`) p.P(`for i := 0; i < `, p.varGen.Current(), `; i++ {`) p.In() p.P(p.varGen.Next(), `:= `, funcCall) p.P(`this.`, generator.SizerName(fieldname), ` += 1`) p.P(`this.`, fieldname, `[i] = `, p.varGen.Current()) p.Out() p.P(`}`) } else { p.P(p.varGen.Next(), `:= `, funcCall) p.P(`this.`, generator.SetterName(fieldname), ` = true`) p.P(`this.`, fieldname, ` = `, p.varGen.Current()) } } else { if field.IsEnum() { enum := p.ObjectNamed(field.GetTypeName()).(*generator.EnumDescriptor) l := len(enum.Value) values := make([]string, l) for i := range enum.Value { values[i] = strconv.Itoa(int(*enum.Value[i].Number)) } arr := "[]int32{" + strings.Join(values, ",") + "}" val := strings.Join([]string{generator.GoTypeToName(goTyp), `(`, arr, `[r.Intn(`, fmt.Sprintf("%d", l), `)])`}, "") if field.IsRepeated() { p.P(p.varGen.Next(), ` := r.Intn(10)`) p.P(`this.`, fieldname, ` = make(`, goTyp, `, `, p.varGen.Current(), `)`) p.P(`for i := 0; i < `, p.varGen.Current(), `; i++ {`) p.In() p.P(`this.`, generator.SizerName(fieldname), ` += 1`) p.P(`this.`, fieldname, `[i] = `, val) p.Out() p.P(`}`) } else { p.P(`this.`, generator.SetterName(fieldname), ` = true`) p.P(`this.`, fieldname, ` = `, val) } } else if field.IsString() { val := fmt.Sprintf("randString%v(r)", p.localName) if field.IsRepeated() { p.P(p.varGen.Next(), ` := r.Intn(10)`) p.P(`this.`, fieldname, ` = make(`, goTyp, `, `, p.varGen.Current(), `)`) p.P(`for i := 0; i < `, p.varGen.Current(), `; i++ {`) p.In() p.P(`this.`, generator.SizerName(fieldname), ` += 1`) p.P(`this.`, fieldname, `[i] = `, ctype, `(`, val, `)`) p.Out() p.P(`}`) } else { p.P(`this.`, generator.SetterName(fieldname), ` = true`) p.P(`this.`, fieldname, ` = `, ctype, `(`, val, `)`) } } else if field.IsBytes() { if field.IsRepeated() { p.P(p.varGen.Next(), ` := r.Intn(100)`) p.P(`this.`, fieldname, ` = make(`, goTyp, `, `, p.varGen.Current(), `)`) p.P(`for i := 0; i < `, p.varGen.Current(), `; i++ {`) p.In() p.P(p.varGen.Next(), ` := r.Intn(100)`) p.P(`this.`, generator.SizerName(fieldname), ` += 1`) p.P(`this.`, fieldname, `[i] = make([]byte,`, p.varGen.Current(), `)`) p.P(`for j := 0; j < `, p.varGen.Current(), `; j++ {`) p.In() p.P(`this.`, fieldname, `[i][j] = byte(r.Intn(256))`) p.Out() p.P(`}`) p.Out() p.P(`}`) } else { p.P(p.varGen.Next(), ` := r.Intn(100)`) p.P(`this.`, fieldname, ` = make(`, goTyp, `, `, p.varGen.Current(), `)`) p.P(`for i := 0; i < `, p.varGen.Current(), `; i++ {`) p.In() p.P(`this.`, generator.SetterName(fieldname), ` = true`) p.P(`this.`, fieldname, `[i] = byte(r.Intn(256))`) p.Out() p.P(`}`) } } else { if field.IsRepeated() { p.P(p.varGen.Next(), ` := r.Intn(100)`) p.P(`this.`, fieldname, ` = make(`, goTyp, `, `, p.varGen.Current(), `)`) p.P(`for i := 0; i < `, p.varGen.Current(), `; i++ {`) p.In() p.P(`this.`, generator.SizerName(fieldname), ` += 1`) p.P(`this.`, fieldname, `[i] = `, ctype, `(`, value(field), `)`) if negative(field) { p.P(`if r.Intn(2) == 0 {`) p.In() p.P(`this.`, fieldname, `[i] *= `, ctype, `(-1)`) p.Out() p.P(`}`) } p.Out() p.P(`}`) } else { p.P(`this.`, generator.SetterName(fieldname), ` = true`) p.P(`this.`, fieldname, ` = `, ctype, `(`, value(field), `)`) if negative(field) { p.P(`if r.Intn(2) == 0 {`) p.In() p.P(`this.`, fieldname, ` *= `, ctype, `(-1)`) p.Out() p.P(`}`) } } } } }