put namespace inside the tri-state

kube-rs · Mar 28, 2020 · fb3224d · fb3224d
1 parent 4d2d1c1
commit fb3224d
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 12 deletions.
diff --git a/kube/src/api/crds.rs b/kube/src/api/crds.rs
@@ -110,13 +110,18 @@ impl CrBuilder {
 /// Make Resource useable on CRDs without k8s_openapi
 impl From<CustomResource> for Resource {
     fn from(c: CustomResource) -> Self {
+        use crate::api::resource::ResourceScope;
+        let scope = if let Some(ns) = c.namespace {
+            ResourceScope::Namespace(ns)
+        } else {
+            ResourceScope::All
+        };
         Self {
             api_version: c.api_version,
             kind: c.kind,
             group: c.group,
             version: c.version,
-            namespace: c.namespace, // yeah, needs to be merged into scope
-            scope: crate::api::resource::ResourceScope::Namespace, // otherwise can clash with ::All
+            scope,
         }
     }
 }

diff --git a/kube/src/api/resource.rs b/kube/src/api/resource.rs
@@ -25,16 +25,13 @@ pub struct Resource {
     /// The version of the resource.
     pub version: String,
 
-    /// The namespace if the resource resides (if namespaced)
-    pub namespace: Option<String>,
-
     pub scope: ResourceScope,
 }
 
 #[derive(Clone, Debug, PartialEq)]
 pub enum ResourceScope {
     Cluster,
-    Namespace, // could maybe put the resource.namespace string in here
+    Namespace(String),
     All,
 }
 
@@ -88,7 +85,6 @@ impl Resource {
             kind: K::KIND.to_string(),
             group: K::GROUP.to_string(),
             version: K::VERSION.to_string(),
-            namespace: None,
             scope: ResourceScope::Cluster,
         }
     }
@@ -103,7 +99,6 @@ impl Resource {
             kind: K::KIND.to_string(),
             group: K::GROUP.to_string(),
             version: K::VERSION.to_string(),
-            namespace: None,
             scope: ResourceScope::All
         }
     }
@@ -115,8 +110,7 @@ impl Resource {
             kind: K::KIND.to_string(),
             group: K::GROUP.to_string(),
             version: K::VERSION.to_string(),
-            namespace: Some(ns.to_string()),
-            scope: ResourceScope::Namespace,
+            scope: ResourceScope::Namespace(ns.to_string()),
         }
     }
 }
@@ -125,7 +119,7 @@ impl Resource {
 
 impl Resource {
     pub(crate) fn make_url(&self) -> String {
-        let n = if let Some(ns) = &self.namespace {
+        let n = if let ResourceScope::Namespace(ns) = &self.scope {
             format!("namespaces/{}/", ns)
         } else {
             "".into()

diff --git a/kube/src/runtime/reflector.rs b/kube/src/runtime/reflector.rs
@@ -97,9 +97,15 @@ where
     /// If you are using a non-namespaced resources with name clashes,
     /// Try `Reflector::get_within` instead.
     pub fn get(&self, name: &str) -> Result<Option<K>> {
+        use crate::api::resource::ResourceScope;
+        let namespace = match &self.resource.scope {
+            ResourceScope::Namespace(ns) => Some(ns.to_owned()),
+            _ => None,
+        };
         let id = ObjectId {
             name: name.into(),
-            namespace: self.resource.namespace.clone(),
+            // TODO: impl From<Resource> for ObjectId
+            namespace,
         };
 
         futures::executor::block_on(async { Ok(self.state.lock().await.data.get(&id).map(Clone::clone)) })