Files
tasks/common/persistence/sql/sqlplugin/matching_task_queue_user_data.go
T
ba33c258f7 Matching task fairness SQL support (#8099)
## What changed?
Add SQL support for matching task fairness.

Based on #8043 with some changes to reduce code duplication (can be
applied to cassandra later).

## Why?
Support all persistence implementations.

## How did you test it?
- [x] covered by existing tests
- [x] added new unit test(s)

---------

Co-authored-by: sivagirish81 <[email protected]>
2025-07-31 05:01:00 -07:00

55 lines
1.5 KiB
Go

package sqlplugin
import "context"
type (
GetTaskQueueUserDataRequest struct {
NamespaceID []byte
TaskQueueName string
}
ListTaskQueueUserDataEntriesRequest struct {
NamespaceID []byte
LastTaskQueueName string
Limit int
}
AddToBuildIdToTaskQueueMapping struct {
NamespaceID []byte
TaskQueueName string
BuildIds []string
}
RemoveFromBuildIdToTaskQueueMapping struct {
NamespaceID []byte
TaskQueueName string
BuildIds []string
}
GetTaskQueuesByBuildIdRequest struct {
NamespaceID []byte
BuildID string
}
CountTaskQueuesByBuildIdRequest struct {
NamespaceID []byte
BuildID string
}
TaskQueueUserDataEntry struct {
TaskQueueName string
VersionedBlob
}
MatchingTaskQueueUserData interface {
GetTaskQueueUserData(ctx context.Context, request *GetTaskQueueUserDataRequest) (*VersionedBlob, error)
UpdateTaskQueueUserData(ctx context.Context, request *UpdateTaskQueueDataRequest) error
ListTaskQueueUserDataEntries(ctx context.Context, request *ListTaskQueueUserDataEntriesRequest) ([]TaskQueueUserDataEntry, error)
AddToBuildIdToTaskQueueMapping(ctx context.Context, request AddToBuildIdToTaskQueueMapping) error
RemoveFromBuildIdToTaskQueueMapping(ctx context.Context, request RemoveFromBuildIdToTaskQueueMapping) error
GetTaskQueuesByBuildId(ctx context.Context, request *GetTaskQueuesByBuildIdRequest) ([]string, error)
CountTaskQueuesByBuildId(ctx context.Context, request *CountTaskQueuesByBuildIdRequest) (int, error)
}
)