org.jboss.weld.bootstrap.ConcurrentBeanDeployer Maven / Gradle / Ivy

Go to download
/*
 * JBoss, Home of Professional Open Source
 * Copyright 2010, Red Hat, Inc., and individual contributors
 * by the @authors tag. See the copyright.txt in the distribution for a
 * full listing of individual 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.jboss.weld.bootstrap;

import java.util.List;

import jakarta.enterprise.inject.spi.Bean;

import org.jboss.weld.annotated.slim.SlimAnnotatedType;
import org.jboss.weld.annotated.slim.SlimAnnotatedTypeContext;
import org.jboss.weld.bean.AbstractClassBean;
import org.jboss.weld.bean.RIBean;
import org.jboss.weld.bootstrap.api.ServiceRegistry;
import org.jboss.weld.executor.IterativeWorkerTaskFactory;
import org.jboss.weld.manager.BeanManagerImpl;
import org.jboss.weld.manager.api.ExecutorServices;
import org.jboss.weld.util.collections.SetMultimap;

/**
 * BeanDeployer that processes some of the deployment tasks in parallel. A threadsafe instance of
 * {@link BeanDeployerEnvironment} is used.
 *
 * @author Jozef Hartinger
 *
 */
public class ConcurrentBeanDeployer extends BeanDeployer {

    private final ExecutorServices executor;

    public ConcurrentBeanDeployer(BeanManagerImpl manager, ServiceRegistry services) {
        super(manager, services, BeanDeployerEnvironmentFactory.newConcurrentEnvironment(manager));
        this.executor = services.get(ExecutorServices.class);
    }

    @Override
    public BeanDeployer addClasses(Iterable c) {
        final AnnotatedTypeLoader loader = createAnnotatedTypeLoader();
        executor.invokeAllAndCheckForExceptions(new IterativeWorkerTaskFactory(c) {
            @Override
            protected void doWork(String className) {
                addClass(className, loader);
            }
        });
        return this;
    }

    @Override
    public void createClassBeans() {
        final SetMultimap, SlimAnnotatedType> otherWeldClasses = SetMultimap.newConcurrentSetMultimap();

        executor.invokeAllAndCheckForExceptions(
                new IterativeWorkerTaskFactory>(getEnvironment().getAnnotatedTypes()) {
                    @Override
                    protected void doWork(SlimAnnotatedTypeContext ctx) {
                        createClassBean(ctx.getAnnotatedType(), otherWeldClasses);
                    }
                });

        // create session beans
        ejbSupport.createSessionBeans(getEnvironment(), otherWeldClasses, getManager());
    }

    @Override
    public void createProducersAndObservers() {
        executor.invokeAllAndCheckForExceptions(
                new IterativeWorkerTaskFactory>(getEnvironment().getClassBeans()) {
                    @Override
                    protected void doWork(AbstractClassBean bean) {
                        createObserversProducersDisposers(bean);
                    }
                });
    }

    @Override
    public void doAfterBeanDiscovery(List> beanList) {
        executor.invokeAllAndCheckForExceptions(new AfterBeanDiscoveryInitializerFactory(beanList));
    }

    @Override
    public AbstractBeanDeployer initializeBeans() {
        executor.invokeAllAndCheckForExceptions(new IterativeWorkerTaskFactory>(getEnvironment().getBeans()) {
            @Override
            protected void doWork(RIBean bean) {
                bean.initialize(getEnvironment());
            }
        });
        return this;
    }

    private static class AfterBeanDiscoveryInitializerFactory extends IterativeWorkerTaskFactory> {

        public AfterBeanDiscoveryInitializerFactory(Iterable> iterable) {
            super(iterable);
        }

        @Override
        protected void doWork(Bean bean) {
            if (bean instanceof RIBean) {
                ((RIBean) bean).initializeAfterBeanDiscovery();
            }
        }
    }
}