Add DNS Policy to fluent controller (#372)

Signed-off-by: Wenjie Ma <[email protected]>

deploy/fluent/crd.yaml

@@ -21,6 +21,8 @@ spec:
           properties:
             spec:
               properties:
+                DNSPolicy:
+                  type: string
                 affinity:
                   description: Affinity is a group of affinity scheduling rules.
                   nullable: true
@@ -557,6 +559,7 @@ spec:
                   nullable: true
                   type: array
               required:
+                - DNSPolicy
                 - fluentBitConfigName
                 - image
                 - runtimeClassName

e2e/src/fluent_e2e.rs

@@ -34,6 +34,7 @@ pub fn fluent_bit() -> String {
         fluentBitConfigName: fluent-bit-config
         image: kubesphere/fluent-bit:v2.1.7
         runtimeClassName: ''
+        DNSPolicy: ''
         tolerations:
             - operator: Exists
     "

src/controller_examples/fluent_controller/fluentbit/exec/resource/daemon_set.rs

@@ -301,6 +301,9 @@ fn make_fluentbit_pod_spec(fb: &FluentBit) -> (pod_spec: PodSpec)
     if !fb.spec().runtime_class_name().eq(&new_strlit("").to_string()) {
         pod_spec.set_runtime_class_name(fb.spec().runtime_class_name());
     }
+    if !fb.spec().dns_policy().eq(&new_strlit("").to_string()) {
+        pod_spec.set_dns_policy(fb.spec().dns_policy());
+    }
     pod_spec
 }
 

src/controller_examples/fluent_controller/fluentbit/exec/types.rs

@@ -223,6 +223,14 @@ impl FluentBitSpec {
     {
         String::from_rust_string(self.inner.runtime_class_name.clone())
     }
+
+    #[verifier(external_body)]
+    pub fn dns_policy(&self) -> (dns_policy: String)
+        ensures
+            dns_policy@ == self@.dns_policy,
+    {
+        String::from_rust_string(self.inner.dns_policy.clone())
+    }
 }
 
 }

src/controller_examples/fluent_controller/fluentbit/spec/resource/daemon_set.rs

@@ -218,6 +218,11 @@ pub open spec fn make_fluentbit_pod_spec(fb: FluentBitView) -> PodSpecView
             } else {
                 PodSpecView::default().runtime_class_name
             },
+        dns_policy: if fb.spec.dns_policy != new_strlit("")@ {
+                Some(fb.spec.dns_policy) 
+            } else {
+                PodSpecView::default().dns_policy
+            },
         ..PodSpecView::default()
     }
 }

src/controller_examples/fluent_controller/fluentbit/spec/types.rs

@@ -148,6 +148,7 @@ pub struct FluentBitSpecView {
     pub affinity: Option<AffinityView>,
     pub node_selector: Map<StringView, StringView>,
     pub runtime_class_name: StringView,
+    pub dns_policy: StringView,
 }
 
 impl FluentBitSpecView {}

src/deps_hack/src/lib.rs

@@ -214,6 +214,8 @@ pub struct FluentBitSpec {
     pub node_selector: std::collections::BTreeMap<String, String>,
     #[serde(rename = "runtimeClassName")]
     pub runtime_class_name: String,
+    #[serde(rename = "DNSPolicy")]
+    pub dns_policy: String,
 }
 
 #[derive(