DefaultLambdaFields.java

/*
 * Copyright 2020 Amazon.com, Inc. or its affiliates.
 * Licensed under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *     http://www.apache.org/licenses/LICENSE-2.0
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 */
package software.amazon.lambda.logging.internal;

import com.amazonaws.services.lambda.runtime.Context;

import java.util.HashMap;
import java.util.Map;

enum DefaultLambdaFields {
    FUNCTION_NAME("functionName"),
    FUNCTION_VERSION("functionVersion"),
    FUNCTION_ARN("functionArn"),
    FUNCTION_MEMORY_SIZE("functionMemorySize");

    private final String name;

    DefaultLambdaFields(String name) {
        this.name = name;
    }

    public String getName() {
        return name;
    }

    static Map<String, String> values(Context context) {
        Map<String, String> hashMap = new HashMap<>();

        hashMap.put(FUNCTION_NAME.name, context.getFunctionName());
        hashMap.put(FUNCTION_VERSION.name, context.getFunctionVersion());
        hashMap.put(FUNCTION_ARN.name, context.getInvokedFunctionArn());
        hashMap.put(FUNCTION_MEMORY_SIZE.name, String.valueOf(context.getMemoryLimitInMB()));

        return hashMap;
    }
}