software.xdev.mockserver.httpclient.HttpClientHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of core Show documentation
MockServer NeoLight Core
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.httpclient;

import static software.xdev.mockserver.httpclient.NettyHttpClient.RESPONSE_FUTURE;
import static software.xdev.mockserver.util.StringUtils.isNotBlank;

import java.util.Arrays;
import java.util.List;

import javax.net.ssl.SSLException;

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

import io.netty.channel.ChannelHandler;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;
import io.netty.handler.codec.DecoderException;
import io.netty.handler.ssl.NotSslRecordException;
import software.xdev.mockserver.model.Message;


@ChannelHandler.Sharable
public class HttpClientHandler extends SimpleChannelInboundHandler
{
	private static final Logger LOG = LoggerFactory.getLogger(HttpClientHandler.class);
	
	private final List connectionClosedStrings = Arrays.asList(
		"Broken pipe",
		"(broken pipe)",
		"Connection reset"
	);
	
	HttpClientHandler()
	{
		super(false);
	}
	
	@Override
	public void channelRead0(final ChannelHandlerContext ctx, final Message response)
	{
		ctx.channel().attr(RESPONSE_FUTURE).get().complete(response);
		ctx.close();
	}
	
	@Override
	public void exceptionCaught(final ChannelHandlerContext ctx, final Throwable cause)
	{
		if(this.isNotSslException(cause) && this.isNotConnectionReset(cause))
		{
			LOG.error("", cause);
		}
		ctx.channel().attr(RESPONSE_FUTURE).get().completeExceptionally(cause);
		ctx.close();
	}
	
	private boolean isNotSslException(final Throwable cause)
	{
		return !(cause.getCause() instanceof SSLException
			|| cause instanceof DecoderException || cause instanceof NotSslRecordException);
	}
	
	private boolean isNotConnectionReset(final Throwable cause)
	{
		return this.connectionClosedStrings.stream().noneMatch(connectionClosedString ->
			(isNotBlank(cause.getMessage()) && cause.getMessage().contains(connectionClosedString))
				|| (cause.getCause() != null && isNotBlank(cause.getCause().getMessage()) && cause.getCause()
				.getMessage()
				.contains(connectionClosedString)));
	}
}