java Socket通信

这个应用程序示范了怎样用ServerSocket类和Socket类来和远程计算机进行通信

• HttpServer (Web服务器)
• Request (Http请求)
• Response (HTTP响应)

package web;

import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;

public class Response {

	private static final int BUFFER_SIZE = 1024;
	Request request;
	OutputStream output;
	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];
		//根据客户端的请求去服务器上查找资源
		Path path = Paths.get(System.getProperty("user.dir"),"webroot", request.getUri());
		if(Files.exists(path)) {
			try(InputStream inputStream = Files.newInputStream(path)){
				//输入流：将服务器上的文件以字节流的方式读到bytes中
				int ch = inputStream.read(bytes,0,BUFFER_SIZE);
				while(ch!=-1) {
					//输出流：将查询到的资源输出给客户端
					output.write(bytes,0,ch);
					//输入流：将服务器上的文件以字节流的方式读到bytes中
					ch = inputStream.read(bytes,0,BUFFER_SIZE);
				}
			}catch(IOException e) {
				e.printStackTrace();
			}
		}else {
			//文件路径不存在，返回错误页面
			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());
		}
	}
}

package web;

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

public class Request {
	/**
	 * 输入流，服务器读取用户发送过来的包
	 */
	private InputStream input;
	private String uri;
	public String getUri() {
		return uri;
	}
	public void setUri(String uri) {
		this.uri = uri;
	}
	public Request(InputStream input) {
		this.input = input;
	}
	//初始化uri
	public void parse() {
		StringBuilder request = new StringBuilder(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(request.toString());
		uri = parseUri(request.toString());
	}
  //截取Web目录下的访问路径
	private String parseUri(String requestString) {
		int index1 = requestString.indexOf(' ');
		int index2;
		if(index1!=-1) {
			index2 = requestString.indexOf(' ', index1 + 1);
			if(index2 > index1) {
				return requestString.substring(index1+1, index2);
			}
		}
		return null;
	}
}

package web;

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 HttpServer {
	private static final String SHUTDOWN_COMMAND = "/SHUTDOWN";
	private boolean shutdown = false;
	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);
		}
		while(!shutdown) {
			Socket socket = null;
			InputStream input = null;
			OutputStream output = null;
			try {
				socket = serverSocket.accept();
				input = socket.getInputStream();
				output = socket.getOutputStream();
				Request request = new Request(input);
				request.parse();
				Response response = new Response(output);
				response.setRequest(request);
				response.sendStaticResource();
				socket.close();
//				System.out.println("\n**********" + request.getUri());
				shutdown = request.getUri().contentEquals(SHUTDOWN_COMMAND);
			} catch (IOException e) {
				e.printStackTrace();
				continue;
			}
		}
	}
	public static void main(String[] args) {
		HttpServer server = new HttpServer();
		server.await();
	}
}