用Flow写同步异步代码结构清晰

2023-10-30 01:47:01

问题背景

朋友最近在面试蚂蚁金服，遇到了一道面试题，乍一看感觉挺简单的，但是实现起来发现内部值得一提的点还是挺多的。题目是这样的：

flow 是指一系列 effects 组成的逻辑片段。 flow 支持嵌套。 effects 的执行只需要支持串行。 用数组包裹的上述三项。 先把参数浅拷贝一份（编写库函数，将数组展开，再用 浅拷贝赋值 ），作为第一项流的数组参数。然后把新的第一个参数和数组后面的其他参数用递归的方式嵌套在一个 flow 中。 嵌套 flow 的每个最内层数组中的第一个参数和第二个参数都是上级流返回的结果。

听起来有点绕口，我把它整理一下，流程是这样的：

把参数浅拷贝一份。
把参数的第一项和数组后面的其他参数递归地嵌套在一个 flow 中。
嵌套 flow 的每个最内层数组中的第一个参数和第二个参数都是上级流返回的结果。
效果是嵌套 flow 的最内层数组中的最后一个参数返回的结果。

比如，给定一个数组 [1, 2, 3, 4, 5]，这个算法返回的结果是 5。

实现

function flat(arr) {
  return arr.reduce((acc, cur) => acc.concat(cur), []);
}

function flow(args) {
  let [first, ...rest] = args;
  let res = first;
  while (rest.length > 0) {
    res = first(...res, rest[0]);
    [first, ...rest] = rest;
  }
  return res;
}

function myFlow(...args) {
  let params = JSON.parse(JSON.stringify(args));
  let first = params[0];
  let rest = flat(params.slice(1));
  return flow([first, rest]);
}

console.log(myFlow(
  (arg1) => arg1 + 1,
  (arg1, arg2) => arg1 + arg2,
  (arg1, arg2, arg3) => arg1 + arg2 + arg3,
  (arg1, arg2, arg3, arg4) => arg1 + arg2 + arg3 + arg4,
  (arg1, arg2, arg3, arg4, arg5) => arg1 + arg2 + arg3 + arg4 + arg5
)); // 15