org.drools.compiler.builder.AbstractAssemblerService Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of drools-compiler
The newest version!
/*
 * Copyright 2018 Red Hat, Inc. and/or its affiliates.
 *
 * 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.builder;

import java.util.Collection;

import org.kie.api.internal.assembler.KieAssemblerService;
import org.kie.api.internal.assembler.ProcessedResource;
import org.kie.api.internal.io.ResourceTypePackage;
import org.kie.api.io.Resource;
import org.kie.api.io.ResourceConfiguration;
import org.kie.api.io.ResourceType;
import org.kie.internal.builder.AssemblerContext;
import org.kie.internal.builder.KnowledgeBuilderError;

/**
 * A base implementation for an {@link KieAssemblerService}, following simple conventions.
 * @param  type of the package that is being assembled
 * @param  type of the processed resource that will be assembled for each given {@link Resource}
 */
public abstract class AbstractAssemblerService, U extends ProcessedResource> implements KieAssemblerService {

    /**
     * Factory for a package of type T
     * @param namespace namespace of the package (e.g. a Java package name)
     */
    protected abstract T createPackage(String namespace);

    /**
     * Factory for the ResourceProcessor subclass that will process Resources that this assembler supports
     */
    protected abstract ResourceProcessor createResourceProcessor(Resource resource);

    @Override
    public final void addResourceAfterRules(Object kbuilder, Resource resource, ResourceType type, ResourceConfiguration configuration) throws Exception {
        AssemblerContext kb = (AssemblerContext) kbuilder;
        ResourceProcessor processor = createResourceProcessor(resource);
        processor.process();
        Collection errors = processor.getErrors();

        if (errors.isEmpty()) {
            U compiledResource = processor.getProcessedResource();
            kb.computeIfAbsent(
                    getResourceType(),
                    compiledResource.getNamespace(),
                    resourceType -> createPackage(compiledResource.getNamespace()))
                    .add(compiledResource);
        } else {
            errors.forEach(kb::reportError);
        }
    }
}