parent
3bf07e0b28
commit
62a894e89e
@ -0,0 +1,72 @@ |
||||
package es6 |
||||
|
||||
import ( |
||||
"bytes" |
||||
"context" |
||||
"fmt" |
||||
"gitea.drugeyes.vip/pharnexbase/utils/glog/v1" |
||||
"github.com/elastic/go-elasticsearch/v6/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-v6.7.0") |
||||
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 |
||||
} |
Loading…
Reference in new issue