add custom pacer and worker ticker configs

runner/options.go

@@ -15,6 +15,7 @@ import (
 	"text/template"
 	"time"
 
+	"github.com/bojand/ghz/load"
 	"github.com/jhump/protoreflect/desc"
 	"github.com/pkg/errors"
 	"google.golang.org/grpc/credentials"
@@ -63,6 +64,8 @@ type RunConfig struct {
 	loadDuration     time.Duration
 	loadStepDuration time.Duration
 
+	pacer load.Pacer
+
 	// concurrency
 	c             int
 	cStart        uint
@@ -72,6 +75,8 @@ type RunConfig struct {
 	cMaxDuration  time.Duration
 	cStepDuration time.Duration
 
+	workerTicker load.WorkerTicker
+
 	// test
 	n     int
 	async bool
@@ -886,6 +891,24 @@ func WithConcurrencyDuration(duration time.Duration) Option {
 	}
 }
 
+// WithPacer specified the custom pacer to use
+func WithPacer(p load.Pacer) Option {
+	return func(o *RunConfig) error {
+		o.pacer = p
+
+		return nil
+	}
+}
+
+// WithWorkerTicker specified the custom worker ticker to use
+func WithWorkerTicker(ticker load.WorkerTicker) Option {
+	return func(o *RunConfig) error {
+		o.workerTicker = ticker
+
+		return nil
+	}
+}
+
 func createClientTransportCredentials(skipVerify bool, cacertFile, clientCertFile, clientKeyFile, cname string) (credentials.TransportCredentials, error) {
 	var tlsConf tls.Config
 

runner/requester.go

@@ -479,6 +479,10 @@ func min(a, b int) int {
 }
 
 func createWorkerTicker(config *RunConfig) load.WorkerTicker {
+	if config.workerTicker != nil {
+		return config.workerTicker
+	}
+
 	var wt load.WorkerTicker
 	switch config.cSchedule {
 	case ScheduleLine:
@@ -506,6 +510,10 @@ func createWorkerTicker(config *RunConfig) load.WorkerTicker {
 }
 
 func createPacer(config *RunConfig) load.Pacer {
+	if config.pacer != nil {
+		return config.pacer
+	}
+
 	var p load.Pacer
 	switch config.loadSchedule {
 	case ScheduleLine: