mirror of
https://gitee.com/johng/gf.git
synced 2024-12-02 20:28:17 +08:00
75 lines
2.1 KiB
Go
75 lines
2.1 KiB
Go
package jaeger
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/gogf/gf/v2/errors/gcode"
|
|
"github.com/gogf/gf/v2/errors/gerror"
|
|
"github.com/gogf/gf/v2/net/gipv4"
|
|
"github.com/gogf/gf/v2/text/gregex"
|
|
"go.opentelemetry.io/otel"
|
|
"go.opentelemetry.io/otel/attribute"
|
|
"go.opentelemetry.io/otel/exporters/jaeger"
|
|
"go.opentelemetry.io/otel/sdk/resource"
|
|
"go.opentelemetry.io/otel/sdk/trace"
|
|
semconv "go.opentelemetry.io/otel/semconv/v1.4.0"
|
|
)
|
|
|
|
const (
|
|
tracerHostnameTagKey = "hostname"
|
|
)
|
|
|
|
// Init initializes and registers jaeger to global TracerProvider.
|
|
//
|
|
// The output parameter `flush` is used for waiting exported trace spans to be uploaded,
|
|
// which is useful if your program is ending, and you do not want to lose recent spans.
|
|
func Init(serviceName, endpoint string) (*trace.TracerProvider, error) {
|
|
// Create the Jaeger exporter
|
|
var endpointOption jaeger.EndpointOption
|
|
if strings.HasPrefix(endpoint, "http") {
|
|
// HTTP.
|
|
endpointOption = jaeger.WithCollectorEndpoint(jaeger.WithEndpoint(endpoint))
|
|
} else {
|
|
// UDP.
|
|
match, err := gregex.MatchString(`(.+):(\d+)`, endpoint)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if len(match) < 3 {
|
|
return nil, gerror.NewCodef(
|
|
gcode.CodeInvalidParameter, `invalid endpoint "%s"`, endpoint,
|
|
)
|
|
}
|
|
var (
|
|
host = match[1]
|
|
port = match[2]
|
|
)
|
|
endpointOption = jaeger.WithAgentEndpoint(
|
|
jaeger.WithAgentHost(host), jaeger.WithAgentPort(port),
|
|
)
|
|
}
|
|
hostIP, err := gipv4.GetIntranetIp()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
exp, err := jaeger.New(endpointOption)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
tp := trace.NewTracerProvider(
|
|
// Always be sure to batch in production.
|
|
trace.WithBatcher(exp),
|
|
// Record information about this application in a Resource.
|
|
trace.WithResource(resource.NewWithAttributes(
|
|
semconv.SchemaURL,
|
|
semconv.ServiceNameKey.String(serviceName),
|
|
semconv.HostNameKey.String(hostIP),
|
|
attribute.String(tracerHostnameTagKey, hostIP),
|
|
)),
|
|
)
|
|
// Register our TracerProvider as the global, so any imported
|
|
// instrumentation in the future will default to using it.
|
|
otel.SetTracerProvider(tp)
|
|
return tp, nil
|
|
}
|