Skip to main content

Selection functions

stateMachine

Commentary

added in 0.0.5

stateful function

Generates data according to a set of states and transitions.

State machines can be used in two ways: either as regular functions, or as a top-level field that controls the entire generator.

For the former, when a state is triggered, its value is used in place. 1 For the latter, its value is merged into the generator to override its existing behavior. 2

For many use cases, it's common that each successive state needs to merge its previously generated event. Top-level state machines may be configured with a merge property to enable this behavior. 3


Examples

Basic state machine

Use a state machine like any other function. ShadowTraffic keeps track of the current state and advances it over time. States can transition indefinitely, or terminate, in which case the generator will go dormant and stop generating new events.

{
"_gen": "stateMachine",
"initial": "a",
"transitions": {
"a": "b",
"b": "c",
"c": "a"
},
"states": {
"a": 1,
"b": 2,
"c": 3
}
}
[
1,
2,
3,
1,
2
]

Top-level state machine

An optional top-level state machine may be defined for the generator. When present, the value of its state is merged into the entire generator output. This is useful when modeling change that affects many attributes, even including the output collection.

{
"generators": [
{
"topic": "customers",
"vars": {
"id": {
"_gen": "uniformDistribution",
"bounds": [
1,
10
]
}
},
"key": {
"_gen": "var",
"var": "id"
},
"value": {},
"stateMachine": {
"_gen": "stateMachine",
"initial": "a",
"transitions": {
"a": "b",
"b": "c",
"c": "a"
},
"states": {
"a": {
"value": {
"action": "start"
}
},
"b": {
"value": {
"action": "running"
}
},
"c": {
"value": {
"action": "stopped"
}
}
}
}
}
],
"connections": {
"kafka": {
"kind": "kafka",
"producerConfigs": {
"bootstrap.servers": "localhost:9092",
"key.serializer": "io.shadowtraffic.kafka.serdes.JsonSerializer",
"value.serializer": "io.shadowtraffic.kafka.serdes.JsonSerializer"
}
}
}
}
[
{
"topic": "customers",
"key": 1.3906161747424364,
"value": {
"action": "start"
},
"headers": null
},
{
"topic": "customers",
"key": 6.102332126254197,
"value": {
"action": "running"
},
"headers": null
},
{
"topic": "customers",
"key": 5.243105539761816,
"value": {
"action": "stopped"
},
"headers": null
},
{
"topic": "customers",
"key": 9.367015862769998,
"value": {
"action": "start"
},
"headers": null
},
{
"topic": "customers",
"key": 8.321629163914373,
"value": {
"action": "running"
},
"headers": null
}
]

Merge previous events

Top-level state machines can force their previously generated events into their body.

{
"topic": "sandbox",
"value": {
"a": 1
},
"stateMachine": {
"_gen": "stateMachine",
"initial": "s1",
"merge": {
"previous": true
},
"transitions": {
"s1": "s2",
"s2": "s3"
},
"states": {
"s1": {
"value": {
"b": 2
}
},
"s2": {
"value": {
"c": 3
}
},
"s3": {
"value": {
"d": 4
}
}
}
}
}
[
{
"a": 1,
"b": 2
},
{
"a": 1,
"b": 2,
"c": 3
},
{
"a": 1,
"b": 2,
"c": 3,
"d": 4
}
]

Sequential transitions

Instead of a map, transitions can also be specified as an array, describing the exact path to take through the state machine.

When specified in this form, you don't need to specify an initial state since it's obvious where to begin.

{
"_gen": "stateMachine",
"transitions": [
"s1",
"s2",
"s2",
"s3"
],
"states": {
"s1": {
"value": {
"a": 2
}
},
"s2": {
"value": {
"b": 3
}
},
"s3": {
"value": {
"c": 4
}
}
}
}
[
{
"a": 2
},
{
"b": 3
},
{
"b": 3
},
{
"c": 4
}
]

Repeated transitions

Optionally use state and times in an array of transitions to repeat states.

{
"_gen": "stateMachine",
"transitions": [
"s1",
{
"state": "s2",
"times": 3
},
"s3"
],
"states": {
"s1": {
"value": {
"a": 2
}
},
"s2": {
"value": {
"b": 3
}
},
"s3": {
"value": {
"c": 4
}
}
}
}
[
{
"a": 2
},
{
"b": 3
},
{
"b": 3
},
{
"b": 3
},
{
"c": 4
}
]

Functions in transitions

Both state and times can be functions. times is evaluated only once, and its value locked for the lifetime of the state machine. state is evaluated on each execution.

In this example, state s1 runs once, and then either s2 or s3 are visited as many as 4 times total.

{
"topic": "sandbox",
"varsOnce": {
"n": {
"_gen": "uniformDistribution",
"bounds": [
1,
5
]
}
},
"value": {},
"stateMachine": {
"_gen": "stateMachine",
"transitions": [
"s1",
{
"state": {
"_gen": "oneOf",
"choices": [
"s2",
"s3"
]
},
"times": {
"_gen": "var",
"var": "n"
}
}
],
"states": {
"s1": {
"value": {
"x": 1
}
},
"s2": {
"value": {
"x": 2
}
},
"s3": {
"value": {
"x": 3
}
}
}
}
}
[
{
"x": 1
},
{
"x": 2
},
{
"x": 3
},
{
"x": 3
},
{
"x": 3
}
]

Terminating state machines

Transition to the value null to terminate a state machine, and thus the generator.

{
"topic": "sandbox",
"value": {},
"stateMachine": {
"_gen": "stateMachine",
"initial": "s1",
"transitions": {
"s1": "s2",
"s2": null,
"s3": "s4"
},
"states": {
"s1": {
"value": {
"x": 1
}
},
"s2": {
"value": {
"x": 2
}
},
"s3": {
"value": {
"x": 3
}
},
"s4": {
"value": {
"x": 4
}
}
}
}
}
[
{
"x": 1
},
{
"x": 2
}
]

Overriding throttle values

Use localConfigs with throttleMs inside of a state to override the generator's default throttle value. This is useful if you want to make a particular state wait longer or shorter before transitioning to the next.

In this example, state s1 throttles for 2000 ms, the default, and s2 throttles for 500 ms.

{
"topic": "sandbox",
"value": {},
"stateMachine": {
"_gen": "stateMachine",
"initial": "s1",
"transitions": {
"s1": "s2",
"s2": "s1"
},
"states": {
"s1": {
"value": {
"x": 1
}
},
"s2": {
"value": {
"x": 2
},
"localConfigs": {
"throttleMs": 500
}
}
}
},
"localConfigs": {
"throttleMs": 2000
}
}

Specification

JSON schema

{
"type": "object",
"properties": {
"merge": {
"type": "object",
"properties": {
"previous": {
"type": "boolean"
}
},
"required": [
"previous"
]
},
"states": {
"type": "object"
}
},
"required": [
"transitions",
"states"
],
"anyOf": [
{
"properties": {
"initial": {
"type": "string"
},
"transitions": {
"type": "object"
}
},
"required": [
"initial"
]
},
{
"properties": {
"transitions": {
"type": "array",
"minItems": 1,
"items": {
"oneOf": [
{
"type": "string"
},
{
"type": "object",
"properties": {
"state": {
"oneOf": [
{
"type": "string"
},
{
"type": "object",
"properties": {
"_gen": {
"type": "string"
}
},
"required": [
"_gen"
]
}
]
},
"times": {
"oneOf": [
{
"type": "integer",
"minimum": 0
},
{
"type": "object",
"properties": {
"_gen": {
"type": "string"
}
},
"required": [
"_gen"
]
}
]
}
},
"required": [
"state",
"times"
]
}
]
}
}
}
}
]
}