本文深入探讨了如何在typescript中创建高度类型安全的函数，该函数接收一个对象和一个键，但仅允许选择那些对应值为特定类型（例如字符串）的键。通过引入自定义工具类型`keysoftype`，文章详细解释了如何利用映射类型和条件类型来精确约束键的类型，从而在编译时捕获错误，并提升ide的代码补全体验，最终实现更健壮、更易用的api设计。
在TypeScript开发中，我们经常需要编写通用函数来处理对象及其属性。然而，有时我们希望函数只接受特定类型的属性键，例如，只允许访问值为字符串的属性，或只允许访问值为布尔值的属性。直接使用 keyof T 这样的泛型约束，虽然可以确保传入的键是对象 T 的有效属性，但它无法进一步约束这些键所对应的值的类型。这可能导致运行时错误，并且在开发过程中无法提供精确的类型提示。
挑战：泛型键与值类型的精确约束
考虑以下场景，我们希望创建一个 extractStringValue 函数，它从一个对象中提取一个字符串类型的值。一个直观的尝试可能如下：
function extractStringValue(obj: T, key: K): string { // 错误：类型 'T[K]' 不能赋值给类型 'string' // 因为 K 可能是任何 keyof T，所以 T[K] 可能是任何类型，不一定是 string return obj[key]; }const myObj = { stringKey: "hi", boolKey: false };// 期望的用法：const stringVal = extractStringValue(myObj, "stringKey"); // 期望通过// const stringVal2 = extractStringValue(myObj, "boolKey"); // 期望报错
在这个例子中，extractStringValue 函数的定义会立即产生一个类型错误。原因是 T[K] 的类型是 T 中所有属性值类型的联合，它可能包含 string、boolean 或其他类型。TypeScript 编译器无法保证 obj[key] 总是 string 类型，因此拒绝了赋值。更重要的是，即使我们强制转换了类型，extractStringValue(myObj, “boolKey”) 在编译时也不会报错，这与我们期望的行为不符。
为了解决这个问题，我们需要一种机制来动态地生成一个类型，该类型只包含那些对应值为 string 的键。
解决方案：利用映射类型和条件类型创建 KeysOfType
TypeScript 的映射类型（Mapped Types）和条件类型（Conditional Types）提供了强大的能力来转换和筛选类型。我们可以定义一个通用的工具类型 KeysOfType，它能从对象 T 中提取所有值为类型 O 的键。
首先，我们定义一个更通用的 KeysOfType 工具类型：
type KeysOfType = Exclude;
让我们分解这个复杂的类型定义：
[P in keyof T]: 这是一个映射类型，它遍历对象 T 的所有属性键 P。T[P] extends O ? P : never: 这是一个条件类型。对于每个属性 P，它检查 T[P]（即属性 P 的值类型）是否可以赋值给目标类型 O。如果 T[P] 是 O 的子类型，那么这个属性的映射结果就是 P（即键名本身）。如果 T[P] 不是 O 的子类型，那么映射结果就是 never。{ … }[keyof T]: 映射类型的结果是一个新的对象类型，其中包含了筛选后的键和 never。通过 [keyof T] 索引这个新的对象类型，我们得到了一个所有键名的联合类型，例如 (“stringKey” | never | “anotherStringKey” | never)。Exclude: 这是一个内置的工具类型，用于从联合类型中排除特定成员。在这里，我们排除 never 类型（never 会在联合类型中被视为 undefined 的一种特殊形式，或者说 never 最终会从联合类型中消失，但 Exclude 是一种常见的安全做法，以确保我们只剩下实际的键名）。最终，我们得到一个只包含那些值类型匹配 O 的键的联合类型。
实现类型安全的 extractStringValue 函数
有了 KeysOfType，我们可以轻松地定义 StringKeys 类型，并将其应用于 extractStringValue 函数：
// 定义 StringKeys，用于获取所有值为 string 类型的键type StringKeys = KeysOfType;function extractStringValue<T extends Record, K extends StringKeys>( obj: T, key: K,): string { // 现在 T[K] 明确是 string 类型，因为 K 已经被约束为 StringKeys // 并且 T 被约束为 Record，确保了 obj[key] 必然是 string return obj[key]; }
在这个改进后的 extractStringValue 函数中：
K extends StringKeys：这个约束确保了传入的 key 参数必须是 T 对象中值为 string 的属性名之一。T extends Record：这个额外的约束是可选但推荐的，它进一步强化了类型安全性。它表明 T 必须是一个至少包含键 K 且其值为 string 的对象。这使得函数体内部 obj[key] 的类型推断更加准确，直接就是 string，从而消除了之前的类型错误。
现在，我们再次尝试之前的例子：
const myObj = { stringKey: "hi", boolKey: false, numKey: 123 };// 正确用法：编译通过const stringVal = extractStringValue(myObj, "stringKey"); console.log(stringVal); // 输出: hi// 错误用法：编译时报错// Argument of type '"boolKey"' is not assignable to parameter of type '"stringKey"'.// Argument of type '"numKey"' is not assignable to parameter of type '"stringKey"'.// const stringVal2 = extractStringValue(myObj, "boolKey"); // const stringVal3 = extractStringValue(myObj, "numKey");
通过这种方式，TypeScript 编译器能够精确地识别出 boolKey 和 numKey 不是 myObj 中值为 string 的键，从而在编译阶段就报告错误。
提升开发体验：IDE 代码补全
这种方法不仅提供了强大的类型检查，还极大地提升了开发体验。当你在调用 extractStringValue 函数并输入 key 参数时，IDE（如 VS Code）的代码补全功能将只会显示那些符合 StringKeys 约束的键，即 myObj 中值为 string 的键。这大大减少了开发者的心智负担，避免了因输入错误键名或选择错误类型键而导致的潜在问题。
推广到其他类型
KeysOfType 的通用性意味着你可以轻松地为其他类型创建类似的键约束。例如，如果你想提取布尔类型的值：
type BooleanKeys = KeysOfType;function extractBooleanValue<T extends Record, K extends BooleanKeys>( obj: T, key: K,): boolean { return obj[key];}const myOtherObj = { isActive: true, name: "Alice", age: 30 };const activeStatus = extractBooleanValue(myOtherObj, "isActive"); // 编译通过console.log(activeStatus); // 输出: true// 编译时报错：Argument of type '"name"' is not assignable to parameter of type '"isActive"'.// const invalidBoolean = extractBooleanValue(myOtherObj, "name");
总结
通过巧妙地结合 TypeScript 的映射类型和条件类型，我们创建了一个强大的 KeysOfType 工具类型，它允许我们精确地约束泛型函数中键的类型，使其只接受那些对应值为特定类型的键。这种方法不仅显著增强了代码的类型安全性，在编译时捕获潜在错误，而且通过提供智能的代码补全，极大地优化了开发者的体验。掌握这些高级类型技巧，能够帮助我们构建更加健壮、可维护且易于使用的 TypeScript 应用。
以上就是TypeScript中按值类型约束对象键的技巧与实践的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/1534321.html
微信扫一扫 
支付宝扫一扫 
