面试题：TypeScript更精确字符串类型在函数重载中的应用

Question

假设你有一个函数`printMessage`，根据传入的字符串类型不同执行不同操作。字符串类型有'error'、'warning'、'info'。如果是'error'，打印错误信息并记录到日志文件；如果是'warning'，打印警告信息；如果是'info'，只打印信息。请使用TypeScript的更精确字符串类型结合函数重载来实现这个功能，并解释为什么使用这种方式而不是简单的if - else判断。

Answer 1

function printMessage(type: 'error', message: string): void;
function printMessage(type: 'warning', message: string): void;
function printMessage(type: 'info', message: string): void;
function printMessage(type: 'error' | 'warning' | 'info', message: string): void {
    if (type === 'error') {
        console.error(`Error: ${message}`);
        // 记录到日志文件的逻辑，例如：
        // const fs = require('fs');
        // fs.appendFileSync('error.log', `Error: ${message}\n`);
    } else if (type === 'warning') {
        console.warn(`Warning: ${message}`);
    } else if (type === 'info') {
        console.log(`Info: ${message}`);
    }
}

为什么使用函数重载而不是简单的if - else判断：

1. 类型安全性：函数重载利用TypeScript的类型系统，在编译时就能确保传入的类型是正确的。例如，如果误传入了一个其他字符串，编译器会报错，而if - else判断只有在运行时才能发现错误。
2. 代码可读性：函数重载在声明处就明确了不同类型参数对应的行为，代码阅读者可以直观地了解到该函数根据不同类型执行不同操作，而if - else判断可能需要深入到函数内部去理解逻辑。
3. 可维护性：当需要添加新的字符串类型时，使用函数重载只需要在函数声明处添加新的重载定义，而if - else判断则需要在函数内部添加新的判断分支，可能影响原有的逻辑结构。