- 一個簡單的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" + "<h1>File Not Found</h1>";
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+" 已經被其他服務器進程占用");
}
}
}
}
