Skip to content
项目
群组
代码片段
帮助
当前项目
正在载入...
登录 / 注册
切换导航面板
Y
yudao-cloud
项目
项目
详情
活动
周期分析
仓库
仓库
文件
提交
分支
标签
贡献者
图表
比较
统计图
议题
0
议题
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
CI / CD
CI / CD
流水线
作业
日程
统计图
Wiki
Wiki
代码片段
代码片段
成员
成员
折叠边栏
关闭边栏
活动
图像
聊天
创建新问题
作业
提交
问题看板
Open sidebar
hblj
yudao-cloud
Commits
f4c83c01
提交
f4c83c01
authored
6月 05, 2022
作者:
YunaiV
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
gateway:增加访问日志,待优化~
上级
f32e4386
全部展开
隐藏空白字符变更
内嵌
并排
正在显示
3 个修改的文件
包含
140 行增加
和
0 行删除
+140
-0
AccessLogFilter.java
...iocoder/yudao/gateway/filter/logging/AccessLogFilter.java
+0
-0
GatewayLog.java
...a/cn/iocoder/yudao/gateway/filter/logging/GatewayLog.java
+29
-0
CacheRequestBodyFilter.java
...oder/yudao/gateway/filter/web/CacheRequestBodyFilter.java
+111
-0
没有找到文件。
yudao-gateway/src/main/java/cn/iocoder/yudao/gateway/filter/logging/AccessLogFilter.java
0 → 100644
浏览文件 @
f4c83c01
差异被折叠。
点击展开。
yudao-gateway/src/main/java/cn/iocoder/yudao/gateway/filter/logging/GatewayLog.java
0 → 100644
浏览文件 @
f4c83c01
package
cn
.
iocoder
.
yudao
.
gateway
.
filter
.
logging
;
import
lombok.Data
;
import
java.util.Date
;
@Data
public
class
GatewayLog
{
/**访问实例*/
private
String
targetServer
;
/**请求路径*/
private
String
requestPath
;
/**请求方法*/
private
String
requestMethod
;
/**协议 */
private
String
schema
;
/**请求体*/
private
String
requestBody
;
/**响应体*/
private
String
responseData
;
/**请求ip*/
private
String
ip
;
/**请求时间*/
private
Date
requestTime
;
/**响应时间*/
private
Date
responseTime
;
/**执行时间*/
private
long
executeTime
;
}
yudao-gateway/src/main/java/cn/iocoder/yudao/gateway/filter/web/CacheRequestBodyFilter.java
0 → 100644
浏览文件 @
f4c83c01
package
cn
.
iocoder
.
yudao
.
gateway
.
filter
.
web
;
import
cn.hutool.core.util.ReflectUtil
;
import
org.springframework.cloud.gateway.filter.GatewayFilterChain
;
import
org.springframework.cloud.gateway.filter.GlobalFilter
;
import
org.springframework.cloud.gateway.filter.factory.rewrite.CachedBodyOutputMessage
;
import
org.springframework.cloud.gateway.support.BodyInserterContext
;
import
org.springframework.core.Ordered
;
import
org.springframework.core.io.buffer.DataBuffer
;
import
org.springframework.core.io.buffer.DataBufferUtils
;
import
org.springframework.http.HttpHeaders
;
import
org.springframework.http.codec.HttpMessageReader
;
import
org.springframework.http.server.reactive.ServerHttpRequest
;
import
org.springframework.http.server.reactive.ServerHttpRequestDecorator
;
import
org.springframework.stereotype.Component
;
import
org.springframework.web.reactive.function.BodyInserter
;
import
org.springframework.web.reactive.function.BodyInserters
;
import
org.springframework.web.reactive.function.server.HandlerStrategies
;
import
org.springframework.web.reactive.function.server.ServerRequest
;
import
org.springframework.web.server.ServerWebExchange
;
import
reactor.core.publisher.Flux
;
import
reactor.core.publisher.Mono
;
import
java.util.List
;
import
java.util.function.Function
;
/**
* 缓存 Request Body 和 的过滤器
*
* 小知识:Request Body 都是无法重复读取的,所以需要实现一个缓存的。
* 从功能上,它类似 yudao-spring-boot-starter-web 的 CacheRequestBodyFilter 过滤器
*
* 实现基本是拷贝 {@link org.springframework.cloud.gateway.filter.factory.rewrite.ModifyRequestBodyGatewayFilterFactory} 类
*
* @author 芋道源码
*/
@Component
public
class
CacheRequestBodyFilter
implements
GlobalFilter
,
Ordered
{
private
final
List
<
HttpMessageReader
<?>>
messageReaders
=
HandlerStrategies
.
withDefaults
().
messageReaders
();
@Override
public
Mono
<
Void
>
filter
(
ServerWebExchange
exchange
,
GatewayFilterChain
chain
)
{
ServerRequest
serverRequest
=
ServerRequest
.
create
(
exchange
,
messageReaders
);
// TODO: flux or mono
Mono
<
String
>
modifiedBody
=
serverRequest
.
bodyToMono
(
String
.
class
);
// .flatMap()
// .switchIfEmpty(Mono.just(""));
BodyInserter
bodyInserter
=
BodyInserters
.
fromPublisher
(
modifiedBody
,
String
.
class
);
HttpHeaders
headers
=
new
HttpHeaders
();
headers
.
putAll
(
exchange
.
getRequest
().
getHeaders
());
// the new content type will be computed by bodyInserter
// and then set in the request decorator
headers
.
remove
(
HttpHeaders
.
CONTENT_LENGTH
);
CachedBodyOutputMessage
outputMessage
=
new
CachedBodyOutputMessage
(
exchange
,
headers
);
return
bodyInserter
.
insert
(
outputMessage
,
new
BodyInserterContext
())
// .log("modify_request", Level.INFO)
.
then
(
Mono
.
defer
(()
->
{
ServerHttpRequest
decorator
=
decorate
(
exchange
,
headers
,
outputMessage
);
return
chain
.
filter
(
exchange
.
mutate
().
request
(
decorator
).
build
());
})).
onErrorResume
((
Function
<
Throwable
,
Mono
<
Void
>>)
throwable
->
release
(
exchange
,
outputMessage
,
throwable
));
}
@Override
public
int
getOrder
()
{
// 必须小于等于 -2,否则无法获取响应结果. 因为 WebClientWriteResponseFilter 和 NettyWriteResponseFilter 是 -1,优先级要高于它们
return
-
2
;
}
protected
Mono
<
Void
>
release
(
ServerWebExchange
exchange
,
CachedBodyOutputMessage
outputMessage
,
Throwable
throwable
)
{
// add by 芋道源码:由于 CachedBodyOutputMessage 的 isCached 非 public 方法,所以只能反射调用
if
((
boolean
)
ReflectUtil
.
getFieldValue
(
outputMessage
,
"cached"
))
{
return
outputMessage
.
getBody
().
map
(
DataBufferUtils:
:
release
).
then
(
Mono
.
error
(
throwable
));
}
return
Mono
.
error
(
throwable
);
}
ServerHttpRequestDecorator
decorate
(
ServerWebExchange
exchange
,
HttpHeaders
headers
,
CachedBodyOutputMessage
outputMessage
)
{
return
new
ServerHttpRequestDecorator
(
exchange
.
getRequest
())
{
@Override
public
HttpHeaders
getHeaders
()
{
long
contentLength
=
headers
.
getContentLength
();
HttpHeaders
httpHeaders
=
new
HttpHeaders
();
httpHeaders
.
putAll
(
headers
);
if
(
contentLength
>
0
)
{
httpHeaders
.
setContentLength
(
contentLength
);
}
else
{
// TODO: this causes a 'HTTP/1.1 411 Length Required' // on
// httpbin.org
httpHeaders
.
set
(
HttpHeaders
.
TRANSFER_ENCODING
,
"chunked"
);
}
return
httpHeaders
;
}
@Override
public
Flux
<
DataBuffer
>
getBody
()
{
return
outputMessage
.
getBody
();
}
};
}
}
编写
预览
Markdown
格式
0%
重试
或
添加新文件
添加附件
取消
您添加了
0
人
到此讨论。请谨慎行事。
请先完成此评论的编辑!
取消
请
注册
或者
登录
后发表评论