andreacavalli/netty5
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[#691] Correctly handle SpdyRstStreamFrame

Browse Source
This commit is contained in:
Norman Maurer 2012-10-30 08:48:09 +01:00
parent e726ff63d9
commit 4696fcde1f
1 changed files with 14 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
codec-http/src/main/java/io/netty/handler/codec/spdy/SpdyHttpResponseStreamIdHandler.java
View File

@ -20,6 +20,7 @@ import java.util.Queue;
import io.netty.channel.ChannelHandlerContext; import io.netty.channel.ChannelHandlerContext;
import io.netty.handler.codec.MessageToMessageCodec; import io.netty.handler.codec.MessageToMessageCodec;
import io.netty.handler.codec.http.HttpMessage;
import io.netty.handler.codec.http.HttpRequest; import io.netty.handler.codec.http.HttpRequest;
import io.netty.handler.codec.http.HttpResponse; import io.netty.handler.codec.http.HttpResponse;
@ -29,22 +30,27 @@ import io.netty.handler.codec.http.HttpResponse;
* for HTTP. * for HTTP.
*/ */
public class SpdyHttpResponseStreamIdHandler extends public class SpdyHttpResponseStreamIdHandler extends
MessageToMessageCodec<HttpRequest, HttpRequest, HttpResponse, HttpResponse> { MessageToMessageCodec<HttpRequest, HttpRequest, Object, Object> {
private static final Integer NO_ID = -1; private static final Integer NO_ID = -1;
private final Queue<Integer> ids = new LinkedList<Integer>(); private final Queue<Integer> ids = new LinkedList<Integer>();
public SpdyHttpResponseStreamIdHandler() { public SpdyHttpResponseStreamIdHandler() {
super(new Class<?>[] { HttpRequest.class }, new Class<?>[] { HttpResponse.class }); super(new Class<?>[] { HttpRequest.class, SpdyRstStreamFrame.class }, new Class<?>[] { HttpResponse.class });
} }
@Override @Override
public HttpResponse encode(ChannelHandlerContext ctx, HttpResponse msg) throws Exception { public Object encode(ChannelHandlerContext ctx, Object msg) throws Exception {
boolean contains = msg.containsHeader(SpdyHttpHeaders.Names.STREAM_ID); if (msg instanceof HttpMessage) {
boolean contains = ((HttpMessage)msg).containsHeader(SpdyHttpHeaders.Names.STREAM_ID);
if (!contains) { if (!contains) {
ids.add(NO_ID); ids.add(NO_ID);
} else { } else {
ids.add(SpdyHttpHeaders.getStreamId(msg)); ids.add(SpdyHttpHeaders.getStreamId((HttpMessage) msg));
} }
} else if (msg instanceof SpdyRstStreamFrame) {
ids.remove(((SpdyRstStreamFrame)msg).getStreamId());
}
return msg; return msg;
} }