runner

package
v0.120.0 Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: May 30, 2024 License: Apache-2.0 Imports: 51 Imported by: 41

Documentation

Index

Examples

Constants

View Source 
const (
	// ReasonNormalEnd indicates a normal end to the run
	ReasonNormalEnd = StopReason("normal")

	// ReasonCancel indicates end due to cancellation
	ReasonCancel = StopReason("cancel")

	// ReasonTimeout indicates run ended due to Z parameter timeout
	ReasonTimeout = StopReason("timeout")
)
View Source 
const ScheduleConst = "const"

ScheduleConst is a constant load schedule

View Source 
const ScheduleLine = "line"

ScheduleLine is the line load schedule

View Source 
const ScheduleStep = "step"

ScheduleStep is the step load schedule

Variables

View Source 
var ErrEndStream = errors.New("ending stream")

ErrEndStream is a signal from message providers that worker should close the stream It should not be used for erronous states

View Source 
var ErrLastMessage = errors.New("last message")

ErrLastMessage is a signal from message providers that the returned payload is the last one of the stream This is optional but encouraged for optimized performance Message payload returned along with this error must be valid and may not be nil

Functions

func LoadConfig added in v0.55.0 

func LoadConfig(p string, c *Config) error

LoadConfig loads the config from a file

Types

type BinaryDataFunc added in v0.62.0 

type BinaryDataFunc func(mtd *desc.MethodDescriptor, callData *CallData) []byte

BinaryDataFunc is a function that can be used for provide binary data for request programatically. MethodDescriptor of the call is passed to the data function. CallData for the request is passed and can be used to access worker id, request number, etc...

type Bucket 

type Bucket struct {
	// The Mark for histogram bucket in seconds
	Mark float64 `json:"mark"`

	// The count in the bucket
	Count int `json:"count"`

	// The frequency of results in the bucket as a decimal percentage
	Frequency float64 `json:"frequency"`
}

Bucket holds histogram data

type CallData added in v0.62.0 

type CallData struct {
	WorkerID           string // unique worker ID
	RequestNumber      int64  // unique incremented request number for each request
	FullyQualifiedName string // fully-qualified name of the method call
	MethodName         string // shorter call method name
	ServiceName        string // the service name
	InputName          string // name of the input message type
	OutputName         string // name of the output message type
	IsClientStreaming  bool   // whether this call is client streaming
	IsServerStreaming  bool   // whether this call is server streaming
	Timestamp          string // timestamp of the call in RFC3339 format
	TimestampUnix      int64  // timestamp of the call as unix time in seconds
	TimestampUnixMilli int64  // timestamp of the call as unix time in milliseconds
	TimestampUnixNano  int64  // timestamp of the call as unix time in nanoseconds
	UUID               string // generated UUIDv4 for each call
	// contains filtered or unexported fields
}

CallData represents contextualized data available for templating

func (*CallData) ExecuteData added in v0.80.0 

func (td *CallData) ExecuteData(data string) ([]byte, error)

ExecuteData applies the call data's parsed template and data string and returns the resulting buffer

func (*CallData) Regenerate added in v0.80.0 

func (td *CallData) Regenerate() *CallData

Regenerate generates a new instance of call data from this parent instance The dynamic data like timestamps and UUIDs are re-filled

type Config added in v0.52.0 

type Config struct {
	Proto                 string            `json:"proto" toml:"proto" yaml:"proto"`
	Protoset              string            `json:"protoset" toml:"protoset" yaml:"protoset"`
	Call                  string            `json:"call" toml:"call" yaml:"call"`
	RootCert              string            `json:"cacert" toml:"cacert" yaml:"cacert"`
	Cert                  string            `json:"cert" toml:"cert" yaml:"cert"`
	Key                   string            `json:"key" toml:"key" yaml:"key"`
	CountErrors           bool              `json:"count-errors" toml:"count-errors" yaml:"count-errors"`
	SkipTLSVerify         bool              `json:"skipTLS" toml:"skipTLS" yaml:"skipTLS"`
	SkipFirst             uint              `json:"skipFirst" toml:"skipFirst" yaml:"skipFirst"`
	CName                 string            `json:"cname" toml:"cname" yaml:"cname"`
	Authority             string            `json:"authority" toml:"authority" yaml:"authority"`
	Insecure              bool              `json:"insecure,omitempty" toml:"insecure,omitempty" yaml:"insecure,omitempty"`
	N                     uint              `json:"total" toml:"total" yaml:"total" default:"200"`
	Async                 bool              `json:"async,omitempty" toml:"async,omitempty" yaml:"async,omitempty"`
	C                     uint              `json:"concurrency" toml:"concurrency" yaml:"concurrency" default:"50"`
	CSchedule             string            `json:"concurrency-schedule" toml:"concurrency-schedule" yaml:"concurrency-schedule" default:"const"`
	CStart                uint              `json:"concurrency-start" toml:"concurrency-start" yaml:"concurrency-start" default:"1"`
	CEnd                  uint              `json:"concurrency-end" toml:"concurrency-end" yaml:"concurrency-end" default:"0"`
	CStep                 int               `json:"concurrency-step" toml:"concurrency-step" yaml:"concurrency-step" default:"0"`
	CStepDuration         Duration          `json:"concurrency-step-duration" toml:"concurrency-step-duration" yaml:"concurrency-step-duration" default:"0"`
	CMaxDuration          Duration          `json:"concurrency-max-duration" toml:"concurrency-max-duration" yaml:"concurrency-max-duration" default:"0"`
	Connections           uint              `json:"connections" toml:"connections" yaml:"connections" default:"1"`
	RPS                   uint              `json:"rps" toml:"rps" yaml:"rps"`
	Z                     Duration          `json:"duration" toml:"duration" yaml:"duration"`
	ZStop                 string            `json:"duration-stop" toml:"duration-stop" yaml:"duration-stop" default:"close"`
	X                     Duration          `json:"max-duration" toml:"max-duration" yaml:"max-duration"`
	Timeout               Duration          `json:"timeout" toml:"timeout" yaml:"timeout" default:"20s"`
	Data                  interface{}       `json:"data,omitempty" toml:"data,omitempty" yaml:"data,omitempty"`
	DataPath              string            `json:"data-file" toml:"data-file" yaml:"data-file"`
	BinData               []byte            `json:"-" toml:"-" yaml:"-"`
	BinDataPath           string            `json:"binary-file" toml:"binary-file" yaml:"binary-file"`
	Metadata              map[string]string `json:"metadata,omitempty" toml:"metadata,omitempty" yaml:"metadata,omitempty"`
	MetadataPath          string            `json:"metadata-file" toml:"metadata-file" yaml:"metadata-file"`
	SI                    Duration          `json:"stream-interval" toml:"stream-interval" yaml:"stream-interval"`
	StreamCallDuration    Duration          `json:"stream-call-duration" toml:"stream-call-duration" yaml:"stream-call-duration"`
	StreamCallCount       uint              `json:"stream-call-count" toml:"stream-call-count" yaml:"stream-call-count"`
	StreamDynamicMessages bool              `json:"stream-dynamic-messages" toml:"stream-dynamic-messages" yaml:"stream-dynamic-messages"`
	Output                string            `json:"output" toml:"output" yaml:"output"`
	Format                string            `json:"format" toml:"format" yaml:"format" default:"summary"`
	DialTimeout           Duration          `json:"connect-timeout" toml:"connect-timeout" yaml:"connect-timeout" default:"10s"`
	KeepaliveTime         Duration          `json:"keepalive" toml:"keepalive" yaml:"keepalive"`
	CPUs                  uint              `json:"cpus" toml:"cpus" yaml:"cpus"`
	ImportPaths           []string          `json:"import-paths,omitempty" toml:"import-paths,omitempty" yaml:"import-paths,omitempty"`
	Name                  string            `json:"name,omitempty" toml:"name,omitempty" yaml:"name,omitempty"`
	Tags                  map[string]string `json:"tags,omitempty" toml:"tags,omitempty" yaml:"tags,omitempty"`
	ReflectMetadata       map[string]string `json:"reflect-metadata,omitempty" toml:"reflect-metadata,omitempty" yaml:"reflect-metadata,omitempty"`
	Debug                 string            `json:"debug,omitempty" toml:"debug,omitempty" yaml:"debug,omitempty"`
	Host                  string            `json:"host" toml:"host" yaml:"host"`
	EnableCompression     bool              `json:"enable-compression,omitempty" toml:"enable-compression,omitempty" yaml:"enable-compression,omitempty"`
	LoadSchedule          string            `json:"load-schedule" toml:"load-schedule" yaml:"load-schedule" default:"const"`
	LoadStart             uint              `json:"load-start" toml:"load-start" yaml:"load-start"`
	LoadEnd               uint              `json:"load-end" toml:"load-end" yaml:"load-end"`
	LoadStep              int               `json:"load-step" toml:"load-step" yaml:"load-step"`
	LoadStepDuration      Duration          `json:"load-step-duration" toml:"load-step-duration" yaml:"load-step-duration"`
	LoadMaxDuration       Duration          `json:"load-max-duration" toml:"load-max-duration" yaml:"load-max-duration"`
	LBStrategy            string            `json:"lb-strategy" toml:"lb-strategy" yaml:"lb-strategy"`
	MaxCallRecvMsgSize    string            `json:"max-recv-message-size" toml:"max-recv-message-size" yaml:"max-recv-message-size"`
	MaxCallSendMsgSize    string            `json:"max-send-message-size" toml:"max-send-message-size" yaml:"max-send-message-size"`
	DisableTemplateFuncs  bool              `json:"disable-template-functions" toml:"disable-template-functions" yaml:"disable-template-functions"`
	DisableTemplateData   bool              `json:"disable-template-data" toml:"disable-template-data" yaml:"disable-template-data"`
}

Config for the run. TODO fix casing and consistency.

type Counter added in v0.70.0 

type Counter struct {
	// contains filtered or unexported fields
}

Counter is an implementation of the request counter

func (*Counter) Get added in v0.70.0 

func (c *Counter) Get() uint64

Get retrieves the current count

func (*Counter) Inc added in v0.70.0 

func (c *Counter) Inc() uint64

Inc increases the current count

type DataProviderFunc added in v0.80.0 

type DataProviderFunc func(*CallData) ([]*dynamic.Message, error)

DataProviderFunc is the interface for providing data for calls For unary and server streaming calls it should return an array with a single element For client and bidi streaming calls it should return an array of messages to be used

type Duration added in v0.52.0 

type Duration time.Duration

Duration is our duration with TOML support

func (Duration) MarshalJSON added in v0.55.0 

func (d Duration) MarshalJSON() ([]byte, error)

MarshalJSON implements encoding JSONMarshaler

func (Duration) MarshalText added in v0.52.0 

func (d Duration) MarshalText() ([]byte, error)

MarshalText implements encoding.TextMarshaler

func (Duration) String added in v0.52.0 

func (d Duration) String() string

func (*Duration) UnmarshalJSON added in v0.55.0 

func (d *Duration) UnmarshalJSON(text []byte) error

UnmarshalJSON is our custom unmarshaller with JSON support

func (*Duration) UnmarshalText added in v0.52.0 

func (d *Duration) UnmarshalText(text []byte) error

UnmarshalText is our custom unmarshaller with TOML support

type LatencyDistribution 

type LatencyDistribution struct {
	Percentage int           `json:"percentage"`
	Latency    time.Duration `json:"latency"`
}

LatencyDistribution holds latency distribution data

type Logger added in v0.47.0 

type Logger interface {
	Debug(args ...interface{})
	Debugf(template string, args ...interface{})
	Debugw(msg string, keysAndValues ...interface{})
	Error(args ...interface{})
	Errorf(template string, args ...interface{})
	Errorw(msg string, keysAndValues ...interface{})
}

Logger interface is the common logger interface for all of web

type MetadataProviderFunc added in v0.80.0 

type MetadataProviderFunc func(*CallData) (*metadata.MD, error)

MetadataProviderFunc is the interface for providing metadadata for calls

type Option 

type Option func(*RunConfig) error

Option controls some aspect of run

func WithAsync added in v0.70.0 

func WithAsync(async bool) Option

WithAsync specifies the async option

func WithAuthority added in v0.25.0 

func WithAuthority(authority string) Option

WithAuthority specifies the value to be used as the :authority pseudo-header. This only works with WithInsecure option.

func WithBinaryData 

func WithBinaryData(data []byte) Option

WithBinaryData specifies the binary data 

msg := &helloworld.HelloRequest{}
msg.Name = "bob"
binData, _ := proto.Marshal(msg)
WithBinaryData(binData)

func WithBinaryDataFromFile 

func WithBinaryDataFromFile(path string) Option

WithBinaryDataFromFile specifies the binary data 

WithBinaryDataFromFile("request_data.bin")

func WithBinaryDataFunc added in v0.62.0 

func WithBinaryDataFunc(data func(mtd *desc.MethodDescriptor, callData *CallData) []byte) Option

WithBinaryDataFunc specifies the binary data func which will be called on each request 

WithBinaryDataFunc(changeFunc)

func WithCPUs 

func WithCPUs(c uint) Option

WithCPUs specifies the number of CPU's to be used 

WithCPUs(4)

func WithCertificate 

func WithCertificate(cert, key string) Option

WithCertificate specifies the certificate options for the run 

WithCertificate("client.crt", "client.key")

func WithClientLoadBalancing added in v0.64.0 

func WithClientLoadBalancing(strategy string) Option

WithClientLoadBalancing specifies the LB strategy to use The strategies has to be self written and pre defined

func WithConcurrency 

func WithConcurrency(c uint) Option

WithConcurrency specifies the C (number of concurrent requests) option 

WithConcurrency(20)

func WithConcurrencyDuration added in v0.70.0 

func WithConcurrencyDuration(duration time.Duration) Option

WithConcurrencyDuration specifies the total concurrency adjustment duration

func WithConcurrencyEnd added in v0.70.0 

func WithConcurrencyEnd(v uint) Option

WithConcurrencyEnd specifies the concurrency end value for line or step schedule 

WithConcurrencyEnd(25)

func WithConcurrencySchedule added in v0.70.0 

func WithConcurrencySchedule(schedule string) Option

WithConcurrencySchedule specifies the concurrency adjustment schedule 

WithConcurrencySchedule("const")

func WithConcurrencyStart added in v0.70.0 

func WithConcurrencyStart(v uint) Option

WithConcurrencyStart specifies the concurrency start for line or step schedule 

WithConcurrencyStart(5)

func WithConcurrencyStep added in v0.70.0 

func WithConcurrencyStep(step int) Option

WithConcurrencyStep specifies the concurrency step value or slope 

WithConcurrencyStep(5)

func WithConcurrencyStepDuration added in v0.70.0 

func WithConcurrencyStepDuration(duration time.Duration) Option

WithConcurrencyStepDuration specifies the concurrency step duration for step schedule

func WithConfig added in v0.52.0 

func WithConfig(cfg *Config) Option

WithConfig uses the configuration to populate the RunConfig See also: WithConfigFromFile, WithConfigFromReader

func WithConfigFromFile added in v0.52.0 

func WithConfigFromFile(file string) Option

WithConfigFromFile uses a configuration JSON file to populate the RunConfig 

WithConfigFromFile("config.json")

func WithConfigFromReader added in v0.52.0 

func WithConfigFromReader(reader io.Reader) Option

WithConfigFromReader uses a reader containing JSON data to populate the RunConfig See also: WithConfigFromFile

func WithConnections added in v0.35.0 

func WithConnections(c uint) Option

WithConnections specifies the number of gRPC connections to use 

WithConnections(5)

func WithCountErrors added in v0.80.0 

func WithCountErrors(v bool) Option

WithCountErrors is the count errors option

func WithData 

func WithData(data interface{}) Option

WithData specifies data as generic data that can be serailized to JSON

func WithDataFromFile 

func WithDataFromFile(path string) Option

WithDataFromFile loads JSON data from file 

WithDataFromFile("data.json")

func WithDataFromJSON 

func WithDataFromJSON(data string) Option

WithDataFromJSON loads JSON data from string 

WithDataFromJSON(`{"name":"bob"}`)

func WithDataFromReader 

func WithDataFromReader(r io.Reader) Option

WithDataFromReader loads JSON data from reader 

file, _ := os.Open("data.json")
WithDataFromReader(file)

func WithDataProvider added in v0.80.0 

func WithDataProvider(fn DataProviderFunc) Option

WithDataProvider provides custom data provider 

WithDataProvider(func(*CallData) ([]*dynamic.Message, error) {
	protoMsg := &helloworld.HelloRequest{Name: "Bob"}
	dynamicMsg, err := dynamic.AsDynamicMessage(protoMsg)
	if err != nil {
		return nil, err
	}
	return []*dynamic.Message{dynamicMsg}, nil
}),

func WithDefaultCallOptions added in v0.100.0 

func WithDefaultCallOptions(opts []grpc.CallOption) Option

WithDefaultCallOptions sets the default CallOptions for calls over the connection.

func WithDialTimeout 

func WithDialTimeout(dt time.Duration) Option

WithDialTimeout specifies the initial connection dial timeout 

WithDialTimeout(time.Duration(20*time.Second))

func WithDisableTemplateData added in v0.111.0 

func WithDisableTemplateData(v bool) Option

WithDisableTemplateData disables template data execution in call data

func WithDisableTemplateFuncs added in v0.111.0 

func WithDisableTemplateFuncs(v bool) Option

WithDisableTemplateFuncs disables template functions in call data

func WithDurationStopAction added in v0.42.0 

func WithDurationStopAction(action string) Option

WithDurationStopAction specifies how run duration (Z) timeout is handled Possible options are "close", "ignore", and "wait" 

WithDurationStopAction("ignore")

func WithEnableCompression added in v0.50.0 

func WithEnableCompression(enableCompression bool) Option

WithEnableCompression specifies that requests should be done using gzip Compressor 

WithEnableCompression(true)

func WithInsecure 

func WithInsecure(insec bool) Option

WithInsecure specifies that this run should be done using insecure mode 

WithInsecure(true)

func WithKeepalive 

func WithKeepalive(k time.Duration) Option

WithKeepalive specifies the keepalive timeout 

WithKeepalive(time.Duration(1*time.Minute))

func WithLoadDuration added in v0.70.0 

func WithLoadDuration(duration time.Duration) Option

WithLoadDuration specifies the load duration

func WithLoadEnd added in v0.70.0 

func WithLoadEnd(end uint) Option

WithLoadEnd specifies the load end 

WithLoadEnd(25)

func WithLoadSchedule added in v0.70.0 

func WithLoadSchedule(schedule string) Option

WithLoadSchedule specifies the load schedule 

WithLoadSchedule("const")

func WithLoadStart added in v0.70.0 

func WithLoadStart(start uint) Option

WithLoadStart specifies the load start 

WithLoadStart(5)

func WithLoadStep added in v0.70.0 

func WithLoadStep(step int) Option

WithLoadStep specifies the load step 

WithLoadStep(5)

func WithLoadStepDuration added in v0.70.0 

func WithLoadStepDuration(duration time.Duration) Option

WithLoadStepDuration specifies the load step duration for step schedule

func WithLogger added in v0.47.0 

func WithLogger(log Logger) Option

WithLogger specifies the logging option

func WithMetadata 

func WithMetadata(md map[string]string) Option

WithMetadata specifies the metadata to be used as a map 

md := make(map[string]string)
md["token"] = "foobar"
md["request-id"] = "123"
WithMetadata(&md)

func WithMetadataFromFile 

func WithMetadataFromFile(path string) Option

WithMetadataFromFile loads JSON metadata from file 

WithMetadataFromJSON("metadata.json")

func WithMetadataFromJSON 

func WithMetadataFromJSON(md string) Option

WithMetadataFromJSON specifies the metadata to be read from JSON string 

WithMetadataFromJSON(`{"request-id":"123"}`)

func WithMetadataProvider added in v0.80.0 

func WithMetadataProvider(fn MetadataProviderFunc) Option

WithMetadataProvider provides custom metadata provider 

WithMetadataProvider(ctd *CallData) (*metadata.MD, error) {
	return &metadata.MD{"token": []string{"secret"}}, nil
}),

func WithName 

func WithName(name string) Option

WithName sets the name of the test run 

WithName("greeter service test")

func WithPacer added in v0.70.0 

func WithPacer(p load.Pacer) Option

WithPacer specified the custom pacer to use

func WithProtoFile 

func WithProtoFile(proto string, importPaths []string) Option

WithProtoFile specified proto file path and optionally import paths We will automatically add the proto file path's directory and the current directory 

WithProtoFile("greeter.proto", []string{"/home/protos"})

func WithProtoset 

func WithProtoset(protoset string) Option

WithProtoset specified protoset file path 

WithProtoset("bundle.protoset")

func WithProtosetBinary added in v0.100.0 

func WithProtosetBinary(b []byte) Option

func WithRPS added in v0.70.0 

func WithRPS(v uint) Option

WithRPS specifies the RPS (requests per second) limit option 

WithRPS(10)

func WithReflectionMetadata added in v0.30.0 

func WithReflectionMetadata(md map[string]string) Option

WithReflectionMetadata specifies the metadata to be used as a map 

md := make(map[string]string)
md["token"] = "foobar"
md["request-id"] = "123"
WithReflectionMetadata(&md)

func WithRootCertificate added in v0.25.0 

func WithRootCertificate(cert string) Option

WithRootCertificate specifies the root certificate options for the run 

WithRootCertificate("ca.crt")

func WithRunDuration 

func WithRunDuration(z time.Duration) Option

WithRunDuration specifies the Z (total test duration) option 

WithRunDuration(time.Duration(2*time.Minute))

func WithServerNameOverride added in v0.25.0 

func WithServerNameOverride(cname string) Option

WithServerNameOverride specifies the certificate options for the run

func WithSkipFirst added in v0.56.0 

func WithSkipFirst(c uint) Option

WithSkipFirst is the skipFirst option

func WithSkipTLSVerify added in v0.25.0 

func WithSkipTLSVerify(skip bool) Option

WithSkipTLSVerify skip client side TLS verification of server certificate

func WithStreamCallCount added in v0.80.0 

func WithStreamCallCount(c uint) Option

WithStreamCallCount sets the stream close count

func WithStreamCallDuration added in v0.80.0 

func WithStreamCallDuration(d time.Duration) Option

WithStreamCallDuration sets the maximum stream call duration at which point the client will close the stream

func WithStreamDynamicMessages added in v0.80.0 

func WithStreamDynamicMessages(v bool) Option

WithStreamDynamicMessages sets the stream dynamic message generation

func WithStreamInterceptorProviderFunc added in v0.118.0 

func WithStreamInterceptorProviderFunc(interceptor StreamInterceptorProviderFunc) Option

WithStreamInterceptor specifies the stream interceptor provider function

func WithStreamInterval added in v0.28.0 

func WithStreamInterval(d time.Duration) Option

WithStreamInterval sets the stream interval

func WithStreamMessageProvider added in v0.90.0 

func WithStreamMessageProvider(fn StreamMessageProviderFunc) Option

WithStreamMessageProvider sets custom stream message provider 

WithStreamMessageProvider(func(cd *CallData) (*dynamic.Message, error) {
	protoMsg := &helloworld.HelloRequest{Name: cd.WorkerID + ": " + strconv.FormatInt(cd.RequestNumber, 10)}
	dynamicMsg, err := dynamic.AsDynamicMessage(protoMsg)
	if err != nil {
		return nil, err
	}

	callCounter++

	if callCounter == 5 {
		err = ErrLastMessage
	}

	return dynamicMsg, err
}),

func WithStreamRecvMsgIntercept added in v0.80.0 

func WithStreamRecvMsgIntercept(fn StreamRecvMsgInterceptFunc) Option

WithStreamRecvMsgIntercept specified the stream receive intercept function 

WithStreamRecvMsgIntercept(func(msg *dynamic.Message, err error) error {
	if err == nil && msg != nil {
		reply := &helloworld.HelloReply{}
		convertErr := msg.ConvertTo(reply)
		if convertErr == nil {
			if reply.GetMessage() == "Hello bar" {
				return ErrEndStream
			}
		}
	}
	return nil
})

func WithTags added in v0.22.0 

func WithTags(tags map[string]string) Option

WithTags specifies the user defined tags as a map 

tags := make(map[string]string)
tags["env"] = "staging"
tags["created by"] = "joe developer"
WithTags(&tags)

func WithTemplateFuncs added in v0.58.0 

func WithTemplateFuncs(funcMap template.FuncMap) Option

WithTemplateFuncs adds additional template functions

func WithTimeout 

func WithTimeout(timeout time.Duration) Option

WithTimeout specifies the timeout for each request 

WithTimeout(time.Duration(20*time.Second))

func WithTotalRequests 

func WithTotalRequests(n uint) Option

WithTotalRequests specifies the N (number of total requests) setting 

WithTotalRequests(1000)

func WithWorkerTicker added in v0.70.0 

func WithWorkerTicker(ticker load.WorkerTicker) Option

WithWorkerTicker specified the custom worker ticker to use

type Options 

type Options struct {
	Call              string   `json:"call,omitempty"`
	Host              string   `json:"host,omitempty"`
	Proto             string   `json:"proto,omitempty"`
	Protoset          string   `json:"protoset,omitempty"`
	ImportPaths       []string `json:"import-paths,omitempty"`
	EnableCompression bool     `json:"enable-compression,omitempty"`

	CACert    string `json:"cacert,omitempty"`
	Cert      string `json:"cert,omitempty"`
	Key       string `json:"key,omitempty"`
	CName     string `json:"cname,omitempty"`
	SkipTLS   bool   `json:"skipTLS,omitempty"`
	Insecure  bool   `json:"insecure"`
	Authority string `json:"authority,omitempty"`

	RPS              int           `json:"rps,omitempty"`
	LoadSchedule     string        `json:"load-schedule"`
	LoadStart        int           `json:"load-start"`
	LoadEnd          int           `json:"load-end"`
	LoadStep         int           `json:"load-step"`
	LoadStepDuration time.Duration `json:"load-step-duration"`
	LoadMaxDuration  time.Duration `json:"load-max-duration"`

	Concurrency   int           `json:"concurrency,omitempty"`
	CSchedule     string        `json:"concurrency-schedule"`
	CStart        int           `json:"concurrency-start"`
	CEnd          int           `json:"concurrency-end"`
	CStep         int           `json:"concurrency-step"`
	CStepDuration time.Duration `json:"concurrency-step-duration"`
	CMaxDuration  time.Duration `json:"concurrency-max-duration"`

	Total int  `json:"total,omitempty"`
	Async bool `json:"async,omitempty"`

	Connections   int           `json:"connections,omitempty"`
	Duration      time.Duration `json:"duration,omitempty"`
	Timeout       time.Duration `json:"timeout,omitempty"`
	DialTimeout   time.Duration `json:"dial-timeout,omitempty"`
	KeepaliveTime time.Duration `json:"keepalive,omitempty"`

	Data     interface{}        `json:"data,omitempty"`
	Binary   bool               `json:"binary"`
	Metadata *map[string]string `json:"metadata,omitempty"`

	CPUs int    `json:"CPUs"`
	Name string `json:"name,omitempty"`

	SkipFirst   int  `json:"skipFirst,omitempty"`
	CountErrors bool `json:"count-errors,omitempty"`
}

Options represents the request options TODO fix casing and consistency

type Report 

type Report struct {
	Name      string     `json:"name,omitempty"`
	EndReason StopReason `json:"endReason,omitempty"`
	Date      time.Time  `json:"date"`
	Options   Options    `json:"options,omitempty"`

	Count   uint64        `json:"count"`
	Total   time.Duration `json:"total"`
	Average time.Duration `json:"average"`
	Fastest time.Duration `json:"fastest"`
	Slowest time.Duration `json:"slowest"`
	Rps     float64       `json:"rps"`

	ErrorDist      map[string]int `json:"errorDistribution"`
	StatusCodeDist map[string]int `json:"statusCodeDistribution"`

	LatencyDistribution []LatencyDistribution `json:"latencyDistribution"`
	Histogram           []Bucket              `json:"histogram"`
	Details             []ResultDetail        `json:"details"`

	Tags map[string]string `json:"tags,omitempty"`
}

Report holds the data for the full test

func Run 

func Run(call, host string, options ...Option) (*Report, error)

Run executes the test 

report, err := runner.Run(
	"helloworld.Greeter.SayHello",
	"localhost:50051",
	WithProtoFile("greeter.proto", []string{}),
	WithDataFromFile("data.json"),
	WithInsecure(true),
)
Example

ExampleRun demonstrates how to use runner package to perform a gRPC load test programmatically. We use the printer package to print the report in pretty JSON format. 

package main

import (
	"fmt"
	"os"

	"github.com/bojand/ghz/printer"
	"github.com/bojand/ghz/runner"
)

func main() {
	report, err := runner.Run(
		"helloworld.Greeter.SayHello",
		"localhost:50051",
		runner.WithProtoFile("greeter.proto", []string{}),
		runner.WithDataFromFile("data.json"),
		runner.WithInsecure(true),
	)

	if err != nil {
		fmt.Println(err.Error())
		os.Exit(1)
	}

	printer := printer.ReportPrinter{
		Out:    os.Stdout,
		Report: report,
	}

	printer.Print("pretty")
}

Output:

func (Report) MarshalJSON 

func (r Report) MarshalJSON() ([]byte, error)

MarshalJSON is custom marshal for report to properly format the date

type Reporter 

type Reporter struct {
	// contains filtered or unexported fields
}

Reporter gathers all the results

func (*Reporter) Finalize 

func (r *Reporter) Finalize(stopReason StopReason, total time.Duration) *Report

Finalize all the gathered data into a final report

func (*Reporter) Run 

func (r *Reporter) Run()

Run runs the reporter

type RequestCounter added in v0.70.0 

type RequestCounter interface {
	Get() uint64
}

RequestCounter gets the request count

type Requester 

type Requester struct {
	// contains filtered or unexported fields
}

Requester is used for doing the requests

func NewRequester added in v0.60.0 

func NewRequester(c *RunConfig) (*Requester, error)

NewRequester creates a new requestor from the passed RunConfig

func (*Requester) Finish 

func (b *Requester) Finish() *Report

Finish finishes the test run

func (*Requester) Run 

func (b *Requester) Run() (*Report, error)

Run makes all the requests and returns a report of results It blocks until all work is done.

func (*Requester) Stop 

func (b *Requester) Stop(reason StopReason)

Stop stops the test

type ResultDetail 

type ResultDetail struct {
	Timestamp time.Time     `json:"timestamp"`
	Latency   time.Duration `json:"latency"`
	Error     string        `json:"error"`
	Status    string        `json:"status"`
}

ResultDetail data for each result

type RunConfig 

type RunConfig struct {
	// contains filtered or unexported fields
}

RunConfig represents the request Configs

func NewConfig added in v0.61.0 

func NewConfig(call, host string, options ...Option) (*RunConfig, error)

NewConfig creates a new RunConfig from the options passed

type StopReason 

type StopReason string

StopReason is a reason why the run ended

func ReasonFromString 

func ReasonFromString(str string) StopReason

ReasonFromString creates a Status from a string

func (StopReason) MarshalJSON 

func (s StopReason) MarshalJSON() ([]byte, error)

MarshalJSON formats a Threshold value into a JSON string

func (StopReason) String 

func (s StopReason) String() string

String() is the string representation of threshold

func (*StopReason) UnmarshalJSON 

func (s *StopReason) UnmarshalJSON(b []byte) error

UnmarshalJSON prases a Threshold value from JSON string

type StreamInterceptor added in v0.118.0 

type StreamInterceptor interface {
	Recv(*dynamic.Message, error) error
	Send(*CallData) (*dynamic.Message, error)
}

StreamInterceptor is an interface for sending and receiving stream messages. The interceptor can keep shared state for the send and receive calls.

type StreamInterceptorProviderFunc added in v0.118.0 

type StreamInterceptorProviderFunc func(*CallData) StreamInterceptor

StreamInterceptorProviderFunc is an interface for a function invoked to generate a stream interceptor

type StreamMessageProviderFunc added in v0.80.0 

type StreamMessageProviderFunc func(*CallData) (*dynamic.Message, error)

StreamMessageProviderFunc is the interface for providing a message for every message send in the course of a streaming call

type StreamRecvMsgInterceptFunc added in v0.80.0 

type StreamRecvMsgInterceptFunc func(*dynamic.Message, error) error

StreamRecvMsgInterceptFunc is an interface for function invoked when we receive a stream message Clients can return ErrEndStream to end the call early

type TickValue added in v0.70.0 

type TickValue struct {
	// contains filtered or unexported fields
}

TickValue is the tick value

type Worker added in v0.35.0 

type Worker struct {
	// contains filtered or unexported fields
}

Worker is used for doing a single stream of requests in parallel

func (*Worker) Stop added in v0.70.0 

func (w *Worker) Stop()

Stop stops the worker. It has to be started with Run() again.

Source Files

View all Source files

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.