-
Notifications
You must be signed in to change notification settings - Fork 17
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Rename package name of applycrds to make it useable (#53)
* Rename package name of applycrds to make it useable Signed-off-by: Tobias Giese <[email protected]> * Rename applycrds pkg to crdutil Signed-off-by: Tobias Giese <[email protected]> --------- Signed-off-by: Tobias Giese <[email protected]>
- Loading branch information
1 parent
d37ba75
commit 1ad8938
Showing
7 changed files
with
187 additions
and
166 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
File renamed without changes.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,183 @@ | ||
| /* | ||
| Copyright 2024 NVIDIA CORPORATION & AFFILIATES | ||
| Licensed under the Apache License, Version 2.0 (the "License"); | ||
| you may not use this file except in compliance with the License. | ||
| You may obtain a copy of the License at | ||
| http://www.apache.org/licenses/LICENSE-2.0 | ||
| Unless required by applicable law or agreed to in writing, software | ||
| distributed under the License is distributed on an "AS IS" BASIS, | ||
| WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
| See the License for the specific language governing permissions and | ||
| limitations under the License. | ||
| */ | ||
|
|
||
| package crdutil | ||
|
|
||
| import ( | ||
| "context" | ||
| "flag" | ||
| "fmt" | ||
| "io" | ||
| "log" | ||
| "os" | ||
| "path/filepath" | ||
| "strings" | ||
|
|
||
| apiextensionsv1 "k8s.io/apiextensions-apiserver/pkg/apis/apiextensions/v1" | ||
| "k8s.io/apiextensions-apiserver/pkg/client/clientset/clientset" | ||
| v1 "k8s.io/apiextensions-apiserver/pkg/client/clientset/clientset/typed/apiextensions/v1" | ||
| apierrors "k8s.io/apimachinery/pkg/api/errors" | ||
| metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" | ||
| "k8s.io/apimachinery/pkg/util/wait" | ||
| "k8s.io/apimachinery/pkg/util/yaml" | ||
| "k8s.io/client-go/util/retry" | ||
| ctrl "sigs.k8s.io/controller-runtime" | ||
| ) | ||
|
|
||
| type StringList []string | ||
|
|
||
| func (s *StringList) String() string { | ||
| return strings.Join(*s, ", ") | ||
| } | ||
|
|
||
| func (s *StringList) Set(value string) error { | ||
| *s = append(*s, value) | ||
| return nil | ||
| } | ||
|
|
||
| var ( | ||
| crdsDir StringList | ||
| ) | ||
|
|
||
| func initFlags() { | ||
| flag.Var(&crdsDir, "crds-dir", "Path to the directory containing the CRD manifests") | ||
| flag.Parse() | ||
|
|
||
| if len(crdsDir) == 0 { | ||
| log.Fatalf("CRDs directory is required") | ||
| } | ||
|
|
||
| for _, crdDir := range crdsDir { | ||
| if _, err := os.Stat(crdDir); os.IsNotExist(err) { | ||
| log.Fatalf("CRDs directory %s does not exist", crdsDir) | ||
| } | ||
| } | ||
| } | ||
|
|
||
| // EnsureCRDsCmd reads each YAML file in the directory, splits it into documents, and applies each CRD to the cluster. | ||
| // The parameter --crds-dir is required and should point to the directory containing the CRD manifests. | ||
| func EnsureCRDsCmd() { | ||
| ctx := context.Background() | ||
|
|
||
| initFlags() | ||
|
|
||
| config, err := ctrl.GetConfig() | ||
| if err != nil { | ||
| log.Fatalf("Failed to get Kubernetes config: %v", err) | ||
| } | ||
|
|
||
| client, err := clientset.NewForConfig(config) | ||
| if err != nil { | ||
| log.Fatalf("Failed to create API extensions client: %v", err) | ||
| } | ||
|
|
||
| if err := walkCrdsDir(ctx, client.ApiextensionsV1().CustomResourceDefinitions()); err != nil { | ||
| log.Fatalf("Failed to apply CRDs: %v", err) | ||
| } | ||
| } | ||
|
|
||
| // walkCrdsDir walks the CRDs directory and applies each YAML file. | ||
| func walkCrdsDir(ctx context.Context, crdClient v1.CustomResourceDefinitionInterface) error { | ||
| for _, crdDir := range crdsDir { | ||
| // Walk the directory recursively and apply each YAML file. | ||
| err := filepath.Walk(crdDir, func(path string, info os.FileInfo, err error) error { | ||
| if err != nil { | ||
| return err | ||
| } | ||
| if info.IsDir() || filepath.Ext(path) != ".yaml" { | ||
| return nil | ||
| } | ||
|
|
||
| log.Printf("Apply CRDs from file: %s", path) | ||
| if err := applyCRDsFromFile(ctx, crdClient, path); err != nil { | ||
| return fmt.Errorf("apply CRD %s: %w", path, err) | ||
| } | ||
| return nil | ||
| }) | ||
| if err != nil { | ||
| return fmt.Errorf("walk the path %s: %w", crdsDir, err) | ||
| } | ||
| } | ||
| return nil | ||
| } | ||
|
|
||
| // applyCRDsFromFile reads a YAML file, splits it into documents, and applies each CRD to the cluster. | ||
| func applyCRDsFromFile(ctx context.Context, crdClient v1.CustomResourceDefinitionInterface, filePath string) error { | ||
| file, err := os.Open(filePath) | ||
| if err != nil { | ||
| return fmt.Errorf("open file %q: %w", filePath, err) | ||
| } | ||
| defer file.Close() | ||
|
|
||
| // Create a decoder that reads multiple YAML documents. | ||
| decoder := yaml.NewYAMLOrJSONDecoder(file, 4096) | ||
| var crdsToApply []*apiextensionsv1.CustomResourceDefinition | ||
| for { | ||
| crd := &apiextensionsv1.CustomResourceDefinition{} | ||
| if err := decoder.Decode(crd); err != nil { | ||
| if err == io.EOF { | ||
| break | ||
| } | ||
| return fmt.Errorf("decode YAML: %w", err) | ||
| } | ||
| if crd.GetObjectKind().GroupVersionKind().Kind != "CustomResourceDefinition" { | ||
| log.Printf("Skipping non-CRD object %s", crd.GetName()) | ||
| continue | ||
| } | ||
| crdsToApply = append(crdsToApply, crd) | ||
| } | ||
|
|
||
| // Apply each CRD separately. | ||
| for _, crd := range crdsToApply { | ||
| err := wait.ExponentialBackoffWithContext(ctx, retry.DefaultBackoff, func(context.Context) (bool, error) { | ||
| if err := applyCRD(ctx, crdClient, crd); err != nil { | ||
| log.Printf("Failed to apply CRD %s: %v", crd.Name, err) | ||
| return false, nil | ||
| } | ||
| return true, nil | ||
| }) | ||
| if err != nil { | ||
| return fmt.Errorf("apply CRD %s: %w", crd.Name, err) | ||
| } | ||
| } | ||
| return nil | ||
| } | ||
|
|
||
| // applyCRD creates or updates the CRD. | ||
| func applyCRD( | ||
| ctx context.Context, | ||
| crdClient v1.CustomResourceDefinitionInterface, | ||
| crd *apiextensionsv1.CustomResourceDefinition, | ||
| ) error { | ||
| // Check if CRD already exists in cluster and create if not found. | ||
| curCRD, err := crdClient.Get(ctx, crd.Name, metav1.GetOptions{}) | ||
| if apierrors.IsNotFound(err) { | ||
| log.Printf("Create CRD %s", crd.Name) | ||
| _, err = crdClient.Create(ctx, crd, metav1.CreateOptions{}) | ||
| if err != nil { | ||
| return fmt.Errorf("create CRD %s: %w", crd.Name, err) | ||
| } | ||
| } else { | ||
| log.Printf("Update CRD %s", crd.Name) | ||
| // Set resource version to update an existing CRD. | ||
| crd.SetResourceVersion(curCRD.GetResourceVersion()) | ||
| _, err = crdClient.Update(ctx, crd, metav1.UpdateOptions{}) | ||
| if err != nil { | ||
| return fmt.Errorf("update CRD %s: %w", crd.Name, err) | ||
| } | ||
| } | ||
| return nil | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
File renamed without changes.
File renamed without changes.