package/mq
1
0
Fork 0
mirror of https://github.com/oarkflow/mq.git synced 2025-11-01 23:52:35 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: Add connection

Browse Source
This commit is contained in:
sujit
2024-10-29 07:41:45 +05:45
parent b5ca85a469
commit c7a8ad466e
6 changed files with 65 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
dag/task_manager.go
View File

@@ -92,6 +92,11 @@ func (tm *TaskManager) getConditionalEdges(node *Node, result mq.Result) []Edge
}
func (tm *TaskManager) handleNextTask(ctx context.Context, result mq.Result) mq.Result {
if result.Ctx != nil {
if headers, ok := mq.GetHeaders(ctx); ok {
ctx = mq.SetHeaders(result.Ctx, headers.AsMap())
}
}
defer func() {
tm.wg.Done()
mq.RecoverPanic(mq.RecoverTitle)