com.lmax.disruptor.dsl.ExceptionHandlerSetting Maven / Gradle / Ivy
/*
* Copyright 2011 LMAX Ltd.
*
* 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.lmax.disruptor.dsl;
import com.lmax.disruptor.BatchEventProcessor;
import com.lmax.disruptor.EventHandler;
import com.lmax.disruptor.EventProcessor;
import com.lmax.disruptor.ExceptionHandler;
/**
* A support class used as part of setting an exception handler for a specific event handler.
* For example:
* disruptorWizard.handleExceptionsIn(eventHandler).with(exceptionHandler);
*
* @param the type of event being handled.
*/
public class ExceptionHandlerSetting
{
private final EventHandler eventHandler;
private final ConsumerRepository consumerRepository;
ExceptionHandlerSetting(
final EventHandler eventHandler,
final ConsumerRepository consumerRepository)
{
this.eventHandler = eventHandler;
this.consumerRepository = consumerRepository;
}
/**
* Specify the {@link ExceptionHandler} to use with the event handler.
*
* @param exceptionHandler the exception handler to use.
*/
@SuppressWarnings("unchecked")
public void with(ExceptionHandler exceptionHandler)
{
final EventProcessor eventProcessor = consumerRepository.getEventProcessorFor(eventHandler);
if (eventProcessor instanceof BatchEventProcessor)
{
((BatchEventProcessor) eventProcessor).setExceptionHandler(exceptionHandler);
consumerRepository.getBarrierFor(eventHandler).alert();
}
else
{
throw new RuntimeException(
"EventProcessor: " + eventProcessor + " is not a BatchEventProcessor " +
"and does not support exception handlers");
}
}
}