mirror of
https://github.com/hanzoai/tasks.git
synced 2026-10-04 02:04:03 +00:00
Stephan BehnkeandGitHub
c9e360a3fb
## What changed? Integrate Go 1.26's new `go fix` into workflow. NOTE that the changes caused our linter to fire; a [separate commit](https://github.com/temporalio/temporal/pull/9337/commits/1b23f787aef370098b35829c4d17dcc06230b4c1) addresses those. ## Why? Ensure Go code is standardized/modernized. ## How did you test it? - [ ] built - [ ] run locally and tested manually - [x] covered by existing tests - [ ] added new unit test(s) - [ ] added new functional test(s)
84 lines
2.9 KiB
Go
84 lines
2.9 KiB
Go
package testing
|
|
|
|
type (
|
|
// Model represents a state transition graph that contains all the relationships of Vertex
|
|
Model interface {
|
|
AddEdge(...Edge)
|
|
ListEdges() []Edge
|
|
}
|
|
|
|
// Generator generates a sequence of vertices based on the defined models
|
|
// It must define InitialEntryVertex and ExitVertex
|
|
// To use a generator:
|
|
// for generator.HasNextVertex {
|
|
// generator.GetNextVertices
|
|
// }
|
|
Generator interface {
|
|
// InitialEntryVertex is the beginning vertices of the graph
|
|
// Only one vertex will be picked as the entry
|
|
AddInitialEntryVertex(...Vertex)
|
|
// ExitVertex is the terminate vertices of the graph
|
|
AddExitVertex(...Vertex)
|
|
// RandomEntryVertex is a random entry point which can be access at any state of the generator
|
|
AddRandomEntryVertex(...Vertex)
|
|
// AddModel loads model into the generator
|
|
// AddModel can load multiple models and models will be joint if there is common vertices
|
|
AddModel(Model)
|
|
// HasNextVertex determines if there is more vertex to generate
|
|
HasNextVertex() bool
|
|
// GetNextVertices generates next vertex batch
|
|
GetNextVertices() []Vertex
|
|
// ListGeneratedVertices lists the pasted generated vertices
|
|
ListGeneratedVertices() []Vertex
|
|
// Reset cleans up all the internal states and reset to a brand new generator
|
|
Reset()
|
|
// DeepCopy copy a new instance of generator
|
|
DeepCopy() Generator
|
|
// SetBatchGenerationRule sets a function that used in GetNextVertex to return batch result
|
|
SetBatchGenerationRule(func([]Vertex, []Vertex) bool)
|
|
// SetVersion sets the event version
|
|
SetVersion(int64)
|
|
// GetVersion gets the event version
|
|
GetVersion() int64
|
|
}
|
|
|
|
// Vertex represents a state in the model. A state represents a type of an Temporal event
|
|
Vertex interface {
|
|
// The name of the vertex. Usually, this will be the Temporal event type
|
|
SetName(string)
|
|
GetName() string
|
|
// Equals(Vertex) bool
|
|
// IsStrictOnNextVertex means if the vertex must be followed by its children
|
|
// When IsStrictOnNextVertex set to true, it means this event can only follow by its neighbors
|
|
SetIsStrictOnNextVertex(bool)
|
|
IsStrictOnNextVertex() bool
|
|
// MaxNextVertex means the max neighbors can branch out from this vertex
|
|
SetMaxNextVertex(int)
|
|
GetMaxNextVertex() int
|
|
|
|
// SetVertexDataFunc sets a function to generate end vertex data
|
|
SetDataFunc(func(...any) any)
|
|
GetDataFunc() func(...any) any
|
|
GenerateData(...any) any
|
|
GetData() any
|
|
DeepCopy() Vertex
|
|
}
|
|
|
|
// Edge is the connection between two vertices
|
|
Edge interface {
|
|
// StartVertex is the head of the connection
|
|
SetStartVertex(Vertex)
|
|
GetStartVertex() Vertex
|
|
// EndVertex is the end of the connection
|
|
SetEndVertex(Vertex)
|
|
GetEndVertex() Vertex
|
|
// Condition defines a function to determine if this connection is accessible
|
|
SetCondition(func(...any) bool)
|
|
GetCondition() func(...any) bool
|
|
// Action defines function to perform when the end vertex reached
|
|
SetAction(func())
|
|
GetAction() func()
|
|
DeepCopy() Edge
|
|
}
|
|
)
|