自己写的小型静态服务器

最近,由于计算机网络课程,自己动手写了一个静态服务器,感觉挺好玩的,慢慢了解到tomcat服务器的原理,而我这次做的只做了GET和PUT方法，废话不多说,下面上代码:

package com.xanxus.http; import java.io.BufferedReader; import java.io.File; import java.io.FileInputStream; import java.io.FileNotFoundException; import java.io.FileOutputStream; import java.io.IOException; import java.io.InputStreamReader; import java.io.OutputStream; import java.net.ServerSocket; import java.net.Socket; public class ThreadedServer { /** * @param args */ public static void main(String[] args) { // TODO Auto-generated method stub ServerSocket serverSocket = null; try { serverSocket = new ServerSocket(8000); while (true) { Socket socket = serverSocket.accept(); new Thread(new SocketRunnable(socket, args[0])).start();// 注意，这里用内部类不行 } } catch (IOException e) { // TODO Auto-generated catch block e.printStackTrace(); } } public static void putOutput(OutputStream outputStream, int stateCode, String fileType, String fileName, BufferedReader requestReader) throws IOException { String clientRequest = null; int contentLength = 0;// 请求实体长度 while ((clientRequest = requestReader.readLine()) != null) { // 获取实体长度 if (clientRequest.toLowerCase().startsWith("content-length")) { contentLength = Integer.parseInt(clientRequest .substring(clientRequest.indexOf(":") + 2)); } else if (clientRequest.equals("")) { // 空行，首部行结束 break; } } // 读取实体 File putFile = new File(fileName); FileOutputStream fos = new FileOutputStream(putFile); char[] requestEntity = new char[1024]; int count = 0; while (contentLength > 0 && (count = requestReader.read(requestEntity)) != -1) { fos.write(requestEntity.toString().getBytes()); contentLength -= count; } fos.close(); // 写响应报文 if (fileType != null) { // 输出首部行 StringBuilder stringBuilder = new StringBuilder(); switch (stateCode) { case 200: stringBuilder.append("HTTP/1.0 200 OK\r\n"); // mime type switch (fileType) { case "jpg": stringBuilder.append("Content-Type: image/jpeg" + "\r\n"); break; case "html": stringBuilder.append("Content-Type: text/html" + "\r\n"); break; case "gif": stringBuilder.append("Content-Type: image/gif" + "\r\n"); break; case "png": stringBuilder.append("Content-Type: image/png" + "\r\n"); break; default: stringBuilder.append("Content-Type: text/plain" + "\r\n"); break; } break; case 404: stringBuilder.append("HTTP/1.0 404 Not Found\r\n"); break; case 201: stringBuilder.append("HTTP/1.0 201 created\r\n"); break; } stringBuilder.append("Server: LFY/0.1\r\n"); stringBuilder.append("Connection: close\r\n"); stringBuilder.append("\r\n"); outputStream.write(stringBuilder.toString().getBytes()); outputStream.flush(); } outputStream.close(); } public static void getOutput(OutputStream outputStream, int stateCode, String fileType, String fileName) throws IOException { if (fileType != null) { // 输出首部行 StringBuilder stringBuilder = new StringBuilder(); switch (stateCode) { case 200: stringBuilder.append("HTTP/1.0 200 OK\r\n"); // mime type switch (fileType) { case "jpg": stringBuilder.append("Content-Type: image/jpeg" + "\r\n"); break; case "html": stringBuilder.append("Content-Type: text/html" + "\r\n"); break; case "gif": stringBuilder.append("Content-Type: image/gif" + "\r\n"); break; case "png": stringBuilder.append("Content-Type: image/png" + "\r\n"); break; case "avi": stringBuilder.append("Content-Type: video/x-msvideo" + "\r\n"); break; case "mp4": stringBuilder.append("Content-Type: video/mpeg" + "\r\n"); break; case "ogg": stringBuilder.append("Content-Type: audio/ogg" + "\r\n"); break; default: stringBuilder.append("Content-Type: text/plain" + "\r\n"); break; } break; case 404: stringBuilder.append("HTTP/1.0 404 Not Found\r\n"); break; case 403: stringBuilder.append("HTTP/1.0 403 Forbidden\r\n"); break; } stringBuilder.append("Server: LFY/0.1\r\n"); stringBuilder.append("Connection: close\r\n"); stringBuilder.append("\r\n"); outputStream.write(stringBuilder.toString().getBytes()); outputStream.flush(); // 文件存在才输出文件 if (stateCode == 200) { // 输出请求文件 int length = -1; byte[] fileContent = new byte[1024]; FileInputStream fis = new FileInputStream(fileName); // 一律用字节流，因为有可能请求二进制文件 while ((length = fis.read(fileContent)) != -1) { outputStream.write(fileContent, 0, length); } } else if (stateCode == 404) { // 输出404返回页面 String notFoundString = "<html><head><title>404</title></head><body><center><h1>Error:404 The requested URL was not found on this server.</center></body></html>"; outputStream.write(notFoundString.getBytes("UTF-8")); outputStream.flush(); } outputStream.flush(); } outputStream.close(); } } class SocketRunnable implements Runnable { Socket socket = null; String rootPath = null; public SocketRunnable(Socket socket, String rootPath) { // TODO Auto-generated constructor stub this.socket = socket; this.rootPath = rootPath; } @Override public void run() { // TODO Auto-generated method stub try { BufferedReader bufferedReader = new BufferedReader( new InputStreamReader(socket.getInputStream())); OutputStream responseStream = socket.getOutputStream(); // 客户端发送的请求字符串，方法，请求文件路径，文件类型，主机名 String clientRequest = null, method = null, filePath = null, fileType = null, hostName = null; Boolean isConnection = false; StringBuffer responseBuilder = new StringBuffer(), fileContentBuilder = new StringBuffer();// 响应报文,请求文件内容 int stateCode = 200;// 状态码 Boolean isPut = false, isClf = false; // 获取首部行 if ((clientRequest = bufferedReader.readLine()) != null) { System.out.println("From:" + socket.getInetAddress().getHostAddress() + "\r\n\t" + clientRequest); // 第一行就返回 if (clientRequest.startsWith("GET") || clientRequest.startsWith("PUT")) { method = clientRequest.substring(0, 3); // 第一个斜杠的位置 int slashPosition = clientRequest.indexOf("/"); filePath = rootPath + clientRequest.substring(slashPosition, clientRequest.indexOf(" ", slashPosition)) .replace("/", "\\");// 把路径分隔符‘/’替换为‘\’ // 默认请求文件 if (filePath.equals(rootPath + "\\")) { filePath = rootPath + "\\index.html"; } File requestFile = new File(filePath); // 判断请求文件是否存在 if (!requestFile.exists()) { stateCode = 404; } fileType = filePath .substring(filePath.lastIndexOf(".") + 1); // 判断请求文件类型，设置输出的mime type switch (fileType) { case "html": case "htm": fileType = "html"; break; case "jpg": case "jpeg": fileType = "jpg"; break; case "gif": fileType = "gif"; break; } if (method.equals("GET")) { ThreadedServer.getOutput(responseStream, stateCode, fileType, filePath); socket.close(); } else if (method.equals("PUT")) { ThreadedServer.putOutput(responseStream, stateCode, fileType, filePath, bufferedReader); socket.close(); } } else if (clientRequest.equals("\r\n") || clientRequest.equals("")) { } } } catch (FileNotFoundException e2) { // TODO Auto-generated catch block e2.printStackTrace(); } catch (IOException e) { // TODO Auto-generated catch block e.printStackTrace(); } } }

资源下载可以到http://download.csdn.net/detail/xanxus46/4745329

是否觉得很给力，以后有时间，一定会继续写动态的内容，如jsp，想一下都觉得兴奋。