mirror of
https://github.com/hanzoai/tasks.git
synced 2026-11-03 03:56:04 +00:00
Alex ShtinandGitHub
8e8f1b0f7c
## What changed?
<!-- Describe what has changed in this PR -->
Remove license header from every file. Because it is really hard to
follow in this PR here is the summary:
1. License header is removed from all `.go` and `.proto` files
:fireworks::fireworks:🎆.
2. `LICENSE` file in the root directory has only Temporal and Uber
copyrights.
3. 5 other `LICENSE` files added to the packages which have copyrights
different from Temporal and Uber: Datadog, Xargin, "Mat Ryer, Tyler
Bunnell and contributors".
4. `license_file` flag is removed from all code generation tools.
5. `copyright_file` flag is removed from `go:generate mockgen`
directive.
6. All copyright related targets are removed from `Makefile`.
7. Updated Temporal copyright year to 2025 everywhere.
## Why?
<!-- Tell your future self why have you made these changes -->
I double checked with legal department that it is not needed to have
license header in every file. One file per repo is enough. I put all
copyrights to the root `LICENSE` file and removed header from all other
files. Also updated tools and `Makefile`.
98 lines
1.6 KiB
Go
98 lines
1.6 KiB
Go
package hsmtest
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
"time"
|
|
|
|
"go.temporal.io/server/service/history/hsm"
|
|
)
|
|
|
|
var (
|
|
errInvalidStateType = fmt.Errorf("invalid state type")
|
|
)
|
|
|
|
type State string
|
|
|
|
const (
|
|
State1 State = "state1"
|
|
State2 State = "state2"
|
|
State3 State = "state3"
|
|
State4 State = "state4"
|
|
)
|
|
|
|
type Data struct {
|
|
state State
|
|
}
|
|
|
|
func NewData(
|
|
state State,
|
|
) *Data {
|
|
return &Data{
|
|
state: state,
|
|
}
|
|
}
|
|
|
|
func (d *Data) State() State {
|
|
return d.state
|
|
}
|
|
|
|
func (d *Data) SetState(s State) {
|
|
d.state = s
|
|
}
|
|
|
|
func (d *Data) RegenerateTasks(node *hsm.Node) ([]hsm.Task, error) {
|
|
return []hsm.Task{
|
|
NewTask(
|
|
hsm.TaskAttributes{Deadline: time.Now().Add(time.Hour)},
|
|
false,
|
|
),
|
|
NewTask(
|
|
hsm.TaskAttributes{Destination: string(d.state)},
|
|
false,
|
|
),
|
|
}, nil
|
|
}
|
|
|
|
type Definition struct {
|
|
typeName string
|
|
}
|
|
|
|
func NewDefinition(typeName string) Definition {
|
|
return Definition{
|
|
typeName: typeName,
|
|
}
|
|
}
|
|
|
|
func (d Definition) Deserialize(b []byte) (any, error) {
|
|
return &Data{State(string(b))}, nil
|
|
}
|
|
|
|
// Serialize implements hsm.StateMachineDefinition.
|
|
func (d Definition) Serialize(s any) ([]byte, error) {
|
|
t, ok := s.(*Data)
|
|
if !ok {
|
|
return nil, errInvalidStateType
|
|
}
|
|
return []byte(t.state), nil
|
|
}
|
|
|
|
// Type implements hsm.StateMachineDefinition.
|
|
func (d Definition) Type() string {
|
|
return d.typeName
|
|
}
|
|
|
|
func (d Definition) CompareState(s1 any, s2 any) (int, error) {
|
|
t1, ok := s1.(*Data)
|
|
if !ok {
|
|
return 0, errInvalidStateType
|
|
}
|
|
|
|
t2, ok := s2.(*Data)
|
|
if !ok {
|
|
return 0, errInvalidStateType
|
|
}
|
|
|
|
return strings.Compare(string(t1.State()), string(t2.State())), nil
|
|
}
|