WordCount.java

package com.oreilly.springdata.hadoop.wordcount;

import java.util.Date;
import java.util.LinkedHashMap;
import java.util.Map;
import java.util.Scanner;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.springframework.batch.core.BatchStatus;
import org.springframework.batch.core.Job;
import org.springframework.batch.core.JobParameter;
import org.springframework.batch.core.JobParameters;
import org.springframework.batch.core.JobParametersBuilder;
import org.springframework.batch.core.launch.JobLauncher;
import org.springframework.beans.factory.BeanInitializationException;
import org.springframework.context.ApplicationContext;
import org.springframework.context.support.AbstractApplicationContext;
import org.springframework.context.support.ClassPathXmlApplicationContext;
import org.springframework.util.Assert;

public class WordCount {

	private static final Log log = LogFactory.getLog(WordCount.class);

	public static void main(String[] args) throws Exception {
		AbstractApplicationContext context = new ClassPathXmlApplicationContext("classpath:/META-INF/spring/*-context.xml");
				//new ClassPathXmlApplicationContext("classpath*:/META-INF/spring/*-context.xml", WordCount.class);
		log.info("Batch WordCount Application Running");
		context.registerShutdownHook();

		JobLauncher jobLauncher = context.getBean(JobLauncher.class);
		Job job = context.getBean(Job.class);

		// wordcount.input.path=/user/gutenberg/input/word/
		// wordcount.output.path=/user/gutenberg/output/word/
		/*
		 * Map<String, JobParameter> p = new LinkedHashMap<String,
		 * JobParameter>(); p.put("mr.input", new
		 * JobParameter("/user/gutenberg/input/word/")); p.put("mr.output", new
		 * JobParameter("/user/gutenberg/output/word/")); p.put("localData", new
		 * JobParameter("./data/nietzsche-chapter-1.txt"));
		 */
		jobLauncher.run(
				job,
				new JobParametersBuilder()
						.addString("mr.input", "/user/gutenberg/input/word/")
						.addString("mr.output", "/user/gutenberg/output/word/")
						.addString("localData",
								"./data/nietzsche-chapter-1.txt")
						.addDate("date", new Date()).toJobParameters());
		// startJobs(context);

	}

	public static void startJobs(ApplicationContext ctx) {
		JobLauncher launcher = ctx.getBean(JobLauncher.class);
		Map<String, Job> jobs = ctx.getBeansOfType(Job.class);

		for (Map.Entry<String, Job> entry : jobs.entrySet()) {
			System.out.println("Executing job " + entry.getKey());
			try {
				if (launcher.run(entry.getValue(), new JobParameters())
						.getStatus().equals(BatchStatus.FAILED)) {
					throw new BeanInitializationException(
							"Failed executing job " + entry.getKey());
				}
			} catch (Exception ex) {
				throw new BeanInitializationException("Cannot execute job "
						+ entry.getKey(), ex);
			}
		}
	}
}