@@ -28,6 +28,7 @@ import (
2828 "k8s.io/apimachinery/pkg/api/errors"
2929 metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
3030 "k8s.io/apimachinery/pkg/util/intstr"
31+ "sigs.k8s.io/controller-runtime/pkg/log"
3132
3233 "github.com/kubeflow/spark-operator/v2/api/v1beta2"
3334 "github.com/kubeflow/spark-operator/v2/pkg/common"
@@ -83,18 +84,19 @@ func getDriverIngressURL(ingressURLFormat string, app *v1beta2.SparkApplication)
8384 return parsedURL , nil
8485}
8586
86- func (r * Reconciler ) createDriverIngress (app * v1beta2.SparkApplication , driverIngressConfiguration * v1beta2.DriverIngressConfiguration , service SparkService , ingressURL * url.URL , ingressClassName string ) (* SparkIngress , error ) {
87+ func (r * Reconciler ) createDriverIngress (ctx context. Context , app * v1beta2.SparkApplication , driverIngressConfiguration * v1beta2.DriverIngressConfiguration , service SparkService , ingressURL * url.URL , ingressClassName string ) (* SparkIngress , error ) {
8788 if driverIngressConfiguration .ServicePort == nil {
8889 return nil , fmt .Errorf ("cannot create Driver Ingress for application %s/%s due to empty ServicePort on driverIngressConfiguration" , app .Namespace , app .Name )
8990 }
9091 ingressName := fmt .Sprintf ("%s-ing-%d" , app .Name , * driverIngressConfiguration .ServicePort )
9192 if util .IngressCapabilities .Has ("networking.k8s.io/v1" ) {
92- return r .createDriverIngressV1 (app , service , ingressName , ingressURL , ingressClassName , []networkingv1.IngressTLS {}, map [string ]string {})
93+ return r .createDriverIngressV1 (ctx , app , service , ingressName , ingressURL , ingressClassName , []networkingv1.IngressTLS {}, map [string ]string {})
9394 }
94- return r .createDriverIngressLegacy (app , service , ingressName , ingressURL )
95+ return r .createDriverIngressLegacy (ctx , app , service , ingressName , ingressURL )
9596}
9697
97- func (r * Reconciler ) createDriverIngressV1 (app * v1beta2.SparkApplication , service SparkService , ingressName string , ingressURL * url.URL , ingressClassName string , defaultIngressTLS []networkingv1.IngressTLS , defaultIngressAnnotations map [string ]string ) (* SparkIngress , error ) {
98+ func (r * Reconciler ) createDriverIngressV1 (ctx context.Context , app * v1beta2.SparkApplication , service SparkService , ingressName string , ingressURL * url.URL , ingressClassName string , defaultIngressTLS []networkingv1.IngressTLS , defaultIngressAnnotations map [string ]string ) (* SparkIngress , error ) {
99+ logger := log .FromContext (ctx )
98100 ingressResourceAnnotations := util .GetWebUIIngressAnnotations (app )
99101 if len (ingressResourceAnnotations ) == 0 && len (defaultIngressAnnotations ) != 0 {
100102 ingressResourceAnnotations = defaultIngressAnnotations
@@ -160,17 +162,17 @@ func (r *Reconciler) createDriverIngressV1(app *v1beta2.SparkApplication, servic
160162 ingress .Spec .IngressClassName = & ingressClassName
161163 }
162164
163- if err := r .client .Create (context . TODO () , ingress ); err != nil {
165+ if err := r .client .Create (ctx , ingress ); err != nil {
164166 if ! errors .IsAlreadyExists (err ) {
165167 return nil , fmt .Errorf ("failed to create ingress %s/%s: %v" , ingress .Namespace , ingress .Name , err )
166168 }
167169
168- if err := r .client .Update (context . TODO () , ingress ); err != nil {
170+ if err := r .client .Update (ctx , ingress ); err != nil {
169171 return nil , fmt .Errorf ("failed to update ingress %s/%s: %v" , ingress .Namespace , ingress .Name , err )
170172 }
171- logger .Info ("Updated networking.v1/Ingress for SparkApplication" , "name" , app . Name , "namespace" , app . Namespace , " ingressName" , ingress .Name )
173+ logger .Info ("Updated networking.v1/Ingress for SparkApplication" , "ingressName" , ingress .Name )
172174 } else {
173- logger .Info ("Created networking.v1/Ingress for SparkApplication" , "name" , app . Name , "namespace" , app . Namespace , " ingressName" , ingress .Name )
175+ logger .Info ("Created networking.v1/Ingress for SparkApplication" , "ingressName" , ingress .Name )
174176 }
175177
176178 return & SparkIngress {
@@ -182,7 +184,8 @@ func (r *Reconciler) createDriverIngressV1(app *v1beta2.SparkApplication, servic
182184 }, nil
183185}
184186
185- func (r * Reconciler ) createDriverIngressLegacy (app * v1beta2.SparkApplication , service SparkService , ingressName string , ingressURL * url.URL ) (* SparkIngress , error ) {
187+ func (r * Reconciler ) createDriverIngressLegacy (ctx context.Context , app * v1beta2.SparkApplication , service SparkService , ingressName string , ingressURL * url.URL ) (* SparkIngress , error ) {
188+ logger := log .FromContext (ctx )
186189 ingressResourceAnnotations := util .GetWebUIIngressAnnotations (app )
187190 // var ingressTLSHosts networkingv1.IngressTLS[]
188191 // That we convert later for extensionsv1beta1, but return as is in SparkIngress.
@@ -236,17 +239,17 @@ func (r *Reconciler) createDriverIngressLegacy(app *v1beta2.SparkApplication, se
236239 if len (ingressTLSHosts ) != 0 {
237240 ingress .Spec .TLS = convertIngressTLSHostsToLegacy (ingressTLSHosts )
238241 }
239- if err := r .client .Create (context . TODO () , ingress ); err != nil {
242+ if err := r .client .Create (ctx , ingress ); err != nil {
240243 if ! errors .IsAlreadyExists (err ) {
241244 return nil , fmt .Errorf ("failed to create ingress %s/%s: %v" , ingress .Namespace , ingress .Name , err )
242245 }
243246
244- if err := r .client .Update (context . TODO () , ingress ); err != nil {
247+ if err := r .client .Update (ctx , ingress ); err != nil {
245248 return nil , fmt .Errorf ("failed to update ingress %s/%s: %v" , ingress .Namespace , ingress .Name , err )
246249 }
247- logger .Info ("Updated extensions.v1beta1/Ingress for SparkApplication" , "name" , app . Name , "namespace" , app . Namespace , " ingressName" , ingress .Name )
250+ logger .Info ("Updated extensions.v1beta1/Ingress for SparkApplication" , "ingressName" , ingress .Name )
248251 } else {
249- logger .Info ("Created extensions.v1beta1/Ingress for SparkApplication" , "name" , app . Name , "namespace" , app . Namespace , " ingressName" , ingress .Name )
252+ logger .Info ("Created extensions.v1beta1/Ingress for SparkApplication" , "ingressName" , ingress .Name )
250253 }
251254
252255 return & SparkIngress {
@@ -269,6 +272,7 @@ func convertIngressTLSHostsToLegacy(ingressTLSHosts []networkingv1.IngressTLS) [
269272}
270273
271274func (r * Reconciler ) createDriverIngressService (
275+ ctx context.Context ,
272276 app * v1beta2.SparkApplication ,
273277 portName string ,
274278 port int32 ,
@@ -278,6 +282,7 @@ func (r *Reconciler) createDriverIngressService(
278282 serviceAnnotations map [string ]string ,
279283 serviceLabels map [string ]string ,
280284) (* SparkService , error ) {
285+ logger := log .FromContext (ctx )
281286 service := & corev1.Service {
282287 ObjectMeta : metav1.ObjectMeta {
283288 Name : serviceName ,
@@ -312,18 +317,18 @@ func (r *Reconciler) createDriverIngressService(
312317 service .ObjectMeta .Annotations = serviceAnnotations
313318 }
314319
315- if err := r .client .Create (context . TODO () , service ); err != nil {
320+ if err := r .client .Create (ctx , service ); err != nil {
316321 if ! errors .IsAlreadyExists (err ) {
317322 return nil , err
318323 }
319324
320325 // Update the service if it already exists.
321- if err := r .client .Update (context . TODO () , service ); err != nil {
326+ if err := r .client .Update (ctx , service ); err != nil {
322327 return nil , err
323328 }
324- logger .Info ("Updated service for SparkApplication" , "name" , app . Name , "namespace" , app . Namespace , "serviceName" , service .Name )
329+ logger .Info ("Updated service for SparkApplication" , "name" , service .Name )
325330 } else {
326- logger .Info ("Created service for SparkApplication" , "name" , app . Name , "namespace" , app . Namespace , "serviceName" , service .Name )
331+ logger .Info ("Created service for SparkApplication" , "name" , service .Name )
327332 }
328333
329334 return & SparkService {
@@ -390,6 +395,7 @@ func getDriverIngressServiceLabels(driverIngressConfiguration *v1beta2.DriverIng
390395}
391396
392397func (r * Reconciler ) createDriverIngressServiceFromConfiguration (
398+ ctx context.Context ,
393399 app * v1beta2.SparkApplication ,
394400 driverIngressConfiguration * v1beta2.DriverIngressConfiguration ,
395401) (* SparkService , error ) {
@@ -402,5 +408,5 @@ func (r *Reconciler) createDriverIngressServiceFromConfiguration(
402408 serviceType := getDriverIngressServiceType (driverIngressConfiguration )
403409 serviceAnnotations := getDriverIngressServiceAnnotations (driverIngressConfiguration )
404410 serviceLabels := getDriverIngressServiceLabels (driverIngressConfiguration )
405- return r .createDriverIngressService (app , portName , port , port , serviceName , serviceType , serviceAnnotations , serviceLabels )
411+ return r .createDriverIngressService (ctx , app , portName , port , port , serviceName , serviceType , serviceAnnotations , serviceLabels )
406412}
0 commit comments