62 lines
3.0 KiB
Go
62 lines
3.0 KiB
Go
package mysql
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"errors"
|
|
"math"
|
|
|
|
"git.yuntuo.hk/k2safe/orbit/internal/jsonwire"
|
|
core "git.yuntuo.hk/k2safe/orbit/model"
|
|
"git.yuntuo.hk/k2safe/orbit/platform/events"
|
|
)
|
|
|
|
// appendResultEvent 必须在来源任务更新和事件写入之后、原事务提交之前调用。
|
|
// 流行一直锁到提交；后来的事务无法提交更大序号并越过尚未提交的较小序号。
|
|
// 不访问订阅或交付行：慢消费者不能反向阻塞任务的结果持久化。
|
|
func appendResultEvent(ctx context.Context, tx *sql.Tx, ns, eventID, taskID core.ID, revision uint64, eventType string) error {
|
|
var snapshot events.TaskResult
|
|
var raw []byte
|
|
var link workflowTaskLink
|
|
snapshot.SchemaVersion = 1
|
|
err := tx.QueryRowContext(ctx, `SELECT t.id,t.queue_id,t.state,t.revision,t.result,t.effect_uncertain,t.current_attempt_id,t.owner_kind,t.owner_id,t.gate_id,
|
|
EXISTS(SELECT 1 FROM weave_activations a WHERE a.namespace_id=t.namespace_id AND a.task_id=t.id)
|
|
OR EXISTS(SELECT 1 FROM weave_poll_rounds r WHERE r.namespace_id=t.namespace_id AND r.task_id=t.id)
|
|
OR EXISTS(SELECT 1 FROM weave_instances i WHERE i.namespace_id=t.namespace_id AND i.gate_id=t.gate_id)
|
|
FROM pulse_tasks t WHERE t.namespace_id=? AND t.id=?`, ns, taskID).Scan(&snapshot.TaskID, &snapshot.QueueID, &snapshot.State, &snapshot.Revision, &raw, &snapshot.EffectUncertain, &snapshot.CurrentAttemptID, &link.Owner, &link.Activation, &link.Gate, &link.Linked)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if snapshot.Revision != revision || eventType != "task."+map[string]string{"SUCCEEDED": "succeeded", "FAILED": "failed", "CANCELLED": "cancelled", "NEEDS_REVIEW": "needs_review"}[snapshot.State] {
|
|
return errors.New("task result event does not match persisted task state")
|
|
}
|
|
snapshot.Result = raw
|
|
body, err := jsonwire.Marshal(snapshot)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
// 内部待处理记录和结果同事务，外部结果流锁仍最后取得。
|
|
if err := appendWorkflowResult(ctx, tx, ns, eventID, taskID, revision, body, link); err != nil {
|
|
return err
|
|
}
|
|
head, _, err := lockEventStream(ctx, tx, ns)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if head == math.MaxUint64 {
|
|
return errors.New("event sequence exhausted")
|
|
}
|
|
if _, err = tx.ExecContext(ctx, `INSERT INTO orbit_result_events(namespace_id,id,sequence,queue_id,snapshot) VALUES (?,?,?,?,?)`, ns, eventID, head+1, snapshot.QueueID, string(body)); err != nil {
|
|
return err
|
|
}
|
|
_, err = tx.ExecContext(ctx, `UPDATE orbit_event_streams SET last_sequence=?,updated_at=UTC_TIMESTAMP(6) WHERE namespace_id=?`, head+1, ns)
|
|
return err
|
|
}
|
|
func lockEventStream(ctx context.Context, tx *sql.Tx, ns core.ID) (head, retained uint64, err error) {
|
|
if _, err = tx.ExecContext(ctx, `INSERT INTO orbit_event_streams(namespace_id) VALUES (?) ON DUPLICATE KEY UPDATE namespace_id=orbit_event_streams.namespace_id`, ns); err != nil {
|
|
return
|
|
}
|
|
err = tx.QueryRowContext(ctx, `SELECT last_sequence,retained_from_sequence FROM orbit_event_streams WHERE namespace_id=? FOR UPDATE`, ns).Scan(&head, &retained)
|
|
return
|
|
}
|