简易版基于Java的Servlet容器实现

最新推荐文章于 2023-01-08 23:09:15 发布

zhangjinpeng66

最新推荐文章于 2023-01-08 23:09:15 发布

阅读量2k

点赞数

分类专栏： Tomcat Java 文章标签： servlet tomcat Catalina

本文链接：https://blog.csdn.net/zhangjinpeng66/article/details/14089945

版权

Java 同时被 2 个专栏收录

25 篇文章 0 订阅

订阅专栏

Tomcat

3 篇文章 0 订阅

订阅专栏

如上文简易版基于Java的Web服务器实现中该服务器可以处理静态页面，在上文基础上实现既可以处理静态页面和Servlet的简易服务器。

HttpServer1类

package com.vipshop.test.ext02;

import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.InetAddress;
import java.net.ServerSocket;
import java.net.Socket;
import java.net.UnknownHostException;

public class HttpServer1 {

	// shutdown command
	private static final String SHUTDOWN_COMMAND = "/SHUTDOWN";
	
	// the shutdown command received
	private boolean shutdown = false;
	
	public static void main(String[] args) {
		
		HttpServer1 server = new HttpServer1();
		server.await();
		
	}
	
	public void await() {
		
		ServerSocket serverSocket = null;
		int port = 8080;
		
		try {
			
			serverSocket = new ServerSocket(port, 1, InetAddress.getByName("127.0.0.1"));
			
		} catch (UnknownHostException e) {
			
			e.printStackTrace();
			
		} catch (IOException e) {

			e.printStackTrace();
			System.exit(1);
			
		}
		
		// Loop waiting for a request
		while(!shutdown) {
			
			Socket socket = null;
			InputStream input = null;
			OutputStream output = null;
			
			try {
				
				socket = serverSocket.accept();
				input = socket.getInputStream();
				output = socket.getOutputStream();
				
			} catch (IOException e1) {
				
				e1.printStackTrace();
				
			}
			
			// create Request object and parse
			Request request = new Request(input);
			request.parse();
			
			// create Response object
			Response response = new Response(output);
			response.setRequest(request);
			
			// check if this is a request for a servlet or a static resource
			// a request for a servlet begings with "/servlet"
			if (request.
					getUri().
					startsWith("/servlet/")) {
				
				ServletProcessor1 processor = new ServletProcessor1();
				processor.process(request, response);
				
			} else {
				
				StaticResourceProcessor processor = new StaticResourceProcessor();
				processor.process(request, response);
				
			}
			
			try {
				// Close the socket
				socket.close();
				
				// check if the previous URI is a shutdown command
				shutdown = request.getUri().equals(SHUTDOWN_COMMAND);
			} catch (IOException e) {
				
				e.printStackTrace();
				System.exit(1);
				
			}
			
		}
		
	}
	
}

处理静态资源的处理器StaticResourceProcessor

package com.vipshop.test.ext02;

import java.io.IOException;

public class StaticResourceProcessor {

	public void process(Request request, Response response) {
		
		try {
			
			response.sendStaticResource();
			
		} catch (IOException e) {

			e.printStackTrace();
			
		}
		
	}
	
}

Servlet的处理器ServletProcessor

package com.vipshop.test.ext02;

import java.io.IOException;

import javax.servlet.Servlet;
import javax.servlet.ServletException;
import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse;

public class ServletProcessor1 {

	public void process (Request request, Response response) {
		
		String uri = request.getUri();
		String servletName = uri.substring(uri.lastIndexOf("/") + 1);
				
		Object myClass = null;
		
		try {
			
			myClass = Class.forName("com.vipshop.test.ext02.servlet." + servletName).newInstance();
			// = loader.loadClass(servletName);
			
		} catch (ClassNotFoundException e) {

			e.printStackTrace();
			
		} catch (InstantiationException e) {
			
			e.printStackTrace();
			
		} catch (IllegalAccessException e) {

			e.printStackTrace();
			
		}
		
		Servlet servlet = null;
		
		try {
			
			servlet = (Servlet) myClass;
			System.out.println(servlet);
			servlet.service(
					(ServletRequest) request, 
					(ServletResponse) response);
			
		} catch (ServletException e) {

			e.printStackTrace();
			
		} catch (IOException e) {

			e.printStackTrace();
			
		}
		
	}
	
}

请求类Request

package com.vipshop.test.ext02;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.UnsupportedEncodingException;
import java.util.Enumeration;
import java.util.Locale;
import java.util.Map;

import javax.servlet.RequestDispatcher;
import javax.servlet.ServletInputStream;
import javax.servlet.ServletRequest;

public class Request implements ServletRequest {

	private InputStream input;
	
	private String uri;
	
	public Request(InputStream input) {
		
		this.input = input;
		
	}
	
	public String getUri() {
		
		return uri;
		
	}
	
	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;
		
	}
	
	public void parse() {
		
		// Read a set of characters from the socket
		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.println("----------------");
		System.out.print(request.toString());
		System.out.println("----------------");
		
		uri = parseUri(request.toString());
	}
	
	@Override
	public Object getAttribute(String name) {
		
		return null;
	}

	@Override
	public Enumeration getAttributeNames() {
		
		return null;
	}

	@Override
	public String getCharacterEncoding() {
		
		return null;
	}

	@Override
	public int getContentLength() {
		
		return 0;
	}

	@Override
	public String getContentType() {
		
		return null;
	}

	@Override
	public ServletInputStream getInputStream() throws IOException {
		
		return null;
	}

	@Override
	public String getLocalAddr() {
		
		return null;
	}

	@Override
	public Locale getLocale() {
		
		return null;
	}

	@Override
	public Enumeration getLocales() {
		
		return null;
	}

	@Override
	public String getLocalName() {
		
		return null;
	}

	@Override
	public int getLocalPort() {
		
		return 0;
	}

	@Override
	public String getParameter(String name) {
		
		return null;
	}

	@Override
	public Map getParameterMap() {
		
		return null;
	}

	@Override
	public Enumeration getParameterNames() {
		
		return null;
	}

	@Override
	public String[] getParameterValues(String name) {
		
		return null;
	}

	@Override
	public String getProtocol() {
		
		return null;
	}

	@Override
	public BufferedReader getReader() throws IOException {
		
		return null;
	}

	@Override
	public String getRealPath(String path) {
		
		return null;
	}

	@Override
	public String getRemoteAddr() {
		
		return null;
	}

	@Override
	public String getRemoteHost() {
		
		return null;
	}

	@Override
	public int getRemotePort() {
		
		return 0;
	}

	@Override
	public RequestDispatcher getRequestDispatcher(String path) {
		
		return null;
	}

	@Override
	public String getScheme() {
		
		return null;
	}

	@Override
	public String getServerName() {
		
		return null;
	}

	@Override
	public int getServerPort() {
		
		return 0;
	}

	@Override
	public boolean isSecure() {
		
		return false;
	}

	@Override
	public void removeAttribute(String name) {
		
		
	}

	@Override
	public void setAttribute(String name, Object o) {
		
		
	}

	@Override
	public void setCharacterEncoding(String env)
			throws UnsupportedEncodingException {
		
		
	}

}

响应类类Response

package com.vipshop.test.ext02;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.OutputStream;
import java.io.PrintWriter;
import java.util.Locale;

import javax.servlet.ServletOutputStream;
import javax.servlet.ServletResponse;

public class Response implements ServletResponse {

	private static final int BUFFER_SIZE = 1024;
	private Request request;
	private OutputStream output;
	private PrintWriter writer;
	
	public Response(OutputStream output) {
		
		this.output = output;
		
	}
	
	public void setRequest(Request request) {
		
		this.request = request;
		
	}
	
	/**
	 * This method is used to save a static page
	 * @throws IOException
	 */
	public void sendStaticResource() throws IOException {
		
		byte[] bytes = new byte[BUFFER_SIZE];
		FileInputStream fis = null;
		
		try {
			
			File file = new File(Constants.WEB_ROOT, request.getUri());
			fis = new FileInputStream(file);
			
			int ch = fis.read(bytes, 0, BUFFER_SIZE);
			
			while(ch != -1) {
				
				output.write(bytes, 0, BUFFER_SIZE);
				ch = fis.read(bytes, 0, BUFFER_SIZE);
				
			}
			
		} catch (FileNotFoundException e) {
			
			String errorMessage = "HTTP/1.1 404 File Not Found\r\n"
				+ "Content-Type: text/html\r\n"
				+ "\r\n" + "<h1>File Not Found</h1>";
			output.write(errorMessage.getBytes());
			
		} finally {
			
			if (fis != null) {
				
				fis.close();
				this.flushBuffer();
				this.getOutputStream().close();
				this.getWriter().close();
			}
			
		}
		
	}
	
	@Override
	public void flushBuffer() throws IOException {
				
	}

	@Override
	public int getBufferSize() {
		
		return 0;
	}

	@Override
	public String getCharacterEncoding() {
		
		return null;
	}

	@Override
	public String getContentType() {
		
		return null;
	}

	@Override
	public Locale getLocale() {
		
		return null;
	}

	@Override
	public ServletOutputStream getOutputStream() throws IOException {
		
		return null;
	}

	@Override
	public PrintWriter getWriter() throws IOException {
		
		return new PrintWriter(this.output, true);
	}

	@Override
	public boolean isCommitted() {
		
		return false;
	}

	@Override
	public void reset() {
		
		
	}

	@Override
	public void resetBuffer() {
		
		
	}

	@Override
	public void setBufferSize(int size) {
		
		
	}

	@Override
	public void setCharacterEncoding(String charset) {
		
		
	}

	@Override
	public void setContentLength(int len) {
		
		
	}

	@Override
	public void setContentType(String type) {
		
		
	}

	@Override
	public void setLocale(Locale loc) {
		
		
	}

}

上述简易版可以处理Servlet的容器已经实现了考虑到HttpServer1调度StaticResourceProcessor类和ServletProcessor1类，Processor直接调用Response和Request，耦合度较高。

在此引入装饰器模式

RequestFacade类

package com.vipshop.test.ext02;

import java.io.IOException;
import java.io.PrintWriter;
import java.util.Locale;

import javax.servlet.ServletOutputStream;
import javax.servlet.ServletResponse;

public class ResponseFacade implements ServletResponse {

	private ServletResponse response;
	
	public ResponseFacade(Response response) {
		
		this.response = response;
		
	}

	@Override
	public void flushBuffer() throws IOException {
		
		response.flushBuffer();
		
	}

	@Override
	public int getBufferSize() {
		
		return response.getBufferSize();
	}

	@Override
	public String getCharacterEncoding() {
		
		return response.getCharacterEncoding();
		
	}

	@Override
	public String getContentType() {
		
		return response.getContentType();
		
	}

	@Override
	public Locale getLocale() {
		
		return response.getLocale();
	}

	@Override
	public ServletOutputStream getOutputStream() throws IOException {
		
		return response.getOutputStream();
	}

	@Override
	public PrintWriter getWriter() throws IOException {
		
		return response.getWriter();
	}

	@Override
	public boolean isCommitted() {
		
		return response.isCommitted();
	}

	@Override
	public void reset() {
		
		response.reset();
		
	}

	@Override
	public void resetBuffer() {
		
		response.resetBuffer();
		
	}

	@Override
	public void setBufferSize(int size) {
		
		response.setBufferSize(size);
		
	}

	@Override
	public void setCharacterEncoding(String charset) {
		
		response.setCharacterEncoding(charset);
		
	}

	@Override
	public void setContentLength(int len) {
		
		response.setContentLength(len);
		
	}

	@Override
	public void setContentType(String type) {
		
		response.setContentType(type);
		
	}

	@Override
	public void setLocale(Locale loc) {
		
		response.setLocale(loc);
		
	}
	
}

RequestFacade类

package com.vipshop.test.ext02;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.util.Enumeration;
import java.util.Locale;
import java.util.Map;

import javax.servlet.RequestDispatcher;
import javax.servlet.ServletInputStream;
import javax.servlet.ServletRequest;

public class RequestFacade implements ServletRequest {

	private ServletRequest request = null;
	
	public RequestFacade(Request request) {
		
		this.request = request;
		
	}
	
	@Override
	public Object getAttribute(String name) {
		
		return request.getAttribute(name);
	}

	@Override
	public Enumeration getAttributeNames() {
		
		return request.getAttributeNames();
	}

	@Override
	public String getCharacterEncoding() {
		
		return request.getCharacterEncoding();
	}

	@Override
	public int getContentLength() {
		
		return request.getContentLength();
	}

	@Override
	public String getContentType() {
		
		return request.getCharacterEncoding();
	}

	@Override
	public ServletInputStream getInputStream() throws IOException {
		
		return request.getInputStream();
	}

	@Override
	public String getLocalAddr() {
		
		return request.getLocalAddr();
	}

	@Override
	public Locale getLocale() {
		
		return request.getLocale();
	}

	@Override
	public Enumeration getLocales() {
		
		return request.getLocales();
	}

	@Override
	public String getLocalName() {
		
		return request.getLocalName();
	}

	@Override
	public int getLocalPort() {
		
		return request.getLocalPort();
	}

	@Override
	public String getParameter(String name) {
		
		return request.getParameter(name);
	}

	@Override
	public Map getParameterMap() {
		
		return request.getParameterMap();
	}

	@Override
	public Enumeration getParameterNames() {
		
		return request.getAttributeNames();
	}

	@Override
	public String[] getParameterValues(String name) {
		
		return request.getParameterValues(name);
	}

	@Override
	public String getProtocol() {
		
		return request.getProtocol();
	}

	@Override
	public BufferedReader getReader() throws IOException {
		
		return request.getReader();
	}

	@Override
	public String getRealPath(String path) {
		
		return request.getRealPath(path);
	}

	@Override
	public String getRemoteAddr() {
		
		return request.getRemoteAddr();
	}

	@Override
	public String getRemoteHost() {
		
		return request.getRemoteHost();
	}

	@Override
	public int getRemotePort() {
		
		return request.getRemotePort();
	}

	@Override
	public RequestDispatcher getRequestDispatcher(String path) {
		
		return request.getRequestDispatcher(path);
	}

	@Override
	public String getScheme() {
		
		return request.getScheme();
	}

	@Override
	public String getServerName() {
		
		return request.getServerName();
	}

	@Override
	public int getServerPort() {
		
		return request.getServerPort();
	}

	@Override
	public boolean isSecure() {
		
		return request.isSecure();
	}

	@Override
	public void removeAttribute(String name) {
		
		request.removeAttribute(name);
		
	}

	@Override
	public void setAttribute(String name, Object o) {
		
		request.setAttribute(name, o);
		
	}

	@Override
	public void setCharacterEncoding(String env)
			throws UnsupportedEncodingException {
		
		request.setCharacterEncoding(env);
		
	}

}

Servlet处理器只和RequestFacade类和ResponseFacade类通话

package com.vipshop.test.ext02;

import java.io.IOException;

import javax.servlet.Servlet;
import javax.servlet.ServletException;
import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse;

public class ServletProcessor2 {

	public void process (Request request, Response response) {
		
		String uri = request.getUri();
		String servletName = uri.substring(uri.lastIndexOf("/") + 1);
				
		Object myClass = null;
		
		try {
			
			myClass = Class.forName("com.vipshop.test.ext02.servlet." + servletName).newInstance();
			// = loader.loadClass(servletName);
			
		} catch (ClassNotFoundException e) {

			e.printStackTrace();
			
		} catch (InstantiationException e) {
			
			e.printStackTrace();
			
		} catch (IllegalAccessException e) {

			e.printStackTrace();
			
		}
		
		Servlet servlet = null;
		RequestFacade requestFacade = new RequestFacade(request);
		ResponseFacade responseFacade = new ResponseFacade(response);
		
		try {
			
			servlet = (Servlet) myClass;
			System.out.println(servlet);
			servlet.service(
					(ServletRequest) requestFacade, 
					(ServletResponse) responseFacade);
			
		} catch (ServletException e) {

			e.printStackTrace();
			
		} catch (IOException e) {

			e.printStackTrace();
			
		}
		
	}
	
	
}

至此简易版Servlet容器已经实现