com.cloudimpl.outstack.outstack.spring.test.ResourceLoader Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of outstack-spring-test Show documentation
There is a newer version: 4.0.551
/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package com.cloudimpl.outstack.outstack.spring.test;

import com.cloudimpl.outstack.runtime.EntityCommandHandler;
import com.cloudimpl.outstack.runtime.EntityEventHandler;
import com.cloudimpl.outstack.runtime.domainspec.Command;
import com.cloudimpl.outstack.runtime.domainspec.Entity;
import com.cloudimpl.outstack.runtime.domainspec.Event;
import io.github.classgraph.ClassGraph;
import io.github.classgraph.ClassInfoList;
import io.github.classgraph.ScanResult;
import java.util.HashMap;
import java.util.Map;

/**
 *
 * @author nuwan
 */
public class ResourceLoader {

    private ScanResult scanResult;
    private final Map> mapEntities = new HashMap<>();
    private final Map> mapEvents = new HashMap<>();
    private final Map> mapCommands = new HashMap<>();
    private final Map> mapCmdHandlers = new HashMap<>();
    private final Map> mapEvtHandlers = new HashMap<>();
    
    public ResourceLoader() {
        loadEntities();
        loadEvents();
        loadCommands();
        loadCmdHandlers();
        loadEvtHandlers();
    }

    private ScanResult getScanResult() {
        if (scanResult == null) {
            scanResult = new ClassGraph().enableClassInfo().scan();
        }
        return scanResult;
    }

    private void loadEntities() {
        ClassInfoList controlClasses = getScanResult().getSubclasses(Entity.class.getName());
        for (Class type : controlClasses.loadClasses()) {
            mapEntities.put(type.getSimpleName(), (Class) type);
        }
    }

    private void loadEvents() {
        ClassInfoList controlClasses = getScanResult().getSubclasses(Event.class.getName());
        for (Class type : controlClasses.loadClasses()) {
            mapEvents.put(type.getSimpleName(), (Class) type);
        }
    }

    private void loadCommands() {
        ClassInfoList controlClasses = getScanResult().getSubclasses(Command.class.getName());
        for (Class type : controlClasses.loadClasses()) {
            mapCommands.put(type.getSimpleName(), (Class) type);
        }
    }
    
    private void loadCmdHandlers() {
        ClassInfoList controlClasses = getScanResult().getSubclasses(EntityCommandHandler.class.getName());
        for (Class type : controlClasses.loadClasses()) {
            mapCmdHandlers.put(type.getSimpleName(), (Class) type);
        }
    }
    
    private void loadEvtHandlers() {
        ClassInfoList controlClasses = getScanResult().getSubclasses(EntityEventHandler.class.getName());
        for (Class type : controlClasses.loadClasses()) {
            mapEvtHandlers.put(type.getSimpleName(), (Class) type);
        }
    }
}