/* * ************************************************************************************* * 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.core.context.stmt; import com.espertech.esper.client.EventBean; import com.espertech.esper.epl.expression.ExprEvaluatorContext; import com.espertech.esper.epl.expression.ExprSubselectStrategy; import java.util.Collection; import java.util.HashMap; import java.util.Map; public class AIRegistrySubselectMap implements AIRegistrySubselect, ExprSubselectStrategy { private final Map<Integer, ExprSubselectStrategy> strategies; public AIRegistrySubselectMap() { strategies = new HashMap<Integer, ExprSubselectStrategy>(); } public void assignService(int num, ExprSubselectStrategy subselectStrategy) { strategies.put(num, subselectStrategy); } public void deassignService(int num) { strategies.remove(num); } public Collection<EventBean> evaluateMatching(EventBean[] eventsPerStream, ExprEvaluatorContext exprEvaluatorContext) { int agentInstanceId = exprEvaluatorContext.getAgentInstanceId(); ExprSubselectStrategy strategy = strategies.get(agentInstanceId); return strategy.evaluateMatching(eventsPerStream, exprEvaluatorContext); } public int getAgentInstanceCount() { return strategies.size(); } }