103 lines
3.5 KiB
Go
103 lines
3.5 KiB
Go
package helper
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
apierrors "k8s.io/apimachinery/pkg/api/errors"
|
|
"k8s.io/apimachinery/pkg/api/meta"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
"k8s.io/apimachinery/pkg/runtime/schema"
|
|
"k8s.io/client-go/dynamic"
|
|
"k8s.io/klog/v2"
|
|
|
|
"github.com/karmada-io/karmada/pkg/util/fedinformer/genericmanager"
|
|
"github.com/karmada-io/karmada/pkg/util/fedinformer/keys"
|
|
"github.com/karmada-io/karmada/pkg/util/restmapper"
|
|
)
|
|
|
|
// GetObjectFromCache gets full object information from cache by key in worker queue.
|
|
func GetObjectFromCache(
|
|
restMapper meta.RESTMapper,
|
|
manager genericmanager.MultiClusterInformerManager,
|
|
fedKey keys.FederatedKey,
|
|
) (*unstructured.Unstructured, error) {
|
|
gvr, err := restmapper.GetGroupVersionResource(restMapper, fedKey.GroupVersionKind())
|
|
if err != nil {
|
|
klog.Errorf("Failed to get GVR from GVK %s. Error: %v", fedKey.GroupVersionKind(), err)
|
|
return nil, err
|
|
}
|
|
|
|
singleClusterManager := manager.GetSingleClusterManager(fedKey.Cluster)
|
|
if singleClusterManager == nil {
|
|
// That may happen in case of multi-controllers sharing one informer. For example:
|
|
// controller-A takes responsibility of initialize informer for clusters, but controller-B consumes
|
|
// the informer before the initialization.
|
|
// Usually this error will be eliminated during the controller reconciling loop.
|
|
return nil, fmt.Errorf("the informer of cluster(%s) has not been initialized", fedKey.Cluster)
|
|
}
|
|
|
|
if !singleClusterManager.IsInformerSynced(gvr) {
|
|
// fall back to call api server in case the cache has not been synchronized yet
|
|
return getObjectFromSingleCluster(gvr, &fedKey.ClusterWideKey, singleClusterManager.GetClient())
|
|
}
|
|
|
|
var obj runtime.Object
|
|
lister := singleClusterManager.Lister(gvr)
|
|
obj, err = lister.Get(fedKey.NamespaceKey())
|
|
if err != nil {
|
|
if apierrors.IsNotFound(err) {
|
|
return nil, err
|
|
}
|
|
|
|
// print logs only for real error.
|
|
klog.Errorf("Failed to get obj %s. error: %v.", fedKey.String(), err)
|
|
return nil, err
|
|
}
|
|
return obj.(*unstructured.Unstructured), nil
|
|
}
|
|
|
|
// GetObjectFromSingleClusterCache gets full object information from single cluster cache by key in worker queue.
|
|
func GetObjectFromSingleClusterCache(restMapper meta.RESTMapper, manager genericmanager.SingleClusterInformerManager,
|
|
cwk *keys.ClusterWideKey) (*unstructured.Unstructured, error) {
|
|
gvr, err := restmapper.GetGroupVersionResource(restMapper, cwk.GroupVersionKind())
|
|
if err != nil {
|
|
klog.Errorf("Failed to get GVR from GVK %s. Error: %v", cwk.GroupVersionKind(), err)
|
|
return nil, err
|
|
}
|
|
|
|
if !manager.IsInformerSynced(gvr) {
|
|
// fall back to call api server in case the cache has not been synchronized yet
|
|
return getObjectFromSingleCluster(gvr, cwk, manager.GetClient())
|
|
}
|
|
|
|
lister := manager.Lister(gvr)
|
|
obj, err := lister.Get(cwk.NamespaceKey())
|
|
if err != nil {
|
|
if apierrors.IsNotFound(err) {
|
|
return nil, err
|
|
}
|
|
|
|
// print logs only for real error.
|
|
klog.Errorf("Failed to get obj %s. error: %v.", cwk.String(), err)
|
|
|
|
return nil, err
|
|
}
|
|
return obj.(*unstructured.Unstructured), nil
|
|
}
|
|
|
|
// getObjectFromSingleCluster will try to get resource from single cluster by DynamicClientSet.
|
|
func getObjectFromSingleCluster(
|
|
gvr schema.GroupVersionResource,
|
|
cwk *keys.ClusterWideKey,
|
|
dynamicClient dynamic.Interface,
|
|
) (*unstructured.Unstructured, error) {
|
|
obj, err := dynamicClient.Resource(gvr).Namespace(cwk.Namespace).Get(context.TODO(), cwk.Name, metav1.GetOptions{})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return obj, nil
|
|
}
|