Supporting reconnecting to /exec socket (#434)

internal/command/controller/run.go

@@ -35,6 +35,7 @@ var noExperimentalRPCV2 bool
 var experimentalPingInterval time.Duration
 var experimentalDisableDBCompression bool
 var workerOfflineTimeout time.Duration
+var execSessionExitTTL time.Duration
 var synthetic bool
 
 func newRunCommand() *cobra.Command {
@@ -87,6 +88,8 @@ func newRunCommand() *cobra.Command {
 		"duration (e.g. 60s or 5m30s) after which a worker is considered offline for the purposes "+
 			"of scheduling (no new VMs will be scheduled on such worker and already assigned VMs will be "+
 			"marked as failed)")
+	cmd.Flags().DurationVar(&execSessionExitTTL, "exec-session-exit-ttl", 10*time.Minute,
+		"duration to retain reconnectable exec session history after the command exits")
 
 	// Hidden flags
 	cmd.Flags().BoolVar(&synthetic, "synthetic", false, "")
@@ -147,6 +150,7 @@ func runController(cmd *cobra.Command, args []string) (err error) {
 		controller.WithListenAddr(address),
 		controller.WithDataDir(dataDir),
 		controller.WithWorkerOfflineTimeout(workerOfflineTimeout),
+		controller.WithExecSessionExitTTL(execSessionExitTTL),
 		controller.WithLogger(logger),
 	}