diff --git a/src/annotations/classes/aws/aws.ts b/src/annotations/classes/aws/aws.ts index 599239b..01aa5de 100644 --- a/src/annotations/classes/aws/aws.ts +++ b/src/annotations/classes/aws/aws.ts @@ -1,8 +1,13 @@ import { CLASS_AWSMEMORYSIZEKEY, CLASS_AWSTIMEOUTKEY, CLASS_AWSRUNTIMEKEY } from '../../constants' import { defineMetadata } from '../../metadata' +export enum AwsRuntimeType { + NodeJs6 = 'nodejs6.10', + NodeJs8 = 'nodejs8.10' +} + export const aws = (config: { - type?: 'nodejs6.10', + type?: string | AwsRuntimeType, memorySize?: number, timeout?: number }) => (target: Function) => { @@ -15,4 +20,4 @@ export const aws = (config: { if (typeof config.timeout === 'number') { defineMetadata(CLASS_AWSTIMEOUTKEY, config.timeout, target); } -} \ No newline at end of file +}