com.github.tomakehurst.wiremock.http.HttpServerFactoryLoader Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of wiremock-standalone Show documentation
A web service test double for all occasions - standalone edition
There is a newer version: 3.9.2
/*
 * Copyright (C) 2024 Thomas Akehurst
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.github.tomakehurst.wiremock.http;

import static java.util.stream.Collectors.toList;
import static java.util.stream.Collectors.toUnmodifiableList;

import com.github.tomakehurst.wiremock.common.FatalStartupException;
import com.github.tomakehurst.wiremock.core.Options;
import com.github.tomakehurst.wiremock.extension.Extension;
import com.github.tomakehurst.wiremock.extension.Extensions;
import java.util.List;
import java.util.ServiceLoader;
import java.util.function.Supplier;
import org.eclipse.jetty.util.Jetty;

public class HttpServerFactoryLoader {

  private final Options options;
  private final Extensions extensions;
  private final Supplier> serviceLoader;
  private final boolean isJetty11;

  public HttpServerFactoryLoader(
      Options options,
      Extensions extensions,
      Supplier> serviceLoader,
      boolean isJetty11) {
    this.options = options;
    this.extensions = extensions;
    this.serviceLoader = serviceLoader;
    this.isJetty11 = isJetty11;
  }

  public HttpServerFactory load() {
    final List extensionCandidates =
        extensions.ofType(HttpServerFactory.class).values().stream().collect(toUnmodifiableList());
    if (!extensionCandidates.isEmpty()) {
      return pickMostAppropriateFrom(extensionCandidates);
    }

    if (options.hasDefaultHttpServerFactory() && !isJetty11) {
      final List serviceLoadedCandidates = serviceLoader.get();
      return pickMostAppropriateFrom(serviceLoadedCandidates);
    }

    return options.httpServerFactory();
  }

  public static Supplier> systemServiceLoader() {
    return () ->
        ServiceLoader.load(Extension.class).stream()
            .filter(extension -> HttpServerFactory.class.isAssignableFrom(extension.type()))
            .map(e -> (HttpServerFactory) e.get())
            .collect(toList());
  }

  private static HttpServerFactory pickMostAppropriateFrom(List candidates) {
    return candidates.size() > 1
        ? candidates.stream()
            .filter(factory -> !DefaultFactory.class.isAssignableFrom(factory.getClass()))
            .findFirst()
            .orElseThrow(HttpServerFactoryLoader::couldNotFindSuitableServerException)
        : candidates.get(0);
  }

  private static FatalStartupException couldNotFindSuitableServerException() {
    return new FatalStartupException(
        "Jetty 11 is not present and no suitable HttpServerFactory extension was found. Please ensure that the classpath includes a WireMock extension that provides an HttpServerFactory implementation. See http://wiremock.org/docs/extending-wiremock/ for more information.");
  }

  public static boolean isJetty11() {
    try {
      return Jetty.VERSION.startsWith("11");
    } catch (Throwable e) {
      return false;
    }
  }
}