[FLINK-17332][k8s] Fix restart policy not equals to Never for native …

…task manager pods This closes apache#11949 .
static-max · May 6, 2020 · 04ab8d2 · 04ab8d2
1 parent fef974d
commit 04ab8d2
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/...main/java/org/apache/flink/kubernetes/kubeclient/decorators/InitTaskManagerDecorator.java b/...main/java/org/apache/flink/kubernetes/kubeclient/decorators/InitTaskManagerDecorator.java
@@ -59,6 +59,7 @@ public FlinkPod decorateFlinkPod(FlinkPod flinkPod) {
  .withAnnotations(kubernetesTaskManagerParameters.getAnnotations())
  .endMetadata()
  .editOrNewSpec()
+ .withRestartPolicy(Constants.RESTART_POLICY_OF_NEVER)
  .withImagePullSecrets(kubernetesTaskManagerParameters.getImagePullSecrets())
  .withNodeSelector(kubernetesTaskManagerParameters.getNodeSelector())
  .withTolerations(kubernetesTaskManagerParameters.getTolerations().stream()

diff --git a/flink-kubernetes/src/main/java/org/apache/flink/kubernetes/utils/Constants.java b/flink-kubernetes/src/main/java/org/apache/flink/kubernetes/utils/Constants.java
@@ -78,4 +78,6 @@ public class Constants {
  public static final String HEADLESS_SERVICE_CLUSTER_IP = "None";
 
  public static final int MAXIMUM_CHARACTERS_OF_CLUSTER_ID = 45;
+
+ public static final String RESTART_POLICY_OF_NEVER = "Never";
 }
diff --git a/.../java/org/apache/flink/kubernetes/kubeclient/decorators/InitTaskManagerDecoratorTest.java b/.../java/org/apache/flink/kubernetes/kubeclient/decorators/InitTaskManagerDecoratorTest.java
@@ -162,6 +162,13 @@ public void testPodAnnotations() {
  assertThat(resultAnnotations, is(equalTo(ANNOTATIONS)));
  }
 
+ @Test
+ public void testRestartPolicy() {
+ final String resultRestartPolicy = this.resultPod.getSpec().getRestartPolicy();
+
+ assertThat(resultRestartPolicy, is(Constants.RESTART_POLICY_OF_NEVER));
+ }
+
  @Test
  public void testImagePullSecrets() {
  final List<String> resultSecrets = this.resultPod.getSpec().getImagePullSecrets()