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 new(rootPackage string, rootMessage string, descSet *descriptor.FileDescriptorSet, path string) (*fdesc, error) { fieldpaths := strings.Split(path, ".") keys := make([]uint64, len(fieldpaths)) fields := make([]*descriptor.FieldDescriptorProto, len(fieldpaths)) curPackage := rootPackage curMessage := rootMessage last := len(fieldpaths) - 1 var fieldDesc *descriptor.FieldDescriptorProto for i, f := range fieldpaths { fieldName := f fieldDesc = descSet.GetField(curPackage, curMessage, fieldName) if fieldDesc == nil { curPackage, fieldDesc = descSet.FindExtension(curPackage, curMessage, fieldName) if fieldDesc == nil { return nil, &errChild{fieldName: fieldName, pkg: curPackage, msg: curMessage} } typeNames := strings.Split(fieldDesc.GetTypeName(), ".") curMessage = fieldDesc.GetTypeName() if len(typeNames) > 1 { curPackage = typeNames[1] curMessage = typeNames[2] } fieldKey := fieldDesc.GetKeyUint64() keys[i] = fieldKey fields[i] = fieldDesc } else { fieldKey := fieldDesc.GetKeyUint64() if fieldDesc.IsMessage() { curPackage, curMessage = descSet.FindMessage(curPackage, curMessage, fieldName) } else if i != last { return nil, &errMessage{fieldName} } keys[i] = fieldKey fields[i] = fieldDesc } } fd := &fdesc{curPackage, curMessage, fields, fieldDesc, keys, 0} if fieldDesc.GetType() == descriptor.FieldDescriptorProto_TYPE_ENUM { typeNames := strings.Split(fieldDesc.GetTypeName(), ".") enumMessage := fieldDesc.GetTypeName() enumPackage := curPackage if len(typeNames) > 1 { enumPackage = typeNames[1] enumMessage = typeNames[2] } enum := descSet.GetEnum(enumPackage, enumMessage) if enum == nil { return nil, &errChild{fieldName: fieldDesc.GetName(), pkg: enumPackage, msg: enumMessage} } for _, v := range enum.GetValue() { if v.GetNumber() < fd.firstEnumValue { fd.firstEnumValue = v.GetNumber() } } } return fd, nil }
func (g *Generator) IsGroup(field *descriptor.FieldDescriptorProto) bool { if d, ok := g.typeNameToObject[field.GetTypeName()].(*Descriptor); ok { return d.IsGroup() } return false }
func (p *unmarshal) field(field *descriptor.FieldDescriptorProto, fieldname string) { repeated := field.IsRepeated() nullable := gogoproto.IsNullable(field) switch *field.Type { case descriptor.FieldDescriptorProto_TYPE_DOUBLE: if !p.unsafe { if repeated { p.P(`var v uint64`) p.decodeFixed64("v", "uint64") p.P(`v2 := `, p.mathPkg.Use(), `.Float64frombits(v)`) p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, v2)`) } else if nullable { p.P(`var v uint64`) p.decodeFixed64("v", "uint64") p.P(`v2 := `, p.mathPkg.Use(), `.Float64frombits(v)`) p.P(`m.`, fieldname, ` = &v2`) } else { p.P(`var v uint64`) p.decodeFixed64("v", "uint64") p.P(`m.`, fieldname, ` = `, p.mathPkg.Use(), `.Float64frombits(v)`) } } else { if repeated { p.P(`var v float64`) p.unsafeFixed64("v", "float64") p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, v)`) } else if nullable { p.P(`var v float64`) p.unsafeFixed64("v", "float64") p.P(`m.`, fieldname, ` = &v`) } else { p.unsafeFixed64(`m.`+fieldname, "float64") } } case descriptor.FieldDescriptorProto_TYPE_FLOAT: if !p.unsafe { if repeated { p.P(`var v uint32`) p.decodeFixed32("v", "uint32") p.P(`v2 := `, p.mathPkg.Use(), `.Float32frombits(v)`) p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, v2)`) } else if nullable { p.P(`var v uint32`) p.decodeFixed32("v", "uint32") p.P(`v2 := `, p.mathPkg.Use(), `.Float32frombits(v)`) p.P(`m.`, fieldname, ` = &v2`) } else { p.P(`var v uint32`) p.decodeFixed32("v", "uint32") p.P(`m.`, fieldname, ` = `, p.mathPkg.Use(), `.Float32frombits(v)`) } } else { if repeated { p.P(`var v float32`) p.unsafeFixed32("v", "float32") p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, v)`) } else if nullable { p.P(`var v float32`) p.unsafeFixed32("v", "float32") p.P(`m.`, fieldname, ` = &v`) } else { p.unsafeFixed32("m."+fieldname, "float32") } } case descriptor.FieldDescriptorProto_TYPE_INT64: if repeated { p.P(`var v int64`) p.decodeVarint("v", "int64") p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, v)`) } else if nullable { p.P(`var v int64`) p.decodeVarint("v", "int64") p.P(`m.`, fieldname, ` = &v`) } else { p.decodeVarint("m."+fieldname, "int64") } case descriptor.FieldDescriptorProto_TYPE_UINT64: if repeated { p.P(`var v uint64`) p.decodeVarint("v", "uint64") p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, v)`) } else if nullable { p.P(`var v uint64`) p.decodeVarint("v", "uint64") p.P(`m.`, fieldname, ` = &v`) } else { p.decodeVarint("m."+fieldname, "uint64") } case descriptor.FieldDescriptorProto_TYPE_INT32: if repeated { p.P(`var v int32`) p.decodeVarint("v", "int32") p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, v)`) } else if nullable { p.P(`var v int32`) p.decodeVarint("v", "int32") p.P(`m.`, fieldname, ` = &v`) } else { p.decodeVarint("m."+fieldname, "int32") } case descriptor.FieldDescriptorProto_TYPE_FIXED64: if !p.unsafe { if repeated { p.P(`var v uint64`) p.decodeFixed64("v", "uint64") p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, v)`) } else if nullable { p.P(`var v uint64`) p.decodeFixed64("v", "uint64") p.P(`m.`, fieldname, ` = &v`) } else { p.decodeFixed64("m."+fieldname, "uint64") } } else { if repeated { p.P(`var v uint64`) p.unsafeFixed64("v", "uint64") p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, v)`) } else if nullable { p.P(`var v uint64`) p.unsafeFixed64("v", "uint64") p.P(`m.`, fieldname, ` = &v`) } else { p.unsafeFixed64("m."+fieldname, "uint64") } } case descriptor.FieldDescriptorProto_TYPE_FIXED32: if !p.unsafe { if repeated { p.P(`var v uint32`) p.decodeFixed32("v", "uint32") p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, v)`) } else if nullable { p.P(`var v uint32`) p.decodeFixed32("v", "uint32") p.P(`m.`, fieldname, ` = &v`) } else { p.decodeFixed32("m."+fieldname, "uint32") } } else { if repeated { p.P(`var v uint32`) p.unsafeFixed32("v", "uint32") p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, v)`) } else if nullable { p.P(`var v uint32`) p.unsafeFixed32("v", "uint32") p.P(`m.`, fieldname, ` = &v`) } else { p.unsafeFixed32("m."+fieldname, "uint32") } } case descriptor.FieldDescriptorProto_TYPE_BOOL: if repeated { p.P(`var v int`) p.decodeVarint("v", "int") p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, bool(v != 0))`) } else if nullable { p.P(`var v int`) p.decodeVarint("v", "int") p.P(`b := bool(v != 0)`) p.P(`m.`, fieldname, ` = &b`) } else { p.P(`var v int`) p.decodeVarint("v", "int") p.P(`m.`, fieldname, ` = bool(v != 0)`) } case descriptor.FieldDescriptorProto_TYPE_STRING: p.P(`var stringLen uint64`) p.decodeVarint("stringLen", "uint64") p.P(`postIndex := index + int(stringLen)`) p.P(`if postIndex > l {`) p.In() p.P(`return `, p.ioPkg.Use(), `.ErrUnexpectedEOF`) p.Out() p.P(`}`) if repeated { p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, string(data[index:postIndex]))`) } else if nullable { p.P(`s := string(data[index:postIndex])`) p.P(`m.`, fieldname, ` = &s`) } else { p.P(`m.`, fieldname, ` = string(data[index:postIndex])`) } p.P(`index = postIndex`) case descriptor.FieldDescriptorProto_TYPE_GROUP: panic(fmt.Errorf("unmarshaler does not support group %v", fieldname)) case descriptor.FieldDescriptorProto_TYPE_MESSAGE: desc := p.ObjectNamed(field.GetTypeName()) msgname := p.TypeName(desc) p.P(`var msglen int`) p.decodeVarint("msglen", "int") p.P(`postIndex := index + msglen`) p.P(`if postIndex > l {`) p.In() p.P(`return `, p.ioPkg.Use(), `.ErrUnexpectedEOF`) p.Out() p.P(`}`) if repeated { if nullable { p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, &`, msgname, `{})`) } else { p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, `, msgname, `{})`) } p.P(`m.`, fieldname, `[len(m.`, fieldname, `)-1].Unmarshal(data[index:postIndex])`) } else if nullable { p.P(`if m.`, fieldname, ` == nil {`) p.In() p.P(`m.`, fieldname, ` = &`, msgname, `{}`) p.Out() p.P(`}`) p.P(`if err := m.`, fieldname, `.Unmarshal(data[index:postIndex]); err != nil {`) p.In() p.P(`return err`) p.Out() p.P(`}`) } else { p.P(`if err := m.`, fieldname, `.Unmarshal(data[index:postIndex]); err != nil {`) p.In() p.P(`return err`) p.Out() p.P(`}`) } p.P(`index = postIndex`) case descriptor.FieldDescriptorProto_TYPE_BYTES: p.P(`var byteLen int`) p.decodeVarint("byteLen", "int") p.P(`postIndex := index + byteLen`) p.P(`if postIndex > l {`) p.In() p.P(`return `, p.ioPkg.Use(), `.ErrUnexpectedEOF`) p.Out() p.P(`}`) if !gogoproto.IsCustomType(field) { if repeated { p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, make([]byte, postIndex-index))`) p.P(`copy(m.`, fieldname, `[len(m.`, fieldname, `)-1], data[index:postIndex])`) } else if nullable { p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, data[index:postIndex]...)`) } else { p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, data[index:postIndex]...)`) } } else { _, ctyp, err := generator.GetCustomType(field) if err != nil { panic(err) } if repeated { p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, `, ctyp, `{})`) p.P(`m.`, fieldname, `[len(m.`, fieldname, `)-1].Unmarshal(data[index:postIndex])`) } else if nullable { p.P(`m.`, fieldname, ` = &`, ctyp, `{}`) p.P(`if err := m.`, fieldname, `.Unmarshal(data[index:postIndex]); err != nil {`) p.In() p.P(`return err`) p.Out() p.P(`}`) } else { p.P(`if err := m.`, fieldname, `.Unmarshal(data[index:postIndex]); err != nil {`) p.In() p.P(`return err`) p.Out() p.P(`}`) } } p.P(`index = postIndex`) case descriptor.FieldDescriptorProto_TYPE_UINT32: if repeated { p.P(`var v uint32`) p.decodeVarint("v", "uint32") p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, v)`) } else if nullable { p.P(`var v uint32`) p.decodeVarint("v", "uint32") p.P(`m.`, fieldname, ` = &v`) } else { p.decodeVarint("m."+fieldname, "uint32") } case descriptor.FieldDescriptorProto_TYPE_ENUM: typName := p.TypeName(p.ObjectNamed(field.GetTypeName())) if repeated { p.P(`var v `, typName) p.decodeVarint("v", typName) p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, v)`) } else if nullable { p.P(`var v `, typName) p.decodeVarint("v", typName) p.P(`m.`, fieldname, ` = &v`) } else { p.decodeVarint("m."+fieldname, typName) } case descriptor.FieldDescriptorProto_TYPE_SFIXED32: if !p.unsafe { if repeated { p.P(`var v int32`) p.decodeFixed32("v", "int32") p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, v)`) } else if nullable { p.P(`var v int32`) p.decodeFixed32("v", "int32") p.P(`m.`, fieldname, ` = &v`) } else { p.decodeFixed32("m."+fieldname, "int32") } } else { if repeated { p.P(`var v int32`) p.unsafeFixed32("v", "int32") p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, v)`) } else if nullable { p.P(`var v int32`) p.unsafeFixed32("v", "int32") p.P(`m.`, fieldname, ` = &v`) } else { p.unsafeFixed32("m."+fieldname, "int32") } } case descriptor.FieldDescriptorProto_TYPE_SFIXED64: if !p.unsafe { if repeated { p.P(`var v int64`) p.decodeFixed64("v", "int64") p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, v)`) } else if nullable { p.P(`var v int64`) p.decodeFixed64("v", "int64") p.P(`m.`, fieldname, ` = &v`) } else { p.decodeFixed64("m."+fieldname, "int64") } } else { if repeated { p.P(`var v int64`) p.unsafeFixed64("v", "int64") p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, v)`) } else if nullable { p.P(`var v int64`) p.unsafeFixed64("v", "int64") p.P(`m.`, fieldname, ` = &v`) } else { p.unsafeFixed64("m."+fieldname, "int64") } } case descriptor.FieldDescriptorProto_TYPE_SINT32: p.P(`var v int32`) p.decodeVarint("v", "int32") p.P(`v = int32((uint32(v) >> 1) ^ uint32(((v&1)<<31)>>31))`) if repeated { p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, v)`) } else if nullable { p.P(`m.`, fieldname, ` = &v`) } else { p.P(`m.`, fieldname, ` = v`) } case descriptor.FieldDescriptorProto_TYPE_SINT64: p.P(`var v uint64`) p.decodeVarint("v", "uint64") p.P(`v = (v >> 1) ^ uint64((int64(v&1)<<63)>>63)`) if repeated { p.P(`m.`, fieldname, ` = append(m.`, fieldname, `, int64(v))`) } else if nullable { p.P(`v2 := int64(v)`) p.P(`m.`, fieldname, ` = &v2`) } else { p.P(`m.`, fieldname, ` = int64(v)`) } default: panic("not implemented") } }
func (p *plugin) GenerateField(message *generator.Descriptor, field *descriptor.FieldDescriptorProto) { goTyp, _ := p.GoType(message, field) fieldname := generator.CamelCase(*field.Name) if field.IsMessage() && gogoproto.IsEmbed(field) { fieldname = generator.EmbedFieldName(goTyp) } goTypName := generator.GoTypeToName(goTyp) 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() if gogoproto.IsNullable(field) { p.P(`this.`, fieldname, `[i] = `, funcCall) } else { p.P(p.varGen.Next(), `:= `, funcCall) p.P(`this.`, fieldname, `[i] = *`, p.varGen.Current()) } p.Out() p.P(`}`) } else { if gogoproto.IsNullable(field) { p.P(`this.`, fieldname, ` = `, funcCall) } else { p.P(p.varGen.Next(), `:= `, funcCall) 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.`, fieldname, `[i] = `, val) p.Out() p.P(`}`) } else if gogoproto.IsNullable(field) { p.P(p.varGen.Next(), ` := `, val) p.P(`this.`, fieldname, ` = &`, p.varGen.Current()) } else { p.P(`this.`, fieldname, ` = `, val) } } else if gogoproto.IsCustomType(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)" 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.`, fieldname, `[i] = *`, p.varGen.Current()) p.Out() p.P(`}`) } else if gogoproto.IsNullable(field) { p.P(`this.`, fieldname, ` = `, funcCall) } else { p.P(p.varGen.Next(), `:= `, funcCall) p.P(`this.`, fieldname, ` = *`, p.varGen.Current()) } } 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.`, 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.`, fieldname, `[i] = byte(r.Intn(256))`) p.Out() p.P(`}`) } } 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.`, fieldname, `[i] = `, val) p.Out() p.P(`}`) } else if gogoproto.IsNullable(field) { p.P(p.varGen.Next(), `:= `, val) p.P(`this.`, fieldname, ` = &`, p.varGen.Current()) } else { p.P(`this.`, fieldname, ` = `, val) } } else { typName := generator.GoTypeToName(goTyp) 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.`, fieldname, `[i] = `, value(typName)) p.Out() p.P(`}`) } else if gogoproto.IsNullable(field) { p.P(p.varGen.Next(), ` := `, value(typName)) p.P(`this.`, fieldname, ` = &`, p.varGen.Current()) } else { p.P(`this.`, fieldname, ` = `, value(typName)) } } } }