Websocket con API Gateway y AWS Lambda

En el siguiente código muestro como utilizar Java para crear un Lambda de AWS que permite utilizar Web Socket desde API Gateway.

  1. public class WebSocketLambda implements RequestStreamHandler {
  3.   private static final AWSCredentials CREDENCIALES
  4.           = new EnvironmentVariableCredentialsProvider().getCredentials();
  6.   @Override
  7.   public void handleRequest(InputStream in, OutputStream out, Context cntxt) throws IOException {
  8.     // obtenemos la representación como JSON del request generado por aws
  9.     JsonObject json = Json.createReader(in).readObject();
  10.     if (json.getJsonObject("requestContext").getString("routeKey").equals("$default")) {
  11.       final String payload = json.getString("body"); // contenido que el cliente nos envia
  12.       Request request = new DefaultRequest<>("execute-api");
  13.       request.setHttpMethod(HttpMethodName.POST);
  14.       request.setEndpoint(URI.create(new StringBuilder()
  15.               .append(System.getenv("url_api")) // el API gateway la define
  16.               .append("/")
  17.               // identificador del cliente al que le respondemos
  18.               .append(json.getJsonObject("requestContext").getString("connectionId")) 
  19.               .toString()));
  20.       // escribimos el mismo mensaje que el cliente nos envía
  21.       request.setContent(new ByteArrayInputStream(payload.getBytes(StandardCharsets.UTF_8)));
  22.       AWS4Signer signer = new AWS4Signer();
  23.       signer.setServiceName(request.getServiceName());
  24.       signer.sign(request, CREDENCIALES);
  25.       // enviar request al api gateway
  26.       new AmazonHttpClient(new ClientConfiguration())
  27.               .requestExecutionBuilder()
  28.               .executionContext(new ExecutionContext(true))
  29.               .request(request)
  30.               .errorResponseHandler(new HttpResponseHandler() {
  31.                 @Override
  32.                 public AmazonClientException handle(HttpResponse rsp) throws Exception {
  33.                   return new AmazonClientException(
  34.                           new StringBuilder()
  35.                                   .append("error al enviar mensaje")
  36.                                   .append(System.getenv("url_api"))
  37.                                   .append(rsp.getStatusCode())
  38.                                   .append(rsp.getStatusText())
  39.                                   .append(leerError(rsp.getContent())).toString());
  40.                 }
  42.                 @Override
  43.                 public boolean needsConnectionLeftOpen() {
  44.                   return false;
  45.                 }
  46.               }).execute(new HttpResponseHandler() {
  47.         @Override
  48.         public Void handle(HttpResponse response) {
  49.           return null;
  50.         }
  52.         @Override
  53.         public boolean needsConnectionLeftOpen() {
  54.           return false;
  55.         }
  56.       });
  57.     }
  58.     try (PrintWriter pw = new PrintWriter(out)) {
  59.       JsonObject response = Json.createObjectBuilder()
  60.               .add("statusCode", 200)
  61.               .build();
  62.       pw.write(response.toString());
  63.     }
  64.   }
  66.   private static String leerError(InputStream inputStream) throws IOException {
  67.     final ByteArrayOutputStream bos = new ByteArrayOutputStream();
  68.     byte[] buffer = new byte[1024];
  69.     int length;
  70.     while ((length = inputStream.read(buffer)) != -1) {
  71.       bos.write(buffer, 0, length);
  72.     }
  73.     return bos.toString(StandardCharsets.UTF_8.name());
  74.   }
  75. }


Algunas notas sobre el código:

  • Recibe conexiones y responde el mismo mensaje recibido al cliente utilizando la URL que entrega el API Gateway.
  • La URL Connection que entrega el API Gateway se define como variable de ambiente (url_api).
  • En connectionId esta definido el identificador del cliente entregado por AWS.


En github he dejado el ejemplo completo utilizando maven para construirlo y algunos comentarios adicionales sobre la configuración en AWS.

Algunos enlaces MUY útiles para conocer los Web Sockets en API Gateway:




a la/s
Etiquetas: , , ,

No hay comentarios.:

Publicar un comentario

Suscribirse a: Comentarios de la entrada (Atom)