Skip to content
Snippets Groups Projects
Commit cdafc81e authored by Clayton, Brandon Scott's avatar Clayton, Brandon Scott
Browse files

clean

parent b8193dc4
No related branches found
No related tags found
1 merge request!806AWS Lambda
This commit is part of merge request !806. Comments created here will be created in the context of that merge request.
package gov.usgs.earthquake.nshmp.www;
import java.nio.file.Path;
import java.util.Optional;
import org.crac.Context;
import org.crac.Resource;
......@@ -13,10 +12,8 @@ import software.amazon.awssdk.services.ssm.model.SsmException;
import com.amazonaws.serverless.proxy.internal.testutils.MockLambdaContext;
import com.amazonaws.serverless.proxy.model.AwsProxyRequest;
import com.amazonaws.serverless.proxy.model.MultiValuedTreeMap;
import gov.usgs.earthquake.nshmp.gmm.Gmm;
import gov.usgs.earthquake.nshmp.gmm.Imt;
import io.micronaut.context.annotation.Value;
import io.micronaut.crac.OrderedResource;
......@@ -41,31 +38,19 @@ public class PrimingResource implements OrderedResource {
handler.handleRequest(
getAwsProxyRequest("/hazard"),
new MockLambdaContext());
Optional<String> location = Optional.ofNullable(System.getenv("TEST_LOCATION"));
if (location.isPresent()) {
handler.handleRequest(
getAwsProxyRequest(String.format("/hazard/%s/760", location.get())),
new MockLambdaContext());
}
}
}
@Override
public void afterRestore(Context<? extends Resource> context) throws Exception {
// Reset executors on restore as Shutdown event does not fire on AWS lambda
// ServletUtil.resetExecutors();
ServletUtil.resetExecutors();
}
private static AwsProxyRequest getAwsProxyRequest(String path) {
final AwsProxyRequest awsProxyRequest = new AwsProxyRequest();
awsProxyRequest.setHttpMethod("GET");
awsProxyRequest.setPath(path);
var parameters = new MultiValuedTreeMap<String, String>();
parameters.add("imt", Imt.PGA.toString());
awsProxyRequest.setMultiValueQueryStringParameters(parameters);
return awsProxyRequest;
}
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment