简单JAVA多线程WEB服务器

  • 一个简单的JAVA多线程WEB服务器,有时间的话我会慢慢更新的。
  • 结构很简单,HttpServer serverThread Request Response Util
  • 直接上源码,有注释。

HttpServer.java

import java.net.Socket;
import java.net.ServerSocket;
import java.net.InetAddress;
import java.io.IOException;
import java.io.File;

public class HttpServer {
  /*
   * WEB_ROOT是HTML和其它文件存放的目录. 这里的WEB_ROOT为工作目录下的webroot目录
   */
  public static final String WEB_ROOT = System.getProperty("user.dir") + File.separator + "webroot";
  public static final String SERVERIP = "127.0.0.1";
  public static final int PORT = 8088;

  // 关闭服务命令
  public static final String SHUTDOWN_COMMAND = "/SHUTDOWN";
  public static boolean closeServer;

  public static void main(String[] args) {
    HttpServer server = new HttpServer();
    //等待连接请求
//    Util.scanPort();
    server.swait();
  }

  public void swait() {
    ServerSocket serverSocket = null;
    Thread receiveThread = null;
    System.out.println("WEB_ROOT:"+ WEB_ROOT + "port:" + PORT);
    try {
      //服务器套接字对象
      serverSocket = new ServerSocket(PORT, 10, InetAddress.getByName(SERVERIP));
    } catch (IOException e) {
      e.printStackTrace();
      System.exit(1);
    }
    // 循环等待一个请求
    while (true) {
      Socket socket = null;
      try {
        //等待连接,连接成功后,返回一个Socket对象
        socket = serverSocket.accept();
        receiveThread =new Thread(new serverThread(socket));
        receiveThread.start();
        // 检查是否是关闭服务命令
        if (closeServer) {
          break;
        }
      } catch (Exception e) {
        e.printStackTrace();
        continue;
      }
    }
  }
}

serverThread

package example;

import java.io.*;
import java.net.Socket;

public class serverThread implements Runnable {
    InputStream input;
    OutputStream output;
    Socket clientRequest;

    public serverThread(Socket socket) {
        this.clientRequest = socket;
        try{
            input = clientRequest.getInputStream() ;
            output = clientRequest.getOutputStream() ;
        }catch (IOException e){
            e.printStackTrace();
        }
    }

    @Override
    public void run() {
        // 创建Request对象并解析
        Request request = new Request(input);
        request.parse();
        if (request.getUri().equals(HttpServer.SHUTDOWN_COMMAND)) {
            HttpServer.closeServer = true;//接受关闭服务器指令
            try {
                clientRequest.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
        if (HttpServer.closeServer !=true){
            // 创建 Response 对象
            Response response = new Response(output);
            response.setRequest(request);
            try {
                response.sendStaticResource();
            } catch (IOException e) {
                e.printStackTrace();
            }
            try{
                // 关闭 socket 对象
                clientRequest.close();
            }catch (IOException e){
                System.out.println(e.getMessage());
            }
        }

    }

}

Request

package example;

import java.io.InputStream;
import java.io.IOException;

public class Request {
    private InputStream input;
    private String uri;

    public Request(InputStream input) {
        this.input = input;
    }

    //从InputStream中读取request信息,并从request中获取uri值
    public void parse() {
        StringBuffer request = new StringBuffer(2048);
        int i;
        byte[] buffer = new byte[2048];
        try {
            i = input.read(buffer);
        } catch (IOException e) {
            e.printStackTrace();
            i = -1;
        }
        for (int j = 0; j < i; j++) {
            request.append((char) buffer[j]);
        }
        System.out.print(request.toString());
        uri = parseUri(request.toString());
    }
    /*
     *
     * requestString形式如下:
     * GET /index.html HTTP/1.1
     * Host: localhost:8080
     * Connection: keep-alive
     * Cache-Control: max-age=0
     * ...
     * 该函数目的就是为了获取/index.html字符串
     */
    private String parseUri(String requestString) {
        int index1, index2;
        index1 = requestString.indexOf(' ');
        if (index1 != -1) {
            index2 = requestString.indexOf(' ', index1 + 1);
            if (index2 > index1)
                return requestString.substring(index1 + 1, index2);
        }
        return null;
    }

    private String parseMethod(String requestString) {//GET、POST、HEAD、PUT、DELETE、TRACE、CONNECT、OPTIONS
        int index1 = requestString.indexOf(' ');
        if (index1 != -1) {
                return requestString.substring(0, index1);
        }
        return null;
    }

    public String getUri() {
        return uri;
    }

}

Response

package example;

import java.io.OutputStream;
import java.io.IOException;
import java.io.FileInputStream;
import java.io.File;

/*
  HTTP Response = Status-Line
    *(( general-header | response-header | entity-header ) CRLF)
    CRLF
    [ message-body ]
    Status-Line = HTTP-Version SP Status-Code SP Reason-Phrase CRLF
*/

public class Response {

    private static final int BUFFER_SIZE = 1024;
    Request request;
    OutputStream output;
    int lenghtStream;

    public Response(OutputStream output) {
        this.output = output;
    }

    public void setRequest(Request request) {
        this.request = request;
    }

    public void sendStaticResource() throws IOException {
        byte[] bytes = new byte[BUFFER_SIZE];
        FileInputStream fis = null;
        try {
//            将web文件写入到OutputStream字节流中
            File file = new File(HttpServer.WEB_ROOT, request.getUri());
            if (file.exists()) { //如果访问路径存在
                fis = new FileInputStream(file);
                lenghtStream = fis.available();//最大获取2GB,File的length()方法可获取大于2GB大小,或者用java.nio
                int ch = fis.read(bytes, 0, BUFFER_SIZE);
                if (ch != -1){//添加header
                    String headerMessage = "HTTP/1.1 200 OK\r\n" + "Content-Type: text/html; charset=UTF-8\r\n"
                            + "Content-Length: "+ lenghtStream +"\r\n" + "Connection: keep-alive"+"\r\n" + "\r\n";
                    byte[] newBytes = Util.byteSum(headerMessage,bytes);
                    output.write(newBytes, 0, newBytes.length);
                    ch = fis.read(bytes, 0, BUFFER_SIZE);
                    while (ch != -1) {
                        output.write(bytes, 0, ch);
                        ch = fis.read(bytes, 0, BUFFER_SIZE);
                    }
                }
            } else {
                // file not found
                String errorMessage = "HTTP/1.1 404 File Not Found\r\n" + "Content-Type: text/html\r\n"
                        + "Content-Length: 23\r\n" + "\r\n" + "

File Not Found

"; output.write(errorMessage.getBytes()); } } catch (Exception e) { // thrown if cannot instantiate a File object System.out.println(e.toString()); } finally { if (fis != null) fis.close(); } } }

Util

package example;

import java.io.IOException;
import java.net.ServerSocket;

public class Util {
    public static byte[] byteSum (byte[] a,byte[] b){
        byte[] c= new byte[a.length+b.length];
        System.arraycopy(a,0,c,0,a.length);
        System.arraycopy(b,0,c,a.length,b.length);
        return c;
    }

    public static byte[] byteSum (String a_string,byte[] b){
        byte[] a = a_string.getBytes();
        byte[] c= new byte[a.length+b.length];
        System.arraycopy(a,0,c,0,a.length);
        System.arraycopy(b,0,c,a.length,b.length);
        return c;
    }

    public static void scanPort (){
        for(int port=1;port<=65535;port++){
            try{
                ServerSocket serverSocket=new ServerSocket(port);
                serverSocket.close();   //及时关闭ServerSocket
            }catch(IOException e){
                System.out.println("端口"+port+" 已经被其他服务器进程占用");
            }
        }
    }
}

源码下载

你可能感兴趣的:(简单JAVA多线程WEB服务器)