org.smartboot.http.client.HttpResponseProtocol Maven / Gradle / Ivy
/*******************************************************************************
* Copyright (c) 2017-2021, org.smartboot. All rights reserved.
* project name: smart-http
* file name: HttpResponseProtocol.java
* Date: 2021-02-04
* Author: sandao ([email protected])
******************************************************************************/
package org.smartboot.http.client;
import org.smartboot.http.client.decode.HeaderDecoder;
import org.smartboot.http.client.decode.HttpProtocolDecoder;
import org.smartboot.socket.Protocol;
import org.smartboot.socket.transport.AioSession;
import java.nio.ByteBuffer;
/**
* @author 三刀（[email protected]）
* @version V1.0 , 2021/2/2
*/
final class HttpResponseProtocol implements Protocol {
public static HttpResponseProtocol INSTANCE = new HttpResponseProtocol();
private final HttpProtocolDecoder httpMethodDecoder = new HttpProtocolDecoder();
@Override
public AbstractResponse decode(ByteBuffer buffer, AioSession session) {
ResponseAttachment attachment = session.getAttachment();
HeaderDecoder decodeChain = attachment.getDecoder();
if (decodeChain == null) {
decodeChain = httpMethodDecoder;
}
AbstractResponse response = attachment.getResponse();
// 数据还未就绪，继续读
if (decodeChain == HeaderDecoder.BODY_CONTINUE_DECODER) {
attachment.setDecoder(HeaderDecoder.BODY_READY_DECODER);
return null;
} else if (decodeChain == HeaderDecoder.BODY_READY_DECODER) {
return response;
}
decodeChain = decodeChain.decode(buffer, session, response);
attachment.setDecoder(decodeChain);
// 响应式流
if (decodeChain == HeaderDecoder.BODY_READY_DECODER) {
return response;
}
if (buffer.remaining() == buffer.capacity()) {
throw new RuntimeException("buffer is too small when decode " + decodeChain.getClass().getName() + " ," + response);
}
return null;
}
}