public_git
/
oauth2-proxy
mirror of https://github.com/oauth2-proxy/oauth2-proxy.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Integrate klog into requests package

This commit is contained in:
Joel Speed 2021-10-06 16:36:56 +01:00
parent ecb6bb5277
commit c2acf47199
No known key found for this signature in database
GPG Key ID: 6E80578D6751DEFB
2 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
pkg/requests/builder.go
View File

@ -95,13 +95,16 @@ func (r *builder) Do() Result {
func (r *builder) do() Result { func (r *builder) do() Result {
req, err := http.NewRequestWithContext(r.context, r.method, r.endpoint, r.body) req, err := http.NewRequestWithContext(r.context, r.method, r.endpoint, r.body)
if err != nil { if err != nil {
debugLogger.Infof("Error creating request: %v", err)
r.result = &result{err: fmt.Errorf("error creating request: %v", err)} r.result = &result{err: fmt.Errorf("error creating request: %v", err)}
return r.result return r.result
} }
req.Header = r.header req.Header = r.header
traceLogger.Infof("Executing request: Method: %v, Endpoint: %v, Header: %+v", r.method, r.endpoint, r.header)
resp, err := http.DefaultClient.Do(req) resp, err := http.DefaultClient.Do(req)
if err != nil { if err != nil {
debugLogger.Infof("Error performing request: %v", err)
r.result = &result{err: fmt.Errorf("error performing request: %v", err)} r.result = &result{err: fmt.Errorf("error performing request: %v", err)}
return r.result return r.result
} }
@ -109,10 +112,12 @@ func (r *builder) do() Result {
defer resp.Body.Close() defer resp.Body.Close()
body, err := ioutil.ReadAll(resp.Body) body, err := ioutil.ReadAll(resp.Body)
if err != nil { if err != nil {
debugLogger.Infof("Error reading response body: %v", err)
r.result = &result{err: fmt.Errorf("error reading response body: %v", err)} r.result = &result{err: fmt.Errorf("error reading response body: %v", err)}
return r.result return r.result
} }
traceLogger.Infof("Request result: Status: %s (%d), Body: %s", resp.Status, resp.StatusCode, string(body))
r.result = &result{response: resp, body: body} r.result = &result{response: resp, body: body}
return r.result return r.result
} }

11
pkg/requests/logger.go Normal file
View File

@ -0,0 +1,11 @@
package requests
import (
"github.com/oauth2-proxy/oauth2-proxy/v7/pkg/logger"
"k8s.io/klog/v2"
)
var (
debugLogger = klog.V(logger.RequestDebug)
traceLogger = klog.V(logger.RequestTrace)
)