mirror of
https://github.com/hanzoai/tasks.git
synced 2026-10-04 02:04:03 +00:00
ba33c258f7
## 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]>
96 lines
2.5 KiB
Go
96 lines
2.5 KiB
Go
package postgresql
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
|
|
"go.temporal.io/api/serviceerror"
|
|
"go.temporal.io/server/common/persistence/sql/sqlplugin"
|
|
)
|
|
|
|
const (
|
|
getTaskV2Qry = `SELECT task_id, data, data_encoding ` +
|
|
`FROM tasks_v2 ` +
|
|
`WHERE range_hash = $1 ` +
|
|
`AND task_queue_id = $2 ` +
|
|
`AND (pass, task_id) >= ($3, $4) ` +
|
|
`ORDER BY pass, task_id`
|
|
|
|
getTaskV2QryWithLimit = getTaskV2Qry + ` LIMIT $5`
|
|
|
|
createTaskV2Qry = `INSERT INTO ` +
|
|
`tasks_v2 ( range_hash, task_queue_id, task_id, pass, data, data_encoding) ` +
|
|
`VALUES (:range_hash, :task_queue_id, :task_id, :task_pass, :data, :data_encoding)`
|
|
|
|
rangeDeleteTaskV2Qry = `DELETE FROM tasks_v2 ` +
|
|
`WHERE range_hash = $1 AND task_queue_id = $2 AND task_id IN (SELECT task_id FROM ` +
|
|
`tasks_v2 WHERE range_hash = $1 AND task_queue_id = $2 AND (pass, task_id) < ($3, $4) ` +
|
|
`ORDER BY task_queue_id,pass,task_id LIMIT $5 )`
|
|
)
|
|
|
|
// InsertIntoTasks inserts one or more rows into tasks table
|
|
func (pdb *db) InsertIntoTasksV2(
|
|
ctx context.Context,
|
|
rows []sqlplugin.TasksRowV2,
|
|
) (sql.Result, error) {
|
|
return pdb.NamedExecContext(ctx,
|
|
createTaskV2Qry,
|
|
rows,
|
|
)
|
|
}
|
|
|
|
// SelectFromTasks reads one or more rows from tasks table
|
|
func (pdb *db) SelectFromTasksV2(
|
|
ctx context.Context,
|
|
filter sqlplugin.TasksFilterV2,
|
|
) ([]sqlplugin.TasksRowV2, error) {
|
|
if filter.InclusiveMinLevel == nil {
|
|
return nil, serviceerror.NewInternal("missing InclusiveMinLevel")
|
|
}
|
|
var err error
|
|
var rows []sqlplugin.TasksRowV2
|
|
switch {
|
|
case filter.PageSize != nil:
|
|
err = pdb.SelectContext(ctx,
|
|
&rows,
|
|
getTaskV2QryWithLimit,
|
|
filter.RangeHash,
|
|
filter.TaskQueueID,
|
|
filter.InclusiveMinLevel.TaskPass,
|
|
filter.InclusiveMinLevel.TaskID,
|
|
*filter.PageSize,
|
|
)
|
|
default:
|
|
err = pdb.SelectContext(ctx,
|
|
&rows,
|
|
getTaskV2Qry,
|
|
filter.RangeHash,
|
|
filter.TaskQueueID,
|
|
filter.InclusiveMinLevel.TaskPass,
|
|
filter.InclusiveMinLevel.TaskID,
|
|
)
|
|
}
|
|
return rows, err
|
|
}
|
|
|
|
// DeleteFromTasks deletes multiple rows from tasks table
|
|
func (pdb *db) DeleteFromTasksV2(
|
|
ctx context.Context,
|
|
filter sqlplugin.TasksFilterV2,
|
|
) (sql.Result, error) {
|
|
if filter.ExclusiveMaxLevel == nil {
|
|
return nil, serviceerror.NewInternal("missing ExclusiveMaxTaskLevel")
|
|
}
|
|
if filter.Limit == nil || *filter.Limit == 0 {
|
|
return nil, serviceerror.NewInternal("missing limit parameter")
|
|
}
|
|
return pdb.ExecContext(ctx,
|
|
rangeDeleteTaskV2Qry,
|
|
filter.RangeHash,
|
|
filter.TaskQueueID,
|
|
filter.ExclusiveMaxLevel.TaskPass,
|
|
filter.ExclusiveMaxLevel.TaskID,
|
|
*filter.Limit,
|
|
)
|
|
}
|