uk.ac.kcl.Main.java Source code

Introduction

Here is the source code for uk.ac.kcl.Main.java

Source

/* 
 * Copyright 2016 King's College London, Richard Jackson <richgjackson@gmail.com>.
 *
 * 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 uk.ac.kcl;

import org.springframework.context.annotation.AnnotationConfigApplicationContext;
import org.springframework.core.env.ConfigurableEnvironment;
import org.springframework.core.env.MapPropertySource;
import org.springframework.core.env.MutablePropertySources;
import org.springframework.core.env.StandardEnvironment;
import uk.ac.kcl.batch.JobConfiguration;
import uk.ac.kcl.scheduling.ScheduledJobLauncher;
import uk.ac.kcl.scheduling.SingleJobLauncher;
import uk.ac.kcl.utils.TcpHelper;

import java.io.*;
import java.util.HashMap;
import java.util.Map;
import java.util.Properties;

/**
 *
 * @author King's College London, Richard Jackson <richgjackson@gmail.com>
 */
public class Main {

    public static void main(String[] args) {
        File folder = new File(args[0]);
        File[] listOfFiles = folder.listFiles();
        assert listOfFiles != null;
        for (File listOfFile : listOfFiles) {
            if (listOfFile.isFile()) {
                if (listOfFile.getName().endsWith(".properties")) {
                    System.out.println("Properties sile found:" + listOfFile.getName()
                            + ". Attempting to launch application context");
                    Properties properties = new Properties();
                    InputStream input;
                    try {
                        input = new FileInputStream(listOfFile);
                        properties.load(input);
                        if (properties.getProperty("globalSocketTimeout") != null) {
                            TcpHelper.setSocketTimeout(
                                    Integer.valueOf(properties.getProperty("globalSocketTimeout")));
                        }
                        Map<String, Object> map = new HashMap<>();
                        properties.forEach((k, v) -> {
                            map.put(k.toString(), v);
                        });
                        ConfigurableEnvironment environment = new StandardEnvironment();
                        MutablePropertySources propertySources = environment.getPropertySources();
                        propertySources.addFirst(new MapPropertySource(listOfFile.getName(), map));
                        @SuppressWarnings("resource")
                        AnnotationConfigApplicationContext ctx = new AnnotationConfigApplicationContext();
                        ctx.registerShutdownHook();
                        ctx.setEnvironment(environment);
                        String scheduling;
                        try {
                            scheduling = properties.getProperty("scheduler.useScheduling");
                            if (scheduling.equalsIgnoreCase("true")) {
                                ctx.register(ScheduledJobLauncher.class);
                                ctx.refresh();
                            } else if (scheduling.equalsIgnoreCase("false")) {
                                ctx.register(SingleJobLauncher.class);
                                ctx.refresh();
                                SingleJobLauncher launcher = ctx.getBean(SingleJobLauncher.class);
                                launcher.launchJob();
                            } else if (scheduling.equalsIgnoreCase("slave")) {
                                ctx.register(JobConfiguration.class);
                                ctx.refresh();
                            } else {
                                throw new RuntimeException(
                                        "useScheduling not configured. Must be true, false or slave");
                            }
                        } catch (NullPointerException ex) {
                            throw new RuntimeException(
                                    "useScheduling not configured. Must be true, false or slave");
                        }
                    } catch (IOException e) {
                        e.printStackTrace();
                    }
                }
            }
        }
    }
}