Skip to content

Commit 71158ee

Browse files
committed
Pick service path from environment
1 parent 357427a commit 71158ee

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

src/main/java/in/erail/amazon/lambda/AWSLambda.java

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22

33
import com.amazonaws.services.lambda.runtime.Context;
44
import com.amazonaws.services.lambda.runtime.RequestStreamHandler;
5+
import com.google.common.base.Strings;
56
import com.google.common.io.ByteStreams;
67
import in.erail.glue.Glue;
78
import in.erail.service.RESTService;
@@ -26,10 +27,16 @@
2627
*/
2728
public class AWSLambda implements RequestStreamHandler {
2829

30+
private static final String SERVICE_ENV = "SERVICE";
31+
private static final String SERVICE_SYS_PROP = "service";
2932
private final RESTService mService;
3033

31-
public AWSLambda(String pComponent) {
32-
mService = Glue.instance().resolve(pComponent);
34+
public AWSLambda() {
35+
String component = System.getenv(SERVICE_ENV);
36+
if (Strings.isNullOrEmpty(component)) {
37+
component = System.getProperty(SERVICE_SYS_PROP);
38+
}
39+
mService = Glue.instance().resolve(component);
3340
}
3441

3542
@Override

0 commit comments

Comments
 (0)