首页 > 编程语言 >Tomcat源码分析使用NIO接收HTTP请求(五)----响应请求

Tomcat源码分析使用NIO接收HTTP请求(五)----响应请求

时间:2022-12-10 14:22:34浏览次数:56  
标签:HTTP 请求 bb ByteChunk convertToBytes 源码 headerBuffer put byte

本章的主要目标是为请求添加响应
第一步:新建Http11OutputBuffer类
public class Http11OutputBuffer {
    protected final ByteBuffer headerBuffer;

    public Http11OutputBuffer () {
        headerBuffer = ByteBuffer.allocate(2048);
    }

    public ByteBuffer getHeaderBuffer() {
        sendStatus();
        byte[] body = ByteChunk.convertToBytes("<html><h1>Hello World!</h1></html>");
        String format = "EEE,dd MMM yyyy HH:mm:ss z";
        SimpleDateFormat sdf = new SimpleDateFormat(format, Locale.US);
        headerBuffer.put(ByteChunk.convertToBytes("Content-Type: text/html"));
        headerBuffer.put((byte) '\r').put((byte) '\n');
        headerBuffer.put(ByteChunk.convertToBytes("Content-Length:" + body.length));
        headerBuffer.put((byte) '\r').put((byte) '\n');
        headerBuffer.put(ByteChunk.convertToBytes("Date:" + sdf.format(new Date())));
        headerBuffer.put((byte) '\r').put((byte) '\n');
        headerBuffer.put(ByteChunk.convertToBytes("Server:Apache/1.1"));
        headerBuffer.put((byte) '\r').put((byte) '\n');
        headerBuffer.put(ByteChunk.convertToBytes("Connection:keep-alive"));
        headerBuffer.put((byte) '\r').put((byte) '\n');
        headerBuffer.put((byte) '\r').put((byte) '\n');
        headerBuffer.put(body);
        headerBuffer.flip();
        byte[] bb = Arrays.copyOfRange(headerBuffer.array(), 0, headerBuffer.limit());
        ByteBuffer result =ByteBuffer.allocate(bb.length);
        result.put(bb);
        return result;
    }

    public void sendStatus() {
        write(ByteChunk.convertToBytes("HTTP/1.1"));
        headerBuffer.put((byte) ' ');

        write(ByteChunk.convertToBytes("200"));

        headerBuffer.put((byte) ' ');
        write(ByteChunk.convertToBytes("OK"));


        headerBuffer.put((byte) '\r').put((byte) '\n');
    }


    public void write(byte[] b) {
        headerBuffer.put(b);
    }
}
第二步:在ByteChunk类中添加convertToBytes方法
public static final byte[] convertToBytes(String value) {
    byte[] result = new byte[value.length()];
    for (int i = 0; i < value.length(); i++) {
        result[i] = (byte) value.charAt(i);
    }
    return result;
}
第三步:在Poller中新建cancelledKey方法。并在processKey方法中进行调用
public void cancelledKey(SelectionKey key) {
    NioSocketWrapper ka = null;
    try {
        ka = (NioSocketWrapper) key.attachment();
        if (ka != null) {
            System.out.println("************************************************");
            NioChannel nioChannel = ka.getSocket();
            Http11OutputBuffer ho = new Http11OutputBuffer();
            ByteBuffer bb = ho.getHeaderBuffer();
            System.out.println("=====" + bb.capacity());
            System.out.println(new String(bb.array(), StandardCharsets.UTF_8));
            int cout=0;
            bb.flip();
            while(bb.hasRemaining()) {
                cout = nioChannel.write(bb);
                Thread.sleep(2000);
                System.out.println("等待。。。。。。" + cout);
            }
            nioChannel.close();
        }
    } catch (Exception e) {

    }
}

 

 第四步:修改NioChannel的write方法

@Override
public int write(ByteBuffer src) throws IOException {
     return sc.write(src);
}
使用浏览器访问http://localhost:8000/结果如下

 

 

结束!!!

标签:HTTP,请求,bb,ByteChunk,convertToBytes,源码,headerBuffer,put,byte
From: https://www.cnblogs.com/yishi-san/p/16971524.html

相关文章