org.tomitribe.tribestream.registryng.bootstrap.Provisioning.java Source code

Introduction

Here is the source code for org.tomitribe.tribestream.registryng.bootstrap.Provisioning.java

Source

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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.tomitribe.tribestream.registryng.bootstrap;

import com.fasterxml.jackson.databind.ObjectMapper;
import io.swagger.models.Swagger;
import org.apache.deltaspike.core.api.config.ConfigProperty;
import org.tomitribe.tribestream.registryng.cdi.Tribe;
import org.tomitribe.tribestream.registryng.documentation.Description;
import org.tomitribe.tribestream.registryng.entities.OpenApiDocument;
import org.tomitribe.tribestream.registryng.repository.Repository;
import org.tomitribe.tribestream.registryng.security.LoginContext;
import org.tomitribe.tribestream.registryng.service.search.SearchEngine;

import javax.annotation.PostConstruct;
import javax.ejb.Singleton;
import javax.ejb.Startup;
import javax.inject.Inject;
import javax.script.Bindings;
import javax.script.ScriptEngine;
import javax.script.ScriptEngineManager;
import javax.script.ScriptException;
import java.io.File;
import java.io.FileReader;
import java.io.IOException;
import java.io.Reader;
import java.lang.management.ManagementFactory;
import java.net.URL;
import java.util.List;
import java.util.logging.Level;
import java.util.logging.Logger;
import java.util.stream.Stream;

import static java.util.Optional.ofNullable;

/**
 * Seeds the database at startup with Swagger documents stored in META-INF/classes/seed-db.
 */
@Singleton
@Startup
public class Provisioning {

    private static final Logger LOGGER = Logger.getLogger(Provisioning.class.getName());

    @Inject
    private LoginContext loginContext;

    @Inject
    private Repository repository;

    @Inject
    private SearchEngine searchEngine;

    @Inject
    @Tribe
    private ObjectMapper mapper;

    @Inject // allow to switch it off or to use an external source for testing
    @Description("Where automatic seeding takes the swagger documents, either at classpath or a directory path")
    @ConfigProperty(name = "tribe.registry.seeding.location")
    private String location;

    @Inject
    @Description("Optional script executed at startup (javascript)")
    @ConfigProperty(name = "tribe.registry.seeding.script")
    private String script;

    @PostConstruct
    public void init() {
        loginContext.setUsername("system");

        ofNullable(script).ifPresent(s -> {
            final ScriptEngine engine = new ScriptEngineManager().getEngineByExtension("js");
            final Bindings bindings = engine.createBindings();
            bindings.put("props", System.getProperties());

            final File file = new File(s);
            if (file.isFile()) {
                try (final Reader reader = new FileReader(file)) {
                    engine.eval(reader, bindings);
                } catch (final IOException | ScriptException e) {
                    throw new IllegalArgumentException(e);
                }
            } else {
                try {
                    engine.eval(s, bindings);
                } catch (final ScriptException e) {
                    throw new IllegalArgumentException(e);
                }
            }
        });
        restore();
    }

    private void seedDatabase() {
        final File dir = new File(location);
        if (dir.isDirectory()) {
            doSeeding(dir);
            return;
        }

        // else try classpath

        final URL res = Thread.currentThread().getContextClassLoader().getResource(location);
        if (res == null) {
            LOGGER.log(Level.WARNING, "Cannot find seed-db resource in the classpath.");
            return;
        }
        if (!"file".equals(res.getProtocol())) {
            LOGGER.log(Level.WARNING, "Cannot load initial OpenAPI documents because seed-db is at {0}!", res);
            return;
        }
        doSeeding(new File(res.getFile()));

    }

    private void doSeeding(final File f) {

        // before seeding, start wiping out all the applications and endpoints
        final List<OpenApiDocument> apps = repository.findAllApplications();
        apps.forEach(d -> repository.deleteApplication(d.getId()));

        Stream.of(ofNullable(f.listFiles((dir, name) -> name.endsWith(".json"))).orElseGet(() -> new File[0]))
                .forEach(this::seedFile);
    }

    private void seedFile(final File swaggerFile) {
        LOGGER.info("Seeding " + swaggerFile.getName());

        try {
            final Swagger swagger = mapper.readValue(swaggerFile, Swagger.class);

            if (repository.findApplicationByNameAndVersion(swagger.getInfo().getTitle(),
                    swagger.getInfo().getVersion()) == null) {
                OpenApiDocument openApiDocument = repository.insert(swagger);
                LOGGER.log(Level.INFO, "Persisted application {0}-{1}",
                        new Object[] { openApiDocument.getName(), openApiDocument.getVersion() });

            } else {
                LOGGER.log(Level.INFO, "Application {0}-{1} already available in DB ",
                        new Object[] { swagger.getInfo().getTitle(), swagger.getInfo().getVersion() });
            }

        } catch (final Exception e) {
            LOGGER.log(Level.WARNING, e, () -> String.format("Seeding %s failed!", swaggerFile.getName()));
        }
        LOGGER.info("Memory = " + ManagementFactory.getMemoryMXBean().getHeapMemoryUsage());
    }

    public void restore() {
        if (location == null || location.trim().isEmpty()) {
            return;
        }

        seedDatabase();
    }
}