如何在忽略可选属性的情况下强制路径类型？Set辅助函数疑问

嘿，这个问题我之前在写类似的属性设置工具函数时也碰到过！核心原因是TypeScript会保留可选属性的undefined可能性，导致当路径包含可选键时，后续的属性类型推导会因为可能的undefined而卡住（比如E[K1]可能是undefined，那keyof E[K1]就会变成never）。咱们可以通过强制基于「所有属性都必选」的类型来推导路径，就能忽略原类型中的可选标记，实现你要的效果。

解决方案代码 #

先直接给你修改后的函数代码，我会一步步解释为什么这么改：

// 工具类型：将对象的所有属性转为必选
type RequiredEntity<T> = Required<T>;

function set<
  C extends Collection<any>,
  // 从Collection中提取实体类型
  E = C extends Collection<infer U> ? U : never,
  // 生成实体的「全必选」版本，用于路径推导
  RE = RequiredEntity<E>,
  // 第一层路径键：基于全必选实体的键
  K1 extends keyof RE = keyof RE,
  // 第二层路径键：基于第一层属性的全必选类型的键
  K2 extends keyof RE[K1] = keyof RE[K1],
  // 值的类型：匹配第二层属性的类型
  V extends RE[K1][K2] = RE[K1][K2]
>(
  collection: C,
  entityId: string,
  path: [K1, K2],
  value: V
) {
  // 你的实现逻辑，注意处理属性不存在的情况：
  const entity = collection.get(entityId);
  if (!entity) return;
  
  // 如果第一层属性不存在，先初始化（根据你的实体类型调整默认值）
  if (!entity[path[0] as keyof E]) {
    entity[path[0] as keyof E] = {} as RE[K1];
  }
  
  // 设置最终值
  (entity[path[0] as keyof E] as RE[K1])[path[1]] = value;
}

关键改动说明 #

1. 引入RequiredEntity工具类型：Required<T>是TypeScript内置工具类型，它会把T中所有可选属性（带?标记的）转为必选属性。我们用它生成实体类型的「全必选」版本RE，作为路径推导的基础。
2. 基于RE推导路径键：原来的K1和K2是基于原实体类型E，现在改为基于RE，这样不管原属性是否可选，都能被纳入路径的可选范围，同时类型推导不会因为undefined而中断。
3. 运行时的属性初始化：因为我们在类型层面忽略了可选性，但实际运行时属性可能还是undefined，所以需要在设置值之前检查并初始化对应的属性，避免抛出Cannot set properties of undefined的错误。

示例验证 #

假设你的实体类型是这样的（包含可选属性）：

interface Comment {
  id: string;
  content: string;
  // 可选的author属性
  author?: {
    // 可选的name属性
    name?: string;
    age: number;
  };
}

type CommentCollection = Collection<Comment>;

现在调用修改后的set函数，即使路径包含可选属性，类型检查也能正常通过：

// 允许设置可选的author.name属性
Collection.set(myCommentCollection, someComment.id, ['author', 'name'], 'jack');
// 允许设置必选的author.age属性
Collection.set(myCommentCollection, someComment.id, ['author', 'age'], 28);

扩展思考 #

如果以后需要支持更长的路径（比如三层、四层），可以用递归的工具类型来处理任意深度的路径，核心思路还是基于「全必选」的实体类型来推导每一层的键。不过针对你当前的两层路径需求，上面的方案已经完全够用啦。

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