@@ -203,7 +203,10 @@ protected void configure() {
203
203
@ Override
204
204
protected void containerIsStarting (InspectContainerResponse containerInfo ) {
205
205
String command = "#!/bin/bash\n " ;
206
- command += "export LAMBDA_DOCKER_FLAGS=" + configureLambdaContainerLabels () + "\n " ;
206
+ command += "export LAMBDA_DOCKER_FLAGS=" + configureServiceContainerLabels ("LAMBDA_DOCKER_FLAGS" ) + "\n " ;
207
+ command += "export ECS_DOCKER_FLAGS=" + configureServiceContainerLabels ("ECS_DOCKER_FLAGS" ) + "\n " ;
208
+ command += "export EC2_DOCKER_FLAGS=" + configureServiceContainerLabels ("EC2_DOCKER_FLAGS" ) + "\n " ;
209
+ command += "export BATCH_DOCKER_FLAGS=" + configureServiceContainerLabels ("BATCH_DOCKER_FLAGS" ) + "\n " ;
207
210
command += "/usr/local/bin/docker-entrypoint.sh\n " ;
208
211
copyFileToContainer (Transferable .of (command , 0777 ), STARTER_SCRIPT );
209
212
}
@@ -214,13 +217,13 @@ protected void containerIsStarting(InspectContainerResponse containerInfo) {
214
217
* chance.
215
218
* @return the lambda container labels as a string
216
219
*/
217
- private String configureLambdaContainerLabels ( ) {
218
- String lambdaDockerFlags = internalMarkerLabels ();
219
- String existingLambdaDockerFlags = getEnvMap ().get ("LAMBDA_DOCKER_FLAGS" );
220
- if (existingLambdaDockerFlags != null ) {
221
- lambdaDockerFlags = existingLambdaDockerFlags + " " + lambdaDockerFlags ;
220
+ private String configureServiceContainerLabels ( String existingEnvFlagKey ) {
221
+ String internalMarkerFlags = internalMarkerLabels ();
222
+ String existingFlags = getEnvMap ().get (existingEnvFlagKey );
223
+ if (existingFlags != null ) {
224
+ internalMarkerFlags = existingFlags + " " + internalMarkerFlags ;
222
225
}
223
- return "\" " + lambdaDockerFlags + "\" " ;
226
+ return "\" " + internalMarkerFlags + "\" " ;
224
227
}
225
228
226
229
/**
0 commit comments