From cb6ca6211f32ea9a4ebc812c2de3f1d9bfd15735 Mon Sep 17 00:00:00 2001
From: Yifan Xiong <yifan.xiong@microsoft.com>
Date: Wed, 9 Oct 2019 10:44:10 +0800
Subject: [PATCH] [Rest Server] Add node selector for user jobs in k8s (#3694)

Add node selector for user jobs in k8s,
only run user pods on pai-worker nodes.
---
 src/rest-server/src/models/v2/job/k8s.js | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/src/rest-server/src/models/v2/job/k8s.js b/src/rest-server/src/models/v2/job/k8s.js
index 21cef6d920..080b078578 100644
--- a/src/rest-server/src/models/v2/job/k8s.js
+++ b/src/rest-server/src/models/v2/job/k8s.js
@@ -405,6 +405,23 @@ const generateTaskRole = (taskRole, labels, config) => {
               },
             },
           ],
+          affinity: {
+            nodeAffinity: {
+              requiredDuringSchedulingIgnoredDuringExecution: {
+                nodeSelectorTerms: [
+                  {
+                    matchExpressions: [
+                      {
+                        key: 'pai-worker',
+                        operator: 'In',
+                        values: ['true'],
+                      },
+                    ],
+                  },
+                ],
+              },
+            },
+          },
           imagePullSecrets: [
             {
               name: launcherConfig.runtimeImagePullSecrets,