Anthrove/plug-sdk
5
0
Fork 0
Code Issues Pull Requests Actions Projects Releases 10 Activity

feat: Assign new context with span context to address gRPC closure issue
All checks were successful
Gitea Build Check / Build (pull_request) Successful in 41s
Details

Browse Source
This commit is contained in:
David Janowski 2024-08-26 16:44:50 +02:00
parent 6b91b4ef09
commit b7af573a0c
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pkg/plug/grpc.go
View File

@ -60,6 +60,8 @@ func (s *server) TaskStart(ctx context.Context, creation *gRPC.PlugTaskCreation)
// gRPC closes the context after the call ended. So the whole scrapping stopped without waiting
// by using this method we assign a new context to each new request we get.
// This can be used for example to close the context with the given id
ctx = trace.ContextWithSpanContext(context.Background(), trace.NewSpanContext(trace.SpanContextConfig{TraceID: span.SpanContext().TraceID()}))
taskCtx, cancel := context.WithCancel(ctx)
s.ctx[id] = cancel
span.AddEvent("Created new context for task", trace.WithAttributes(attribute.String("task_id", id)))