如何使用TypeScript装饰器修改方法的返回值？

你的思路是对的，用装饰器封装「检查参数+给返回值加属性」的横切逻辑，非常适合多方法复用的场景。先帮你修正现有代码的问题，再聊聊优化方向。

一、修正现有装饰器的实现 #

你原来的代码判断时机不对，应该先检查参数，再执行原方法拿到结果，最后给结果添加属性。另外要考虑参数为空的情况，避免报错：

function ParseDiscounts() {
    return function (
        target: any,
        propertyKey: string,
        descriptor: PropertyDescriptor,
    ) {
        const originalMethod = descriptor.value;
        // 兼容同步/异步方法
        descriptor.value = async function (...args: any[]) {
            // 检查第一个参数是否存在discounts（做空值保护）
            const hasDiscount = args[0]?.discounts !== undefined;
            // 执行原方法（同步/异步统一用await处理）
            const result = await originalMethod.apply(this, args);
            
            // 给返回值添加属性（处理result为null/undefined的情况）
            if (result != null) {
                result.isDiscountApplied = hasDiscount;
            }
            
            return result;
        };
        return descriptor;
    };
}

这样修改后，不管原方法是同步还是异步，都能正确给返回值加上isDiscountApplied属性，同时避免了参数为空时的报错。

二、原方案是否最优？ #

这个方案的核心思路没问题：用装饰器实现AOP（面向切面编程），把和业务无关的横切逻辑（判断折扣、添加标记）抽离出来，让业务方法只关注核心逻辑。这种方式的优势是：

• 复用性强：一个装饰器可以给任意多个方法添加相同逻辑
• 侵入性低：业务代码不需要修改，只需要加个装饰器注解
• 维护方便：后续要修改判断规则或属性名，只需要改装饰器的实现

所以从复用和架构的角度，这个方向是最优的之一，如果你只需要针对discounts参数和isDiscountApplied属性，当前方案已经足够用。

三、更优的通用化实现建议 #

如果以后需要给不同的参数、添加不同的属性，或者需要更灵活的配置，可以把装饰器改成通用版本，支持自定义参数：

// 通用装饰器：检查指定参数是否存在，给返回值添加指定属性
function AddFlagOnParamExists<T extends object>(
    paramKey: keyof T, // 要检查的参数键名
    flagKey: string,   // 要添加的属性名
    flagValueWhenExists: boolean = true, // 参数存在时的属性值
    flagValueWhenNotExists: boolean = false // 参数不存在时的属性值
) {
    return function (
        target: any,
        propertyKey: string,
        descriptor: PropertyDescriptor,
    ) {
        const originalMethod = descriptor.value;
        descriptor.value = async function (...args: any[]) {
            // 检查指定位置的参数是否包含目标键（做空值保护）
            const paramExists = args[0] != null && paramKey in args[0];
            const result = await originalMethod.apply(this, args);
            
            if (result != null) {
                // 给返回值添加自定义属性
                (result as any)[flagKey] = paramExists ? flagValueWhenExists : flagValueWhenNotExists;
            }
            
            return result;
        };
        return descriptor;
    };
}

使用时可以灵活配置不同场景：

class Order {
    // 原来的折扣场景
    @AddFlagOnParamExists('discounts', 'isDiscountApplied')
    createOrder(discountPayload) {
        return resultToBeParsed;
    }

    // 扩展场景：检查是否有优惠券，添加isCouponApplied属性
    @AddFlagOnParamExists('coupons', 'isCouponApplied')
    createPromotionOrder(couponPayload) {
        return promotionResult;
    }
}

如果需要更强的类型安全，还可以给装饰器加上泛型约束，确保返回值类型合规：

// 定义返回值必须包含要添加的属性类型
type WithFlag<T extends object, K extends string, V> = T & { [key in K]: V };

// 带类型约束的通用装饰器
function AddFlagOnParamExists<T extends object, K extends string, V = boolean>(
    paramKey: keyof T,
    flagKey: K,
    flagValueWhenExists: V = true as V,
    flagValueWhenNotExists: V = false as V
) {
    return function (
        target: any,
        propertyKey: string,
        descriptor: TypedPropertyDescriptor<(...args: [T]) => Promise<WithFlag<any, K, V>> | WithFlag<any, K, V>>
    ) {
        const originalMethod = descriptor.value!;
        descriptor.value = async function (param: T) {
            const paramExists = param != null && paramKey in param;
            const result = await originalMethod.call(this, param);
            
            if (result != null) {
                (result as WithFlag<any, K, V>)[flagKey] = paramExists ? flagValueWhenExists : flagValueWhenNotExists;
            }
            
            return result;
        };
        return descriptor;
    };
}

这样TypeScript会在编译期帮你检查参数和返回值的类型，避免类型错误。

四、额外注意事项 #

• 如果原方法的返回值是不可变对象（比如用Object.freeze冻结的），直接添加属性会报错，这时候可以返回新对象：return { ...result, isDiscountApplied: hasDiscount }
• 如果需要检查的不是第一个参数，可以在装饰器里添加paramIndex: number = 0配置项，用args[paramIndex]获取目标参数

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