mirror of
https://github.com/moby/moby.git
synced 2022-11-09 12:21:53 -05:00
bc67a78862
Signed-off-by: Tonis Tiigi <tonistiigi@gmail.com> Signed-off-by: Tibor Vass <tibor@docker.com>
29 lines
508 B
Go
29 lines
508 B
Go
package buildid
|
|
|
|
import (
|
|
"context"
|
|
|
|
"google.golang.org/grpc/metadata"
|
|
)
|
|
|
|
var metadataKey = "buildkit-controlapi-buildid"
|
|
|
|
func AppendToOutgoingContext(ctx context.Context, id string) context.Context {
|
|
if id != "" {
|
|
return metadata.AppendToOutgoingContext(ctx, metadataKey, id)
|
|
}
|
|
return ctx
|
|
}
|
|
|
|
func FromIncomingContext(ctx context.Context) string {
|
|
md, ok := metadata.FromIncomingContext(ctx)
|
|
if !ok {
|
|
return ""
|
|
}
|
|
|
|
if ids := md.Get(metadataKey); len(ids) == 1 {
|
|
return ids[0]
|
|
}
|
|
|
|
return ""
|
|
}
|