Function Profile

packages/core/src/Utils/Profiler/AutoProfiler.ts:590–630

(
    name?: string,
    category: ProfileCategory = ProfileCategory.Custom
)

 * ```
 */
export function Profile(
    name?: string,
    category: ProfileCategory = ProfileCategory.Custom
): MethodDecorator {
    return function(
        target: object,
        propertyKey: string | symbol,
        descriptor: PropertyDescriptor
    ): PropertyDescriptor {
        const original = descriptor.value;
        const methodName = name || `${target.constructor.name}.${String(propertyKey)}`;

        descriptor.value = function(this: any, ...args: any[]): any {
            if (!ProfilerSDK.isEnabled()) {
                return original.apply(this, args);
            }

            const handle = ProfilerSDK.beginSample(methodName, category);
            try {
                const result = original.apply(this, args);

                // 处理异步方法
                if (result instanceof Promise) {
                    return result.finally(() => {
                        ProfilerSDK.endSample(handle);
                    });
                }

                // 同步方法，立即结束采样
                ProfilerSDK.endSample(handle);
                return result;
            } catch (error) {
                // 发生错误时也要结束采样
                ProfilerSDK.endSample(handle);
                throw error;
            }
        };

        return descriptor;
    };
}

/**
 * @ProfileClass 装饰器