com.ottogroup.bi.spqr.resman.server.SPQRResourceManagerServer.java Source code

Introduction

Here is the source code for com.ottogroup.bi.spqr.resman.server.SPQRResourceManagerServer.java

Source

/**
 * Copyright 2015 Otto (GmbH & Co KG)
 *
 * Licensed 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 com.ottogroup.bi.spqr.resman.server;

import io.dropwizard.Application;
import io.dropwizard.client.JerseyClientBuilder;
import io.dropwizard.setup.Environment;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.util.Map;

import org.apache.commons.lang3.StringUtils;
import org.apache.log4j.Logger;
import org.apache.log4j.PropertyConfigurator;

import com.fasterxml.jackson.core.JsonParseException;
import com.fasterxml.jackson.databind.JsonMappingException;
import com.ottogroup.bi.spqr.exception.RequiredInputMissingException;
import com.ottogroup.bi.spqr.repository.ComponentDescriptor;
import com.ottogroup.bi.spqr.repository.ComponentRepository;
import com.ottogroup.bi.spqr.resman.node.SPQRNodeManager;
import com.ottogroup.bi.spqr.resman.resource.node.SPQRNodeManagementResource;
import com.ottogroup.bi.spqr.resman.resource.pipeline.SPQRPipelineManagementResource;

/**
 * @author mnxfst
 * @since Apr 10, 2015
 */
public class SPQRResourceManagerServer extends Application<SPQRResourceManagerConfiguration> {

    /** our faithful logging facility ... ;-) */
    private static final Logger logger = Logger.getLogger(SPQRResourceManagerServer.class);

    /** reference to component repository - required for deploying new artifacts to nodes on request */
    private ComponentRepository componentRepository = null;
    /** reference to spqr node manager */
    private SPQRNodeManager spqrNodeManager = null;

    /**
     * @see io.dropwizard.Application#run(io.dropwizard.Configuration, io.dropwizard.setup.Environment)
     */
    public void run(SPQRResourceManagerConfiguration configuration, Environment environment) throws Exception {
        initializeLog4j(configuration.getLog4jConfiguration());
        this.componentRepository = loadAndDeployApplicationRepository(configuration.getComponentRepositoryFolder());
        this.spqrNodeManager = new SPQRNodeManager(5,
                new JerseyClientBuilder(environment).using(configuration.getHttpClient()));
        environment.jersey().register(new SPQRPipelineManagementResource());
        environment.jersey().register(new SPQRNodeManagementResource(spqrNodeManager));
        Runtime.getRuntime().addShutdownHook(new SPQRResourceManagerShutdownHandler());
    }

    /**
     * Initializes the log4j framework by pointing it to the configuration file referenced by parameter
     * @param log4jConfigurationFile
     */
    protected void initializeLog4j(final String log4jConfigurationFile) {
        String log4jPropertiesFile = StringUtils.lowerCase(StringUtils.trim(log4jConfigurationFile));
        if (StringUtils.isNoneBlank(log4jConfigurationFile)) {
            File log4jFile = new File(log4jPropertiesFile);
            if (log4jFile.isFile()) {
                try {
                    PropertyConfigurator.configure(new FileInputStream(log4jFile));
                } catch (FileNotFoundException e) {
                    System.out.println("No log4j configuration found at '" + log4jConfigurationFile + "'");
                }
            } else {
                System.out.println("No log4j configuration found at '" + log4jConfigurationFile + "'");
            }
        } else {
            System.out.println("No log4j configuration file provided");
        }
    }

    /**
     * Reads contents from {@link ApplicationRepositoryConfiguration referenced repository} and deploys them to given
     * {@link ActorRef deployment receiver}.
     * @param cfg
     * @param jarDeploymentReceiverRef
     * @throws RequiredInputMissingException
     * @throws JsonParseException
     * @throws JsonMappingException
     * @throws IOException
     */
    public ComponentRepository loadAndDeployApplicationRepository(final String repositoryPath)
            throws RequiredInputMissingException, JsonParseException, JsonMappingException, IOException {

        ////////////////////////////////////////////////////////////////////////////
        // validate provided input and ensure that folder exists
        if (StringUtils.isBlank(repositoryPath))
            throw new RequiredInputMissingException("Missing required input for parameter 'repositoryPath'");

        File repositoryFolder = new File(StringUtils.trim(repositoryPath));
        if (repositoryFolder == null || !repositoryFolder.isDirectory())
            throw new RequiredInputMissingException(
                    "No repository found at provided folder '" + repositoryPath + "'");
        //
        ////////////////////////////////////////////////////////////////////////////

        ////////////////////////////////////////////////////////////////////////////

        final ComponentRepository library = new ComponentRepository();

        ////////////////////////////////////////////////////////////////////////////
        // find component repositories below repository folder 
        File[] repoFolders = repositoryFolder.listFiles();
        if (repoFolders == null || repoFolders.length < 1)
            return library;
        //
        ////////////////////////////////////////////////////////////////////////////

        ////////////////////////////////////////////////////////////////////////////
        // for each entry check if it is a folder and pass it on the library
        int folderCount = 0;
        int componentsCount = 0;
        for (File folder : repoFolders) {
            if (folder.isDirectory()) {
                try {
                    logger.info("Processing folder '" + folder.getAbsolutePath() + "'");
                    Map<String, ComponentDescriptor> descriptors = library
                            .addComponentFolder(folder.getAbsolutePath());
                    componentsCount = componentsCount + descriptors.size();
                    folderCount++;
                } catch (Exception e) {
                    logger.error("Failed to add folder '" + folder.getAbsolutePath()
                            + "' to component repository. Error: " + e.getMessage());
                }
            }
        }
        //
        ////////////////////////////////////////////////////////////////////////////

        logger.info("Components deployment finished [repo=" + repositoryPath + ", componentFolders=" + folderCount
                + ", componets=" + componentsCount + "]");
        return library;
    }

    /**
     * Executes the SPQR resource manager
     * @param args
     * @throws Exception
     */
    public static void main(String[] args) throws Exception {
        new SPQRResourceManagerServer().run(args);
    }
}