org.kuali.common.util.main.MainUtils Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of kuali-util Show documentation
Kuali Utilities
There is a newer version: 4.4.17
/**
 * Copyright 2010-2014 The Kuali Foundation
 *
 * Licensed under the Educational Community 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.opensource.org/licenses/ecl2.php
 *
 * 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.kuali.common.util.main;

import java.util.Collections;
import java.util.Map;

import org.kuali.common.util.spring.SpringExecutable;
import org.kuali.common.util.spring.service.SpringContext;
import org.kuali.common.util.spring.service.SpringService;

public class MainUtils {

	public static final String MAIN_CONTEXT_BEAN_NAME = "mainContext";
	public static final String MAIN_PROFILE_NAME = "main";

	/**
	 * Load the @Configuration mainClass using Spring and then terminate the JVM.
	 */
	public static void runAndExit(Class mainClass, String[] args) {
		runAndExit(mainClass, args, false);
	}

	/**
	 * Load the @Configuration mainClass using Spring and then terminate the JVM.
	 */
	public static void runAndExit(Class mainClass, String[] args, boolean stacktrace) {
		run(mainClass, args, stacktrace, true);
	}

	/**
	 * Load the @Configuration mainClass using Spring
	 */
	public static void run(Class mainClass, String[] args, boolean stacktrace) {
		run(mainClass, args, stacktrace, false);
	}

	/**
	 * Load the @Configuration mainClass using Spring
	 */
	public static void run(Class mainClass, String[] args) {
		run(mainClass, args, true, false);
	}

	/**
	 * 
	 */
	public static void run(Class mainClass, String[] args, boolean stacktrace, boolean exit) {
		try {
			// Preserve the context info from the class where main(String[] args) was invoked
			MainContext mainContext = new MainContext(mainClass, args);

			// Create a map containing the context so we can register it with Spring
			Map beans = Collections.singletonMap(MAIN_CONTEXT_BEAN_NAME, (Object) mainContext);

			// Create a SpringContext using the map and main class, with 1 active profile called "main"
			SpringContext context = new SpringContext(beans, mainClass, MAIN_PROFILE_NAME);

			// DefaultSpringService does what we need
			SpringService service = SpringExecutable.DEFAULT_SPRING_SERVICE;

			// This causes Spring to load the @Configuration annotated main class
			new SpringExecutable(service, context).execute();

			// Exit with zero if there is no exception
			if (exit) {
				System.exit(Status.SUCCESS.getValue());
			}
		} catch (Exception e) {
			handleException(e, stacktrace, exit);
		}
	}

	protected static void handleException(Exception e, boolean stacktrace, boolean exit) {
		if (stacktrace) {
			e.printStackTrace();
		} else {
			System.err.print(e.getMessage());
		}
		if (exit) {
			System.exit(Status.FAILURE.getValue());
		}
	}

}