Adding pause duration.

command/memory/leak.go

@@ -9,6 +9,7 @@ import (
 type Params struct {
 	MaxMemoryInBytes   uint64
 	BlockSizeInBytes   uint64
+	StepTimeInSeconds  time.Duration
 	PauseTimeInSeconds time.Duration
 }
 
@@ -24,9 +25,11 @@ func memoryLeak(params *Params) error {
 			list[i][j] = 0
 		}
 		metrics.PrintMemory()
-		time.Sleep(params.PauseTimeInSeconds)
+		time.Sleep(params.StepTimeInSeconds)
 	}
 
+	fmt.Printf("Waiting for %s.\n", params.PauseTimeInSeconds.String())
+	time.Sleep(params.PauseTimeInSeconds)
 	fmt.Println("Done")
 
 	return nil

command/memory/register.go

@@ -6,9 +6,12 @@ import (
 	"github.com/urfave/cli"
 )
 
-const MaxMemoryName = "max-memory"
-const BlockSizeName = "block-size"
-const PauseDurationName = "pause"
+const (
+	MaxMemoryName     = "max-memory"
+	BlockSizeName     = "block-size"
+	StepDurationName  = "step-duration"
+	PauseDurationName = "pause-duration"
+)
 
 func Register() cli.Command {
 	return cli.Command{
@@ -32,10 +35,15 @@ func flags() []cli.Flag {
 			Value: "10MB",
 		},
 		cli.DurationFlag{
-			Name:  PauseDurationName,
+			Name:  StepDurationName,
 			Usage: "Time between allocations in seconds.",
 			Value: 1,
 		},
+		cli.DurationFlag{
+			Name:  PauseDurationName,
+			Usage: "Time to wait, in seconds, after all memory has been allocated.",
+			Value: 0,
+		},
 	}
 }
 
@@ -57,6 +65,7 @@ func action(ctx *cli.Context) error {
 	params := &Params{
 		MaxMemoryInBytes:   uint64(maxMemory),
 		BlockSizeInBytes:   uint64(blockSize),
+		StepTimeInSeconds:  ctx.Duration(StepDurationName),
 		PauseTimeInSeconds: ctx.Duration(PauseDurationName),
 	}