ca.ibodrov.mica.server.data.InitialDataLoader Maven / Gradle / Ivy

Go to download
package ca.ibodrov.mica.server.data;

import ca.ibodrov.mica.api.model.PartialEntity;
import ca.ibodrov.mica.db.MicaDB;
import ca.ibodrov.mica.server.YamlMapper;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.walmartlabs.concord.server.security.UserPrincipal;
import org.jooq.DSLContext;
import org.reflections.Reflections;
import org.reflections.scanners.ResourcesScanner;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.inject.Inject;
import java.io.IOException;

import static ca.ibodrov.mica.server.data.UserEntryUtils.systemUser;
import static java.nio.charset.StandardCharsets.UTF_8;
import static java.util.Objects.requireNonNull;

/**
 * Responsible for loading the initial data set, e.g. the default entity kinds
 * and such.
 */
public class InitialDataLoader {

    private static final Logger log = LoggerFactory.getLogger(InitialDataLoader.class);

    private final DSLContext dsl;
    private final EntityStore entityStore;
    private final ObjectMapper objectMapper;

    @Inject
    public InitialDataLoader(@MicaDB DSLContext dsl,
                             EntityStore entityStore,
                             ObjectMapper objectMapper) {

        this.dsl = requireNonNull(dsl);
        this.entityStore = requireNonNull(entityStore);
        this.objectMapper = requireNonNull(objectMapper);

        // no out of the box support for @PostConstruct in Guice
        load();
    }

    public void load() {
        var session = new UserPrincipal("system", systemUser());
        // load example files
        loadPackage(session, "ca.ibodrov.mica.server.examples");
        // load other stuff
        loadPackage(session, "ca.ibodrov.mica.server.entities");
    }

    private void loadPackage(UserPrincipal session, String packageName) {
        var cl = getClass().getClassLoader();
        var yamlMapper = new YamlMapper(objectMapper);
        var reflections = new Reflections(packageName, new ResourcesScanner());
        reflections.getResources(s -> s.endsWith(".yaml")).forEach(resourceName -> {
            try (var in = cl.getResourceAsStream(resourceName)) {
                assert in != null;
                var doc = new String(in.readAllBytes(), UTF_8);
                var entity = yamlMapper.readValue(doc, PartialEntity.class);
                createOrReplace(session, entity, doc);
            } catch (IOException e) {
                throw new RuntimeException("Error loading " + resourceName, e);
            }
        });
    }

    private void createOrReplace(UserPrincipal session, PartialEntity entity, String doc) {
        dsl.transaction(cfg -> {
            var tx = cfg.dsl();

            entityStore.getByName(tx, entity.name())
                    .flatMap(existingEntity -> entityStore.deleteById(tx, session, existingEntity.id()))
                    .ifPresent(deleted -> log.info("Removed old version of {}: {}", entity.name(), deleted));

            entityStore.upsert(tx, session, entity, doc);
        });

        log.info("Created or replaced an entity: {}", entity.name());
    }
}