-
Notifications
You must be signed in to change notification settings - Fork 1.2k
/
runner.go
82 lines (66 loc) · 2.21 KB
/
runner.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
/*
*
* k6 - a next-generation load testing tool
* Copyright (C) 2016 Load Impact
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation, either version 3 of the
* License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*
*/
package lib
import (
"context"
"github.com/loadimpact/k6/stats"
)
// Ensure RunnerFunc conforms to Runner.
var _ Runner = RunnerFunc(nil)
// A Runner is a factory for VUs.
type Runner interface {
// Creates a new VU. As much as possible should be precomputed here, to allow a pool
// of prepared VUs to be used to quickly scale up and down.
NewVU() (VU, error)
// Returns the default (root) group.
GetDefaultGroup() *Group
// Returns the option set.
GetOptions() Options
// Applies a set of options.
ApplyOptions(opts Options)
}
// A VU is a Virtual User.
type VU interface {
// Runs the VU once. An iteration should be completely self-contained, and no state
// or open connections should carry over from one iteration to the next.
RunOnce(ctx context.Context) ([]stats.Sample, error)
// Called when the VU's identity changes.
Reconfigure(id int64) error
}
// RunnerFunc adapts a function to be used as both a runner and a VU. NewVU() returns copies of
// the function itself. Mainly useful for testing.
type RunnerFunc func(ctx context.Context) ([]stats.Sample, error)
func (fn RunnerFunc) NewVU() (VU, error) {
return fn, nil
}
func (fn RunnerFunc) GetDefaultGroup() *Group {
return nil
}
func (fn RunnerFunc) GetOptions() Options {
return Options{}
}
func (fn RunnerFunc) ApplyOptions(opts Options) {
}
func (fn RunnerFunc) RunOnce(ctx context.Context) ([]stats.Sample, error) {
return fn(ctx)
}
func (fn RunnerFunc) Reconfigure(id int64) error {
return nil
}