com.starxmind.boot.concurrent.lock.XLockFactoryHolder Maven / Gradle / Ivy

Go to download
package com.starxmind.boot.concurrent.lock;

import com.google.common.collect.Maps;
import com.starxmind.bass.concurrent.lock.XLockFactory;
import com.starxmind.boot.utils.ApplicationContextHolder;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.stereotype.Component;

import java.util.Map;

@Slf4j
@Component
@RequiredArgsConstructor
public class XLockFactoryHolder implements InitializingBean {
    private static final Map CONTAINER = Maps.newHashMap();
    private final ApplicationContextHolder applicationContextHolder;

    @Override
    public void afterPropertiesSet() throws Exception {
        Map map = applicationContextHolder.getApplicationContext().getBeansOfType(XLockFactory.class);
        map.forEach(
                (k, v) -> {
                    String className = v.getClass().getName();
                    CONTAINER.put(className, v);
                    log.info(" has found a bean, className: {}", className);
                }
        );
        log.info(" totally has found <{}> XLockFactories", CONTAINER.size());
    }

    public XLockFactory get(Class clazz) {
        return CONTAINER.get(clazz.getName());
    }
}