adding model mapper caches

Author: yue9944882

e2e/src/test/groovy/io/kubernetes/client/e2e/util/ModelMapperTest.groovy

@@ -33,12 +33,15 @@ class ModelMapperTest extends Specification {
         expect:
         new GroupVersionKind("", "v1", "Pod") == ModelMapper.getGroupVersionKindByClass(V1Pod.class)
         new GroupVersionResource("", "v1", "pods") == ModelMapper.getGroupVersionResourceByClass(V1Pod.class)
+        ModelMapper.isNamespaced(V1Pod.class)
 
         new GroupVersionKind("apps", "v1", "Deployment") == ModelMapper.getGroupVersionKindByClass(V1Deployment.class)
         new GroupVersionResource("apps", "v1", "deployments") == ModelMapper.getGroupVersionResourceByClass(V1Deployment.class)
+        ModelMapper.isNamespaced(V1Deployment.class)
 
         new GroupVersionKind("apiextensions.k8s.io", "v1beta1", "CustomResourceDefinition") == ModelMapper.getGroupVersionKindByClass(V1beta1CustomResourceDefinition.class)
         new GroupVersionResource("apiextensions.k8s.io", "v1beta1", "customresourcedefinitions") == ModelMapper.getGroupVersionResourceByClass(V1beta1CustomResourceDefinition.class)
+        !ModelMapper.isNamespaced(V1beta1CustomResourceDefinition.class)
 
     }
 }

util/src/main/java/io/kubernetes/client/util/ModelMapper.java

@@ -53,12 +53,13 @@ public class ModelMapper {
   // Model's api-version midfix to kubernetes api-version
   private static List<String> preBuiltApiVersions = new ArrayList<>();
 
-  // TODO(yue9944882): make the map bi-directional
   private static Map<GroupVersionKind, Class<?>> classesByGVK = new ConcurrentHashMap<>();
-
-  // TODO(yue9944882): make the map bi-directional
   private static Map<GroupVersionResource, Class<?>> classesByGVR = new ConcurrentHashMap<>();
 
+  private static Map<Class<?>, GroupVersionKind> gvkByClasses = new ConcurrentHashMap<>();
+  private static Map<Class<?>, GroupVersionResource> gvrByClasses = new ConcurrentHashMap<>();
+  private static Map<Class<?>, Boolean> isNamespacedByClasses = new ConcurrentHashMap<>();
+
   private static Set<Discovery.APIResource> lastAPIDiscovery = new HashSet<>();
 
   private static volatile long nextDiscoveryRefreshTimeMillis =
@@ -114,8 +115,34 @@ public static void addModelMap(String group, String version, String kind, Class<
    */
   public static void addModelMap(
       String group, String version, String kind, String resourceNamePlural, Class<?> clazz) {
+    // TODO(yue9944882): consistency between bi-directional maps
     classesByGVK.put(new GroupVersionKind(group, version, kind), clazz);
+    gvkByClasses.put(clazz, new GroupVersionKind(group, version, kind));
+
     classesByGVR.put(new GroupVersionResource(group, version, resourceNamePlural), clazz);
+    gvrByClasses.put(clazz, new GroupVersionResource(group, version, resourceNamePlural));
+  }
+
+  /**
+   * Registering concrete model classes by its group, version, kind and isNamespaced (e.g. "apps",
+   * "v1", "Deployment", true).
+   *
+   * @param group the group
+   * @param version the version
+   * @param kind the kind
+   * @param resourceNamePlural the resource name plural
+   * @param isNamespacedResource the is namespaced resource
+   * @param clazz the clazz
+   */
+  public static void addModelMap(
+      String group,
+      String version,
+      String kind,
+      String resourceNamePlural,
+      Boolean isNamespacedResource,
+      Class<?> clazz) {
+    addModelMap(group, version, kind, resourceNamePlural, clazz);
+    isNamespacedByClasses.put(clazz, isNamespacedResource);
   }
 
   /**
@@ -163,11 +190,7 @@ public static Class<?> getApiTypeClass(String group, String version, String kind
    * @return the group version kind by class
    */
   public static GroupVersionKind getGroupVersionKindByClass(Class<?> clazz) {
-    return classesByGVK.entrySet().stream()
-        .filter(e -> clazz.equals(e.getValue()))
-        .map(e -> e.getKey())
-        .findFirst()
-        .orElse(preBuiltGetGroupVersionKindByClass(clazz));
+    return gvkByClasses.get(clazz);
   }
 
   /**
@@ -177,13 +200,18 @@ public static GroupVersionKind getGroupVersionKindByClass(Class<?> clazz) {
    * @return the group version kind by class
    */
   public static GroupVersionResource getGroupVersionResourceByClass(Class<?> clazz) {
-    return classesByGVR.entrySet().stream()
-        .filter(e -> clazz.equals(e.getValue()))
-        .map(e -> e.getKey())
-        .findFirst()
-        .get();
+    return gvrByClasses.get(clazz);
   }
 
+  /**
+   * Refreshes the model mapping by syncing up w/the api discovery info from the kubernetes
+   * apiserver. These mapping will be cached for {@link
+   * ModelMapper#DEFAULT_DISCOVERY_REFRESH_INTERVAL}.
+   *
+   * @param discovery the discovery
+   * @return the set
+   * @throws ApiException the api exception
+   */
   public static Set<Discovery.APIResource> refresh(Discovery discovery) throws ApiException {
     return refresh(discovery, DEFAULT_DISCOVERY_REFRESH_INTERVAL);
   }
@@ -220,6 +248,7 @@ public static Set<Discovery.APIResource> refresh(Discovery discovery, Duration r
             version,
             apiResource.getKind(),
             apiResource.getResourcePlural(),
+            apiResource.getNamespaced(),
             clazz);
       }
     }
@@ -253,6 +282,16 @@ public static GroupVersionKind preBuiltGetGroupVersionKindByClass(Class<?> clazz
         .get();
   }
 
+  /**
+   * Checks whether the class is connected with a namespaced kubernetes resource.
+   *
+   * @param clazz the clazz
+   * @return the boolean
+   */
+  public static Boolean isNamespaced(Class<?> clazz) {
+    return isNamespacedByClasses.get(clazz);
+  }
+
   private static void initApiGroupMap() {
     preBuiltApiGroups.put("Admissionregistration", "admissionregistration.k8s.io");
     preBuiltApiGroups.put("Apiextensions", "apiextensions.k8s.io");