package es7 import ( "bytes" "context" "fmt" "gitea.drugeyes.vip/pharnexbase/utils/glog/v1" "github.com/elastic/go-elasticsearch/v7/estransport" "go.opentelemetry.io/otel" "go.opentelemetry.io/otel/attribute" semconv "go.opentelemetry.io/otel/semconv/v1.4.0" "go.opentelemetry.io/otel/trace" "net/http" "time" ) var _ estransport.Logger = (*Logger)(nil) type Logger struct { EnableRequestBody bool EnableResponseBody bool } func (l *Logger) LogRoundTrip(request *http.Request, response *http.Response, err error, time time.Time, duration time.Duration) error { ctx := request.Context() index := fmt.Sprintf("%s?%s", request.URL.Path, request.URL.RawQuery) method := request.Method if l.RequestBodyEnabled() && request != nil && request.Body != nil && request.Body != http.NoBody { var buf bytes.Buffer if request.GetBody != nil { b, _ := request.GetBody() _, _ = buf.ReadFrom(b) } else { _, _ = buf.ReadFrom(request.Body) } glog.Glog.WithContext(ctx).Info("url:", method, index, " request:", buf.String()) } if l.ResponseBodyEnabled() && response != nil && response.Body != nil && response.Body != http.NoBody { defer response.Body.Close() var buf bytes.Buffer _, _ = buf.ReadFrom(response.Body) glog.Glog.WithContext(ctx).Info("response:", buf.String()) } l.Tracer(ctx, time, method, index) return nil } func (l *Logger) Tracer(ctx context.Context, begin time.Time, method, index string) { tracer := otel.Tracer("elasticsearch-v7.10") kind := trace.SpanKindInternal ctx, span := tracer.Start(ctx, index, trace.WithSpanKind(kind), trace.WithTimestamp(begin), ) var attrs []attribute.KeyValue attrs = append(attrs, semconv.DBSystemElasticsearch) attrs = append(attrs, attribute.Key("es.index").String(method+" "+index)) span.SetAttributes(attrs...) span.End(trace.WithTimestamp(time.Now())) } func (l *Logger) RequestBodyEnabled() bool { return l.EnableRequestBody } func (l *Logger) ResponseBodyEnabled() bool { return l.EnableResponseBody }