github/client-go
1
0
Fork 0
mirror of https://github.com/kubernetes/client-go.git synced 2025-08-14 05:25:33 +00:00
Code Issues Projects Releases Wiki Activity

basic caching working

Browse Source 
Kubernetes-commit: 7b9757faa4924422bec746b689233ed1ea790b1b
This commit is contained in:
Kevin Delgado 2021-07-14 18:54:16 +00:00 committed by Kubernetes Publisher
parent 7525c0e37a
commit 5d4570c9e0
1 changed files with 49 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
applyconfigurations/meta/v1/unstructured.go
View File

@ -1,6 +1,10 @@
package v1 package v1
import ( import (
"crypto/sha512"
"encoding/json"
"fmt"
"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured" "k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
"k8s.io/apimachinery/pkg/runtime/schema" "k8s.io/apimachinery/pkg/runtime/schema"
"k8s.io/apimachinery/pkg/util/managedfields" "k8s.io/apimachinery/pkg/util/managedfields"
@ -26,7 +30,11 @@ type objectTypeCache interface {
// (i.e. it downloads the OpenAPISchema every time) // (i.e. it downloads the OpenAPISchema every time)
// Useful during the proof-of-concept stage until we agree on a caching solution. // Useful during the proof-of-concept stage until we agree on a caching solution.
type nonCachingObjectTypeCache struct { type nonCachingObjectTypeCache struct {
discoveryClient *discovery.DiscoveryClient // TODO: lock this?
discoveryClient discovery.DiscoveryInterface
docHash [sha512.Size]uint8
gvkParser *fieldmanager.GvkParser
typeForGVK map[schema.GroupVersionKind]*typed.ParseableType
} }
// objectTypeForGVK retrieves the typed.ParseableType for a given gvk from the cache // objectTypeForGVK retrieves the typed.ParseableType for a given gvk from the cache
@ -36,6 +44,28 @@ func (c *nonCachingObjectTypeCache) objectTypeForGVK(gvk schema.GroupVersionKind
return nil, err return nil, err
} }
docBytes, err := json.Marshal(doc)
if err != nil {
return nil, err
}
docHash := sha512.Sum512(docBytes)
// cache hit
if c.docHash == docHash {
fmt.Println("cache hit")
fmt.Printf("docHash = %+v\n", docHash)
objType, ok := c.typeForGVK[gvk]
if ok {
fmt.Println("gvk recognized")
fmt.Printf("gvk = %+v\n", gvk)
return objType, nil
}
objType = c.gvkParser.Type(gvk)
c.typeForGVK[gvk] = objType
return objType, nil
} else {
fmt.Println("cache miss")
// cache miss
models, err := proto.NewOpenAPIData(doc) models, err := proto.NewOpenAPIData(doc)
if err != nil { if err != nil {
return nil, err return nil, err
@ -46,7 +76,15 @@ func (c *nonCachingObjectTypeCache) objectTypeForGVK(gvk schema.GroupVersionKind
return nil, err return nil, err
} }
return gvkParser.Type(gvk), nil objType := gvkParser.Type(gvk)
c.docHash = docHash
c.gvkParser = gvkParser
c.typeForGVK = map[schema.GroupVersionKind]*typed.ParseableType{
gvk: objType,
}
return objType, nil
}
} }
type extractor struct { type extractor struct {
@ -55,9 +93,11 @@ type extractor struct {
// NewUnstructuredExtractor creates the extractor with which you can extract the applied configuration // NewUnstructuredExtractor creates the extractor with which you can extract the applied configuration
// for a given manager from an unstructured object. // for a given manager from an unstructured object.
func NewUnstructuredExtractor(dc *discovery.DiscoveryClient) UnstructuredExtractor { func NewUnstructuredExtractor(dc discovery.DiscoveryInterface) UnstructuredExtractor {
return &extractor{ return &extractor{
cache: &nonCachingObjectTypeCache{dc}, cache: &nonCachingObjectTypeCache{
discoveryClient: dc,
},
} }
} }