Get the names of different stages in an AWS Lambda function

Question

I have a lambda function(without APIGateWay) which I have created using BONES framework. I want same lambda function in both test and prod stage and for that I want to get the stage name during runtime. My code is written in java and I know that we can find out the default AWS region using following:

Regions regions = Regions.fromName(System.getenv("AWS_DEFAULT_REGION"));

So here are my questions:

1- How to get the name of the stage in which my lambda function is running?

2- What does this default region means? will it be always same as the region in which my lambda function is running?

Any lead is highly appreciated.

Answer 1

How to get the name of the stage in which my lambda function is running?

For Lambda Proxy integration, you can get the stage name from the requestContext in the input stream, which contains the API request serialized as a JSON string by API Gateway. The input data can also include the request's stage variables stageVariables (if you use/need any). For example:

JSONParser parser = new JSONParser();

public void handleRequest(InputStream inputStream, OutputStream outputStream, Context context) throws IOException {
  ...
  BufferedReader reader = new BufferedReader(new InputStreamReader(inputStream));
  String stage = null;

  JSONObject event = (JSONObject)parser.parse(reader);
  if (event.get("requestContext") != null) {
    JSONObject requestContext = (JSONObject)parser.parse((String)event.get("requestContext"));
    if (requestContext.get("stage") != null) {
      stage = (String)requestContext.get("stage");
    }
  }
  ...
}

See full example.

What does this default region means? will it be always same as the region in which my lambda function is running?

YES according to docs and NO according to the SDK.

The SDK does not reference it. It uses AWS_REGION only. See SDKGlobalConfiguration.java.