mirror of
https://github.com/hanzoai/tasks.git
synced 2026-10-30 14:56:19 +00:00
Roey BermanandGitHub
243effc6e1
## What Used the `depguard` linter to prevent importing the package and modified all uses of it. Also fixed a bug in the membership code where we were not updating the next page token: https://github.com/temporalio/temporal/pull/8541/files#diff-985dd64ffa4c6bfe0248b982dc47439b725862baa3e1f2c559b7ff522bf21969R350 ## Why? From the package README in GitHub: > This package now leverages the github.com/google/uuid package (which is based off an earlier version of this package). Also, there's no point in having two uuid implementations in the codebase. ## Test Added the invalid import and confirmed it works: ``` ../service/frontend/admin_handler_test.go:13:2: import 'github.com/pborman/uuid' is not allowed from list 'main': Importing github.com/pborman/uuid is disallowed; use github.com/google/uuid instead (depguard) _ "github.com/pborman/uuid" ```
124 lines
3.2 KiB
Go
124 lines
3.2 KiB
Go
package workflow
|
|
|
|
import (
|
|
"sync/atomic"
|
|
|
|
"github.com/google/uuid"
|
|
enumspb "go.temporal.io/api/enums/v1"
|
|
querypb "go.temporal.io/api/query/v1"
|
|
"go.temporal.io/api/serviceerror"
|
|
historyi "go.temporal.io/server/service/history/interfaces"
|
|
)
|
|
|
|
const (
|
|
QueryCompletionTypeSucceeded historyi.QueryCompletionType = iota
|
|
QueryCompletionTypeUnblocked
|
|
QueryCompletionTypeFailed
|
|
)
|
|
|
|
var (
|
|
errCompletionStateInvalid = serviceerror.NewInternal("query completion state invalid")
|
|
errAlreadyInCompletionState = serviceerror.NewInternal("query is already in completion state")
|
|
errQueryNotInCompletionState = serviceerror.NewInternal("query is not in completion state")
|
|
)
|
|
|
|
type (
|
|
query interface {
|
|
getID() string
|
|
getCompletionCh() <-chan struct{}
|
|
getQueryInput() *querypb.WorkflowQuery
|
|
GetCompletionState() (*historyi.QueryCompletionState, error)
|
|
setCompletionState(*historyi.QueryCompletionState) error
|
|
}
|
|
|
|
queryImpl struct {
|
|
id string
|
|
queryInput *querypb.WorkflowQuery
|
|
completionCh chan struct{}
|
|
|
|
completionState atomic.Value
|
|
}
|
|
)
|
|
|
|
func newQuery(queryInput *querypb.WorkflowQuery) query {
|
|
return &queryImpl{
|
|
id: uuid.NewString(),
|
|
queryInput: queryInput,
|
|
completionCh: make(chan struct{}),
|
|
}
|
|
}
|
|
|
|
func (q *queryImpl) getID() string {
|
|
return q.id
|
|
}
|
|
|
|
func (q *queryImpl) getCompletionCh() <-chan struct{} {
|
|
return q.completionCh
|
|
}
|
|
|
|
func (q *queryImpl) getQueryInput() *querypb.WorkflowQuery {
|
|
return q.queryInput
|
|
}
|
|
|
|
func (q *queryImpl) GetCompletionState() (*historyi.QueryCompletionState, error) {
|
|
ts := q.completionState.Load()
|
|
if ts == nil {
|
|
return nil, errQueryNotInCompletionState
|
|
}
|
|
queryState, ok := ts.(*historyi.QueryCompletionState)
|
|
if !ok {
|
|
return nil, errCompletionStateInvalid
|
|
}
|
|
return queryState, nil
|
|
}
|
|
|
|
func (q *queryImpl) setCompletionState(completionState *historyi.QueryCompletionState) error {
|
|
if err := q.validateCompletionState(completionState); err != nil {
|
|
return err
|
|
}
|
|
currCompletionState, _ := q.GetCompletionState()
|
|
if currCompletionState != nil {
|
|
return errAlreadyInCompletionState
|
|
}
|
|
q.completionState.Store(completionState)
|
|
close(q.completionCh)
|
|
return nil
|
|
}
|
|
|
|
func (q *queryImpl) validateCompletionState(
|
|
completionState *historyi.QueryCompletionState,
|
|
) error {
|
|
if completionState == nil {
|
|
return errCompletionStateInvalid
|
|
}
|
|
switch completionState.Type {
|
|
case QueryCompletionTypeSucceeded:
|
|
if completionState.Result == nil || completionState.Err != nil {
|
|
return errCompletionStateInvalid
|
|
}
|
|
queryResult := completionState.Result
|
|
validAnswered := queryResult.GetResultType() == enumspb.QUERY_RESULT_TYPE_ANSWERED &&
|
|
queryResult.Answer != nil &&
|
|
queryResult.GetErrorMessage() == ""
|
|
validFailed := queryResult.GetResultType() == enumspb.QUERY_RESULT_TYPE_FAILED &&
|
|
queryResult.Answer == nil &&
|
|
queryResult.GetErrorMessage() != ""
|
|
if !validAnswered && !validFailed {
|
|
return errCompletionStateInvalid
|
|
}
|
|
return nil
|
|
case QueryCompletionTypeUnblocked:
|
|
if completionState.Result != nil || completionState.Err != nil {
|
|
return errCompletionStateInvalid
|
|
}
|
|
return nil
|
|
case QueryCompletionTypeFailed:
|
|
if completionState.Result != nil || completionState.Err == nil {
|
|
return errCompletionStateInvalid
|
|
}
|
|
return nil
|
|
default:
|
|
return errCompletionStateInvalid
|
|
}
|
|
}
|