/*
* *************************************************************************************
* Copyright (C) 2008 EsperTech, Inc. All rights reserved. *
* http://esper.codehaus.org *
* http://www.espertech.com *
* ---------------------------------------------------------------------------------- *
* The software in this package is published under the terms of the GPL license *
* a copy of which has been included with this distribution in the license.txt file. *
* *************************************************************************************
*/
package com.espertech.esper.dataflow.ops;
import com.espertech.esper.client.EPException;
import com.espertech.esper.client.EventBean;
import com.espertech.esper.client.EventType;
import com.espertech.esper.client.dataflow.EPDataFlowEventBeanCollector;
import com.espertech.esper.client.dataflow.EPDataFlowEventBeanCollectorContext;
import com.espertech.esper.core.context.util.AgentInstanceContext;
import com.espertech.esper.core.context.util.EPStatementAgentInstanceHandle;
import com.espertech.esper.core.service.EPStatementHandleCallback;
import com.espertech.esper.core.service.StatementAgentInstanceFilterVersion;
import com.espertech.esper.dataflow.annotations.DataFlowContext;
import com.espertech.esper.dataflow.annotations.DataFlowOpParameter;
import com.espertech.esper.dataflow.annotations.DataFlowOperator;
import com.espertech.esper.dataflow.interfaces.*;
import com.espertech.esper.epl.core.StreamTypeServiceImpl;
import com.espertech.esper.epl.expression.ExprNode;
import com.espertech.esper.epl.expression.ExprValidationException;
import com.espertech.esper.filter.FilterHandleCallback;
import com.espertech.esper.filter.FilterSpecCompiled;
import com.espertech.esper.filter.FilterSpecCompiler;
import com.espertech.esper.filter.FilterValueSet;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.List;
import java.util.concurrent.LinkedBlockingDeque;
@DataFlowOperator
public class EventBusSource implements DataFlowSourceOperator, DataFlowOpLifecycle, FilterHandleCallback {
@DataFlowOpParameter
private ExprNode filter;
@DataFlowOpParameter
private EPDataFlowEventBeanCollector collector;
@DataFlowContext
private EPDataFlowEmitter graphContext;
private EventType eventType;
private AgentInstanceContext agentInstanceContext;
private EPStatementHandleCallback callbackHandle;
private LinkedBlockingDeque<Object> emittables = new LinkedBlockingDeque<Object>();
private boolean submitEventBean;
private ThreadLocal<EPDataFlowEventBeanCollectorContext> collectorDataTL = new ThreadLocal<EPDataFlowEventBeanCollectorContext>() {
protected synchronized EPDataFlowEventBeanCollectorContext initialValue() {
return null;
}
};
public DataFlowOpInitializeResult initialize(DataFlowOpInitializateContext context) throws Exception {
if (context.getOutputPorts().size() != 1) {
throw new IllegalArgumentException("EventBusSource operator requires one output stream but produces " + context.getOutputPorts().size() + " streams");
}
DataFlowOpOutputPort portZero = context.getOutputPorts().get(0);
if (portZero.getOptionalDeclaredType() == null || portZero.getOptionalDeclaredType().getEventType() == null) {
throw new IllegalArgumentException("EventBusSource operator requires an event type declated for the output stream");
}
if (!portZero.getOptionalDeclaredType().isUnderlying()) {
submitEventBean = true;
}
this.eventType = portZero.getOptionalDeclaredType().getEventType();
this.agentInstanceContext = context.getAgentInstanceContext();
return new DataFlowOpInitializeResult();
}
public void next() throws InterruptedException {
Object next = emittables.take();
graphContext.submit(next);
}
public void matchFound(EventBean theEvent, Collection<FilterHandleCallback> allStmtMatches) {
if (collector != null) {
EPDataFlowEventBeanCollectorContext holder = collectorDataTL.get();
if (holder == null) {
holder = new EPDataFlowEventBeanCollectorContext(graphContext, submitEventBean, theEvent);
collectorDataTL.set(holder);
}
else {
holder.setEvent(theEvent);
}
collector.collect(holder);
}
else if (submitEventBean) {
emittables.add(theEvent);
}
else {
emittables.add(theEvent.getUnderlying());
}
}
public boolean isSubSelect() {
return false;
}
public String getStatementId() {
return agentInstanceContext.getStatementId();
}
public void open(DataFlowOpOpenContext openContext) {
FilterValueSet valueSet;
try {
List<ExprNode> filters = Collections.emptyList();
if (filter != null) {
filters = Collections.singletonList(filter);
}
FilterSpecCompiled spec = FilterSpecCompiler.makeFilterSpec(eventType, eventType.getName(), filters, null,
null, null, new StreamTypeServiceImpl(eventType, eventType.getName(), true, agentInstanceContext.getEngineURI()), null, agentInstanceContext.getStatementContext(), new ArrayList<Integer>());
valueSet = spec.getValueSet(null, agentInstanceContext, null);
}
catch (ExprValidationException ex) {
throw new EPException("Failed to open filter: " + ex.getMessage(), ex);
}
EPStatementAgentInstanceHandle handle = new EPStatementAgentInstanceHandle(agentInstanceContext.getStatementContext().getEpStatementHandle(), agentInstanceContext.getAgentInstanceLock(), 0, new StatementAgentInstanceFilterVersion());
callbackHandle = new EPStatementHandleCallback(handle, this);
agentInstanceContext.getStatementContext().getFilterService().add(valueSet, callbackHandle);
}
public synchronized void close(DataFlowOpCloseContext openContext) {
if (callbackHandle != null) {
agentInstanceContext.getStatementContext().getFilterService().remove(callbackHandle);
callbackHandle = null;
}
}
}