xkm/runbin
1
0
Fork 0
mirror of https://github.com/dreamstarsky/runbin.git synced 2026-05-15 14:23:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor(postgres): Remove explicit transaction in GetTask

Browse Source
This commit is contained in:
xkm
2025-04-15 20:41:08 +08:00
parent a960184433
commit 8f98b021a9
1 changed files with 1 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
internal/repository/database_store.go
View File

@@ -98,15 +98,10 @@ func (s *PostgresStore) DispatchExecutionTask(id string) error {
} }
func (s *PostgresStore) GetTask(ctx context.Context) (*model.Paste, error) { func (s *PostgresStore) GetTask(ctx context.Context) (*model.Paste, error) {
tx, err := s.db.BeginTx(ctx, &sql.TxOptions{Isolation: sql.LevelSerializable})
if err != nil {
return nil, fmt.Errorf("failed to begin transaction: %w", err)
}
defer tx.Rollback()
// 原子性地删除并获取队列中最旧的任务ID // 原子性地删除并获取队列中最旧的任务ID
var taskID string var taskID string
err = tx.QueryRowContext(ctx, err := s.db.QueryRowContext(ctx,
`DELETE FROM queue `DELETE FROM queue
WHERE ctid = ( WHERE ctid = (
SELECT ctid FROM queue SELECT ctid FROM queue
@@ -123,10 +118,6 @@ func (s *PostgresStore) GetTask(ctx context.Context) (*model.Paste, error) {
return nil, fmt.Errorf("failed to get task: %w", err) return nil, fmt.Errorf("failed to get task: %w", err)
} }
if err := tx.Commit(); err != nil {
return nil, fmt.Errorf("failed to commit transaction: %w", err)
}
// 获取完整的任务数据 // 获取完整的任务数据
p, ok := s.GetByID(taskID) p, ok := s.GetByID(taskID)