JavaScript中深度移除嵌套Avro Schema无keepThisField标记字段的方法

实现思路 #

核心逻辑是递归遍历每一层fields数组，过滤掉无keepThisField标记的字段，同时清理保留字段上的标记，遇到嵌套的record类型字段时，递归处理它内部的fields属性即可。

问题修正与完整代码 #

你现有代码存在几个可调整的点：

1. 判断条件o.keepThisField === true和你实际标记的字符串值"true"不匹配，会导致所有字段都被过滤
2. 嵌套record的fields判断逻辑有误，没有正确定位到type下的fields属性
3. 直接操作原对象可能导致原数据被污染，建议先做深拷贝

以下是可直接运行的实现代码：

const _ = require('lodash'); // 你已引入lodash可直接使用

function cleanMarkedSchema(markedSchema) {
  // 深拷贝避免修改原标记Schema
  const schema = _.cloneDeep(markedSchema);

  // 单字段处理函数
  function processField(field) {
    // 无保留标记直接返回空数组，flatMap会自动过滤该字段
    if (!field.keepThisField) return [];
    // 移除保留标记
    const processedField = _.omit(field, 'keepThisField');

    // 如果当前字段的类型是嵌套record，递归处理它的内部fields
    if (_.get(processedField.type, 'type') === 'record' && Array.isArray(processedField.type.fields)) {
      processedField.type.fields = _.flatMap(processedField.type.fields, processField);
    }

    return processedField;
  }

  // 处理根节点的fields
  schema.fields = _.flatMap(schema.fields, processField);
  return schema;
}

使用示例 #

// 传入你已经添加了keepThisField标记的Avro Schema即可得到裁剪后的结果
const reducedSchema = cleanMarkedSchema(yourMarkedSchema);

内容的提问来源于stack exchange，提问作者Sero Daniel