Skip to content

Commit c04852c

Browse files
author
Matrix-X
committed
feat(http): support external trasnport component
1 parent fac2050 commit c04852c

File tree

4 files changed

+26
-18
lines changed

4 files changed

+26
-18
lines changed

src/kernel/accessToken.go

Lines changed: 12 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -358,24 +358,27 @@ func (accessToken *AccessToken) OverrideGetMiddlewareOfLog() {
358358
accessToken.GetMiddlewareOfLog = func(logger contract2.LoggerInterface) contract3.RequestMiddleware {
359359
return func(handle contract3.RequestHandle) contract3.RequestHandle {
360360
return func(request *http.Request) (response *http.Response, err error) {
361-
logger := logger.WithContext(request.Context())
362-
363-
// 此处请求前后日志根据 log 配置中的 level 判断是否打印
364-
request2.LogRequest(logger, request)
365-
361+
defer func() {
362+
config := (accessToken.App).GetConfig()
363+
debug := config.GetBool("http_debug", false)
364+
if debug {
365+
logger := logger.WithContext(request.Context())
366+
// 此处请求前后日志根据 log 配置中的 level 判断是否打印
367+
request2.LogRequest(logger, request)
368+
if response != nil {
369+
response2.LogResponse(logger, response)
370+
}
371+
}
372+
}()
366373
response, err = handle(request)
367374
if err != nil {
368375
return response, err
369376
}
370-
371-
response2.LogResponse(logger, response)
372-
373377
return
374378
}
375379
}
376380
}
377381
}
378-
379382
func (accessToken *AccessToken) OverrideGetEndpoint() {
380383
accessToken.GetEndpoint = func() (string, error) {
381384
if accessToken.EndpointToGetToken == "" {

src/kernel/baseClient.go

Lines changed: 12 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -448,17 +448,22 @@ func (client *BaseClient) OverrideGetMiddlewareOfLog() {
448448
client.GetMiddlewareOfLog = func(logger contract2.LoggerInterface) contract.RequestMiddleware {
449449
return func(handle contract.RequestHandle) contract.RequestHandle {
450450
return func(request *http.Request) (response *http.Response, err error) {
451-
logger = logger.WithContext(request.Context())
452-
453-
// 此处请求前后日志根据 log 配置中的 level 判断是否打印
454-
request2.LogRequest(logger, request)
455-
451+
config := (*client.App).GetConfig()
452+
defer func() {
453+
debug := config.GetBool("http_debug", false)
454+
if debug {
455+
logger := logger.WithContext(request.Context())
456+
// 此处请求前后日志根据 log 配置中的 level 判断是否打印
457+
request2.LogRequest(logger, request)
458+
if response != nil {
459+
response2.LogResponse(logger, response)
460+
}
461+
}
462+
}()
456463
response, err = handle(request)
457464
if err != nil {
458465
return response, err
459466
}
460-
response2.LogResponse(logger, response)
461-
462467
return
463468
}
464469
}

src/kernel/request/request.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,5 +34,5 @@ func LogRequest(logger contract2.LoggerInterface, request *http.Request) {
3434
output.Write(body)
3535
}
3636

37-
logger.Info(output.String())
37+
logger.Debug(output.String())
3838
}

src/kernel/response/response.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,5 +13,5 @@ func LogResponse(logger contract2.LoggerInterface, response *http.Response) {
1313
output.Write([]byte("response content:"))
1414
dumpRes, _ := httputil.DumpResponse(response, true)
1515
output.Write(dumpRes)
16-
logger.Info(output.String())
16+
logger.Debug(output.String())
1717
}

0 commit comments

Comments
 (0)