com.arangodb.shaded.vertx.ext.web.client.impl.predicate.PredicateInterceptor Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of arangodb-java-driver-shaded Show documentation
ArangoDB Java Driver Shaded
There is a newer version: 7.8.0
Show newest version
/*
 * Copyright 2014 Red Hat, Inc.
 *
 *  All rights reserved. This program and the accompanying materials
 *  are made available under the terms of the Eclipse Public License v1.0
 *  and Apache License v2.0 which accompanies this distribution.
 *
 *  The Eclipse Public License is available at
 *  http://www.eclipse.org/legal/epl-v10.html
 *
 *  The Apache License v2.0 is available at
 *  http://www.opensource.org/licenses/apache2.0.php
 *
 *  You may elect to redistribute this code under either of these licenses.
 */
package com.arangodb.shaded.vertx.ext.web.client.impl.predicate;

import com.arangodb.shaded.vertx.core.Handler;
import com.arangodb.shaded.vertx.core.MultiMap;
import com.arangodb.shaded.vertx.core.http.HttpClientResponse;
import com.arangodb.shaded.vertx.core.impl.NoStackTraceThrowable;
import com.arangodb.shaded.vertx.ext.web.client.impl.ClientPhase;
import com.arangodb.shaded.vertx.ext.web.client.impl.HttpContext;
import com.arangodb.shaded.vertx.ext.web.client.impl.HttpRequestImpl;
import com.arangodb.shaded.vertx.ext.web.client.impl.HttpResponseImpl;
import com.arangodb.shaded.vertx.ext.web.client.predicate.ErrorConverter;
import com.arangodb.shaded.vertx.ext.web.client.predicate.ResponsePredicate;

import java.util.ArrayList;
import java.util.List;

public class PredicateInterceptor implements Handler> {

  @Override
  public void handle(HttpContext httpContext) {

    if (httpContext.phase() == ClientPhase.RECEIVE_RESPONSE) {

      // Run expectations
      HttpRequestImpl request = (HttpRequestImpl) httpContext.request();
      HttpClientResponse resp = httpContext.clientResponse();
      List expectations = request.expectations();
      for (ResponsePredicate expectation : expectations) {
        ResponsePredicateResultImpl predicateResult;
        try {
          predicateResult = (ResponsePredicateResultImpl) expectation.apply(responseCopy(resp, httpContext, null));
        } catch (Exception e) {
          httpContext.fail(e);
          return;
        }
        if (!predicateResult.succeeded()) {
          ErrorConverter errorConverter = expectation.errorConverter();
          if (!errorConverter.requiresBody()) {
            predicateResult.setHttpResponse(responseCopy(resp, httpContext, null));
            failOnPredicate(httpContext, errorConverter, predicateResult);
          } else {
            resp.bodyHandler(buffer -> {
              predicateResult.setHttpResponse(responseCopy(resp, httpContext, buffer));
              failOnPredicate(httpContext, errorConverter, predicateResult);
            });
            resp.resume();
          }
          return;
        }
      }
    }

    httpContext.next();
  }

  private  HttpResponseImpl responseCopy(HttpClientResponse resp, HttpContext httpContext, B value) {
    return new HttpResponseImpl<>(
      resp.version(),
      resp.statusCode(),
      resp.statusMessage(),
      MultiMap.caseInsensitiveMultiMap().addAll(resp.headers()),
      null,
      new ArrayList<>(resp.cookies()),
      value,
      httpContext.getRedirectedLocations()
    );
  }

  private void failOnPredicate(HttpContext ctx, ErrorConverter converter, ResponsePredicateResultImpl predicateResult) {
    Throwable result;
    try {
      result = converter.apply(predicateResult);
    } catch (Exception e) {
      result = e;
    }
    if (result != null) {
      ctx.fail(result);
    } else {
      ctx.fail(new NoStackTraceThrowable("Invalid HTTP response"));
    }
  }
}