forked from opendatahub-io/opendatahub-operator
-
Notifications
You must be signed in to change notification settings - Fork 15
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Addding kfto component to odh operator (opendatahub-io#944)
* Addding kfto component to odh operator Signed-off-by: ted chang <[email protected]> * update: add KFTO default into sample config + linter + branch Signed-off-by: Wen Zhou <[email protected]> --------- Signed-off-by: ted chang <[email protected]> Signed-off-by: Wen Zhou <[email protected]> Co-authored-by: Wen Zhou <[email protected]>
- Loading branch information
1 parent
2f45093
commit 4230f6e
Showing
17 changed files
with
313 additions
and
38 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
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
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
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
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
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,108 @@ | ||
| // Package trainingoperator provides utility functions to config trainingoperator as part of the stack | ||
| // which makes managing distributed compute infrastructure in the cloud easy and intuitive for Data Scientists | ||
| package trainingoperator | ||
|
|
||
| import ( | ||
| "context" | ||
| "fmt" | ||
| "path/filepath" | ||
|
|
||
| operatorv1 "github.com/openshift/api/operator/v1" | ||
| metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" | ||
| "sigs.k8s.io/controller-runtime/pkg/client" | ||
|
|
||
| dsciv1 "github.com/opendatahub-io/opendatahub-operator/v2/apis/dscinitialization/v1" | ||
| "github.com/opendatahub-io/opendatahub-operator/v2/components" | ||
| "github.com/opendatahub-io/opendatahub-operator/v2/pkg/deploy" | ||
| "github.com/opendatahub-io/opendatahub-operator/v2/pkg/monitoring" | ||
| ) | ||
|
|
||
| var ( | ||
| ComponentName = "trainingoperator" | ||
| TrainingOperatorPath = deploy.DefaultManifestPath + "/" + ComponentName + "/rhoai" | ||
| ) | ||
|
|
||
| // Verifies that TrainingOperator implements ComponentInterface. | ||
| var _ components.ComponentInterface = (*TrainingOperator)(nil) | ||
|
|
||
| // TrainingOperator struct holds the configuration for the TrainingOperator component. | ||
| // +kubebuilder:object:generate=true | ||
| type TrainingOperator struct { | ||
| components.Component `json:""` | ||
| } | ||
|
|
||
| func (r *TrainingOperator) OverrideManifests(_ string) error { | ||
| // If devflags are set, update default manifests path | ||
| if len(r.DevFlags.Manifests) != 0 { | ||
| manifestConfig := r.DevFlags.Manifests[0] | ||
| if err := deploy.DownloadManifests(ComponentName, manifestConfig); err != nil { | ||
| return err | ||
| } | ||
| // If overlay is defined, update paths | ||
| defaultKustomizePath := "openshift" | ||
| if manifestConfig.SourcePath != "" { | ||
| defaultKustomizePath = manifestConfig.SourcePath | ||
| } | ||
| TrainingOperatorPath = filepath.Join(deploy.DefaultManifestPath, ComponentName, defaultKustomizePath) | ||
| } | ||
|
|
||
| return nil | ||
| } | ||
|
|
||
| func (r *TrainingOperator) GetComponentName() string { | ||
| return ComponentName | ||
| } | ||
|
|
||
| func (r *TrainingOperator) ReconcileComponent(ctx context.Context, cli client.Client, owner metav1.Object, dscispec *dsciv1.DSCInitializationSpec, _ bool) error { | ||
| var imageParamMap = map[string]string{ | ||
| "odh-training-operator-controller-image": "RELATED_IMAGE_ODH_TRAINING_OPERATOR_IMAGE", | ||
| "namespace": dscispec.ApplicationsNamespace, | ||
| } | ||
|
|
||
| enabled := r.GetManagementState() == operatorv1.Managed | ||
| monitoringEnabled := dscispec.Monitoring.ManagementState == operatorv1.Managed | ||
| platform, err := deploy.GetPlatform(cli) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| if enabled { | ||
| if r.DevFlags != nil { | ||
| // Download manifests and update paths | ||
| if err = r.OverrideManifests(string(platform)); err != nil { | ||
| return err | ||
| } | ||
| } | ||
| if (dscispec.DevFlags == nil || dscispec.DevFlags.ManifestsUri == "") && (r.DevFlags == nil || len(r.DevFlags.Manifests) == 0) { | ||
| if err := deploy.ApplyParams(TrainingOperatorPath, imageParamMap, true); err != nil { | ||
| return err | ||
| } | ||
| } | ||
| } | ||
| // Deploy Training Operator | ||
| if err := deploy.DeployManifestsFromPath(cli, owner, TrainingOperatorPath, dscispec.ApplicationsNamespace, ComponentName, enabled); err != nil { | ||
| return err | ||
| } | ||
|
|
||
| // CloudService Monitoring handling | ||
| if platform == deploy.ManagedRhods { | ||
| if enabled { | ||
| // first check if the service is up, so prometheus wont fire alerts when it is just startup | ||
| if err := monitoring.WaitForDeploymentAvailable(ctx, cli, ComponentName, dscispec.ApplicationsNamespace, 20, 2); err != nil { | ||
| return fmt.Errorf("deployment for %s is not ready to server: %w", ComponentName, err) | ||
| } | ||
| fmt.Printf("deployment for %s is done, updating monitoring rules\n", ComponentName) | ||
| } | ||
| if err := r.UpdatePrometheusConfig(cli, enabled && monitoringEnabled, ComponentName); err != nil { | ||
| return err | ||
| } | ||
| if err = deploy.DeployManifestsFromPath(cli, owner, | ||
| filepath.Join(deploy.DefaultManifestPath, "monitoring", "prometheus", "apps"), | ||
| dscispec.Monitoring.Namespace, | ||
| "prometheus", true); err != nil { | ||
| return err | ||
| } | ||
| } | ||
|
|
||
| return nil | ||
| } |
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
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
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
Oops, something went wrong.