/*
* Copyright 2008 Red Hat
*
* 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 org.drools.examples.broker;
import org.drools.KnowledgeBase;
import org.drools.KnowledgeBaseConfiguration;
import org.drools.KnowledgeBaseFactory;
import org.drools.builder.KnowledgeBuilder;
import org.drools.builder.KnowledgeBuilderFactory;
import org.drools.builder.ResourceType;
import org.drools.conf.EventProcessingOption;
import org.drools.examples.broker.events.Event;
import org.drools.examples.broker.events.EventReceiver;
import org.drools.examples.broker.model.Company;
import org.drools.examples.broker.model.CompanyRegistry;
import org.drools.examples.broker.model.StockTick;
import org.drools.examples.broker.ui.BrokerWindow;
import org.drools.io.ResourceFactory;
import org.drools.logger.KnowledgeRuntimeLogger;
import org.drools.logger.KnowledgeRuntimeLoggerFactory;
import org.drools.runtime.StatefulKnowledgeSession;
import org.drools.runtime.rule.WorkingMemoryEntryPoint;
/**
* The broker application
*
* @author etirelli
*/
public class Broker implements EventReceiver, BrokerServices {
private static final String RULES_FILE = "/broker.drl";
private BrokerWindow window;
private CompanyRegistry companies;
private StatefulKnowledgeSession session;
private WorkingMemoryEntryPoint tickStream;
public Broker(BrokerWindow window,
CompanyRegistry companies) {
super();
this.window = window;
this.companies = companies;
this.session = createSession();
this.tickStream = this.session.getWorkingMemoryEntryPoint( "StockTick stream" );
}
@SuppressWarnings("unchecked")
public void receive(Event<?> event) {
try {
StockTick tick = ((Event<StockTick>) event).getObject();
Company company = this.companies.getCompany( tick.getSymbol() );
this.tickStream.insert( tick );
this.session.getAgenda().getAgendaGroup( "evaluation" ).setFocus();
this.session.fireAllRules();
window.updateCompany( company.getSymbol() );
window.updateTick( tick );
} catch ( Exception e ) {
System.err.println("=============================================================");
System.err.println("Unexpected exception caught: "+e.getMessage() );
e.printStackTrace();
}
}
private StatefulKnowledgeSession createSession() {
KnowledgeBase kbase = loadRuleBase();
StatefulKnowledgeSession session = kbase.newStatefulKnowledgeSession();
//KnowledgeRuntimeLogger logger = KnowledgeRuntimeLoggerFactory.newConsoleLogger( session );
session.setGlobal( "services", this );
for( Company company : this.companies.getCompanies() ) {
session.insert( company );
}
session.fireAllRules();
return session;
}
private KnowledgeBase loadRuleBase() {
KnowledgeBuilder builder = KnowledgeBuilderFactory.newKnowledgeBuilder();
try {
builder.add( ResourceFactory.newInputStreamResource( Broker.class.getResourceAsStream( RULES_FILE ) ),
ResourceType.DRL);
} catch ( Exception e ) {
e.printStackTrace();
}
if( builder.hasErrors() ) {
System.err.println(builder.getErrors());
System.exit( 0 );
}
KnowledgeBaseConfiguration conf = KnowledgeBaseFactory.newKnowledgeBaseConfiguration();
conf.setOption( EventProcessingOption.STREAM );
//System.out.println(((RuleBaseConfiguration)conf).getEventProcessingMode());
KnowledgeBase kbase = KnowledgeBaseFactory.newKnowledgeBase( conf );
kbase.addKnowledgePackages( builder.getKnowledgePackages() );
return kbase;
}
public void log(String message) {
window.log( message );
}
}