AppWrapper implements JobWithManagedBy

pkg/controller/jobs/appwrapper/appwrapper_controller.go

@@ -33,6 +33,7 @@ import (
 
 	kueue "sigs.k8s.io/kueue/apis/kueue/v1beta1"
 	"sigs.k8s.io/kueue/pkg/controller/jobframework"
+	"sigs.k8s.io/kueue/pkg/features"
 	"sigs.k8s.io/kueue/pkg/podset"
 )
 
@@ -90,6 +91,7 @@ func SetupIndexes(ctx context.Context, indexer client.FieldIndexer) error {
 type AppWrapper awv1beta2.AppWrapper
 
 var _ jobframework.GenericJob = (*AppWrapper)(nil)
+var _ jobframework.JobWithManagedBy = (*AppWrapper)(nil)
 
 func fromObject(o runtime.Object) *AppWrapper {
 	return (*AppWrapper)(o.(*awv1beta2.AppWrapper))
@@ -155,6 +157,20 @@ func (aw *AppWrapper) PodsReady() bool {
 	return meta.IsStatusConditionTrue(aw.Status.Conditions, string(awv1beta2.PodsReady))
 }
 
+func (j *AppWrapper) CanDefaultManagedBy() bool {
+	jobSpecManagedBy := j.Spec.ManagedBy
+	return features.Enabled(features.MultiKueue) &&
+		(jobSpecManagedBy == nil || *jobSpecManagedBy == awv1beta2.AppWrapperControllerName)
+}
+
+func (j *AppWrapper) ManagedBy() *string {
+	return j.Spec.ManagedBy
+}
+
+func (j *AppWrapper) SetManagedBy(managedBy *string) {
+	j.Spec.ManagedBy = managedBy
+}
+
 func GetWorkloadNameForAppWrapper(jobName string, jobUID types.UID) string {
 	return jobframework.GetWorkloadNameForOwnerWithGVK(jobName, jobUID, gvk)
 }

pkg/controller/jobs/appwrapper/appwrapper_multikueue_adapter.go

@@ -116,7 +116,6 @@ func (b *multikueueAdapter) IsJobManagedByKueue(ctx context.Context, c client.Cl
 		return false, fmt.Sprintf("Expecting spec.managedBy to be %q not %q", kueue.MultiKueueControllerName, awControllerName), nil
 	}
 	return true, "", nil
-
 }
 
 var _ jobframework.MultiKueueWatcher = (*multikueueAdapter)(nil)