software.xdev.mockserver.mock.action.http.HttpResponseClassCallbackActionHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of server Show documentation
MockServer NeoLight Netty Server
There is a newer version: 1.0.8
/*
 * Copyright © 2024 XDEV Software (https://xdev.software)
 *
 * 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 software.xdev.mockserver.mock.action.http;

import static software.xdev.mockserver.model.HttpResponse.notFoundResponse;

import java.lang.reflect.Constructor;
import java.lang.reflect.InvocationTargetException;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import software.xdev.mockserver.mock.action.ExpectationResponseCallback;
import software.xdev.mockserver.model.HttpClassCallback;
import software.xdev.mockserver.model.HttpRequest;
import software.xdev.mockserver.model.HttpResponse;


public class HttpResponseClassCallbackActionHandler
{
	private static final Logger LOG = LoggerFactory.getLogger(HttpResponseClassCallbackActionHandler.class);
	private static ClassLoader contextClassLoader = ClassLoader.getSystemClassLoader();
	
	public static void setContextClassLoader(final ClassLoader contextClassLoader)
	{
		HttpResponseClassCallbackActionHandler.contextClassLoader = contextClassLoader;
	}
	
	public HttpResponse handle(final HttpClassCallback httpClassCallback, final HttpRequest request)
	{
		return this.invokeCallbackMethod(httpClassCallback, request);
	}
	
	@SuppressWarnings({"rawtypes", "unchecked"})
	private ExpectationResponseCallback instantiateCallback(final HttpClassCallback httpClassCallback)
	{
		try
		{
			final Class expectationResponseCallbackClass =
				contextClassLoader.loadClass(httpClassCallback.getCallbackClass());
			if(ExpectationResponseCallback.class.isAssignableFrom(expectationResponseCallbackClass))
			{
				final Constructor constructor =
					expectationResponseCallbackClass.getConstructor();
				return constructor.newInstance();
			}
			else
			{
				LOG.error("{} does not implement {} required for responses using class callback",
					httpClassCallback.getCallbackClass(), ExpectationResponseCallback.class.getName());
			}
		}
		catch(final ClassNotFoundException e)
		{
			LOG.error("ClassNotFoundException - while trying to instantiate ExpectationResponseCallback class \"{}\"",
				httpClassCallback.getCallbackClass(), e);
		}
		catch(final NoSuchMethodException e)
		{
			LOG.error(
				"NoSuchMethodException - while trying to create default constructor on "
					+ "ExpectationResponseCallback class \"{}\"",
				httpClassCallback.getCallbackClass(),
				e);
		}
		catch(final InvocationTargetException | InstantiationException | IllegalAccessException e)
		{
			LOG.error(
				"InvocationTargetException - while trying to execute default constructor on "
					+ "ExpectationResponseCallback class \"{}\"",
				httpClassCallback.getCallbackClass(),
				e);
		}
		return null;
	}
	
	private HttpResponse invokeCallbackMethod(final HttpClassCallback httpClassCallback, final HttpRequest httpRequest)
	{
		if(httpRequest != null)
		{
			final ExpectationResponseCallback expectationResponseCallback =
				this.instantiateCallback(httpClassCallback);
			if(expectationResponseCallback != null)
			{
				try
				{
					return expectationResponseCallback.handle(httpRequest);
				}
				catch(final Exception ex)
				{
					LOG.error("{} throw exception while executing handle callback method",
						httpClassCallback.getCallbackClass(), ex);
					return notFoundResponse();
				}
			}
			else
			{
				return notFoundResponse();
			}
		}
		else
		{
			return notFoundResponse();
		}
	}
}