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
}
