本文探讨了如何在typescript中为一个特定的数组实例添加自定义函数，如`findbyid`和`findbyname`，以替代重复的`array.prototype.find`调用。通过使用`object.assign()`和类型交叉，我们可以优雅地扩展数组实例的功能，提高代码的可读性和复用性，同时详细讨论了处理潜在`undefined`返回值的重要性。
在TypeScript开发中，我们经常会遇到需要对特定数据数组进行重复查找的场景。例如，有一个包含对象数组的数据集，我们需要根据对象的id或name属性频繁地查找其中的元素。虽然Array.prototype.find方法可以完成这项任务，但当查找逻辑在代码库中多处重复出现时，会导致代码冗余且可读性下降。本文将介绍一种在不修改原生Array.prototype的前提下，为特定数组实例添加自定义查找方法（如findById、findByName）的有效方法。
问题场景
假设我们有一个定义了A类型对象的数组：
type A = {
id: number;
name: string;
};
const data: A[] = [
{ id: 1, name: 'Foo' },
{ id: 2, name: 'Bar' },
// ... 更多数据
];
在实际应用中，我们可能需要频繁地进行如下操作：
const entryById = data.find((item) => item.id === 1);
if (entryById) {
console.log(entryById.name);
}
const entryByName = data.find((item) => item.name === 'Foo');
if (entryByName) {
console.log(entryByName.id);
}
这种重复的.find调用及其内部的匿名函数，使得代码不够简洁，尤其是在不同的模块中需要执行相同的查找逻辑时。我们期望能够像下面这样调用：
data.findById(1)?.name;
data.findByName('Foo')?.id;
解决方案：使用 Object.assign() 扩展数组实例
TypeScript允许我们通过类型交叉（Intersection Types）来描述一个既是数组又包含额外属性或方法的对象。结合Object.assign()方法，我们可以在运行时将自定义方法附加到数组实例上。
1. 定义数据类型和扩展类型
首先，定义我们的基础数据类型A。然后，我们需要创建一个新的类型，它既是A[]，又包含了我们希望添加的自定义方法。
type A = {
id: number;
name: string;
};
// 定义扩展后的数组类型
type ExtendedAArray = A[] & {
findById: (id: number) => A | undefined;
findByName: (name: string) => A | undefined;
};
这里，ExtendedAArray是一个类型交叉，它表示一个既是A类型数组，又拥有findById和findByName这两个方法的对象。注意，findById和findByName方法的返回值类型被定义为A | undefined，这是因为Array.prototype.find在找不到匹配项时会返回undefined。
2. 使用 Object.assign() 构造扩展数组
Object.assign()方法用于将所有可枚举的自有属性从一个或多个源对象复制到目标对象。在本例中，目标对象就是我们的A类型数组实例。
在 Java 中初始化和管理阿里云 SDK客户端。包括单例模式、线程安全、endpoint 与 region 配置、VPC 终端节点、同步与异步等。
const initialData: A[] = [
{ id: 1, name: 'Foo' },
{ id: 2, name: 'Bar' },
{ id: 3, name: 'Baz' },
];
// 使用 Object.assign() 构造扩展数组实例
const data: ExtendedAArray = Object.assign(initialData, {
findById: function (this: ExtendedAArray, id: number): A | undefined {
// 注意：这里使用 this 来引用当前 data 数组实例
// 或者直接引用外部的 data 变量（如果 data 已经声明）
return this.find((item) => id === item.id);
},
findByName: function (this: ExtendedAArray, name: string): A | undefined {
return this.find((item) => name === item.name);
},
});
代码解析：
- 我们将原始的A[]数组作为Object.assign()的第一个参数（目标对象）。
- 第二个参数是一个对象字面量，其中包含了我们希望添加的自定义方法findById和findByName。
- 在这些方法内部，我们使用this关键字来引用当前被扩展的数组实例，从而能够调用其原生的find方法。
- 最终，Object.assign()返回的data对象，就同时具备了数组的所有原生功能和我们自定义的查找方法，并且其类型被正确推断为ExtendedAArray。
3. 使用示例
现在，我们可以以更简洁、更具表达力的方式调用这些自定义方法：
console.log(data.findById(1)?.name); // 输出: Foo
console.log(data.findByName('Foo')?.id); // 输出: 1
// 当找不到匹配项时，方法会返回 undefined
console.log(data.findById(99)); // 输出: undefined
console.log(data.findById(99)?.name); // 输出: undefined (安全访问)
注意事项与最佳实践
-
处理 undefined 返回值： Array.prototype.find在找不到匹配项时会返回undefined。因此，我们的自定义方法也应该返回A | undefined。在使用这些方法的返回值时，务必进行空值检查，例如使用可选链操作符（?.）或空值合并操作符（??），或者显式地进行if (value)检查。
const foundItem = data.findById(100); if (foundItem) { console.log(foundItem.name); } else { console.log('Item not found.'); } -
错误处理： 如果你希望在找不到元素时抛出错误而不是返回undefined，可以修改自定义方法的实现：
type ExtendedAArrayWithThrow = A[] & { findById: (id: number) => A; // 返回类型变为 A }; const dataWithThrow: ExtendedAArrayWithThrow = Object.assign(initialData, { findById: function (this: ExtendedAArrayWithThrow, id: number): A { const item = this.find((i) => id === i.id); if (item === undefined) { throw new Error(`Item with id ${id} not found.`); } return item; }, }); try { console.log(dataWithThrow.findById(1)?.name); console.log(dataWithThrow.findById(99).name); // 抛出错误 } catch (e: any) { console.error(e.message); // 输出: Item with id 99 not found. }这种方式在某些场景下可以简化后续代码，但需要调用者捕获异常。
this 的绑定： 在自定义方法中使用function关键字定义时，this会根据调用上下文正确绑定到data实例。如果使用箭头函数，this会 Lexically 绑定到定义时的上下文，这可能不是你想要的。在本例中，使用function关键字是推荐的做法。
避免修改原型链： 这种方法仅扩展了特定的数组实例，而不是全局的Array.prototype。修改全局原型链通常被认为是不良实践，因为它可能导致命名冲突和不可预期的副作用。当前方法完美规避了这一风险。
总结
通过利用TypeScript的类型交叉和JavaScript的Object.assign()方法，我们能够优雅地为特定的数组实例添加自定义功能，如便捷的查找方法。这种模式提高了代码的可读性和可维护性，将重复的查找逻辑封装起来，同时保持了类型安全。在设计API或处理特定数据集时，这种扩展数组实例的能力是非常实用的技巧。
