template.yml

AWSTemplateFormatVersion: '2010-09-09'
Transform: 'AWS::Serverless-2016-10-31'
Description: Graviton2 Lambda Tester

Resources:
  Graviton2RestApi:
    Type: AWS::ApiGateway::RestApi
    Properties:
      Name : graviton2TesterApi

  gravitonCombinations:
    Type: AWS::ApiGateway::Resource
    Properties:
      RestApiId: !Ref Graviton2RestApi
      ParentId: !GetAtt Graviton2RestApi.RootResourceId
      PathPart: 'graviton-combinations'

  gravitonCombinationsN:
    Type: AWS::ApiGateway::Resource
    Properties:
      RestApiId: !Ref Graviton2RestApi
      ParentId: !Ref gravitonCombinations
      PathPart: '{n}'

  gravitonCombinationsK:
    Type: AWS::ApiGateway::Resource
    Properties:
      RestApiId: !Ref Graviton2RestApi
      ParentId: !Ref gravitonCombinationsN
      PathPart: '{k}'

  Method:
    Type: AWS::ApiGateway::Method
    Properties:
      HttpMethod: GET
      ResourceId: !Ref gravitonCombinationsK
      RestApiId: !Ref Graviton2RestApi
      AuthorizationType: NONE
      Integration:
         Type: AWS_PROXY
         Uri: !Sub arn:aws:apigateway:${AWS::Region}:lambda:path/2015-03-31/functions/${gravitonTesterFunction.Arn}/invocations
         IntegrationHttpMethod: POST

  gravitonCombinationsX86:
    Type: AWS::ApiGateway::Resource
    Properties:
      RestApiId: !Ref Graviton2RestApi
      ParentId: !GetAtt Graviton2RestApi.RootResourceId
      PathPart: 'graviton-x86-combinations'

  gravitonCombinationsX86N:
    Type: AWS::ApiGateway::Resource
    Properties:
      RestApiId: !Ref Graviton2RestApi
      ParentId: !Ref gravitonCombinationsX86
      PathPart: '{n}'

  gravitonCombinationsX86K:
    Type: AWS::ApiGateway::Resource
    Properties:
      RestApiId: !Ref Graviton2RestApi
      ParentId: !Ref gravitonCombinationsX86N
      PathPart: '{k}'

  MethodX86:
    Type: AWS::ApiGateway::Method
    Properties:
      HttpMethod: GET
      ResourceId: !Ref gravitonCombinationsX86K
      RestApiId: !Ref Graviton2RestApi
      AuthorizationType: NONE
      Integration:
        Type: AWS_PROXY
        Uri: !Sub arn:aws:apigateway:${AWS::Region}:lambda:path/2015-03-31/functions/${gravitonTesterFunctionX8664.Arn}/invocations
        IntegrationHttpMethod: POST

  ApiGatewayDeployment:
    Type: AWS::ApiGateway::Deployment
    DependsOn:
    - Method
    - MethodX86
    Properties:
      RestApiId: !Ref Graviton2RestApi

  ApiGatewayStage:
    Type: AWS::ApiGateway::Stage
    Properties:
      DeploymentId: !Ref ApiGatewayDeployment
      RestApiId: !Ref Graviton2RestApi
      StageName: 'default'

  LambdaExecutionRole:
    Type: AWS::IAM::Role
    Properties:
      AssumeRolePolicyDocument:
        Version: '2012-10-17'
        Statement:
          - Effect: Allow
            Principal:
              Service:
                - lambda.amazonaws.com
            Action:
              - sts:AssumeRole
      Path: "/"
      Policies:
      - PolicyName: root
        PolicyDocument:
          Version: '2012-10-17'
          Statement:
          - Effect: Allow
            Action:
            - logs:*
            Resource: arn:aws:logs:*:*:*
          - Effect: Allow
            Action:
            - xray:PutTraceSegments
            - xray:PutTelemetryRecords
            - xray:GetSamplingRules
            - xray:GetSamplingTargets
            - xray:GetSamplingStatisticSummaries
            Resource: '*'

  gravitonTesterFunction:
    Type: AWS::Lambda::Function
    Properties:
      Handler: com.slalom.GravitonCombinatorialCalculator
      Description: Function calculating combinations utilizing Graviton V2
      Runtime: java11
      PackageType: Zip
      Code: build/distributions/graviton2.zip
      Role: !GetAtt LambdaExecutionRole.Arn
      Layers:
        - !Ref gravitonLibraries
      Architectures:
        - arm64
      TracingConfig:
        Mode: Active
      Tags:
        - Key: 'CostProfile'
          Value: 'GravitonTester-CF'


#  Using AWS::Serverless::Function it won't be necessary to create API resources and methods separately
#  gravitonTesterFunction:
#    Type: AWS::Serverless::Function
#    Properties:
#      Handler: com.slalom.GravitonCombinatorialCalculator
#      Description: Function calculating combinations utilizing Graviton V2
#      Runtime: java11
#      CodeUri: build/distributions/graviton2.zip
#      Timeout: 30
#      Policies:
#        - AWSXrayWriteOnlyAccess
#        - AWSLambdaBasicExecutionRole
#      Layers:
#        - !Ref gravitonLibraries
#      Architectures:
#        - arm64
#      Tracing: Active
#      Events:
#        ComputeCombinations:
#          Type: Api
#          Properties:
#            Path: /graviton-combinations/{n}/{k}
#            RequestParameters:
#            - method.request.path.n:
#                Required: true
#                Caching: false
#            - method.request.path.k:
#                Required: true
#                Caching: false
#      Tags:
#        'CostProfile': 'GravitonTester-CF'

  gravitonTesterFunctionPermission:
    Type: AWS::Lambda::Permission
    Properties:
      FunctionName: !GetAtt gravitonTesterFunction.Arn
      Action: lambda:InvokeFunction
      Principal: apigateway.amazonaws.com
      SourceArn: !Sub arn:aws:execute-api:${AWS::Region}:${AWS::AccountId}:${Graviton2RestApi}/*/*

  gravitonTesterFunctionX8664:
    Type: AWS::Lambda::Function
    Properties:
      Handler: com.slalom.GravitonCombinatorialCalculator
      Description: Function calculating combinations
      Runtime: java11
      PackageType: Zip
      Code: build/distributions/graviton2.zip
      Role: !GetAtt LambdaExecutionRole.Arn
      Layers:
        - !Ref gravitonLibraries
      TracingConfig:
        Mode: Active
      Tags:
        - Key: 'CostProfile'
          Value: 'GravitonTesterX8664-CF'

  gravitonTesterFunctionX8664Permission:
    Type: AWS::Lambda::Permission
    Properties:
      FunctionName: !GetAtt gravitonTesterFunctionX8664.Arn
      Action: lambda:InvokeFunction
      Principal: apigateway.amazonaws.com
      SourceArn: !Sub arn:aws:execute-api:${AWS::Region}:${AWS::AccountId}:${Graviton2RestApi}/*/*

  gravitonLibraries:
    Type: AWS::Lambda::LayerVersion
    Properties:
      LayerName: graviton2-tester-library
      Description: Dependencies for the graviton-tester functions
      Content: build/distributions/graviton2-lib.zip
      CompatibleRuntimes:
        - java11
      CompatibleArchitectures:
        - arm64
        - x86_64