org.drools.compiler.PackageBuilder Maven / Gradle / Ivy
/*
* Copyright 2005 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.compiler;
import java.beans.IntrospectionException;
import java.io.IOException;
import java.io.InputStream;
import java.io.Reader;
import java.io.Serializable;
import java.io.StringReader;
import java.lang.reflect.Field;
import java.lang.reflect.Method;
import java.lang.reflect.InvocationTargetException;
import java.util.*;
import org.drools.ChangeSet;
import org.drools.PackageIntegrationException;
import org.drools.RuleBase;
import org.drools.RuntimeDroolsException;
import org.drools.base.ClassFieldAccessor;
import org.drools.base.ClassFieldAccessorCache;
import org.drools.base.ClassFieldAccessorStore;
import org.drools.base.TypeResolver;
import org.drools.base.evaluators.TimeIntervalParser;
import org.drools.base.mvel.MVELCompileable;
import org.drools.builder.DecisionTableConfiguration;
import org.drools.builder.ResourceConfiguration;
import org.drools.builder.ResourceType;
import org.drools.builder.conf.impl.JaxbConfigurationImpl;
import org.drools.common.InternalRuleBase;
import org.drools.commons.jci.problems.CompilationProblem;
import org.drools.compiler.xml.XmlPackageReader;
import org.drools.core.util.ClassUtils;
import org.drools.core.util.DroolsStreamUtils;
import org.drools.core.util.StringUtils;
import org.drools.core.util.asm.ClassFieldInspector;
import org.drools.definition.process.Process;
import org.drools.definition.type.FactField;
import org.drools.definition.type.Position;
import org.drools.factmodel.AnnotationDefinition;
import org.drools.factmodel.ClassBuilder;
import org.drools.factmodel.ClassBuilderFactory;
import org.drools.factmodel.ClassDefinition;
import org.drools.factmodel.FieldDefinition;
import org.drools.factmodel.traits.Thing;
import org.drools.factmodel.traits.TraitFactory;
import org.drools.factmodel.traits.TraitRegistry;
import org.drools.factmodel.traits.TraitableBean;
import org.drools.factmodel.traits.Traitable;
import org.drools.factmodel.traits.Trait;
import org.drools.facttemplates.FactTemplate;
import org.drools.facttemplates.FactTemplateImpl;
import org.drools.facttemplates.FieldTemplate;
import org.drools.facttemplates.FieldTemplateImpl;
import org.drools.io.Resource;
import org.drools.io.impl.ByteArrayResource;
import org.drools.io.impl.ClassPathResource;
import org.drools.io.impl.DescrResource;
import org.drools.io.impl.ReaderResource;
import org.drools.io.internal.InternalResource;
import org.drools.lang.descr.AnnotationDescr;
import org.drools.lang.descr.AttributeDescr;
import org.drools.lang.descr.BaseDescr;
import org.drools.lang.descr.EntryPointDeclarationDescr;
import org.drools.lang.descr.FactTemplateDescr;
import org.drools.lang.descr.FieldTemplateDescr;
import org.drools.lang.descr.FunctionDescr;
import org.drools.lang.descr.FunctionImportDescr;
import org.drools.lang.descr.GlobalDescr;
import org.drools.lang.descr.ImportDescr;
import org.drools.lang.descr.PackageDescr;
import org.drools.lang.descr.PatternDescr;
import org.drools.lang.descr.RuleDescr;
import org.drools.lang.descr.TypeDeclarationDescr;
import org.drools.lang.descr.TypeFieldDescr;
import org.drools.lang.dsl.DSLMappingFile;
import org.drools.lang.dsl.DSLTokenizedMappingFile;
import org.drools.lang.dsl.DefaultExpander;
import org.drools.reteoo.ReteooRuleBase;
import org.drools.rule.Function;
import org.drools.rule.ImportDeclaration;
import org.drools.rule.JavaDialectRuntimeData;
import org.drools.rule.MVELDialectRuntimeData;
import org.drools.rule.Package;
import org.drools.rule.Rule;
import org.drools.rule.TypeDeclaration;
import org.drools.rule.builder.RuleBuildContext;
import org.drools.rule.builder.RuleBuilder;
import org.drools.rule.builder.dialect.DialectError;
import org.drools.runtime.pipeline.impl.DroolsJaxbHelperProviderImpl;
import org.drools.runtime.rule.Activation;
import org.drools.spi.InternalReadAccessor;
import org.drools.type.DateFormats;
import org.drools.type.DateFormatsImpl;
import org.drools.util.CompositeClassLoader;
import org.drools.xml.XmlChangeSetReader;
import org.xml.sax.SAXException;
/**
* This is the main compiler class for parsing and compiling rules and
* assembling or merging them into a binary Package instance. This can be done
* by merging into existing binary packages, or totally from source.
*
* If you are using the Java dialect the JavaDialectConfiguration will attempt
* to validate that the specified compiler is in the classpath, using
* ClassLoader.loasClass(String). If you intented to just Janino sa the compiler
* you must either overload the compiler property before instantiating this
* class or the PackageBuilder, or make sure Eclipse is in the classpath, as
* Eclipse is the default.
*
* Normally, a complete package is built using one of the applicable
* addPackageFromXXX methods. It is however possible to construct a package
* incrementally by adding individual component parts. When a package is built
* incrementally package level attributes are cached and applied to Rules
* subsequently added. Caution should be exercised when using the same
* PackageBuilder to construct packages from multiple sources as the cached
* package level attributes will still apply even if the resource added to
* PackageBuilder does not explicitly include package level attributes.
*/
public class PackageBuilder {
private Map pkgRegistryMap;
private List results;
private final PackageBuilderConfiguration configuration;
public static final RuleBuilder ruleBuilder = new RuleBuilder();
/**
* Optional RuleBase for incremental live building
*/
private ReteooRuleBase ruleBase;
/**
* default dialect
*/
private final String defaultDialect;
private CompositeClassLoader rootClassLoader;
private Map> globals;
private Resource resource;
private List dslFiles;
private TimeIntervalParser timeParser;
protected DateFormats dateFormats;
private ProcessBuilder processBuilder;
private PMMLCompiler pmmlCompiler;
private Map builtinTypes;
private Map cacheTypes;
//This list of package level attributes is initialised with the PackageDescr's attributes added to the builder.
//The package level attributes are inherited by individual rules not containing explicit overriding parameters.
//The map is keyed on the PackageDescr's namespace and contains a map of AttributeDescr's keyed on the
//AttributeDescr's name.
private Map> packageAttributes = new HashMap>();
//PackageDescrs' list of ImportDescrs are kept identical as subsequent PackageDescrs are added.
private Map> packages = new HashMap>();
/**
* Use this when package is starting from scratch.
*/
public PackageBuilder() {
this( (RuleBase) null,
null );
}
/**
* This will allow you to merge rules into this pre existing package.
*/
public PackageBuilder(final Package pkg) {
this( pkg,
null );
}
public PackageBuilder(final RuleBase ruleBase) {
this( ruleBase,
null );
}
/**
* Pass a specific configuration for the PackageBuilder
*
* PackageBuilderConfiguration is not thread safe and it also contains
* state. Once it is created and used in one or more PackageBuilders it
* should be considered immutable. Do not modify its properties while it is
* being used by a PackageBuilder.
*
* @param configuration
*/
public PackageBuilder(final PackageBuilderConfiguration configuration) {
this( (RuleBase) null,
configuration );
}
public PackageBuilder(Package pkg,
PackageBuilderConfiguration configuration) {
if ( configuration == null ) {
this.configuration = new PackageBuilderConfiguration();
} else {
this.configuration = configuration;
}
this.dateFormats = null;//(DateFormats) this.environment.get( EnvironmentName.DATE_FORMATS );
if ( this.dateFormats == null ) {
this.dateFormats = new DateFormatsImpl();
//this.environment.set( EnvironmentName.DATE_FORMATS , this.dateFormats );
}
this.rootClassLoader = this.configuration.getClassLoader();
this.rootClassLoader.addClassLoader( getClass().getClassLoader() );
this.defaultDialect = this.configuration.getDefaultDialect();
this.pkgRegistryMap = new LinkedHashMap();
this.results = new ArrayList();
PackageRegistry pkgRegistry = new PackageRegistry( this,
pkg );
pkgRegistry.setDialect( this.defaultDialect );
this.pkgRegistryMap.put( pkg.getName(),
pkgRegistry );
globals = new HashMap>();
processBuilder = createProcessBuilder();
builtinTypes = new HashMap();
initBuiltinTypeDeclarations();
}
public PackageBuilder(RuleBase ruleBase,
PackageBuilderConfiguration configuration) {
if ( configuration == null ) {
this.configuration = new PackageBuilderConfiguration();
} else {
this.configuration = configuration;
}
if ( ruleBase != null ) {
this.rootClassLoader = ((InternalRuleBase) ruleBase).getRootClassLoader();
} else {
this.rootClassLoader = this.configuration.getClassLoader();
}
this.rootClassLoader.addClassLoader( getClass().getClassLoader() );
this.dateFormats = null;//(DateFormats) this.environment.get( EnvironmentName.DATE_FORMATS );
if ( this.dateFormats == null ) {
this.dateFormats = new DateFormatsImpl();
//this.environment.set( EnvironmentName.DATE_FORMATS , this.dateFormats );
}
// FIXME, we need to get drools to support "default" namespace.
//this.defaultNamespace = pkg.getName();
this.defaultDialect = this.configuration.getDefaultDialect();
this.pkgRegistryMap = new LinkedHashMap();
this.results = new ArrayList();
this.ruleBase = (ReteooRuleBase) ruleBase;
globals = new HashMap>();
processBuilder = createProcessBuilder();
builtinTypes = new HashMap();
initBuiltinTypeDeclarations();
}
private void initBuiltinTypeDeclarations() {
TypeDeclaration colType = new TypeDeclaration( "Collection" );
colType.setTypesafe( false );
colType.setTypeClass( Collection.class );
builtinTypes.put( "java.util.Collection",
colType );
TypeDeclaration mapType = new TypeDeclaration( "Map" );
mapType.setTypesafe( false );
mapType.setTypeClass( Map.class );
builtinTypes.put( "java.util.Map",
mapType );
TypeDeclaration activationType = new TypeDeclaration( "Activation" );
activationType.setTypesafe( false );
activationType.setTypeClass( Activation.class );
builtinTypes.put( Activation.class.getCanonicalName(),
activationType );
TypeDeclaration thingType = new TypeDeclaration( Thing.class.getName() );
thingType.setFormat( TypeDeclaration.Format.TRAIT );
thingType.setTypeClass( Thing.class );
builtinTypes.put( Thing.class.getCanonicalName(),
thingType );
ClassDefinition def = new ClassDefinition();
def.setClassName( thingType.getTypeClass().getName() );
def.setDefinedClass( Thing.class );
TraitRegistry.getInstance().addTrait( def );
}
private ProcessBuilder createProcessBuilder() {
try {
return ProcessBuilderFactory.newProcessBuilder( this );
} catch ( IllegalArgumentException e ) {
return null;
}
}
private PMMLCompiler getPMMLCompiler() {
if ( this.pmmlCompiler == null ) {
this.pmmlCompiler = PMMLCompilerFactory.getPMMLCompiler();
}
return this.pmmlCompiler;
}
/**
* Load a rule package from DRL source.
*
* @param reader
* @throws DroolsParserException
* @throws IOException
*/
public void addPackageFromDrl(final Reader reader) throws DroolsParserException,
IOException {
this.resource = new ReaderResource( reader );
final DrlParser parser = new DrlParser();
final PackageDescr pkg = parser.parse( reader );
this.results.addAll( parser.getErrors() );
if ( pkg == null ) {
this.results.add( new ParserError( "Parser returned a null Package",
0,
0 ) );
}
if ( !parser.hasErrors() ) {
addPackage( pkg );
}
this.resource = null;
}
public void addPackageFromDrl(Resource resource) throws DroolsParserException,
IOException {
this.resource = resource;
PackageDescr pkg = null;
boolean hasErrors = false;
if ( resource instanceof DescrResource ) {
pkg = (PackageDescr) ((DescrResource) resource).getDescr();
} else {
final DrlParser parser = new DrlParser();
pkg = parser.parse( resource.getInputStream() );
this.results.addAll( parser.getErrors() );
if ( pkg == null ) {
this.results.add( new ParserError( "Parser returned a null Package",
0,
0 ) );
}
hasErrors = parser.hasErrors();
}
if ( !hasErrors ) {
addPackage( pkg );
}
this.resource = null;
}
/**
* Load a rule package from XML source.
*
* @param reader
* @throws DroolsParserException
* @throws IOException
*/
public void addPackageFromXml(final Reader reader) throws DroolsParserException,
IOException {
this.resource = new ReaderResource( reader );
final XmlPackageReader xmlReader = new XmlPackageReader( this.configuration.getSemanticModules() );
xmlReader.getParser().setClassLoader( this.rootClassLoader );
try {
xmlReader.read( reader );
} catch ( final SAXException e ) {
throw new DroolsParserException( e.toString(),
e.getCause() );
}
addPackage( xmlReader.getPackageDescr() );
this.resource = null;
}
public void addPackageFromXml(final Resource resource) throws DroolsParserException,
IOException {
this.resource = resource;
final XmlPackageReader xmlReader = new XmlPackageReader( this.configuration.getSemanticModules() );
xmlReader.getParser().setClassLoader( this.rootClassLoader );
try {
xmlReader.read( resource.getReader() );
} catch ( final SAXException e ) {
throw new DroolsParserException( e.toString(),
e.getCause() );
}
addPackage( xmlReader.getPackageDescr() );
this.resource = null;
}
/**
* Load a rule package from DRL source using the supplied DSL configuration.
*
* @param source
* The source of the rules.
* @param dsl
* The source of the domain specific language configuration.
* @throws DroolsParserException
* @throws IOException
*/
public void addPackageFromDrl(final Reader source,
final Reader dsl) throws DroolsParserException,
IOException {
this.resource = new ReaderResource( source );
final DrlParser parser = new DrlParser();
final PackageDescr pkg = parser.parse( source,
dsl );
this.results.addAll( parser.getErrors() );
if ( !parser.hasErrors() ) {
addPackage( pkg );
}
this.resource = null;
}
public void addPackageFromDslr(final Resource resource) throws DroolsParserException,
IOException {
this.resource = resource;
final DrlParser parser = new DrlParser();
DefaultExpander expander = getDslExpander();
try {
if ( expander == null ) {
expander = new DefaultExpander();
}
String str = expander.expand( resource.getReader() );
if ( expander.hasErrors() ) {
this.results.addAll( expander.getErrors() );
}
final PackageDescr pkg = parser.parse( str );
this.results.addAll( parser.getErrors() );
if ( !parser.hasErrors() ) {
addPackage( pkg );
}
} catch ( IOException e ) {
throw new RuntimeException( e );
}
this.resource = null;
}
public void addPackageFromBrl(final Resource resource) throws DroolsParserException {
this.resource = resource;
try {
BusinessRuleProvider provider = BusinessRuleProviderFactory.getInstance().getProvider();
Reader knowledge = provider.getKnowledgeReader( resource );
DrlParser parser = new DrlParser();
if ( provider.hasDSLSentences() ) {
DefaultExpander expander = getDslExpander();
if ( null != expander ) {
knowledge = new StringReader( expander.expand( knowledge ) );
if ( expander.hasErrors() ) this.results.addAll( expander.getErrors() );
}
}
PackageDescr pkg = parser.parse( knowledge );
if ( parser.hasErrors() ) {
this.results.addAll( parser.getErrors() );
} else {
addPackage( pkg );
}
} catch ( Exception e ) {
throw new DroolsParserException( e );
} finally {
this.resource = null;
}
}
public void addDsl(Resource resource) throws IOException {
this.resource = resource;
DSLTokenizedMappingFile file = new DSLTokenizedMappingFile();
if ( !file.parseAndLoad( resource.getReader() ) ) {
this.results.addAll( file.getErrors() );
}
if ( this.dslFiles == null ) {
this.dslFiles = new ArrayList();
}
this.dslFiles.add( file );
this.resource = null;
}
/**
* Add a ruleflow (.rfm) asset to this package.
*/
public void addRuleFlow(Reader processSource) {
addProcessFromXml( processSource );
}
public void addProcessFromXml(Resource resource) {
this.resource = resource;
try {
this.results.addAll( processBuilder.addProcessFromXml( resource ) );
} catch ( Exception e ) {
if ( e instanceof RuntimeException ) {
throw (RuntimeException) e;
}
this.results.add( new ProcessLoadError( "Unable to load process.",
e ) );
}
this.results = getResults( this.results );
this.resource = null;
}
public void addProcessFromXml(Reader processSource) {
addProcessFromXml( new ReaderResource( processSource ) );
}
public void addKnowledgeResource(Resource resource,
ResourceType type,
ResourceConfiguration configuration) {
try {
if ( ResourceType.DRL.equals( type ) ) {
((InternalResource) resource).setResourceType( type );
addPackageFromDrl( resource );
} else if ( ResourceType.DESCR.equals( type ) ) {
((InternalResource) resource).setResourceType( type );
addPackageFromDrl( resource );
} else if ( ResourceType.DSLR.equals( type ) ) {
((InternalResource) resource).setResourceType( type );
addPackageFromDslr( resource );
} else if ( ResourceType.DSL.equals( type ) ) {
((InternalResource) resource).setResourceType( type );
addDsl( resource );
} else if ( ResourceType.XDRL.equals( type ) ) {
((InternalResource) resource).setResourceType( type );
addPackageFromXml( resource );
} else if ( ResourceType.BRL.equals( type ) ) {
((InternalResource) resource).setResourceType( type );
addPackageFromBrl( resource );
} else if ( ResourceType.DRF.equals( type ) ) {
((InternalResource) resource).setResourceType( type );
addProcessFromXml( resource );
} else if ( ResourceType.BPMN2.equals( type ) ) {
((InternalResource) resource).setResourceType( type );
BPMN2ProcessFactory.configurePackageBuilder( this );
addProcessFromXml( resource );
} else if ( ResourceType.DTABLE.equals( type ) ) {
((InternalResource) resource).setResourceType( type );
DecisionTableConfiguration dtableConfiguration = (DecisionTableConfiguration) configuration;
String string = DecisionTableFactory.loadFromInputStream( resource.getInputStream(),
dtableConfiguration );
addPackageFromDrl( new StringReader( string ) );
} else if ( ResourceType.PKG.equals( type ) ) {
InputStream is = resource.getInputStream();
Package pkg = (Package) DroolsStreamUtils.streamIn( is,
this.configuration.getClassLoader() );
is.close();
addPackage( pkg );
} else if ( ResourceType.CHANGE_SET.equals( type ) ) {
((InternalResource) resource).setResourceType( type );
XmlChangeSetReader reader = new XmlChangeSetReader( this.configuration.getSemanticModules() );
if ( resource instanceof ClassPathResource ) {
reader.setClassLoader( ((ClassPathResource) resource).getClassLoader(),
((ClassPathResource) resource).getClazz() );
} else {
reader.setClassLoader( this.configuration.getClassLoader(),
null );
}
ChangeSet changeSet = reader.read( resource.getReader() );
if ( changeSet == null ) {
// @TODO should log an error
}
for ( Resource nestedResource : changeSet.getResourcesAdded() ) {
InternalResource iNestedResourceResource = (InternalResource) nestedResource;
if ( iNestedResourceResource.isDirectory() ) {
this.resourceDirectories.add( iNestedResourceResource );
for ( Resource childResource : iNestedResourceResource.listResources() ) {
if ( ((InternalResource) childResource).isDirectory() ) {
continue; // ignore sub directories
}
((InternalResource) childResource).setResourceType( iNestedResourceResource.getResourceType() );
addKnowledgeResource( childResource,
iNestedResourceResource.getResourceType(),
iNestedResourceResource.getConfiguration() );
}
} else {
addKnowledgeResource( iNestedResourceResource,
iNestedResourceResource.getResourceType(),
iNestedResourceResource.getConfiguration() );
}
}
} else if ( ResourceType.XSD.equals( type ) ) {
JaxbConfigurationImpl confImpl = (JaxbConfigurationImpl) configuration;
String[] classes = DroolsJaxbHelperProviderImpl.addXsdModel( resource,
this,
confImpl.getXjcOpts(),
confImpl.getSystemId() );
for ( String cls : classes ) {
confImpl.getClasses().add( cls );
}
} else if ( ResourceType.PMML.equals( type ) ) {
PMMLCompiler compiler = getPMMLCompiler();
if ( compiler != null ) {
String theory = compiler.compile( resource.getInputStream(),
getPackageRegistry() );
addKnowledgeResource( new ByteArrayResource( theory.getBytes() ),
ResourceType.DRL,
configuration );
} else {
throw new RuntimeException( "Unknown resource type: " + type );
}
} else {
ResourceTypeBuilder builder = ResourceTypeBuilderRegistry.getInstance().getResourceTypeBuilder( type );
if ( builder != null ) {
builder.setPackageBuilder( this );
builder.addKnowledgeResource( resource,
type,
configuration );
} else {
throw new RuntimeException( "Unknown resource type: " + type );
}
}
} catch ( RuntimeException e ) {
throw e;
} catch ( Exception e ) {
throw new RuntimeException( e );
}
}
private Set resourceDirectories = new HashSet();
/**
* This adds a package from a Descr/AST This will also trigger a compile, if
* there are any generated classes to compile of course.
*/
public void addPackage(final PackageDescr packageDescr) {
validateUniqueRuleNames( packageDescr );
//Derive namespace
if ( isEmpty( packageDescr.getNamespace() ) ) {
packageDescr.setNamespace( this.configuration.getDefaultPackageName() );
}
if ( !checkNamespace( packageDescr.getNamespace() ) ) {
return;
}
//Gather all imports for all PackageDescrs for the current package and replicate into
//all PackageDescrs for the current package, thus maintaining a complete list of
//ImportDescrs for all PackageDescrs for the current package.
List packageDescrsForPackage = packages.get( packageDescr.getName() );
if ( packageDescrsForPackage == null ) {
packageDescrsForPackage = new ArrayList();
packages.put( packageDescr.getName(),
packageDescrsForPackage );
}
packageDescrsForPackage.add( packageDescr );
Set imports = new HashSet();
for ( PackageDescr pd : packageDescrsForPackage ) {
imports.addAll( pd.getImports() );
}
for ( PackageDescr pd : packageDescrsForPackage ) {
pd.getImports().clear();
//PackageDescr.getImports() can return a Collections.EmptyList which doesn't
//support addAll(). PackageDescr.addImport() ensures the list can be appended
for ( ImportDescr id : imports ) {
pd.addImport( id );
}
}
//Copy package level attributes for inclusion on individual rules
if ( packageDescr.getAttributes().size() > 0 ) {
Map pkgAttributes = packageAttributes.get( packageDescr.getNamespace() );
if ( pkgAttributes == null ) {
pkgAttributes = new HashMap();
this.packageAttributes.put( packageDescr.getNamespace(),
pkgAttributes );
}
for ( AttributeDescr attr : packageDescr.getAttributes() ) {
pkgAttributes.put( attr.getName(),
attr );
}
}
String dialectName = this.defaultDialect;
// see if this packageDescr overrides the current default dialect
for ( Iterator it = packageDescr.getAttributes().iterator(); it.hasNext(); ) {
AttributeDescr value = (AttributeDescr) it.next();
if ( "dialect".equals( value.getName() ) ) {
dialectName = value.getValue();
break;
}
}
PackageRegistry pkgRegistry = this.pkgRegistryMap.get( packageDescr.getNamespace() );
if ( pkgRegistry == null ) {
// initialise the package and namespace if it hasn't been used before
pkgRegistry = newPackage( packageDescr );
} else {
// merge into existing package
mergePackage( packageDescr );
}
// set the default dialect for this package
pkgRegistry.setDialect( dialectName );
// only try to compile if there are no parse errors
if ( !hasErrors() ) {
if ( !packageDescr.getFunctions().isEmpty() ) {
for ( final Iterator it = packageDescr.getFunctions().iterator(); it.hasNext(); ) {
FunctionDescr functionDescr = (FunctionDescr) it.next();
if ( isEmpty( functionDescr.getNamespace() ) ) {
// make sure namespace is set on components
functionDescr.setNamespace( packageDescr.getNamespace() );
}
if ( isEmpty( functionDescr.getDialect() ) ) {
// make sure namespace is set on components
functionDescr.setDialect( pkgRegistry.getDialect() );
}
preCompileAddFunction( functionDescr );
}
// iterate and compile
for ( final Iterator it = packageDescr.getFunctions().iterator(); it.hasNext(); ) {
// inherit the dialect from the package
FunctionDescr functionDescr = (FunctionDescr) it.next();
addFunction( functionDescr );
}
// We need to compile all the functions now, so scripting
// languages like mvel can find them
compileAll();
for ( final Iterator it = packageDescr.getFunctions().iterator(); it.hasNext(); ) {
FunctionDescr functionDescr = (FunctionDescr) it.next();
postCompileAddFunction( functionDescr );
}
}
// iterate and compile
for ( final Iterator it = packageDescr.getRules().iterator(); it.hasNext(); ) {
RuleDescr ruleDescr = (RuleDescr) it.next();
if ( isEmpty( ruleDescr.getNamespace() ) ) {
// make sure namespace is set on components
ruleDescr.setNamespace( packageDescr.getNamespace() );
}
Map pkgAttributes = packageAttributes.get( packageDescr.getNamespace() );
inheritPackageAttributes( pkgAttributes,
ruleDescr );
if ( isEmpty( ruleDescr.getDialect() ) ) {
ruleDescr.addAttribute( new AttributeDescr( "dialect",
pkgRegistry.getDialect() ) );
}
addRule( ruleDescr );
}
}
compileAll();
try {
reloadAll();
} catch ( Exception e ) {
this.results.add( new DialectError( "Unable to wire compiled classes, probably related to compilation failures:" + e.getMessage() ) );
}
updateResults();
// iterate and compile
if ( this.ruleBase != null ) {
for ( final Iterator it = packageDescr.getRules().iterator(); it.hasNext(); ) {
RuleDescr ruleDescr = (RuleDescr) it.next();
pkgRegistry = this.pkgRegistryMap.get( ruleDescr.getNamespace() );
this.ruleBase.addRule( pkgRegistry.getPackage(),
pkgRegistry.getPackage().getRule( ruleDescr.getName() ) );
}
}
}
// test
/**
* This checks to see if it should all be in the one namespace.
*/
private boolean checkNamespace(String newName) {
if ( this.configuration == null ) return true;
if ( (!this.pkgRegistryMap.isEmpty()) && (!this.pkgRegistryMap.containsKey( newName )) ) {
return this.configuration.isAllowMultipleNamespaces();
}
return true;
}
public boolean isEmpty(String string) {
return (string == null || string.trim().length() == 0);
}
public void updateResults() {
// some of the rules and functions may have been redefined
this.results = getResults( this.results );
}
public void compileAll() {
for ( PackageRegistry pkgRegistry : this.pkgRegistryMap.values() ) {
pkgRegistry.compileAll();
}
}
public void reloadAll() {
for ( PackageRegistry pkgRegistry : this.pkgRegistryMap.values() ) {
pkgRegistry.getDialectRuntimeRegistry().onBeforeExecute();
}
}
private List getResults(List results) {
for ( PackageRegistry pkgRegistry : this.pkgRegistryMap.values() ) {
results = pkgRegistry.getDialectCompiletimeRegistry().addResults( results );
}
return results;
}
public synchronized void addPackage(final Package newPkg) {
PackageRegistry pkgRegistry = this.pkgRegistryMap.get( newPkg.getName() );
Package pkg = null;
if ( pkgRegistry != null ) {
pkg = pkgRegistry.getPackage();
}
if ( pkg == null ) {
pkg = newPackage( new PackageDescr( newPkg.getName() ) ).getPackage();
}
// first merge anything related to classloader re-wiring
pkg.getDialectRuntimeRegistry().merge( newPkg.getDialectRuntimeRegistry(),
this.rootClassLoader );
if ( newPkg.getFunctions() != null ) {
for ( Map.Entry entry : newPkg.getFunctions().entrySet() ) {
pkg.addFunction( entry.getValue() );
}
}
pkg.getClassFieldAccessorStore().merge( newPkg.getClassFieldAccessorStore() );
pkg.getDialectRuntimeRegistry().onBeforeExecute();
// we have to do this before the merging, as it does some classloader resolving
TypeDeclaration lastType = null;
try {
// Resolve the class for the type declaation
if ( newPkg.getTypeDeclarations() != null ) {
// add type declarations
for ( TypeDeclaration type : newPkg.getTypeDeclarations().values() ) {
lastType = type;
type.setTypeClass( this.rootClassLoader.loadClass( type.getTypeClassName() ) );
}
}
} catch ( ClassNotFoundException e ) {
throw new RuntimeDroolsException( "unable to resolve Type Declaration class '" + lastType.getTypeName() + "'" );
}
// now merge the new package into the existing one
mergePackage( pkg,
newPkg );
}
/**
* Merge a new package with an existing package. Most of the work is done by
* the concrete implementations, but this class does some work (including
* combining imports, compilation data, globals, and the actual Rule objects
* into the package).
*/
private void mergePackage(final Package pkg,
final Package newPkg) {
// Merge imports
final Map imports = pkg.getImports();
imports.putAll( newPkg.getImports() );
String lastType = null;
try {
// merge globals
if ( newPkg.getGlobals() != null && newPkg.getGlobals() != Collections.EMPTY_MAP ) {
Map globals = pkg.getGlobals();
// Add globals
for ( final Map.Entry entry : newPkg.getGlobals().entrySet() ) {
final String identifier = entry.getKey();
final String type = entry.getValue();
lastType = type;
if ( globals.containsKey( identifier ) && !globals.get( identifier ).equals( type ) ) {
throw new PackageIntegrationException( pkg );
} else {
pkg.addGlobal( identifier,
this.rootClassLoader.loadClass( type ) );
// this isn't a package merge, it's adding to the rulebase, but I've put it here for convenience
this.globals.put( identifier,
this.rootClassLoader.loadClass( type ) );
}
}
}
} catch ( ClassNotFoundException e ) {
throw new RuntimeDroolsException( "Unable to resolve class '" + lastType + "'" );
}
// merge the type declarations
if ( newPkg.getTypeDeclarations() != null ) {
// add type declarations
for ( TypeDeclaration type : newPkg.getTypeDeclarations().values() ) {
// @TODO should we allow overrides? only if the class is not in use.
if ( !pkg.getTypeDeclarations().containsKey( type.getTypeName() ) ) {
// add to package list of type declarations
pkg.addTypeDeclaration( type );
}
}
}
final Rule[] newRules = newPkg.getRules();
for ( int i = 0; i < newRules.length; i++ ) {
final Rule newRule = newRules[i];
pkg.addRule( newRule );
}
//Merge The Rule Flows
if ( newPkg.getRuleFlows() != null ) {
final Map flows = newPkg.getRuleFlows();
for ( final Iterator iter = flows.values().iterator(); iter.hasNext(); ) {
final Process flow = (Process) iter.next();
pkg.addProcess( flow );
}
}
}
//
// private void validatePackageName(final PackageDescr packageDescr) {
// if ( (this.pkg == null || this.pkg.getName() == null || this.pkg.getName().equals( "" )) && (packageDescr.getName() == null || "".equals( packageDescr.getName() )) ) {
// throw new MissingPackageNameException( "Missing package name for rule package." );
// }
// if ( this.pkg != null && packageDescr.getName() != null && !"".equals( packageDescr.getName() ) && !this.pkg.getName().equals( packageDescr.getName() ) ) {
// throw new PackageMergeException( "Can't merge packages with different names. This package: " + this.pkg.getName() + " - New package: " + packageDescr.getName() );
// }
// return;
// }
private void validateUniqueRuleNames(final PackageDescr packageDescr) {
final Set names = new HashSet();
for ( final RuleDescr rule : packageDescr.getRules() ) {
final String name = rule.getName();
if ( names.contains( name ) ) {
this.results.add( new ParserError( "Duplicate rule name: " + name,
rule.getLine(),
rule.getColumn() ) );
}
names.add( name );
}
}
private PackageRegistry newPackage(final PackageDescr packageDescr) {
Package pkg;
if ( this.ruleBase == null || (pkg = this.ruleBase.getPackage( packageDescr.getName() )) == null ) {
// there is no rulebase or it does not define this package so define it
pkg = new Package( packageDescr.getName() );
pkg.setClassFieldAccessorCache( new ClassFieldAccessorCache( this.rootClassLoader ) );
// if there is a rulebase then add the package.
if ( this.ruleBase != null ) {
// Must lock here, otherwise the assumption about addPackage/getPackage behavior below might be violated
this.ruleBase.lock();
try {
this.ruleBase.addPackage( pkg );
pkg = this.ruleBase.getPackage( packageDescr.getName() );
} finally {
this.ruleBase.unlock();
}
} else {
// the RuleBase will also initialise the
pkg.getDialectRuntimeRegistry().onAdd( this.rootClassLoader );
}
}
PackageRegistry pkgRegistry = new PackageRegistry( this,
pkg );
// add default import for this namespace
pkgRegistry.addImport( packageDescr.getNamespace() + ".*" );
this.pkgRegistryMap.put( packageDescr.getName(),
pkgRegistry );
mergePackage( packageDescr );
return pkgRegistry;
}
private void mergePackage(final PackageDescr packageDescr) {
PackageRegistry pkgRegistry = this.pkgRegistryMap.get( packageDescr.getNamespace() );
for ( final ImportDescr importEntry : packageDescr.getImports() ) {
pkgRegistry.addImport( importEntry.getTarget() );
}
processEntryPointDeclarations( packageDescr );
processTypeDeclarations( packageDescr );
for ( final FunctionImportDescr functionImport : packageDescr.getFunctionImports() ) {
String importEntry = functionImport.getTarget();
pkgRegistry.addStaticImport( importEntry );
pkgRegistry.getPackage().addStaticImport( importEntry );
}
for ( final GlobalDescr global : packageDescr.getGlobals() ) {
final String identifier = global.getIdentifier();
String className = global.getType();
// JBRULES-3039: can't handle type name with generic params
while ( className.indexOf( '<' ) >= 0 ) {
className = className.replaceAll( "<[^<>]+?>",
"" );
}
Class< ? > clazz;
try {
clazz = pkgRegistry.getTypeResolver().resolveType( className );
pkgRegistry.getPackage().addGlobal( identifier,
clazz );
this.globals.put( identifier,
clazz );
} catch ( final ClassNotFoundException e ) {
this.results.add( new GlobalError( identifier,
global.getLine() ) );
e.printStackTrace();
}
}
// need to reinsert this to ensure that the package is the first/last one in the ordered map
// this feature is exploited by the knowledgeAgent
this.pkgRegistryMap.remove( packageDescr.getName() );
this.pkgRegistryMap.put( packageDescr.getName(),
pkgRegistry );
}
public TypeDeclaration getTypeDeclaration(Class< ? > cls) {
if ( cls.isPrimitive() || cls.isArray() ) {
return null;
}
// If this class has already been accessed, it'll be in the cache
TypeDeclaration tdecl = null;
PackageRegistry pkgReg = null;
if ( this.cacheTypes == null ) {
this.cacheTypes = new HashMap();
} else {
tdecl = cacheTypes.get( cls.getName() );
if ( tdecl != null ) {
return tdecl;
}
}
// Check if we are in the built-ins
tdecl = this.builtinTypes.get( (cls.getName()) );
if ( tdecl == null ) {
// No built-in
// Check if there is a user specified typedeclr
pkgReg = this.pkgRegistryMap.get( ClassUtils.getPackage( cls ) );
if ( pkgReg != null ) {
tdecl = pkgReg.getPackage().getTypeDeclaration( cls.getSimpleName() );
}
}
if ( tdecl == null ) {
// no typedeclr exists, so create one, which will be added to the cache
tdecl = new TypeDeclaration( cls.getSimpleName() );
// it's a new type declaration, so generate the @Position for it
ClassDefinition clsDef = tdecl.getTypeClassDef();
if ( clsDef == null ) {
clsDef = new ClassDefinition();
Collection fields = new LinkedList();
Class< ? > tempKlass = cls;
while ( tempKlass != null && tempKlass != Object.class ) {
for ( Field f : tempKlass.getDeclaredFields() ) {
fields.add( f );
}
tempKlass = tempKlass.getSuperclass();
}
List orderedFields = new ArrayList( fields.size() );
for ( int i = 0; i < fields.size(); i++ ) {
// as these could be set in any order, initialise first, to allow setting later.
orderedFields.add( null );
}
for ( Field fld : fields ) {
Position pos = fld.getAnnotation( Position.class );
if ( pos != null ) {
FieldDefinition fldDef = new FieldDefinition( fld.getName(),
fld.getType().getName() );
fldDef.setIndex( pos.value() );
orderedFields.set( pos.value(),
fldDef );
}
}
for ( FieldDefinition fld : orderedFields ) {
if ( fld != null ) {
// it's null if there is no @Position
clsDef.addField( fld );
}
}
tdecl.setTypeClassDef( clsDef );
}
}
// build up a set of all the super classes and interfaces
Set tdecls = new LinkedHashSet();
tdecls.add( tdecl );
buildTypeDeclarations(cls, tdecls);
// Iterate and for each typedeclr assign it's value if it's not already set
// We start from the rear as those are the furthest away classes and interfaces
TypeDeclaration[] tarray = tdecls.toArray( new TypeDeclaration[tdecls.size()] );
for ( int i = tarray.length - 1; i >= 0; i-- ) {
TypeDeclaration currentTDecl = tarray[i];
if ( (tdecl.getSetMask() & TypeDeclaration.ROLE_BIT) != TypeDeclaration.ROLE_BIT ) {
tdecl.setRole( currentTDecl.getRole() );
}
if ( (tdecl.getSetMask() & TypeDeclaration.FORMAT_BIT) != TypeDeclaration.FORMAT_BIT ) {
tdecl.setFormat( currentTDecl.getFormat() );
}
if ( (tdecl.getSetMask() & TypeDeclaration.TYPESAFE_BIT) != TypeDeclaration.TYPESAFE_BIT ) {
tdecl.setTypesafe( currentTDecl.isTypesafe() );
}
}
this.cacheTypes.put( cls.getName(),
tdecl );
return tdecl;
}
public void buildTypeDeclarations(Class< ? > cls,
Set tdecls) {
TypeDeclaration tdecl = null;
// Process current interfaces
Class< ? >[] intfs = cls.getInterfaces();
for ( Class< ? > intf : intfs ) {
buildTypeDeclarationInterfaces( intf,
tdecls );
}
// Process super classes and their interfaces
cls = cls.getSuperclass();
while ( tdecl == null && (cls != null && cls != Object.class) ) {
if ( !buildTypeDeclarationInterfaces( cls,
tdecls ) ) {
break;
}
cls = cls.getSuperclass();
}
}
public boolean buildTypeDeclarationInterfaces(Class cls,
Set tdecls) {
PackageRegistry pkgReg = null;
TypeDeclaration tdecl = null;
tdecl = this.builtinTypes.get( (cls.getName()) );
if ( tdecl == null ) {
pkgReg = this.pkgRegistryMap.get( ClassUtils.getPackage( cls ) );
if ( pkgReg != null ) {
tdecl = pkgReg.getPackage().getTypeDeclaration( cls.getSimpleName() );
}
}
if ( tdecl != null ) {
if ( !tdecls.add( tdecl ) ) {
return false; // the interface already exists, return to stop recursion
}
}
Class< ? >[] intfs = cls.getInterfaces();
for ( Class< ? > intf : intfs ) {
pkgReg = this.pkgRegistryMap.get( ClassUtils.getPackage( intf ) );
if ( pkgReg != null ) {
tdecl = pkgReg.getPackage().getTypeDeclaration( intf.getSimpleName() );
}
if ( tdecl != null ) {
tdecls.add( tdecl );
}
}
for ( Class< ? > intf : intfs ) {
if ( !buildTypeDeclarationInterfaces( intf,
tdecls ) ) {
return false;
}
}
return true;
}
/**
* Tries to determine the namespace (package) of a simple type chosen to be
* the superclass of a declared bean. Looks among imports, local
* declarations and previous declarations. Means that a class can't extend
* another class declared in package that has not been loaded yet.
*
* @param sup
* the simple name of the superclass
* @param packageDescr
* the descriptor of the package the base class is declared in
* @param pkgRegistry
* the current package registry
* @return the fully qualified name of the superclass
*/
private String resolveType( String sup,
PackageDescr packageDescr,
PackageRegistry pkgRegistry) {
//look among imports
for ( ImportDescr id : packageDescr.getImports() ) {
if ( id.getTarget().endsWith( "." + sup ) ) {
//System.out.println("Replace supertype " + sup + " with full name " + id.getTarget());
return id.getTarget();
}
}
//look among local declarations
if ( pkgRegistry != null ) {
for ( String declaredName : pkgRegistry.getPackage().getTypeDeclarations().keySet() ) {
if ( declaredName.endsWith( sup ) ) sup = pkgRegistry.getPackage().getTypeDeclaration( declaredName ).getTypeClass().getName();
}
}
if ( (sup != null) && (!sup.contains( "." )) && (packageDescr.getNamespace() != null && packageDescr.getNamespace().length() > 0) ) {
for ( TypeDeclarationDescr td : packageDescr.getTypeDeclarations() ) {
if ( sup.equals( td.getTypeName() ) ) sup = packageDescr.getNamespace() + "." + sup;
}
}
return sup;
}
/**
* Resolves and sets the superclass (name and package) for a given type
* declaration descriptor The declared supertype, if any, may be a simple
* name or a fully qualified one. In the former case, the simple name could
* be the local name of some f.q.n. which has to be resolved
*
* @param typeDescr
* the descriptor of the declared superclass whose superclass
* will be identified
* @param packageDescr
* the descriptor of the package the class is declared in
*/
private void fillSuperType(TypeDeclarationDescr typeDescr,
PackageDescr packageDescr) {
for ( TypeDeclarationDescr.QualifiedName qname : typeDescr.getSuperTypes() ) {
String declaredSuperType = qname.getFullName();
if ( declaredSuperType != null ) {
int separator = declaredSuperType.lastIndexOf( "." );
boolean qualified = separator > 0;
// check if a simple name corresponds to a f.q.n.
if ( !qualified ) {
declaredSuperType =
resolveType( declaredSuperType,
packageDescr,
this.pkgRegistryMap.get( typeDescr.getNamespace() ) );
// sets supertype name and supertype package
separator = declaredSuperType.lastIndexOf( "." );
if ( separator < 0 ) {
this.results.add( new TypeDeclarationError(
"Cannot resolve supertype '" + declaredSuperType + "'",
typeDescr.getLine() ) );
qname.setName( null );
qname.setNamespace( null );
} else {
qname.setName(declaredSuperType.substring(separator + 1));
qname.setNamespace(declaredSuperType.substring(0,
separator));
}
}
}
}
}
private void fillFieldTypes( TypeDeclarationDescr typeDescr,
PackageDescr packageDescr) {
for ( TypeFieldDescr field : typeDescr.getFields().values() ) {
String declaredType = field.getPattern().getObjectType();
if ( declaredType != null ) {
int separator = declaredType.lastIndexOf( "." );
boolean qualified = separator > 0;
// check if a simple name corresponds to a f.q.n.
if ( !qualified ) {
declaredType =
resolveType( declaredType,
packageDescr,
this.pkgRegistryMap.get( typeDescr.getNamespace() ) );
field.getPattern().setObjectType( declaredType );
}
}
}
}
/**
* In order to build a declared class, the fields inherited from its
* superclass(es) are added to its declaration. Inherited descriptors are
* marked as such to distinguish them from native ones. Various scenarioes
* are possible. (i) The superclass has been declared in the DRL as well :
* the fields are cloned as inherited (ii) The superclass is imported
* (external), but some of its fields have been tagged with metadata (iii)
* The superclass is imported.
*
* The search for field descriptors is carried out in the order. (i) and
* (ii+iii) are mutually exclusive. The search is as such: (i) The
* superclass' declared fields are used to build the base class additional
* fields (iii) The superclass is inspected to discover its (public) fields,
* from which descriptors are generated (ii) Both (i) and (iii) are applied,
* but the declared fields override the inspected ones
*
* @param typeDescr
* The base class descriptor, to be completed with the inherited
* fields descriptors
* @return true if all went well
*/
private boolean mergeInheritedFields( TypeDeclarationDescr typeDescr ) {
if ( typeDescr.getSuperTypes().isEmpty() ) return false;
boolean merge = false;
for ( TypeDeclarationDescr.QualifiedName qname : typeDescr.getSuperTypes() ) {
String simpleSuperTypeName = qname.getName();
String superTypePackageName = qname.getNamespace();
String fullSuper = qname.getFullName();
merge = merge || mergeInheritedFields( simpleSuperTypeName, superTypePackageName, fullSuper, typeDescr );
}
return merge;
}
private boolean mergeInheritedFields( String simpleSuperTypeName, String superTypePackageName, String fullSuper, TypeDeclarationDescr typeDescr ) {
Map fieldMap = new LinkedHashMap();
boolean isSuperClassDeclared = true; //in the same package, or in a previous one
boolean isSuperClassTagged = false;
PackageRegistry registry = this.pkgRegistryMap.get( superTypePackageName );
Package pack = null;
if ( registry != null ) {
pack = registry.getPackage();
} else {
// If there is no regisrty the type isn't a DRL-declared type, which is forbidden.
// Avoid NPE JIRA-3041 when trying to access the registry. Avoid subsequent problems.
this.results.add(new TypeDeclarationError("Cannot extend supertype '" + fullSuper + "' (not a declared type)",
typeDescr.getLine()));
typeDescr.setType( null, null );
return false;
}
// if a class is declared in DRL, its package can't be null? The default package is replaced by "defaultpkg"
if ( pack != null ) {
// look for the supertype declaration in available packages
TypeDeclaration superTypeDeclaration = pack.getTypeDeclaration( simpleSuperTypeName );
if ( superTypeDeclaration != null ) {
ClassDefinition classDef = superTypeDeclaration.getTypeClassDef();
// inherit fields
for ( FactField fld : classDef.getFields() ) {
TypeFieldDescr inheritedFlDescr = TypeFieldDescr.buildInheritedFromDefinition( fld );
fieldMap.put( inheritedFlDescr.getFieldName(),
inheritedFlDescr );
}
// new classes are already distinguished from tagged external classes
isSuperClassTagged = !superTypeDeclaration.isNovel();
} else {
isSuperClassDeclared = false;
}
} else {
isSuperClassDeclared = false;
}
// look for the class externally
if ( !isSuperClassDeclared || isSuperClassTagged ) {
try {
ClassFieldInspector inspector = new ClassFieldInspector( registry.getTypeResolver().resolveType( fullSuper ) );
for ( String name : inspector.getGetterMethods().keySet() ) {
// classFieldAccessor requires both getter and setter
if ( inspector.getSetterMethods().containsKey(name) ) {
if ( !inspector.isNonGetter( name ) && !"class".equals( name ) ) {
TypeFieldDescr inheritedFlDescr = new TypeFieldDescr( name,
new PatternDescr( inspector.getFieldTypes().get( name ).getName() ) );
inheritedFlDescr.setInherited( true );
inheritedFlDescr.setIndex( inspector.getFieldNames().size() + inspector.getFieldNames().get( name ) );
if ( !fieldMap.containsKey( inheritedFlDescr.getFieldName() ) ) fieldMap.put( inheritedFlDescr.getFieldName(),
inheritedFlDescr );
}
}
}
} catch ( ClassNotFoundException cnfe ) {
throw new RuntimeDroolsException( "Unable to resolve Type Declaration superclass '" + fullSuper + "'" );
} catch ( IOException e ) {
}
}
// finally, locally declared fields are merged. The map swap ensures that super-fields are added in order, before the subclass' ones
// notice that it is not possible to override a field changing its type
for (String fieldName : typeDescr.getFields().keySet()) {
if (fieldMap.containsKey(fieldName)) {
String type1 = fieldMap.get(fieldName).getPattern().getObjectType();
String type2 = typeDescr.getFields().get(fieldName).getPattern().getObjectType();
if ( type2.lastIndexOf(".") < 0 ) {
try {
type1 = pkgRegistryMap.get(pack.getName()).getTypeResolver().resolveType( type1 ).getName();
type2 = pkgRegistryMap.get(pack.getName()).getTypeResolver().resolveType( type2 ).getName();
// now that we are at it... this will be needed later anyway
fieldMap.get(fieldName).getPattern().setObjectType(type1);
typeDescr.getFields().get(fieldName).getPattern().setObjectType(type2);
} catch (ClassNotFoundException cnfe) {
// will fail later
}
}
if ( ! type1.equals(type2) ) {
this.results.add(new TypeDeclarationError("Cannot redeclare field '" + fieldName + " from " + type1 + " to " + type2, typeDescr.getLine()));
typeDescr.setType(null, null);
return false;
} else {
String initVal = fieldMap.get(fieldName).getInitExpr();
if (typeDescr.getFields().get(fieldName).getInitExpr() == null) {
typeDescr.getFields().get(fieldName).setInitExpr(initVal);
}
}
}
fieldMap.put(fieldName, typeDescr.getFields().get(fieldName));
}
typeDescr.setFields( fieldMap );
return true;
}
/**
* @param packageDescr
*/
private void processEntryPointDeclarations(final PackageDescr packageDescr) {
PackageRegistry pkgRegistry = this.pkgRegistryMap.get( packageDescr.getNamespace() );
for ( EntryPointDeclarationDescr epDescr : packageDescr.getEntryPointDeclarations() ) {
pkgRegistry.getPackage().addEntryPointId( epDescr.getEntryPointId() );
}
}
/**
* @param packageDescr
*/
private void processTypeDeclarations(final PackageDescr packageDescr) {
PackageRegistry defaultRegistry = this.pkgRegistryMap.get( packageDescr.getNamespace() );
PackageRegistry pkgRegistry = null;
for ( TypeDeclarationDescr typeDescr : packageDescr.getTypeDeclarations() ) {
if ( isEmpty( typeDescr.getNamespace() ) ) {
for ( ImportDescr id : packageDescr.getImports() ) {
String imp = id.getTarget();
int separator = imp.lastIndexOf( '.' );
String tail = imp.substring( separator + 1 );
// if ( imp.endsWith( typeDescr.getTypeName() ) ) {
if ( tail.equals(typeDescr.getTypeName()) ) {
typeDescr.setNamespace( imp.substring( 0,
separator ) );
}
}
}
String qName = typeDescr.getType().getFullName();
int dotPos = qName.lastIndexOf( '.' );
if ( dotPos >= 0 ) {
// see if this overwrites an existing bean, which also could be a nested class.
Class cls = null;
try {
cls = Class.forName( typeDescr.getTypeName(),
true,
this.rootClassLoader );
} catch ( ClassNotFoundException e ) {
}
String qualifiedClass = qName;
int lastIndex;
while ( cls == null && (lastIndex = qualifiedClass.lastIndexOf( '.' )) != -1 ) {
try {
qualifiedClass = qualifiedClass.substring( 0,
lastIndex ) + "$" + qualifiedClass.substring( lastIndex + 1 );
cls = Class.forName( qualifiedClass,
true,
this.rootClassLoader );
} catch ( final ClassNotFoundException e ) {
cls = null;
}
}
if ( cls != null ) {
String str = ClassUtils.getPackage( cls );
typeDescr.setNamespace( str );
dotPos = cls.getName().lastIndexOf( '.' ); // reget dotPos, incase there were nested classes
typeDescr.setTypeName( cls.getName().substring( dotPos + 1 ) );
} else {
typeDescr.setNamespace( qName.substring( 0,
dotPos ) );
typeDescr.setTypeName( qName.substring( dotPos + 1 ) );
}
}
if ( isEmpty( typeDescr.getNamespace() ) && typeDescr.getFields().isEmpty() ) {
// might be referencing a class imported with a package import (.*)
PackageRegistry pkgReg = this.pkgRegistryMap.get( packageDescr.getName() );
if( pkgReg != null ) {
try {
Class clz = pkgReg.getTypeResolver().resolveType( typeDescr.getTypeName() );
java.lang.Package pkg = clz.getPackage();
if( pkg != null ) {
typeDescr.setNamespace( pkg.getName() );
int index = typeDescr.getNamespace() != null && typeDescr.getNamespace().length() > 0 ? typeDescr.getNamespace().length()+1 : 0;
typeDescr.setTypeName( clz.getCanonicalName().substring( index ) );
}
} catch ( Exception e ) {
// intentionally eating the exception as we will fallback to default namespace
}
}
}
if ( isEmpty( typeDescr.getNamespace() ) ) {
typeDescr.setNamespace( packageDescr.getNamespace() ); // set the default namespace
}
//identify superclass type and namespace
fillSuperType( typeDescr,
packageDescr );
//identify field types as well
fillFieldTypes( typeDescr,
packageDescr );
if ( !typeDescr.getNamespace().equals( packageDescr.getNamespace() ) ) {
// If the type declaration is for a different namespace, process that separately.
PackageDescr altDescr = new PackageDescr( typeDescr.getNamespace() );
altDescr.addTypeDeclaration( typeDescr );
for ( ImportDescr imp : packageDescr.getImports() ) {
altDescr.addImport( imp );
}
if ( getPackageRegistry().containsKey( altDescr.getNamespace() ) ) {
mergePackage( altDescr );
} else {
newPackage( altDescr );
}
}
}
// sort declarations : superclasses must be generated first
Collection sortedTypeDescriptors = sortByHierarchy( packageDescr.getTypeDeclarations() );
for ( TypeDeclarationDescr typeDescr : sortedTypeDescriptors ) {
if ( !typeDescr.getNamespace().equals( packageDescr.getNamespace() ) ) {
continue;
}
pkgRegistry = this.pkgRegistryMap.get( packageDescr.getNamespace() );
//descriptor needs fields inherited from superclass
for ( TypeDeclarationDescr.QualifiedName qname : typeDescr.getSuperTypes() ) {
//descriptor needs fields inherited from superclass
if ( mergeInheritedFields( typeDescr ) ) {
//descriptor also needs metadata from superclass
Iterator iter = sortedTypeDescriptors.iterator();
while ( iter.hasNext() ) {
// sortedTypeDescriptors are sorted by inheritance order, so we'll always find the superClass (if any) before the subclass
TypeDeclarationDescr descr = iter.next();
if ( qname.equals( descr.getType() ) ) {
typeDescr.getAnnotations().putAll( descr.getAnnotations() );
break;
} else if ( typeDescr.getType().equals( descr.getType() ) ) {
break;
}
}
}
}
// Go on with the build
TypeDeclaration type = new TypeDeclaration( typeDescr.getTypeName() );
if ( resource != null ) {
type.setResource( this.resource );
}
// is it a regular fact or an event?
AnnotationDescr annotationDescr = typeDescr.getAnnotation( TypeDeclaration.Role.ID );
String role = (annotationDescr != null) ? annotationDescr.getSingleValue() : null;
if ( role != null ) {
type.setRole( TypeDeclaration.Role.parseRole( role ) );
}
annotationDescr = typeDescr.getAnnotation( TypeDeclaration.ATTR_TYPESAFE );
String typesafe = (annotationDescr != null) ? annotationDescr.getSingleValue() : null;
if ( typesafe != null ) {
type.setTypesafe( Boolean.parseBoolean( typesafe ) );
}
// is it a pojo or a trait?
annotationDescr = typeDescr.getAnnotation( TypeDeclaration.Format.ID );
String format = (annotationDescr != null) ? annotationDescr.getSingleValue() : null;
if ( format != null ) {
type.setFormat( TypeDeclaration.Format.parseFormat(format) );
}
annotationDescr = typeDescr.getAnnotation( TypeDeclaration.ATTR_CLASS );
String className = (annotationDescr != null) ? annotationDescr.getSingleValue() : null;
if ( StringUtils.isEmpty( className ) ) {
className = type.getTypeName();
}
Class clazz;
try {
// the type declaration is generated in any case (to be used by subclasses, if any)
// the actual class will be generated only if needed
generateDeclaredBean( typeDescr,
type,
pkgRegistry );
clazz = pkgRegistry.getTypeResolver().resolveType( typeDescr.getType().getFullName() );
type.setTypeClass( clazz );
if ( type.getTypeClassDef() != null ) {
try {
buildFieldAccessors( type,
pkgRegistry );
} catch ( Exception e ) {
this.results.add( new TypeDeclarationError( "Error creating field accessors for TypeDeclaration '" + className + "' for type '" + type.getTypeName() + "'",
typeDescr.getLine() ) );
continue;
}
}
} catch ( final ClassNotFoundException e ) {
this.results.add( new TypeDeclarationError( "Class '" + className +
"' not found for type declaration of '" + type.getTypeName() + "'",
typeDescr.getLine() ) );
continue;
}
annotationDescr = typeDescr.getAnnotation( TypeDeclaration.ATTR_TIMESTAMP );
String timestamp = (annotationDescr != null) ? annotationDescr.getSingleValue() : null;
if ( timestamp != null ) {
type.setTimestampAttribute( timestamp );
ClassDefinition cd = type.getTypeClassDef();
Package pkg = pkgRegistry.getPackage();
InternalReadAccessor reader = pkg.getClassFieldAccessorStore().getMVELReader( ClassUtils.getPackage( type.getTypeClass() ),
type.getTypeClass().getName(),
timestamp,
type.isTypesafe() );
MVELDialectRuntimeData data = (MVELDialectRuntimeData) pkg.getDialectRuntimeRegistry().getDialectData( "mvel" );
data.addCompileable( (MVELCompileable) reader );
((MVELCompileable) reader).compile( data );
type.setTimestampExtractor( reader );
}
annotationDescr = typeDescr.getAnnotation( TypeDeclaration.ATTR_DURATION );
String duration = (annotationDescr != null) ? annotationDescr.getSingleValue() : null;
if ( duration != null ) {
type.setDurationAttribute( duration );
ClassDefinition cd = type.getTypeClassDef();
Package pkg = pkgRegistry.getPackage();
InternalReadAccessor reader = pkg.getClassFieldAccessorStore().getMVELReader( ClassUtils.getPackage( type.getTypeClass() ),
type.getTypeClass().getName(),
duration,
type.isTypesafe() );
MVELDialectRuntimeData data = (MVELDialectRuntimeData) pkg.getDialectRuntimeRegistry().getDialectData( "mvel" );
data.addCompileable( (MVELCompileable) reader );
((MVELCompileable) reader).compile( data );
type.setDurationExtractor( reader );
}
annotationDescr = typeDescr.getAnnotation( TypeDeclaration.ATTR_EXPIRE );
String expiration = (annotationDescr != null) ? annotationDescr.getSingleValue() : null;
if ( expiration != null ) {
if ( timeParser == null ) {
timeParser = new TimeIntervalParser();
}
type.setExpirationOffset( timeParser.parse( expiration )[0].longValue() );
}
boolean dynamic = typeDescr.getAnnotationNames().contains( TypeDeclaration.ATTR_PROP_CHANGE_SUPPORT );
type.setDynamic( dynamic );
pkgRegistry.getPackage().addTypeDeclaration( type );
}
}
private void updateTraitDefinition( TypeDeclaration type, Class concrete ) {
try {
ClassFieldInspector inspector = new ClassFieldInspector( concrete );
Map methods = inspector.getGetterMethods();
Map setters = inspector.getSetterMethods();
int j = 0;
for ( String fieldName : methods.keySet() ) {
if ( "core".equals( fieldName ) || "fields".equals( fieldName ) ) {
continue;
}
if ( ! inspector.isNonGetter( fieldName ) && setters.keySet().contains( fieldName )) {
Class ret = methods.get( fieldName ).getReturnType();
FieldDefinition field = new FieldDefinition();
field.setName( fieldName );
field.setTypeName( ret.getName() );
field.setIndex( j++ );
type.getTypeClassDef().addField( field );
}
}
Set interfaces = new HashSet();
for ( String iface : type.getTypeClassDef().getInterfaces() ) {
interfaces.add( iface );
}
for ( Class iKlass : concrete.getInterfaces() ) {
interfaces.add( iKlass.getName() );
}
type.getTypeClassDef().setInterfaces( interfaces.toArray(new String[interfaces.size()] ) );
} catch (IOException e) {
e.printStackTrace();
}
}
/**
* Checks whether a declaration is novel, or is a retagging of an external
* one
*
* @param typeDescr
* @return
*/
private boolean isNovelClass(TypeDeclarationDescr typeDescr) {
try {
PackageRegistry reg = this.pkgRegistryMap.get( typeDescr.getNamespace() );
if ( reg != null ) {
// String availableName = typeDescr.getNamespace() != null
// ? typeDescr.getNamespace() + "." + typeDescr.getTypeName()
// : typeDescr.getTypeName();
String availableName = typeDescr.getType().getFullName();
Class< ? > resolvedType = reg.getTypeResolver().resolveType( availableName );
if ( resolvedType != null && typeDescr.getFields().size() > 1 ) {
this.results.add( new TypeDeclarationError( "Duplicate type definition. A class with the name '" + resolvedType.getName() + "' was found in the classpath while trying to " +
"redefine the fields in the declare statement. Fields can only be defined for non-existing classes.",
typeDescr.getLine() ) );
}
return false;
} else {
return false;
}
} catch ( ClassNotFoundException cnfe ) {
return true;
}
}
/**
* Tries to determine whether a given annotation is properly defined using a
* java.lang.Annotation and can be resolved
*
* Proper annotations will be wired to dynamically generated beans
*
* @param annotation
* @param resolver
* @return
*/
private Class resolveAnnotation(String annotation,
TypeResolver resolver) {
// do not waste time with @role and @format
if ( TypeDeclaration.Role.ID.equals( annotation )
|| TypeDeclaration.Format.ID.equals( annotation ) ) {
return null;
}
// known conflicting annotation
if ( TypeDeclaration.ATTR_CLASS.equals( annotation) ) {
return null;
}
try {
Class ann = resolver.resolveType( annotation.substring(0,1).toUpperCase() + annotation.substring(1) );
return ann;
} catch ( ClassNotFoundException e ) {
// internal annotation, or annotation which can't be resolved.
return null;
}
}
/**
*
* @param pkgRegistry
* @throws SecurityException
* @throws IllegalArgumentException
* @throws InstantiationException
* @throws IllegalAccessException
* @throws IOException
* @throws IntrospectionException
* @throws ClassNotFoundException
* @throws NoSuchMethodException
* @throws InvocationTargetException
* @throws NoSuchFieldException
*/
private final void buildFieldAccessors(final TypeDeclaration type,
final PackageRegistry pkgRegistry) throws SecurityException,
IllegalArgumentException,
InstantiationException,
IllegalAccessException,
IOException,
IntrospectionException,
ClassNotFoundException,
NoSuchMethodException,
InvocationTargetException,
NoSuchFieldException {
ClassDefinition cd = type.getTypeClassDef();
ClassFieldAccessorStore store = pkgRegistry.getPackage().getClassFieldAccessorStore();
for ( FieldDefinition attrDef : cd.getFieldsDefinitions() ) {
ClassFieldAccessor accessor = store.getAccessor( cd.getDefinedClass().getName(),
attrDef.getName() );
attrDef.setReadWriteAccessor( accessor );
}
}
/**
* Generates a bean, and adds it to the composite class loader that
* everything is using.
*/
private void generateDeclaredBean( TypeDeclarationDescr typeDescr,
TypeDeclaration type,
PackageRegistry pkgRegistry ) {
// extracts type, supertype and interfaces
String fullName = typeDescr.getType().getFullName();
if ( type.getFormat().equals( TypeDeclaration.Format.POJO ) ) {
if ( typeDescr.getSuperTypes().size() > 1 ) {
this.results.add( new TypeDeclarationError( "Declared class " + fullName + " - has more than one supertype;",
typeDescr.getLine() ) );
return;
} else if ( typeDescr.getSuperTypes().size() == 0 ) {
typeDescr.addSuperType("java.lang.Object");
}
}
boolean traitable = typeDescr.getAnnotation( Traitable.class.getSimpleName() ) != null;
String[] fullSuperTypes = new String[ typeDescr.getSuperTypes().size() +1 ];
int j = 0;
for ( TypeDeclarationDescr.QualifiedName qname : typeDescr.getSuperTypes() ) {
fullSuperTypes[ j++ ] = qname.getFullName();
}
fullSuperTypes[ j++ ] = Thing.class.getName();
List interfaceList = new ArrayList();
interfaceList.add( Serializable.class.getName() );
if ( traitable ) {
interfaceList.add( TraitableBean.class.getName() );
}
String[] interfaces = interfaceList.toArray( new String[ interfaceList.size() ] );
// prepares a class definition
ClassDefinition def = null;
if ( type.getFormat().equals( TypeDeclaration.Format.TRAIT ) ) {
def = new ClassDefinition( fullName,
"java.lang.Object",
fullSuperTypes );
} else {
def = new ClassDefinition( fullName,
fullSuperTypes[0],
interfaces );
def.setTraitable( traitable );
}
for ( String annotationName : typeDescr.getAnnotationNames() ) {
Class annotation = resolveAnnotation( annotationName,
pkgRegistry.getTypeResolver() );
if ( annotation != null ) {
try {
AnnotationDefinition annotationDefinition = AnnotationDefinition.build(
annotation,
typeDescr.getAnnotations().get( annotationName ).getValueMap(),
pkgRegistry.getTypeResolver() );
def.addAnnotation( annotationDefinition );
} catch ( NoSuchMethodException nsme ) {
this.results.add( new TypeDeclarationError( "Annotated type " + fullName + " - undefined property in @annotation " + annotationName + ": " + nsme.getMessage() + ";",
typeDescr.getLine() ) );
}
}
}
// fields definitions are created. will be used by subclasses, if any.
// Fields are SORTED in the process
if ( typeDescr.getFields().size() > 0 ) {
PriorityQueue fieldDefs = sortFields( typeDescr.getFields(),
pkgRegistry );
while ( fieldDefs.size() > 0 ) {
FieldDefinition fld = fieldDefs.poll();
def.addField( fld );
}
}
// check whether it is necessary to build the class or not
type.setNovel( isNovelClass( typeDescr ) );
// attach the class definition, it will be completed later
type.setTypeClassDef( def );
generateDeclaredBean( typeDescr, type, pkgRegistry, def );
}
private void generateDeclaredBean( TypeDeclarationDescr typeDescr,
TypeDeclaration type,
PackageRegistry pkgRegistry,
ClassDefinition def ) {
if ( typeDescr.getAnnotation( Traitable.class.getSimpleName() ) != null ) {
if ( ! isNovelClass( typeDescr ) ) {
try {
PackageRegistry reg = this.pkgRegistryMap.get( typeDescr.getNamespace() );
String availableName = typeDescr.getType().getFullName();
Class< ? > resolvedType = reg.getTypeResolver().resolveType( availableName );
updateTraitDefinition( type, resolvedType );
} catch ( ClassNotFoundException cnfe ) {
// we already know the class exists
}
}
TraitRegistry.getInstance().addTraitable( def );
} else if ( type.getFormat().equals( TypeDeclaration.Format.TRAIT)
|| typeDescr.getAnnotation( Trait.class.getSimpleName() ) != null ) {
if ( ! type.isNovel() ) {
try {
PackageRegistry reg = this.pkgRegistryMap.get( typeDescr.getNamespace() );
String availableName = typeDescr.getType().getFullName();
Class< ? > resolvedType = reg.getTypeResolver().resolveType( availableName );
if ( ! Thing.class.isAssignableFrom( resolvedType) ) {
updateTraitDefinition( type, resolvedType );
String target = typeDescr.getTypeName() + TraitFactory.SUFFIX;
TypeDeclarationDescr tempDescr = new TypeDeclarationDescr();
tempDescr.setNamespace( typeDescr.getNamespace() );
tempDescr.setFields( typeDescr.getFields() );
tempDescr.setType( target, typeDescr.getNamespace() );
tempDescr.addSuperType( typeDescr.getType() );
TypeDeclaration tempDeclr = new TypeDeclaration( target );
tempDeclr.setFormat( TypeDeclaration.Format.TRAIT );
tempDeclr.setTypesafe( type.isTypesafe() );
tempDeclr.setNovel( true );
tempDeclr.setTypeClassName( tempDescr.getType().getFullName() );
tempDeclr.setResource( type.getResource() );
ClassDefinition tempDef = new ClassDefinition( target );
tempDef.setClassName( tempDescr.getType().getFullName() );
tempDef.setTraitable( false );
for ( FieldDefinition fld : def.getFieldsDefinitions() ) {
tempDef.addField( fld );
}
tempDef.setInterfaces( def.getInterfaces() );
tempDef.setSuperClass( def.getClassName() );
tempDef.setDefinedClass( resolvedType );
type.setFormat( TypeDeclaration.Format.POJO );
generateDeclaredBean(tempDescr, tempDeclr, pkgRegistry, tempDef );
} else {
updateTraitDefinition( type, resolvedType );
TraitRegistry.getInstance().addTrait( def );
}
} catch ( ClassNotFoundException cnfe ) {
// we already know the class exists
}
} else {
if ( def.getClassName().endsWith( "_Trait__Extension" ) ) {
TraitRegistry.getInstance().addTrait( def.getClassName().replace( "_Trait__Extension", "" ) , def );
} else {
TraitRegistry.getInstance().addTrait( def );
}
}
}
if ( type.isNovel() ) {
String fullName = typeDescr.getType().getFullName();
JavaDialectRuntimeData dialect = (JavaDialectRuntimeData) pkgRegistry.getDialectRuntimeRegistry().getDialectData( "java" );
switch (type.getFormat()) {
case TRAIT:
try {
byte[] d;
ClassBuilder tb = ClassBuilderFactory.getTraitBuilderService();
d = tb.buildClass( def );
dialect.write( JavaDialectRuntimeData.convertClassToResourcePath( fullName ),
d );
} catch ( Exception e ) {
this.results.add( new TypeDeclarationError( "Unable to compile declared trait " + fullName + ": " + e.getMessage() + ";",
typeDescr.getLine() ) );
}
break;
case POJO:
default:
try {
ClassBuilder cb = ClassBuilderFactory.getBeanClassBuilderService( );
byte[] d = cb.buildClass( def );
dialect.write( JavaDialectRuntimeData.convertClassToResourcePath( fullName ),
d );
} catch ( Exception e ) {
this.results.add( new TypeDeclarationError( "Unable to create a class for declared type " + fullName + ": " + e.getMessage() + ";",
typeDescr.getLine() ) );
}
break;
}
}
}
/**
* Sorts a bean's fields according to the positional index metadata. The
* order is as follows (i) as defined using the @position metadata (ii) as
* resulting from the inspection of an external java superclass, if
* applicable (iii) in declaration order, superclasses first
*
* @param flds
* @param pkgRegistry
* @return
*/
private PriorityQueue sortFields(Map flds,
PackageRegistry pkgRegistry) {
PriorityQueue queue = new PriorityQueue();
int last = 0;
for ( TypeFieldDescr field : flds.values() ) {
last = Math.max( last,
field.getIndex() );
}
for ( TypeFieldDescr field : flds.values() ) {
if ( field.getIndex() < 0 ) {
field.setIndex( ++last );
}
String fullFieldType;
try {
fullFieldType = pkgRegistry.getTypeResolver().resolveType( field.getPattern().getObjectType() ).getName();
FieldDefinition fieldDef = new FieldDefinition( field.getFieldName(),
fullFieldType );
// field is marked as PK
boolean isKey = field.getAnnotation( TypeDeclaration.ATTR_KEY ) != null;
fieldDef.setKey( isKey );
fieldDef.setIndex( field.getIndex() );
fieldDef.setInherited( field.isInherited() );
fieldDef.setInitExpr( field.getInitExpr() );
for ( String annotationName : field.getAnnotationNames() ) {
Class annotation = resolveAnnotation( annotationName,
pkgRegistry.getTypeResolver() );
if ( annotation != null ) {
try {
AnnotationDefinition annotationDefinition = AnnotationDefinition.build( annotation,
field.getAnnotations().get( annotationName ).getValueMap(),
pkgRegistry.getTypeResolver() );
fieldDef.addAnnotation( annotationDefinition );
} catch ( NoSuchMethodException nsme ) {
this.results.add( new TypeDeclarationError( "Annotated field " + field.getFieldName() + " - undefined property in @annotation " + annotationName + ": " + nsme.getMessage() + ";",
field.getLine() ) );
}
}
}
queue.add( fieldDef );
} catch ( ClassNotFoundException cnfe ) {
this.results.add( new TypeDeclarationError( cnfe.getMessage(),
field.getLine() ) );
}
}
return queue;
}
private void addFunction(final FunctionDescr functionDescr) {
functionDescr.setResource( this.resource );
PackageRegistry pkgRegistry = this.pkgRegistryMap.get( functionDescr.getNamespace() );
Dialect dialect = pkgRegistry.getDialectCompiletimeRegistry().getDialect( functionDescr.getDialect() );
dialect.addFunction( functionDescr,
pkgRegistry.getTypeResolver(),
this.resource );
}
private void preCompileAddFunction(final FunctionDescr functionDescr) {
PackageRegistry pkgRegistry = this.pkgRegistryMap.get( functionDescr.getNamespace() );
Dialect dialect = pkgRegistry.getDialectCompiletimeRegistry().getDialect( functionDescr.getDialect() );
dialect.preCompileAddFunction( functionDescr,
pkgRegistry.getTypeResolver() );
}
private void postCompileAddFunction(final FunctionDescr functionDescr) {
PackageRegistry pkgRegistry = this.pkgRegistryMap.get( functionDescr.getNamespace() );
Dialect dialect = pkgRegistry.getDialectCompiletimeRegistry().getDialect( functionDescr.getDialect() );
dialect.postCompileAddFunction( functionDescr,
pkgRegistry.getTypeResolver() );
}
private void addFactTemplate(final PackageDescr pkgDescr,
final FactTemplateDescr factTemplateDescr) {
final List fields = new ArrayList();
int index = 0;
PackageRegistry pkgRegistry = this.pkgRegistryMap.get( pkgDescr.getNamespace() );
for ( final Iterator it = factTemplateDescr.getFields().iterator(); it.hasNext(); ) {
final FieldTemplateDescr fieldTemplateDescr = (FieldTemplateDescr) it.next();
FieldTemplate fieldTemplate = null;
try {
fieldTemplate = new FieldTemplateImpl( fieldTemplateDescr.getName(),
index++,
pkgRegistry.getTypeResolver().resolveType( fieldTemplateDescr.getClassType() ) );
} catch ( final ClassNotFoundException e ) {
this.results.add( new FieldTemplateError( pkgRegistry.getPackage(),
fieldTemplateDescr,
null,
"Unable to resolve Class '" + fieldTemplateDescr.getClassType() + "'" ) );
}
fields.add( fieldTemplate );
}
final FactTemplate factTemplate = new FactTemplateImpl( pkgRegistry.getPackage(),
factTemplateDescr.getName(),
(FieldTemplate[]) fields.toArray( new FieldTemplate[fields.size()] ) );
}
private void addRule(final RuleDescr ruleDescr) {
ruleDescr.setResource( resource );
PackageRegistry pkgRegistry = this.pkgRegistryMap.get( ruleDescr.getNamespace() );
Package pkg = pkgRegistry.getPackage();
DialectCompiletimeRegistry ctr = pkgRegistry.getDialectCompiletimeRegistry();
RuleBuildContext context = new RuleBuildContext( this,
ruleDescr,
ctr,
pkg,
ctr.getDialect( pkgRegistry.getDialect() ) );
this.ruleBuilder.build( context );
this.results.addAll( context.getErrors() );
if ( resource != null ) {
context.getRule().setResource( resource );
}
context.getDialect().addRule( context );
if ( this.ruleBase != null ) {
if ( pkg.getRule( ruleDescr.getName() ) != null ) {
this.ruleBase.lock();
try {
// XXX: this one notifies listeners
this.ruleBase.removeRule( pkg,
pkg.getRule( ruleDescr.getName() ) );
} finally {
this.ruleBase.unlock();
}
}
}
pkg.addRule( context.getRule() );
}
/**
* @return The compiled package. The package may contain errors, which you
* can report on by calling getErrors or printErrors. If you try to
* add an invalid package (or rule) to a RuleBase, you will get a
* runtime exception.
*
* Compiled packages are serializable.
*/
public Package getPackage() {
PackageRegistry pkgRegistry = null;
if ( !this.pkgRegistryMap.isEmpty() ) {
pkgRegistry = (PackageRegistry) this.pkgRegistryMap.values().toArray()[this.pkgRegistryMap.size()-1];
}
Package pkg = null;
if ( pkgRegistry != null ) {
pkg = pkgRegistry.getPackage();
}
if ( hasErrors() && pkg != null ) {
pkg.setError( getErrors().toString() );
}
return pkg;
}
public Package[] getPackages() {
Package[] pkgs = new Package[this.pkgRegistryMap.size()];
int i = pkgs.length;
String errors = null;
if ( !getErrors().isEmpty() ) {
errors = getErrors().toString();
}
for ( PackageRegistry pkgRegistry : this.pkgRegistryMap.values() ) {
Package pkg = pkgRegistry.getPackage();
pkg.getDialectRuntimeRegistry().onBeforeExecute();
if ( errors != null ) {
pkg.setError( errors );
}
pkgs[--i] = pkg;
}
return pkgs;
}
/**
* Return the PackageBuilderConfiguration for this PackageBuilder session
*
* @return The PackageBuilderConfiguration
*/
public PackageBuilderConfiguration getPackageBuilderConfiguration() {
return this.configuration;
}
public PackageRegistry getPackageRegistry(String name) {
return this.pkgRegistryMap.get( name );
}
public Map getPackageRegistry() {
return this.pkgRegistryMap;
}
public DateFormats getDateFormats() {
return this.dateFormats;
}
/**
* Returns an expander for DSLs (only if there is a DSL configured for this
* package).
*/
public DefaultExpander getDslExpander() {
DefaultExpander expander = new DefaultExpander();
if ( this.dslFiles == null || this.dslFiles.isEmpty() ) {
return null;
}
for ( DSLMappingFile file : this.dslFiles ) {
expander.addDSLMapping( file.getMapping() );
}
return expander;
}
public Map> getGlobals() {
return this.globals;
}
/**
* This will return true if there were errors in the package building and
* compiling phase
*/
public boolean hasErrors() {
return !this.results.isEmpty();
}
/**
* @return A list of Error objects that resulted from building and compiling
* the package.
*/
public PackageBuilderErrors getErrors() {
return new PackageBuilderErrors( this.results.toArray( new DroolsError[this.results.size()] ) );
}
/**
* Reset the error list. This is useful when incrementally building
* packages. Care should be used when building this, if you clear this when
* there were errors on items that a rule depends on (eg functions), then
* you will get spurious errors which will not be that helpful.
*/
protected void resetErrors() {
this.results.clear();
}
public String getDefaultDialect() {
return this.defaultDialect;
}
public static class MissingPackageNameException extends IllegalArgumentException {
private static final long serialVersionUID = 510l;
public MissingPackageNameException(final String message) {
super( message );
}
}
public static class PackageMergeException extends IllegalArgumentException {
private static final long serialVersionUID = 400L;
public PackageMergeException(final String message) {
super( message );
}
}
/**
* This is the super of the error handlers. Each error handler knows how to
* report a compile error of its type, should it happen. This is needed, as
* the compiling is done as one hit at the end, and we need to be able to
* work out what rule/ast element caused the error.
*
* An error handler it created for each class task that is queued to be
* compiled. This doesn't mean an error has occurred, it just means it *may*
* occur in the future and we need to be able to map it back to the AST
* element that originally spawned the code to be compiled.
*/
public abstract static class ErrorHandler {
private final List errors = new ArrayList();
protected String message;
private boolean inError = false;
/** This needes to be checked if there is infact an error */
public boolean isInError() {
return this.inError;
}
public void addError(final CompilationProblem err) {
this.errors.add( err );
this.inError = true;
}
/**
*
* @return A DroolsError object populated as appropriate, should the
* unthinkable happen and this need to be reported.
*/
public abstract DroolsError getError();
/**
* We must use an error of JCI problem objects. If there are no
* problems, null is returned. These errors are placed in the
* DroolsError instances. Its not 1 to 1 with reported errors.
*/
protected CompilationProblem[] collectCompilerProblems() {
if ( this.errors.size() == 0 ) {
return null;
} else {
final CompilationProblem[] list = new CompilationProblem[this.errors.size()];
this.errors.toArray( list );
return list;
}
}
}
public static class RuleErrorHandler extends ErrorHandler {
private BaseDescr descr;
private Rule rule;
public RuleErrorHandler(final BaseDescr ruleDescr,
final Rule rule,
final String message) {
this.descr = ruleDescr;
this.rule = rule;
this.message = message;
}
public DroolsError getError() {
return new RuleBuildError( this.rule,
this.descr,
collectCompilerProblems(),
this.message );
}
}
/**
* There isn't much point in reporting invoker errors, as they are no help.
*/
public static class RuleInvokerErrorHandler extends RuleErrorHandler {
public RuleInvokerErrorHandler(final BaseDescr ruleDescr,
final Rule rule,
final String message) {
super( ruleDescr,
rule,
message );
}
}
public static class FunctionErrorHandler extends ErrorHandler {
private FunctionDescr descr;
public FunctionErrorHandler(final FunctionDescr functionDescr,
final String message) {
this.descr = functionDescr;
this.message = message;
}
public DroolsError getError() {
return new FunctionError( this.descr,
collectCompilerProblems(),
this.message );
}
}
public static class SrcErrorHandler extends ErrorHandler {
public SrcErrorHandler(final String message) {
this.message = message;
}
public DroolsError getError() {
return new SrcError( collectCompilerProblems(),
this.message );
}
}
public static class SrcError extends DroolsError {
private Object object;
private String message;
private int[] errorLines = new int[0];
public SrcError(Object object,
String message) {
this.object = object;
this.message = message;
}
public Object getObject() {
return this.object;
}
public int[] getErrorLines() {
return this.errorLines;
}
public String getMessage() {
return this.message;
}
public String toString() {
final StringBuilder buf = new StringBuilder();
buf.append( this.message );
buf.append( " : " );
buf.append( "\n" );
if ( this.object instanceof CompilationProblem[] ) {
final CompilationProblem[] problem = (CompilationProblem[]) this.object;
for ( int i = 0; i < problem.length; i++ ) {
buf.append( "\t" );
buf.append( problem[i] );
buf.append( "\n" );
}
} else if ( this.object != null ) {
buf.append( this.object );
}
return buf.toString();
}
}
private String ucFirst(final String name) {
return name.toUpperCase().charAt( 0 ) + name.substring( 1 );
}
public CompositeClassLoader getRootClassLoader() {
return this.rootClassLoader;
}
/**
* Utility method to sort declared beans. Linearizes the hierarchy,
* i.e.generates a sequence of declaration such that, if Sub is subclass of
* Sup, then the index of Sub will be > than the index of Sup in the
* resulting collection. This ensures that superclasses are processed before
* their subclasses
*
* @param typeDeclarations
* @return
*/
public static Collection sortByHierarchy(List typeDeclarations) {
Node root = new Node( null );
Map> map = new HashMap>();
for ( TypeDeclarationDescr tdescr : typeDeclarations ) {
String typeName = tdescr.getType().getFullName();
Node node = map.get( typeName );
if ( node == null ) {
node = new Node( typeName,
tdescr );
map.put( typeName,
node );
} else if ( node.getData() == null ) {
node.setData( tdescr );
}
if ( tdescr.getSuperTypes().isEmpty() ) {
root.addChild( node );
} else {
for ( TypeDeclarationDescr.QualifiedName qname : tdescr.getSuperTypes() ) {
String superTypeName = qname.getFullName();
Node superNode = map.get( superTypeName );
if ( superNode == null ) {
superNode = new Node( superTypeName );
map.put( superTypeName,
superNode );
}
superNode.addChild( node );
}
}
for ( TypeFieldDescr field : tdescr.getFields().values() ) {
String fieldTypeName = field.getPattern().getObjectType();
Node superNode = map.get( fieldTypeName );
if ( superNode == null ) {
superNode = new Node( fieldTypeName );
map.put( fieldTypeName,
superNode );
}
superNode.addChild( node );
}
}
Iterator> iter = map.values().iterator();
while ( iter.hasNext() ) {
Node n = iter.next();
if ( n.getData() == null ) root.addChild( n );
}
List sortedList = new LinkedList();
root.accept( sortedList );
return sortedList;
}
/**
* Utility class for the sorting algorithm
*
* @param
*/
private static class Node {
private String key;
private T data;
private List> children;
public Node(String key) {
this.key = key;
this.children = new LinkedList>();
}
public Node(String key,
T content) {
this( key );
this.data = content;
}
public void addChild(Node child) {
this.children.add( child );
}
public List> getChildren() {
return children;
}
public String getKey() {
return key;
}
public T getData() {
return data;
}
public void setData(T content) {
this.data = content;
}
public void accept(List list) {
if ( this.data != null ) {
if ( list.contains( this.data ) ) {
list.remove( this.data );
}
list.add( this.data );
}
for ( int j = 0; j < children.size(); j++ )
children.get( j ).accept( list );
}
}
//Entity rules inherit package attributes
private void inheritPackageAttributes(Map pkgAttributes,
RuleDescr ruleDescr) {
if ( pkgAttributes == null ) {
return;
}
for ( AttributeDescr attrDescr : pkgAttributes.values() ) {
String name = attrDescr.getName();
AttributeDescr ruleAttrDescr = ruleDescr.getAttributes().get( name );
if ( ruleAttrDescr == null ) {
ruleDescr.getAttributes().put( name,
attrDescr );
}
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy