Merge pull request #278 from justinsb/register_builtin_types

mockkubeapiserver: register built-in types

Author: k8s-ci-robot

mockkubeapiserver/kubernetes_builtin_schema.go

@@ -5,35 +5,79 @@ import (
 	"sync"
 
 	"sigs.k8s.io/structured-merge-diff/v4/typed"
+	"sigs.k8s.io/yaml"
 
 	_ "embed"
 )
 
 //go:embed kubernetes_builtin_schema.yaml
 var kubernetesBuiltinSchemaYAML string
 
+//go:embed kubernetes_builtin_schema.meta.yaml
+var kubernetesBuiltinSchemaMetaYAML string
+
+const BuiltinKey = "kubernetes_builtin_schema"
+
 type Schema struct {
 	Parser *typed.Parser
+	Meta   *SchemaMeta
 }
 
-var cacheKubernetesBuiltinSchema *Schema
+type SchemaMeta struct {
+	Resources []SchemaMetaResource `json:"resources"`
+}
 
-var mutexKubernetesBuiltinSchema sync.Mutex
+type SchemaMetaResource struct {
+	Key      string `json:"key"`
+	Group    string `json:"group"`
+	Version  string `json:"version"`
+	Kind     string `json:"kind"`
+	Resource string `json:"resource"`
+	Scope    string `json:"scope"`
+}
+
+type schemaCache struct {
+	mutex   sync.Mutex
+	schemas map[string]*Schema
+}
+
+var globalSchemaCache schemaCache
 
 func KubernetesBuiltInSchema() (*Schema, error) {
-	mutexKubernetesBuiltinSchema.Lock()
-	defer mutexKubernetesBuiltinSchema.Unlock()
+	return globalSchemaCache.Get(BuiltinKey)
+}
+
+func (c *schemaCache) Get(key string) (*Schema, error) {
+	c.mutex.Lock()
+	defer c.mutex.Unlock()
 
-	if cacheKubernetesBuiltinSchema != nil {
-		return cacheKubernetesBuiltinSchema, nil
+	if schema := c.schemas[key]; schema != nil {
+		return schema, nil
 	}
 
-	schemaYAML := kubernetesBuiltinSchemaYAML
+	if key != BuiltinKey {
+		return nil, fmt.Errorf("schema %q not known", key)
+	}
 
+	schemaYAML := kubernetesBuiltinSchemaYAML
 	parser, err := typed.NewParser(typed.YAMLObject(schemaYAML))
 	if err != nil {
 		return nil, fmt.Errorf("error parsing schema: %w", err)
 	}
-	cacheKubernetesBuiltinSchema = &Schema{Parser: parser}
-	return cacheKubernetesBuiltinSchema, nil
+
+	metaYAML := kubernetesBuiltinSchemaMetaYAML
+	meta := &SchemaMeta{}
+	if err := yaml.Unmarshal([]byte(metaYAML), meta); err != nil {
+		return nil, fmt.Errorf("error parsing schema metadata: %w", err)
+	}
+
+	schema := &Schema{
+		Parser: parser,
+		Meta:   meta,
+	}
+	if c.schemas == nil {
+		c.schemas = make(map[string]*Schema)
+	}
+	c.schemas[key] = schema
+	return schema, nil
 }