Merge pull request #98 from shivramsrivastava/affinity_enhancement

Node Affinity and Node Anti-Affinity enhancement related changes.

Author: k8s-ci-robot

Gopkg.lock

@@ -3,13 +3,17 @@ load("@io_bazel_rules_go//go:def.bzl", "go_library", "go_test")
 go_library(
     name = "go_default_library",
     srcs = [
+        "affinity.pb.go",
         "coco_interference_scores.pb.go",
         "firmament_client.go",
         "firmament_scheduler.pb.go",
         "firmament_scheduler_mock.go",
         "job_desc.pb.go",
         "label.pb.go",
         "label_selector.pb.go",
+        "node_affinity.pb.go",
+        "pod_affinity.pb.go",
+        "pod_anti_affinity.pb.go",
         "reference_desc.pb.go",
         "resource_desc.pb.go",
         "resource_stats.pb.go",

pkg/firmament/BUILD

@@ -0,0 +1,18 @@
+syntax = "proto3";
+
+package firmament;
+
+import "node_affinity.proto";
+import "pod_affinity.proto";
+import "pod_anti_affinity.proto";
+
+message Affinity {
+
+    NodeAffinity node_affinity = 1;
+    
+    PodAffinity pod_affinity = 2;
+
+    PodAntiAffinity pod_anti_affinity = 3;
+}
+
+