com.cyngn.vertx.web.handler.RequestIdResponseHandler Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of vertx-util Show documentation

A library for using opentsdb with vert.x

There is a newer version: 0.6.0

package com.cyngn.vertx.web.handler;

import com.cyngn.vertx.web.RestApi;
import io.vertx.core.Handler;
import io.vertx.ext.web.RoutingContext;
import org.apache.commons.lang.StringUtils;

/**
 * Tags a response with the request id we associated to the request if it had the header present.
 *
 * @author [email protected] (Jeremy Truelove) 09/05/15
 */
public class RequestIdResponseHandler implements Handler {

  private RequestIdResponseHandler(){}

  /**
   * Get a RequestIdResponseHandler
   *
   * @return reference to created handler
   */
  public static RequestIdResponseHandler create() { return new RequestIdResponseHandler(); }

  @Override
  public void handle(RoutingContext ctx) {
    ctx.addHeadersEndHandler(fut -> {
        // grab the header from request and add to response if present
        String requestId = ctx.request().getHeader(RestApi.X_REQUEST_ID);
        if (StringUtils.isNotEmpty(requestId)) {
          ctx.response().putHeader(RestApi.X_REQUEST_ID, requestId);
        }
          fut.complete();
      });
      ctx.next();
    }
  }