package org.drools.builder.impl;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import org.drools.KnowledgeBase;
import org.drools.KnowledgeBaseFactory;
import org.drools.builder.CompositeKnowledgeBuilder;
import org.drools.builder.KnowledgeBuilder;
import org.drools.builder.KnowledgeBuilderError;
import org.drools.builder.KnowledgeBuilderErrors;
import org.drools.builder.KnowledgeBuilderResults;
import org.drools.builder.ResultSeverity;
import org.drools.builder.ResourceConfiguration;
import org.drools.builder.ResourceType;
import org.drools.compiler.CompositeKnowledgeBuilderImpl;
import org.drools.compiler.DroolsWarning;
import org.drools.compiler.PackageBuilder;
import org.drools.definition.KnowledgePackage;
import org.drools.definitions.impl.KnowledgePackageImp;
import org.drools.io.Resource;
import org.drools.io.impl.BaseResource;
import org.drools.rule.Package;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
public class KnowledgeBuilderImpl implements KnowledgeBuilder {
protected static transient Logger logger = LoggerFactory.getLogger(KnowledgeBuilderImpl.class);
private PackageBuilder pkgBuilder;
public KnowledgeBuilderImpl(PackageBuilder pkgBuilder) {
this.pkgBuilder = pkgBuilder;
}
public void add(Resource resource, ResourceType type) {
ResourceConfiguration resourceConfiguration = resource instanceof BaseResource ? ((BaseResource) resource).getConfiguration() : null;
add(resource, type, resourceConfiguration) ;
}
public CompositeKnowledgeBuilder batch() {
return new CompositeKnowledgeBuilderImpl(pkgBuilder);
}
public void add(Resource resource,
ResourceType type,
ResourceConfiguration configuration) {
pkgBuilder.registerBuildResource( resource, type );
pkgBuilder.addKnowledgeResource( resource, type, configuration );
}
public void undo() {
pkgBuilder.undo();
}
public Collection<KnowledgePackage> getKnowledgePackages() {
if ( pkgBuilder.hasErrors() ) {
return new ArrayList<KnowledgePackage>( 0 );
}
Package[] pkgs = pkgBuilder.getPackages();
List<KnowledgePackage> list = new ArrayList<KnowledgePackage>( pkgs.length );
for ( Package pkg : pkgs ) {
list.add( new KnowledgePackageImp( pkg ) );
}
return list;
}
public KnowledgeBase newKnowledgeBase() {
KnowledgeBuilderErrors errors = getErrors();
if (errors.size() > 0) {
for (KnowledgeBuilderError error: errors) {
logger.error(error.toString());
}
throw new IllegalArgumentException("Could not parse knowledge.");
}
KnowledgeBase kbase = KnowledgeBaseFactory.newKnowledgeBase();
kbase.addKnowledgePackages(getKnowledgePackages());
return kbase;
}
public boolean hasErrors() {
return pkgBuilder.hasErrors();
}
public KnowledgeBuilderErrors getErrors() {
return pkgBuilder.getErrors();
}
public boolean hasWarnings() {
return pkgBuilder.hasWarnings();
}
public List<DroolsWarning> getWarnings() {
return pkgBuilder.getWarningList();
}
public PackageBuilder getPackageBuilder() {
return pkgBuilder;
}
public KnowledgeBuilderResults getResults(ResultSeverity... severities) {
return pkgBuilder.getProblems(severities);
}
public boolean hasResults(ResultSeverity... severities) {
return pkgBuilder.hasProblems(severities);
}
public void dispose() {
this.pkgBuilder.dispose();
}
}