org.drools.container.spring.namespace.KnowledgeAgentDefinitionParser Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2010 JBoss Inc
 *
 * 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.container.spring.namespace;

import static org.drools.container.spring.namespace.DefinitionParserHelper.emptyAttributeCheck;

import org.drools.container.spring.beans.KnowledgeAgentBeanFactory;
import org.drools.container.spring.beans.StatelessKnowledgeSessionBeanFactory;
import org.springframework.beans.PropertyValue;
import org.springframework.beans.factory.config.BeanDefinition;
import org.springframework.beans.factory.config.RuntimeBeanReference;
import org.springframework.beans.factory.support.AbstractBeanDefinition;
import org.springframework.beans.factory.support.BeanDefinitionBuilder;
import org.springframework.beans.factory.support.ManagedList;
import org.springframework.beans.factory.xml.AbstractBeanDefinitionParser;
import org.springframework.beans.factory.xml.ParserContext;
import org.springframework.util.StringUtils;
import org.w3c.dom.Element;

public class KnowledgeAgentDefinitionParser extends AbstractBeanDefinitionParser {

    @Override
    protected AbstractBeanDefinition parseInternal(Element element,
                                                   ParserContext parserContext) {

        BeanDefinitionBuilder factory = BeanDefinitionBuilder.rootBeanDefinition( KnowledgeAgentBeanFactory.class );

        String id = element.getAttribute( "id" );
        emptyAttributeCheck( element.getLocalName(),
                             "id",
                             id );
        factory.addPropertyValue( "id",
                                  id );

        String kbase = element.getAttribute( "kbase" );
        if ( StringUtils.hasText( kbase ) ) {
            factory.addPropertyReference( "kbase",
                                          kbase );
        }

        String newInstance = element.getAttribute( "new-instance" );
        if ( StringUtils.hasText( newInstance ) ) {
            factory.addPropertyValue( "newInstance",
                                      newInstance );
        }
        
        String useKbaseClassloader = element.getAttribute( "use-kbase-classloader" );
        if (!useKbaseClassloader.trim().equals("")) {
            factory.addPropertyValue( "useKbaseClassloader",
                                      useKbaseClassloader );
        }

        ManagedList resources = KnowledgeBaseDefinitionParser.getResources( element,
                                                                            parserContext,
                                                                            factory );

        if ( resources != null ) {
            factory.addPropertyValue( "resources",
                                      resources );
        }

        // inject the kagent into any stateless sessions
        for ( String beanName : parserContext.getRegistry().getBeanDefinitionNames() ) {
            BeanDefinition def = parserContext.getRegistry().getBeanDefinition( beanName );
            if ( StatelessKnowledgeSessionBeanFactory.class.getName().equals( def.getBeanClassName() ) ) {
                PropertyValue pvalue = def.getPropertyValues().getPropertyValue( "kbase" );
                RuntimeBeanReference tbf = (RuntimeBeanReference) pvalue.getValue();
                if ( kbase.equals( tbf.getBeanName() ) ) {
                    def.getPropertyValues().addPropertyValue( "knowledgeAgent",
                                                              new RuntimeBeanReference( id ) );
                }
            }
        }

        return factory.getBeanDefinition();
    }
}