Graceful shutdown timeout

changelog/fragments/ansible-drain-time.yaml

@@ -0,0 +1,5 @@
+entries:
+  - description: >
+      For Ansible-based operators, allow passing a --drain-time to configure the duration the manager should wait before stopping.
+    kind: "addition"
+    breaking: false

internal/ansible/flags/flag.go

@@ -35,6 +35,7 @@ type Flags struct {
 	ProbeAddr               string
 	LeaderElectionID        string
 	LeaderElectionNamespace string
+	DrainTime               time.Duration
 	AnsibleArgs             string
 }
 
@@ -109,6 +110,12 @@ func (f *Flags) AddTo(flagSet *pflag.FlagSet) {
 			" holding the leader lock (required if running locally with leader"+
 			" election enabled).",
 	)
+	flagSet.DurationVar(&f.DrainTime,
+		"drain-time",
+		30*time.Second,
+		"The amount of time that will be spent waiting"+
+			" for runners to gracefully exit.",
+	)
 	flagSet.StringVar(&f.AnsibleArgs,
 		"ansible-args",
 		"",

internal/cmd/ansible-operator/run/cmd.go

@@ -109,6 +109,7 @@ func run(cmd *cobra.Command, f *flags.Flags) {
 		LeaderElectionResourceLock: resourcelock.ConfigMapsResourceLock,
 		LeaderElectionNamespace:    f.LeaderElectionNamespace,
 		ClientBuilder:              clientbuilder.NewUnstructedCached(),
+		GracefulShutdownTimeout:    &f.DrainTime,
 	}
 
 	namespace, found := os.LookupEnv(k8sutil.WatchNamespaceEnvVar)