Esempio n. 1
0
func patchMsg(input, patch *data.DynamicMessage, indent int, sheetName string) bool {

	return patch.IterateFieldDesc(func(fd *pbmeta.FieldDescriptor) bool {

		if fd.Type() == pbprotos.FieldDescriptorProto_TYPE_MESSAGE {

			if fd.IsRepeated() {

				// 顶级的消息肯定是repeated的, 因此直接进入
				if indent == 0 {

					nextInputMsgArray := input.GetRepeatedMessage(fd)
					nextPatchMsgArray := patch.GetRepeatedMessage(fd)

					if len(nextInputMsgArray) != len(nextPatchMsgArray) {
						log.Errorf("patch rows diff from input msg")
						return false
					}

					// 相同的行数才可以进去继续迭代
					for index, nextInputMsg := range nextInputMsgArray {

						patchMsg(nextInputMsg, nextPatchMsgArray[index], indent+1, sheetName)
					}

				} else {

					// 直接替换值数组
					pathMsgArray := patch.GetRepeatedMessage(fd)

					input.ClearFieldValue(fd)
					for _, msg := range pathMsgArray {
						input.AddRepeatedMessage(fd, msg)
					}

				}

			} else {

				nextInputMsg := input.GetMessage(fd)
				nextPatchMsg := patch.GetMessage(fd)

				patchMsg(nextInputMsg, nextPatchMsg, indent+1, sheetName)
			}

		} else {

			if fd.IsRepeated() {

				log.Infof("patch repeated field: '%s' = '%s'", fd.Name(), patch.GetRepeatedValue(fd))

				// 直接用patch的覆盖多值
				input.ClearFieldValue(fd)
				for _, v := range patch.GetRepeatedValue(fd) {

					input.AddRepeatedValue(fd, v)
				}

			} else {

				v, _ := patch.GetValue(fd)

				log.Infof("patch field: '%s' = '%s'", fd.Name(), v)

				// 单值设置
				input.SetValue(fd, v)

			}

		}

		return true
	})

	return true
}
Esempio n. 2
0
File: pbt.go Progetto: jwk000/tabtoy
func writeString(msg *data.DynamicMessage, indent int) string {
	var out bytes.Buffer

	for i := 0; i < msg.Desc.FieldCount(); i++ {
		fd := msg.Desc.Field(i)

		if i > 0 {
			out.WriteString(" ")
		}

		if fd.Type() == pbprotos.FieldDescriptorProto_TYPE_MESSAGE {

			if fd.IsRepeated() {

				// 多结构
				if msgList := msg.GetRepeatedMessage(fd); msgList != nil {

					for _, msg := range msgList {

						msgStr := makeMsg(msg, indent+1)

						if msgStr != "" {

							out.WriteString(fmt.Sprintf("%s %s", fd.Name(), msgStr))

							if indent == 0 {
								out.WriteString("\n")
							} else {
								out.WriteString(" ")
							}
						}

					}

					continue
				}

			} else {
				// 单结构
				if msg := msg.GetMessage(fd); msg != nil {
					msgStr := makeMsg(msg, indent+1)
					if msgStr != "" {
						out.WriteString(fmt.Sprintf("%s %s", fd.Name(), msgStr))
					}

					continue
				}
			}

		} else {

			// 多值
			if fd.IsRepeated() {

				if valuelist := msg.GetRepeatedValue(fd); valuelist != nil {
					for vIndex, value := range valuelist {

						if vIndex > 0 {
							out.WriteString(" ")
						}

						out.WriteString(makeValue(fd, value))

					}

					continue
				}

			} else {

				// 单值
				if value, ok := msg.GetValue(fd); ok {
					out.WriteString(makeValue(fd, value))
					continue
				}

			}

		}

	}

	return out.String()

}