使用 JavaScript 递归搜索嵌套对象以匹配字符串列表_技术教程

本文介绍了如何使用 JavaScript 递归搜索嵌套对象，以查找与给定字符串列表匹配的键，并返回匹配的对象。我们将探讨使用生成器函数来实现这一目标，以及如何处理更复杂的需求，例如查找顺序键。

递归搜索嵌套对象

在处理复杂的数据结构时，经常需要根据某些条件查找特定的数据。当数据以嵌套对象的形式存在时，递归搜索是一种常用的方法。以下介绍如何使用 JavaScript 实现递归搜索，以查找与给定字符串列表匹配的键。

基本实现：使用生成器函数

首先，我们使用生成器函数 search1 来递归地搜索对象。该函数接受一个数据对象和一个搜索值作为参数。

function* search1(data, value) {
  if (Object(data) === data) {
    for (const key of Object.keys(data)) {
      if (key === value)
        yield data[key];
      else
        yield* search1(data[key], value);
    }
  }
}

这个函数会遍历对象的所有键，如果键与搜索值匹配，则返回该键对应的值。否则，它会递归地调用自身来搜索嵌套的对象。

接下来，我们可以创建一个 search 函数，它接受一个数据对象和一个搜索值列表作为参数。这个函数会遍历搜索值列表，并对每个值调用 search1 函数。

function* search(data, values) {
  for (const value of values)
    yield* search1(data, value);
}

示例

假设我们有以下数据结构：

const data = {
  "tabs-1": {
    "test 1": {
      "test 2": {
        "test 3a": {
          "tab1graph1": {
            "String a": "value a",
            "String b": "value b",
            "String c": "value c"
          }
        },
        "test 3b": {
          "tab1graph2": {
            "String a": "value a",
            "String b": "value b",
            "String c": "value c"
          }
        },
        "test 3c": {
          "tab1graph3": {
            "String a": "value a",
            "String b": "value b",
            "String c": "value c"
          }
        }
      }
    }
  }
};

const searchList = ["apple", "testx3", "test 3b", "test 3a"];

我们可以使用以下代码来搜索与 searchList 中的值匹配的对象：

for (const result of search(data, searchList))
  console.log(result);

这将输出以下结果：

{
  "tab1graph2": {
    "String a": "value a",
    "String b": "value b",
    "String c": "value c"
  }
}
{
  "tab1graph1": {
    "String a": "value a",
    "String b": "value b",
    "String c": "value c"
  }
}

使用谓词的高阶函数

可以使用高阶函数和谓词来使搜索更加灵活。以下是一个使用谓词的 search 函数的示例：

function* search(data, predicate) {
  if (Object(data) === data) {
    for (const entry of Object.entries(data)) {
      if (predicate(entry))
        yield entry[1];
      else
        yield* search(entry[1], predicate);
    }
  }
}

可以使用以下代码来搜索与 searchList 中的值匹配的对象：

Array.from(
  search(
    data,
    ([key, value]) => searchList.includes(key)
  )
)

搜索顺序键

有时，需要查找一系列顺序键。例如，可能需要查找 key1 后跟 key2 的对象。为了支持这一点，可以将单个 value 字段更改为 path，它表示一个字符串数组。

function* search1(data, path) {
  if (path.length === 0)
    yield data;
  else if (Object(data) === data) {
    for (const key of Object.keys(data)) {
      if (key === path[0])
        yield* search1(data[key], path.slice(1));
      else
        yield* search1(data[key], path);
    }
  }
}

然后，可以像这样调用 search 函数：

for (const result of search(data, [
  ["test 1", "test 2", "test 3a"], // test 1 -> test 2 -> test 3a
  ["test 2", "String a"] // test 2 -> String a
]) {
  console.log(result);
}