Time: 2024-04-25 星期四 19:09:01
Author: Jackasher

手写服务器主流程

有个不错的思想 就是,把业务交给用户来做,一个Servlet接口,用户来写,然后固定方法名,这样服务器开发者就可可以反射创建这个类调用使用者的业务代码

package org.example;

import javax2.Servlet;
import org.dom4j.DocumentException;

import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.Map;

public class Main {
    public static void main(String[] args) throws IOException, DocumentException, ClassNotFoundException, InstantiationException, IllegalAccessException {
        ServerSocket serverSocket = new ServerSocket(8080);
        System.out.println("Server started on port 8080");

        while (true) {
            Socket socket = serverSocket.accept();
            System.out.println("Connection established");

            BufferedReader reader = new BufferedReader(new InputStreamReader(socket.getInputStream()));
            PrintWriter writer = new PrintWriter(socket.getOutputStream());

            // 读取HTTP请求的第一行（请求行）
            String requestLine = reader.readLine();
            System.out.println("Received request: " + requestLine);

            // 解析请求URI
            String[] parts = requestLine.split(" ");

            //  oa/login/jack.html
            String requestURI = parts[1];

            // 如果请求的是favicon.ico文件，则直接返回空响应
            if (requestURI.equals("/favicon.ico")) {
                writer.write("HTTP/1.1 200 OK\r\n");
                writer.write("\r\n");
                writer.flush();
                writer.close();
                reader.close();
                socket.close();
                System.out.println("Connection closed");
                continue; // 继续监听下一个连接
            }

            if (requestURI.endsWith(".html")) {
                // 加载资源文件
                InputStream inputStream = Main.class.getResourceAsStream(requestURI);
                BufferedReader fileReader = new BufferedReader(new InputStreamReader(inputStream));

                // 构建HTTP响应
                writer.write("HTTP/1.1 200 OK\r\n");
                writer.write("Content-Type: text/html\r\n");
                writer.write("\r\n"); // 空行，标志头部结束

                // 将HTML内容写入响应
                String line;
                while ((line = fileReader.readLine()) != null) {
                    writer.write(line + "\r\n");
                }

                // 刷新缓冲区，确保数据被发送
                writer.flush();

                // 关闭连接
                writer.close();
                reader.close();
                socket.close();
                System.out.println("Connection closed");
            }else {
                String webName = "/" +  requestURI.split("[/]")[1];

                Map<String, String> servletMap = WebParser.parser(webName);
                String urlPattern = "/" + requestURI.substring(1 + webName.length());
                String servletClass  = servletMap.get(urlPattern);
                Class<?> aClass = Class.forName(servletClass);
                Object o = aClass.newInstance();
                Servlet servlet = (Servlet) o;
                servlet.service();
            }


        }
    }
}