| 107 | } |
| 108 | |
| 109 | func NewStateDriver( |
| 110 | k8sClient client.Client, |
| 111 | namespace string, |
| 112 | scheme *runtime.Scheme, |
| 113 | manifestDir string) (State, error) { |
| 114 | |
| 115 | files, err := utils.GetFilesWithSuffix(manifestDir, render.ManifestFileSuffix...) |
| 116 | if err != nil { |
| 117 | return nil, fmt.Errorf("failed to get files from manifest directory: %v", err) |
| 118 | } |
| 119 | |
| 120 | renderer := render.NewRenderer(files) |
| 121 | state := &stateDriver{ |
| 122 | stateSkel: stateSkel{ |
| 123 | name: "state-driver", |
| 124 | description: "NVIDIA driver deployed in the cluster", |
| 125 | client: k8sClient, |
| 126 | namespace: namespace, |
| 127 | scheme: scheme, |
| 128 | renderer: renderer, |
| 129 | }, |
| 130 | } |
| 131 | return state, nil |
| 132 | } |
| 133 | |
| 134 | func (s *stateDriver) Sync(ctx context.Context, customResource interface{}, infoCatalog InfoCatalog) (SyncState, error) { |
| 135 | cr, ok := customResource.(*nvidiav1alpha1.NVIDIADriver) |