Skip to content

Commit

Permalink
test: generate uuid on startup for load tool (#9383)
Browse files Browse the repository at this point in the history
the `NewClient` method is called by the load test framework for each connection. This means that if multiple connections are instantiated, each connection will erroneously have its own UUID. This PR changes the UUID generation to happen at the _beginning_ of the script instead of on client creation so that each experimental run shares a UUID.

Caught while preparing the script for production readiness.

#### PR checklist

- [ ] Tests written/updated, or no tests needed
- [ ] `CHANGELOG_PENDING.md` updated, or no changelog entry needed
- [ ] Updated relevant documentation (`docs/`) and code comments, or no
      documentation updates needed
  • Loading branch information
williambanfield authored Sep 8, 2022
1 parent 43ebbed commit 59a711e
Showing 1 changed file with 6 additions and 4 deletions.
10 changes: 6 additions & 4 deletions test/loadtime/cmd/load/main.go
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,9 @@ var (
)

// ClientFactory implements the loadtest.ClientFactory interface.
type ClientFactory struct{}
type ClientFactory struct {
ID []byte
}

// TxGenerator is responsible for generating transactions.
// TxGenerator holds the set of information that will be used to generate
Expand All @@ -28,7 +30,8 @@ type TxGenerator struct {
}

func main() {
if err := loadtest.RegisterClientFactory("loadtime-client", &ClientFactory{}); err != nil {
u := [16]byte(uuid.New()) // generate run ID on startup
if err := loadtest.RegisterClientFactory("loadtime-client", &ClientFactory{ID: u[:]}); err != nil {
panic(err)
}
loadtest.Run(&loadtest.CLIConfig{
Expand All @@ -51,9 +54,8 @@ func (f *ClientFactory) ValidateConfig(cfg loadtest.Config) error {
}

func (f *ClientFactory) NewClient(cfg loadtest.Config) (loadtest.Client, error) {
u := [16]byte(uuid.New())
return &TxGenerator{
id: u[:],
id: f.ID,
conns: uint64(cfg.Connections),
rate: uint64(cfg.Rate),
size: uint64(cfg.Size),
Expand Down

0 comments on commit 59a711e

Please sign in to comment.