/*
* Copyright 2013 Red Hat Inc. and/or its affiliates and other contributors.
*
* 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.switchyard.component.common.knowledge.session;
import java.util.Map;
import java.util.Properties;
import org.kie.api.KieBase;
import org.kie.api.KieServices;
import org.kie.api.builder.KieScanner;
import org.kie.api.persistence.jpa.KieStoreServices;
import org.kie.api.runtime.Environment;
import org.kie.api.runtime.KieContainer;
import org.kie.api.runtime.KieSession;
import org.kie.api.runtime.KieSessionConfiguration;
import org.kie.api.runtime.StatelessKieSession;
import org.switchyard.ServiceDomain;
import org.switchyard.component.common.knowledge.CommonKnowledgeMessages;
import org.switchyard.component.common.knowledge.config.model.ContainerModel;
import org.switchyard.component.common.knowledge.config.model.KnowledgeComponentImplementationModel;
import org.switchyard.component.common.knowledge.util.Channels;
import org.switchyard.component.common.knowledge.util.Configurations;
import org.switchyard.component.common.knowledge.util.Containers;
import org.switchyard.component.common.knowledge.util.Disposals;
import org.switchyard.component.common.knowledge.util.Environments;
import org.switchyard.component.common.knowledge.util.Listeners;
import org.switchyard.component.common.knowledge.util.Loggers;
/**
* A Container-based KnowledgeSessionFactory.
*
* @author David Ward <<a href="mailto:dward@jboss.org">dward@jboss.org</a>> © 2012 Red Hat Inc.
*/
class KnowledgeContainerSessionFactory extends KnowledgeSessionFactory {
private final ContainerModel _containerModel;
private final KieContainer _kieContainer;
private final KieSessionConfiguration _sessionConfiguration;
KnowledgeContainerSessionFactory(KnowledgeComponentImplementationModel model, ClassLoader loader, ServiceDomain domain, Properties propertyOverrides) {
super(model, loader, domain, propertyOverrides);
_containerModel = Containers.getContainerModel(model);
_kieContainer = Containers.getContainer(_containerModel);
_sessionConfiguration = Configurations.getSessionConfiguration(getModel(), getLoader(), getPropertyOverrides());
registerScannerForDisposal();
}
/**
* {@inheritDoc}
*/
@Override
public KnowledgeSession newStatelessSession() {
StatelessKieSession stateless = newStatelessKieSession();
KnowledgeDisposal loggersDisposal = Loggers.registerLoggersForDisposal(getModel(), getLoader(), stateless);
Listeners.registerListeners(getModel(), getLoader(), stateless);
Channels.registerChannels(getModel(), getLoader(), getDomain(), stateless);
return new KnowledgeSession(stateless, false, loggersDisposal);
}
/**
* {@inheritDoc}
*/
@Override
public KnowledgeSession newStatefulSession(Map<String, Object> environmentOverrides) {
KieSession stateful = newKieSession(environmentOverrides);
KnowledgeDisposal loggersDisposal = Loggers.registerLoggersForDisposal(getModel(), getLoader(), stateful);
Listeners.registerListeners(getModel(), getLoader(), stateful);
Channels.registerChannels(getModel(), getLoader(), getDomain(), stateful);
return new KnowledgeSession(stateful, false, false, loggersDisposal);
}
/**
* {@inheritDoc}
*/
@Override
public KnowledgeSession getPersistentSession(Map<String, Object> environmentOverrides, Integer sessionId) {
KieSession stateful = getPersistentKieSession(environmentOverrides, sessionId);
KnowledgeDisposal loggersDisposal = Loggers.registerLoggersForDisposal(getModel(), getLoader(), stateful);
Listeners.registerListeners(getModel(), getLoader(), stateful);
Channels.registerChannels(getModel(), getLoader(), getDomain(), stateful);
return new KnowledgeSession(stateful, false, true, loggersDisposal);
}
private StatelessKieSession newStatelessKieSession() {
if (_containerModel != null) {
String sessionName = _containerModel.getSessionName();
if (sessionName != null) {
return _kieContainer.newStatelessKieSession(sessionName, _sessionConfiguration);
}
String baseName = _containerModel.getBaseName();
if (baseName != null) {
return _kieContainer.getKieBase(baseName).newStatelessKieSession(_sessionConfiguration);
}
}
return _kieContainer.newStatelessKieSession(_sessionConfiguration);
}
private KieSession newKieSession(Map<String, Object> environmentOverrides) {
Environment environment = Environments.getEnvironment(environmentOverrides);
if (_containerModel != null) {
String sessionName = _containerModel.getSessionName();
if (sessionName != null) {
return _kieContainer.newKieSession(sessionName, environment, _sessionConfiguration);
}
String baseName = _containerModel.getBaseName();
if (baseName != null) {
return _kieContainer.getKieBase(baseName).newKieSession(_sessionConfiguration, environment);
}
}
return _kieContainer.newKieSession(environment, _sessionConfiguration);
}
private KieSession getPersistentKieSession(Map<String, Object> environmentOverrides, Integer sessionId) {
if (_containerModel != null) {
String baseName = _containerModel.getBaseName();
KieBase base = baseName != null ? _kieContainer.getKieBase(baseName) : _kieContainer.getKieBase();
KieStoreServices kieStoreServices = KieServices.Factory.get().getStoreServices();
Environment environment = Environments.getEnvironment(environmentOverrides);
KieSession session = null;
if (sessionId != null) {
session = kieStoreServices.loadKieSession(sessionId, base, _sessionConfiguration, environment);
}
if (session == null) {
session = kieStoreServices.newKieSession(base, _sessionConfiguration, environment);
}
return session;
}
throw CommonKnowledgeMessages.MESSAGES.manifestContainerRequiredInConfigurationForPersistentSessions();
}
private void registerScannerForDisposal() {
if (_containerModel != null && _containerModel.isScan()) {
Long scanInterval = _containerModel.getScanInterval();
if (scanInterval == null) {
scanInterval = Long.valueOf(60000);
}
long si = scanInterval.longValue();
if (si < 1) {
throw CommonKnowledgeMessages.MESSAGES.containerScanIntervalMustBePositive();
}
KieScanner scanner = KieServices.Factory.get().newKieScanner(_kieContainer);
addDisposals(Disposals.newDisposal(scanner));
scanner.start(si);
}
}
}