Add registrat

Rafal Wojdyla · mxm · commit a1e1c74b301a · 2016-03-15T13:34:09.000+01:00
diff --git a/runner/pom.xml b/runner/pom.xml
@@ -83,6 +83,11 @@
 				</exclusion>
 			</exclusions>
 		</dependency>
+		<dependency>
+			<groupId>com.google.auto.service</groupId>
+			<artifactId>auto-service</artifactId>
+			<version>1.0-rc2</version>
+		</dependency>
 		<!-- Test scoped -->
 		<dependency>
 			<groupId>com.google.cloud.dataflow</groupId>
diff --git a/runner/src/main/java/com/dataartisans/flink/dataflow/FlinkRunnerRegistrar.java b/runner/src/main/java/com/dataartisans/flink/dataflow/FlinkRunnerRegistrar.java
@@ -0,0 +1,52 @@
+package com.dataartisans.flink.dataflow;/*
+ * Copyright 2016 Data Artisans GmbH
+ *
+ * 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.
+ */
+
+import com.google.auto.service.AutoService;
+import com.dataartisans.flink.dataflow.FlinkPipelineOptions;
+import com.dataartisans.flink.dataflow.FlinkPipelineRunner;
+import com.google.cloud.dataflow.sdk.options.PipelineOptions;
+import com.google.cloud.dataflow.sdk.options.PipelineOptionsRegistrar;
+import com.google.cloud.dataflow.sdk.runners.PipelineRunner;
+import com.google.cloud.dataflow.sdk.runners.PipelineRunnerRegistrar;
+import com.google.common.collect.ImmutableList;
+
+
+/**
+ * AuteService registrar - will register FlinkRunner and FlinkOptions
+ * as possible pipeline runner services.
+ *
+ * It ends up in META-INF/services and gets picked up by Dataflow.
+ *
+ */
+public class FlinkRunnerRegistrar {
+	private FlinkRunnerRegistrar() { }
+
+	@AutoService(PipelineRunnerRegistrar.class)
+	public static class Runner implements PipelineRunnerRegistrar {
+		@Override
+		public Iterable<Class<? extends PipelineRunner<?>>> getPipelineRunners() {
+			return ImmutableList.<Class<? extends PipelineRunner<?>>>of(FlinkPipelineRunner.class);
+		}
+	}
+
+	@AutoService(PipelineOptionsRegistrar.class)
+	public static class Options implements PipelineOptionsRegistrar {
+		@Override
+		public Iterable<Class<? extends PipelineOptions>> getPipelineOptions() {
+			return ImmutableList.<Class<? extends PipelineOptions>>of(FlinkPipelineOptions.class);
+		}
+	}
+}
