Skip to content

Commit b139c3c

Browse files
committed
make oauth part of config and remove annotation check
Signed-off-by: Kevin <[email protected]>
1 parent 2a1de75 commit b139c3c

File tree

3 files changed

+6
-48
lines changed

3 files changed

+6
-48
lines changed

controllers/raycluster_controller.go

Lines changed: 1 addition & 47 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,6 @@ import (
2121
"crypto/rand"
2222
"crypto/sha1"
2323
"encoding/base64"
24-
"strconv"
2524

2625
rayv1 "github.com/ray-project/kuberay/ray-operator/apis/ray/v1"
2726

@@ -128,52 +127,7 @@ func (r *RayClusterReconciler) Reconcile(ctx context.Context, req ctrl.Request)
128127
return ctrl.Result{}, nil
129128
}
130129

131-
val, ok := cluster.ObjectMeta.Annotations["codeflare.dev/oauth"]
132-
boolVal, err := strconv.ParseBool(val)
133-
if err != nil {
134-
logger.Error(err, "Could not convert codeflare.dev/oauth value to bool", "codeflare.dev/oauth", val)
135-
}
136-
if !ok || err != nil || !boolVal {
137-
logger.Info("Removing all OAuth Objects")
138-
err := r.deleteIfNotExist(
139-
ctx, types.NamespacedName{Namespace: cluster.Namespace, Name: oauthSecretNameFromCluster(&cluster)}, &corev1.Secret{},
140-
)
141-
if err != nil {
142-
logger.Error(err, "Error deleting OAuth Secret, retrying", logRequeueing, strTrue)
143-
return ctrl.Result{RequeueAfter: requeueTime}, nil
144-
}
145-
err = r.deleteIfNotExist(
146-
ctx, types.NamespacedName{Namespace: cluster.Namespace, Name: oauthServiceNameFromCluster(&cluster)}, &corev1.Service{},
147-
)
148-
if err != nil {
149-
logger.Error(err, "Error deleting OAuth Service, retrying", logRequeueing, strTrue)
150-
return ctrl.Result{RequeueAfter: requeueTime}, nil
151-
}
152-
err = r.deleteIfNotExist(
153-
ctx, types.NamespacedName{Namespace: cluster.Namespace, Name: oauthServiceAccountNameFromCluster(&cluster)}, &corev1.ServiceAccount{},
154-
)
155-
if err != nil {
156-
logger.Error(err, "Error deleting OAuth ServiceAccount, retrying", logRequeueing, strTrue)
157-
return ctrl.Result{RequeueAfter: requeueTime}, nil
158-
}
159-
err = r.deleteIfNotExist(
160-
ctx, types.NamespacedName{Namespace: cluster.Namespace, Name: crbNameFromCluster(&cluster)}, &rbacv1.ClusterRoleBinding{},
161-
)
162-
if err != nil {
163-
logger.Error(err, "Error deleting OAuth CRB, retrying", logRequeueing, strTrue)
164-
return ctrl.Result{RequeueAfter: requeueTime}, nil
165-
}
166-
err = r.deleteIfNotExist(
167-
ctx, types.NamespacedName{Namespace: cluster.Namespace, Name: routeNameFromCluster(&cluster)}, &routev1.Route{},
168-
)
169-
if err != nil {
170-
logger.Error(err, "Error deleting OAuth Route, retrying", logRequeueing, strTrue)
171-
return ctrl.Result{RequeueAfter: requeueTime}, nil
172-
}
173-
return ctrl.Result{}, nil
174-
}
175-
176-
_, err = r.routeClient.Routes(cluster.Namespace).Apply(ctx, desiredClusterRoute(&cluster), metav1.ApplyOptions{FieldManager: controllerName, Force: true})
130+
_, err := r.routeClient.Routes(cluster.Namespace).Apply(ctx, desiredClusterRoute(&cluster), metav1.ApplyOptions{FieldManager: controllerName, Force: true})
177131
if err != nil {
178132
logger.Error(err, "Failed to update OAuth Route")
179133
}

main.go

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -134,6 +134,7 @@ func main() {
134134
MaxScaleoutAllowed: 5,
135135
},
136136
},
137+
RayClusterOAuth: pointer.Bool(true),
137138
}
138139

139140
kubeConfig, err := ctrl.GetConfig()
@@ -180,7 +181,8 @@ func main() {
180181
exitOnError(instaScaleController.SetupWithManager(context.Background(), mgr), "Error setting up InstaScale controller")
181182
}
182183

183-
if v, err := HasAPIResourceForGVK(kubeClient.DiscoveryClient, rayv1.GroupVersion.WithKind("RayCluster")); v {
184+
v, err := HasAPIResourceForGVK(kubeClient.DiscoveryClient, rayv1.GroupVersion.WithKind("RayCluster"))
185+
if v && *cfg.RayClusterOAuth {
184186
rayClusterController := cfoControllers.RayClusterReconciler{Client: mgr.GetClient(), Scheme: mgr.GetScheme()}
185187
exitOnError(rayClusterController.SetupWithManager(mgr), "Error setting up RayCluster controller")
186188
} else if err != nil {

pkg/config/config.go

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,8 @@ type CodeFlareOperatorConfiguration struct {
3636

3737
// The InstaScale controller configuration
3838
InstaScale *InstaScaleConfiguration `json:"instascale,omitempty"`
39+
40+
RayClusterOAuth *bool `json:"rayClusterOAuth,omitempty"`
3941
}
4042

4143
type InstaScaleConfiguration struct {

0 commit comments

Comments
 (0)