如何告诉 TypeScript 我的函数参数是 Enum 键？

Question

我有一个枚举，它将 HTTP 状态映射到他们的代码，如下所示：

我可以通过访问它们的密钥轻松获取这些代码（注意 IntelliSense 如何在右侧面板上显示 HttpStatus.NOT_FOUND = 404）

现在假设我有一个名为 sendStatus 的函数：

我应该如何输入这个函数才能让 IntelliSense 自动完成这些代码？

使用 keyof typeof 不起作用，因为它不会自动完成枚举值。

Answer 1

您可以通过运行这样的示例来测试keyof typeof，这表明 Enum 本身在您的情况下工作正常。

enum HttpStatus {
    OK = 200,
    CREATED = 201
}

function printStatus(code: keyof typeof HttpStatus) {
    const num = HttpStatus[code];
    if (num <= HttpStatus.CREATED) {
       console.log('HTTP Status key is: ', code);
       console.log('HTTP Status value is: ', num);
    }
}

printStatus("OK");

// Prints out
// > HTTP Status key is:  – "OK"
// > HTTP Status value is:  – 200
// Which shows that Enum value is being detected properly

keyof typeof 让您键入将所有 Enum 键表示为字符串的类型，这就是自动完成行为如此的原因。 https://www.typescriptlang.org/docs/handbook/enums.html#enums-at-compile-time

Answer 2

你想要的不是直接可能的。当您执行 keyof typeof 时，您会创建一个联合类型 'CREATED' | 'OK' | 'NOT_FOUND' ...，它与以前的枚举完全分开。

你能得到的最接近的就是做

function sendStatus(code: HttpStatus) {
    // code
}

send(HttpStatus.OK) // This will autocomplete and show the status code numbers

然后在 sendStatus 函数中将代码转换为字符串。

如果不知道你正在寻找的确切用法，很难说出你真正想要的东西，但我会考虑只使用一个普通的旧对象而不是枚举

const HTTP_STATUS = {'OK':200, 'CREATED':201} as const

然后，如果需要，您也可以像这样创建这两种类型

type StringStatus = keyof typeof HTTP_STATUS // 'OK' | 'CREATED'
type NumsStatus = (typeof HTTP_STATUS)[keyof typeof HTTP_STATUS] // 200 | 201

一般来说，在现代 TS 中使用 enums 很少有充分的理由。通常对象和/或联合类型做得更好。