26 lines
508 B
Go
26 lines
508 B
Go
|
package apm
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/SkyAPM/go2sky/propagation"
|
||
|
|
"net/http"
|
||
|
|
)
|
||
|
|
|
||
|
|
func HttpExtractor(req *http.Request) propagation.Extractor {
|
||
|
|
return func() (s string, e error) {
|
||
|
|
if req == nil || req.Header == nil {
|
||
|
|
return "", nil
|
||
|
|
}
|
||
|
|
return req.Header.Get(propagation.Header), nil
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func HttpInjector(req *http.Request) propagation.Injector {
|
||
|
|
return func(header string) error {
|
||
|
|
if req == nil || req.Header == nil {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
req.Header.Set(propagation.Header, header)
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
}
|