diff --git a/proxy/context.go b/proxy/context.go index f0400a65c..2f59dd77b 100644 --- a/proxy/context.go +++ b/proxy/context.go @@ -17,10 +17,12 @@ const ( resolvedIPsKey ) +// ContextWithSource creates a new context with given source. func ContextWithSource(ctx context.Context, src net.Destination) context.Context { return context.WithValue(ctx, sourceKey, src) } +// SourceFromContext retreives source from the given context. func SourceFromContext(ctx context.Context) (net.Destination, bool) { v, ok := ctx.Value(sourceKey).(net.Destination) return v, ok