zuul添加或修改請求參數
一、為什么要用到這個
在基于 springcloud 構建的微服務系統中,通常使用網關zuul來進行一些用戶驗證等過濾的操作,比如 用戶在 header 或者 url 參數中存放了 token ,網關層需要 用該 token 查出用戶 的 userId ,并存放于 request 中,以便后續微服務可以直接使用而避免再去用 token 查詢。
二、基礎知識
在 zuul 中最大的用法的除了路由之外,就是過濾器了,自定義過濾器需實現接口 ZuulFilter ,在 run() 方法中,可以用
1
2
|
RequestContext ctx = RequestContext.getCurrentContext(); HttpServletRequest request = ctx.getRequest(); |
獲取到 request,但是在 request 中只有 getParameter() 而沒有 setParameter() 方法,所以直接修改 url 參數不可行,另外在 reqeust 中雖然可以使用 setAttribute() ,但是可能由于作用域的不同,在這里設置的 attribute 在后續的微服務中是獲取不到的,因此必須考慮另外的方式。
三、具體做法
最后確定的可行的方法是,用
1
|
ctx.setRequest( new HttpServletRequestWrapper(request) {}) |
的方式,重新構造上下文中的 request ,代碼如下:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
|
// 例如在請求參數中添加 userId try { InputStream in = ctx.getRequest().getInputStream(); String body = StreamUtils.copyToString(in, Charset.forName( "UTF-8" )); if (StringUtils.isBlank(body)){ body = "{}" ; } JSONObject jsonObject = JSON.parseObject(body); jsonObject.put( "userId" , 666 ); String newBody = jsonObject.toString(); final byte [] reqBodyBytes = newBody.getBytes(); ctx.setRequest( new HttpServletRequestWrapper(request){ @Override public ServletInputStream getInputStream() throws IOException { return new ServletInputStreamWrapper(reqBodyBytes); } @Override public int getContentLength() { return reqBodyBytes.length; } @Override public long getContentLengthLong() { return reqBodyBytes.length; } }); } catch (IOException e) { e.printStackTrace(); } |
思路就是,獲取請求的輸入流,并重寫,即重寫json參數。
在后續的微服務的 controller 中,通過下面的方式獲取通過zuul添加或修改的請求參數。
1
2
3
4
5
6
|
InputStream in = request().getInputStream(); String body = StreamUtils.copyToString(in, Charset.forName( "UTF-8" )); if (StringUtils.isNotBlank(body)){ JSONObject jsonObject = JSON.parseObject(body); Object userId = jsonObject.get( "userId" ); } |
zuul修改請求url
除了修改請求參數、設置響應header,響應body外,還有一種需求就是url重新,或者是修改url,這里簡述一下怎么在zuul修改url。
轉發配置
1
2
3
4
5
6
7
8
9
10
|
demo: ribbon: NIWSServerListClassName: com.netflix.loadbalancer.ConfigurationBasedServerList listOfServers: 192.168.99.100,192.168.99.101 zuul: routes: demo: path: /demo/** stripPrefix: true serviceId: demo |
filter配置
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
|
@Component public class UrlPathFilter extends ZuulFilter{ @Override public String filterType() { return FilterConstants.PRE_TYPE; } @Override public int filterOrder() { return FilterConstants.PRE_DECORATION_FILTER_ORDER + 1 ; } @Override public boolean shouldFilter() { final String serviceId = (String) RequestContext.getCurrentContext().get( "proxy" ); return "demo" .equals(serviceId); } @Override public Object run() { RequestContext context = RequestContext.getCurrentContext(); Object originalRequestPath = context.get(FilterConstants.REQUEST_URI_KEY); //http://localhost:10000/demo/list/data //-->/api/prefix/list/data String modifiedRequestPath = "/api/prefix" + originalRequestPath; context.put(FilterConstants.REQUEST_URI_KEY, modifiedRequestPath); return null ; } } |
這樣就大功告成了!
以上為個人經驗,希望能給大家一個參考,也希望大家多多支持服務器之家。
原文鏈接:https://blog.csdn.net/kysmkj/article/details/79092781