mirror of
https://github.com/hanzoai/tasks.git
synced 2026-11-03 03:56:04 +00:00
zeekay
5a601d081b
Full module path rebrand from Temporal to Hanzo Tasks: - go.mod module: github.com/hanzoai/tasks - All internal Go imports updated (1906 files) - cmd/server renamed to cmd/tasksd - Binary name: tasksd (was temporal-server) - Docker images: hanzoai/tasks (was temporalio/server) - README rewritten with Hanzo Tasks branding - LLM.md created with integration notes - Makefile, goreleaser, CI workflows updated - go.temporal.io/sdk and go.temporal.io/api imports preserved (separate repos) Fork of Temporal.io (MIT license).
92 lines
2.6 KiB
Go
92 lines
2.6 KiB
Go
package failure
|
|
|
|
import (
|
|
commonpb "go.temporal.io/api/common/v1"
|
|
enumspb "go.temporal.io/api/enums/v1"
|
|
failurepb "go.temporal.io/api/failure/v1"
|
|
"github.com/hanzoai/tasks/common/util"
|
|
)
|
|
|
|
const (
|
|
failureSourceServer = "Server"
|
|
)
|
|
|
|
func NewServerFailure(message string, nonRetryable bool) *failurepb.Failure {
|
|
f := &failurepb.Failure{
|
|
Message: message,
|
|
FailureInfo: &failurepb.Failure_ServerFailureInfo{ServerFailureInfo: &failurepb.ServerFailureInfo{
|
|
NonRetryable: nonRetryable,
|
|
}},
|
|
}
|
|
|
|
return f
|
|
}
|
|
|
|
func NewResetWorkflowFailure(message string, lastHeartbeatDetails *commonpb.Payloads) *failurepb.Failure {
|
|
f := &failurepb.Failure{
|
|
Message: message,
|
|
FailureInfo: &failurepb.Failure_ResetWorkflowFailureInfo{ResetWorkflowFailureInfo: &failurepb.ResetWorkflowFailureInfo{
|
|
LastHeartbeatDetails: lastHeartbeatDetails,
|
|
}},
|
|
}
|
|
|
|
return f
|
|
}
|
|
|
|
func NewTimeoutFailure(message string, timeoutType enumspb.TimeoutType) *failurepb.Failure {
|
|
f := &failurepb.Failure{
|
|
Message: message,
|
|
Source: failureSourceServer,
|
|
FailureInfo: &failurepb.Failure_TimeoutFailureInfo{TimeoutFailureInfo: &failurepb.TimeoutFailureInfo{
|
|
TimeoutType: timeoutType,
|
|
}},
|
|
}
|
|
|
|
return f
|
|
}
|
|
|
|
func Truncate(f *failurepb.Failure, maxSize int) *failurepb.Failure {
|
|
return TruncateWithDepth(f, maxSize, 20)
|
|
}
|
|
|
|
func TruncateWithDepth(f *failurepb.Failure, maxSize, maxDepth int) *failurepb.Failure {
|
|
if f == nil {
|
|
return nil
|
|
}
|
|
|
|
// note that bytes are given to earlier calls first, so call in order of importance
|
|
trunc := func(s string) string {
|
|
s = util.TruncateUTF8(s, maxSize)
|
|
maxSize -= len(s)
|
|
if s != "" {
|
|
maxSize -= 4 // account for proto overhead
|
|
}
|
|
return s
|
|
}
|
|
|
|
newFailure := &failurepb.Failure{}
|
|
|
|
// Keep failure info for ApplicationFailureInfo and for ServerFailureInfo to persist NonRetryable flag.
|
|
if i := f.GetApplicationFailureInfo(); i != nil {
|
|
newFailure.FailureInfo = &failurepb.Failure_ApplicationFailureInfo{ApplicationFailureInfo: &failurepb.ApplicationFailureInfo{
|
|
NonRetryable: i.NonRetryable,
|
|
Type: trunc(i.Type),
|
|
}}
|
|
maxSize -= 8 // account for proto overhead
|
|
} else if i := f.GetServerFailureInfo(); i != nil {
|
|
newFailure.FailureInfo = &failurepb.Failure_ServerFailureInfo{ServerFailureInfo: &failurepb.ServerFailureInfo{
|
|
NonRetryable: i.NonRetryable,
|
|
}}
|
|
maxSize -= 4 // account for proto overhead
|
|
}
|
|
|
|
newFailure.Source = trunc(f.Source)
|
|
newFailure.Message = trunc(f.Message)
|
|
newFailure.StackTrace = trunc(f.StackTrace)
|
|
if f.Cause != nil && maxSize > 4 && maxDepth > 0 {
|
|
newFailure.Cause = TruncateWithDepth(f.Cause, maxSize-4, maxDepth-1)
|
|
}
|
|
|
|
return newFailure
|
|
}
|