Spring Bootのwebアプリとバッチ処理をひとつにまとめる

Dockerで運用しているSpring Bootでできたwebアプリに、あらたにバッチ処理が必要になった。
webアプリと同じサーバで実行するので、同一のjarでwebアプリもバッチ処理も行えるようにしてみた。
仕組みとしては、javaコマンドでjarを実行するときに目的のAppクラスを指定出来るようになっている。

ビルドの設定

pom.xml

<build>
	<plugins>
		<plugin>
			<groupId>org.apache.maven.plugins</groupId>
			<artifactId>maven-jar-plugin</artifactId>
			<configuration>
				<archive>
					<manifest>
						<mainClass>sandbox.Entrypoint</mainClass>
					</manifest>
				</archive>
			</configuration>
		</plugin>
		<plugin>
			<groupId>org.springframework.boot</groupId>
			<artifactId>spring-boot-maven-plugin</artifactId>
			<configuration>
				<classifier>exec</classifier>
			</configuration>
		</plugin>
	</plugins>
</build>

mavenでjarをビルドする時に、javaコマンドで実行されるクラスが指定する。
このクラスで、引数に応じて目的のAppクラスを切り替えられるようにする。

アプリの基底クラス

AppBase.java

@Configuration
public abstract class AppBase {
	
	@Autowired
	DataSource dataSource;

	@Autowired
	Dialect dialect;

	@Autowired
	Naming naming;

}

webアプリ/バッチ処理で共通して利用するデータソースの設定などを行う。

webアプリ

WebApp.java

@EnableAutoConfiguration
@ComponentScan
public class WebApp extends AppBase {
	
	public static void main(String[] args) {
		SpringApplication.run(WebApp.class, args);
	}

}

元のAppクラスから、わかりやすいように名称を変更した。
特に変わったことは行わない。

バッチ処理

Batch.java

@EnableBatchProcessing
@EnableAutoConfiguration
@ComponentScan
public class Batch extends AppBase {

	@Autowired
	JobBuilderFactory jobBuilderFactory;

	@Autowired
	StepBuilderFactory stepBuilderFactory;

	@Autowired
	Crawler crawler;
	
	public static void main(String[] args) {
		SpringApplication application = new SpringApplication(Batch.class);
		application.setWebEnvironment(false);
		System.exit(SpringApplication.exit(application.run(args)));
	}

	@Bean
	public Step crawl() {
		return stepBuilderFactory.get("crawl").tasklet(new Tasklet() {
			public RepeatStatus execute(StepContribution stepContribution,
					ChunkContext chunkContext) throws Exception {
				crawler.crawl();
				return RepeatStatus.FINISHED;
			}
		}).build();
	}

	@Bean
	public Job job(Step crawl) throws Exception {
		return jobBuilderFactory.get("site-crawl")
				.incrementer(new RunIdIncrementer()).start(crawl).build();
	}

}

新たに実装したAppクラス。
実行時に不要なTomcatが起動しないようにしている。

エンドポイント

Entrypoint.java

public class Entrypoint {

	public static void main(String[] args) throws Exception {
		if (args != null && args.length > 0 && "web".equals(args[0])) {
			WebApp.main(args);
			return;
		}

		Batch.main(args);
	}

}

引数に応じて目的のAppクラスを切り替えるクラス。
第１引数が「web」の場合にwebアプリを実行するように定義。
新たなバッチ処理が増えた場合は、この引数のパターンを増やして対処が出来ると思う。

Docker

FROM frolvlad/alpine-oraclejdk8:slim
ENTRYPOINT ["java","-Djava.security.egd=file:/dev/./urandom","-jar","/var/app.jar"]

イメージを作成するDockerfileではjarの実行を指定するのみ。

実行

Dockerを実行するサーバの/var/xxx.jarへ、avenでビルドしたjarを配置する。

run_web.sh

docker run --name site-crawler-web \
  -d \
  -v /etc/localtime:/etc/localtime:ro \
  -v /var/xxx.jar:/var/app.jar \
  -p 80:8080 \
  container \
  web

webアプリを実行するときは、引数に「web」を指定する。

run_batch.sh

docker run --name site-crawler-batch \
  -d \
  -v /etc/localtime:/etc/localtime:ro \
  -v /var/xxx.jar:/var/app.jar \
  container \
  batch

バッチを実行するときは、引数に「batch」（「web」以外）を指定する。
バッチ処理が終わるとコンテナが終了する。