MaysWind
36 lines
841 B
Go
36 lines
841 B
Go
package httpclient
|
|
|
|
import (
|
|
"github.com/mayswind/ezbookkeeping/pkg/core"
|
|
)
|
|
|
|
const (
|
|
logHandleKey = "log_handler"
|
|
)
|
|
|
|
// HttpResponseLogHandlerFunc represents the http response log handler function
|
|
type HttpResponseLogHandlerFunc func([]byte)
|
|
|
|
// httpRequestContext represents the context for http request
|
|
type httpRequestContext struct {
|
|
core.Context
|
|
logHandler HttpResponseLogHandlerFunc
|
|
}
|
|
|
|
// Value returns the value associated with key
|
|
func (c *httpRequestContext) Value(key any) any {
|
|
if key == logHandleKey {
|
|
return c.logHandler
|
|
}
|
|
|
|
return c.Context.Value(key)
|
|
}
|
|
|
|
// CustomHttpResponseLog returns a context with http response log handler
|
|
func CustomHttpResponseLog(c core.Context, responseLogHandler HttpResponseLogHandlerFunc) core.Context {
|
|
return &httpRequestContext{
|
|
Context: c,
|
|
logHandler: responseLogHandler,
|
|
}
|
|
}
|