mirror of
https://github.com/thegeeklab/drone-plugin-lib.git
synced 2024-11-05 02:40:40 +00:00
Add httptrace facility
This commit is contained in:
parent
b693e76822
commit
8ff841fbeb
117
pkg/urfave/httptrace.go
Normal file
117
pkg/urfave/httptrace.go
Normal file
@ -0,0 +1,117 @@
|
||||
// Copyright (c) 2019, the Drone Plugins project authors.
|
||||
// Please see the AUTHORS file for details. All rights reserved.
|
||||
// Use of this source code is governed by an Apache 2.0 license that can be
|
||||
// found in the LICENSE file.
|
||||
|
||||
package urfave
|
||||
|
||||
import (
|
||||
"context"
|
||||
"crypto/tls"
|
||||
"net/http/httptrace"
|
||||
"net/textproto"
|
||||
|
||||
"github.com/sirupsen/logrus"
|
||||
)
|
||||
|
||||
// traceHTTP uses httptrace to log all network activity for HTTP requests.
|
||||
func traceHTTP(ctx context.Context) context.Context {
|
||||
return httptrace.WithClientTrace(ctx, &httptrace.ClientTrace{
|
||||
GetConn: func(hostPort string) {
|
||||
logrus.WithField("host-port", hostPort).Trace("ClientTrace.GetConn")
|
||||
},
|
||||
|
||||
GotConn: func(connInfo httptrace.GotConnInfo) {
|
||||
logrus.WithFields(logrus.Fields{
|
||||
"local-address": connInfo.Conn.LocalAddr(),
|
||||
"remote-address": connInfo.Conn.RemoteAddr(),
|
||||
"reused": connInfo.Reused,
|
||||
"was-idle": connInfo.WasIdle,
|
||||
"idle-time": connInfo.IdleTime,
|
||||
}).Trace("ClientTrace.GoConn")
|
||||
},
|
||||
|
||||
PutIdleConn: func(err error) {
|
||||
logrus.WithField("error", err).Trace("ClientTrace.GoConn")
|
||||
},
|
||||
|
||||
GotFirstResponseByte: func() {
|
||||
logrus.Trace("ClientTrace.GotFirstResponseByte")
|
||||
},
|
||||
|
||||
Got100Continue: func() {
|
||||
logrus.Trace("ClientTrace.Got100Continue")
|
||||
},
|
||||
|
||||
Got1xxResponse: func(code int, header textproto.MIMEHeader) error {
|
||||
logrus.WithFields(logrus.Fields{
|
||||
"code": code,
|
||||
"header": header,
|
||||
}).Trace("ClientTrace.Got1xxxResponse")
|
||||
return nil
|
||||
},
|
||||
|
||||
DNSStart: func(dnsInfo httptrace.DNSStartInfo) {
|
||||
logrus.WithField("host", dnsInfo.Host).Trace("ClientTrace.DNSStart")
|
||||
},
|
||||
|
||||
DNSDone: func(dnsInfo httptrace.DNSDoneInfo) {
|
||||
logrus.WithFields(logrus.Fields{
|
||||
"addresses": dnsInfo.Addrs,
|
||||
"error": dnsInfo.Err,
|
||||
"coalesced": dnsInfo.Coalesced,
|
||||
}).Trace("ClientTrace.DNSDone")
|
||||
},
|
||||
|
||||
ConnectStart: func(network, addr string) {
|
||||
logrus.WithFields(logrus.Fields{
|
||||
"network": network,
|
||||
"address": addr,
|
||||
}).Trace("ClientTrace.ConnectStart")
|
||||
},
|
||||
|
||||
ConnectDone: func(network, addr string, err error) {
|
||||
logrus.WithFields(logrus.Fields{
|
||||
"network": network,
|
||||
"address": addr,
|
||||
"error": err,
|
||||
}).Trace("ClientTrace.ConnectDone")
|
||||
},
|
||||
|
||||
TLSHandshakeStart: func() {
|
||||
logrus.Trace("ClientTrace.TLSHandshakeStart")
|
||||
},
|
||||
|
||||
TLSHandshakeDone: func(cs tls.ConnectionState, err error) {
|
||||
logrus.WithFields(logrus.Fields{
|
||||
"version": cs.Version,
|
||||
"handshake-complete": cs.HandshakeComplete,
|
||||
"did-resume": cs.DidResume,
|
||||
"cipher-suite": cs.CipherSuite,
|
||||
"negotiated-protocol": cs.NegotiatedProtocol,
|
||||
"negotiated-protocol-is-mutual": cs.NegotiatedProtocolIsMutual,
|
||||
"server-name": cs.ServerName,
|
||||
"error": err,
|
||||
}).Trace("ClientTrace.TLSHandshakeDone")
|
||||
},
|
||||
|
||||
WroteHeaderField: func(key string, value []string) {
|
||||
logrus.WithFields(logrus.Fields{
|
||||
"key": key,
|
||||
"values": value,
|
||||
}).Trace("ClientTrace.WroteHeaderField")
|
||||
},
|
||||
|
||||
WroteHeaders: func() {
|
||||
logrus.Trace("ClientTrace.WroteHeaders")
|
||||
},
|
||||
|
||||
Wait100Continue: func() {
|
||||
logrus.Trace("ClientTrace.Wait100Continue")
|
||||
},
|
||||
|
||||
WroteRequest: func(reqInfo httptrace.WroteRequestInfo) {
|
||||
logrus.WithField("error", reqInfo.Err).Trace("ClientTrace.WroteRequest")
|
||||
},
|
||||
})
|
||||
}
|
Loading…
Reference in New Issue
Block a user