1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16 package io.netty.example.spdy.client;
17
18 import io.netty.channel.ChannelHandlerContext;
19 import io.netty.channel.ChannelOutboundHandlerAdapter;
20 import io.netty.channel.ChannelPromise;
21 import io.netty.handler.codec.http.HttpMessage;
22 import io.netty.handler.codec.spdy.SpdyHttpHeaders;
23 import io.netty.handler.codec.spdy.SpdyHttpHeaders.Names;
24
25
26
27
28 public class SpdyClientStreamIdHandler extends ChannelOutboundHandlerAdapter {
29
30 private int currentStreamId = 1;
31
32 public boolean acceptOutboundMessage(Object msg) {
33 return msg instanceof HttpMessage;
34 }
35
36 @Override
37 public void write(ChannelHandlerContext ctx, Object msg, ChannelPromise promise) {
38 if (acceptOutboundMessage(msg)) {
39 HttpMessage httpMsg = (HttpMessage) msg;
40 if (!httpMsg.headers().contains(SpdyHttpHeaders.Names.STREAM_ID)) {
41 httpMsg.headers().setInt(Names.STREAM_ID, currentStreamId);
42
43 currentStreamId += 2;
44 }
45 }
46 ctx.write(msg, promise);
47 }
48 }