com.oneandone.ejbcdiunit.EjbUnitBeanInitializerClass Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of ejb-cdi-unit Show documentation

A module that can be used together with cdiunit to build en ejb-test-environment.

The newest version!

package com.oneandone.ejbcdiunit;

import com.oneandone.ejbcdiunit.internal.EjbExtensionExtended;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.annotation.PostConstruct;
import javax.annotation.PreDestroy;
import javax.enterprise.context.ApplicationScoped;
import javax.enterprise.context.Dependent;
import javax.enterprise.inject.spi.BeanManager;
import javax.inject.Inject;
import javax.naming.NamingException;
import java.util.List;

/**
 * @author aschoerk
 */
@Dependent
public class EjbUnitBeanInitializerClass {
    @Inject
    EjbExtensionExtended ejbExtensionExtended;

    @Inject
    BeanManager bm;

    @Inject
    AsynchronousManager asynchronousManager;

    Logger logger = null;
    private CreationalContexts creationalContexts;

    Logger getLogger() {
        if (logger == null) {
            logger = LoggerFactory.getLogger("LateInitializerClass");
        }
        return logger;
    }



    @PostConstruct
    public void init() {
        List> startups = ejbExtensionExtended.getStartupSingletons();
        if (!startups.isEmpty()) {
            try {
                creationalContexts = new CreationalContexts();
                for (Class clazz: startups) {
                    creationalContexts.create(clazz, ApplicationScoped.class);
                }
            } catch (NamingException e) {
                throw new RuntimeException(e);
            }
        }
        asynchronousManager.addTimerMethods();
    }

    @PreDestroy
    private void preDestroy() {
        if (creationalContexts != null) {
            try {
                creationalContexts.close();
            } catch (Exception e) {
                throw new RuntimeException(e);
            }
        }

    }
}