Http学习之组装报文

out.write(header);        //写入Http报文头部部分
out.write(content);       //写入Http报文数据部分
out.flush();              //刷新输出流，确保缓冲区内数据已经发送完成

package com.dragon;

import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

public class HttpServer {
	private static ServerSocket server;
	
	/**
	 * 启动服务
	 * */
	public void start() {
	    try {
	    	server = new ServerSocket(10000);
	    	System.out.println("服务启动成功...");
	    	this.receiveRequest();
	    } catch (IOException e) {
	    	e.printStackTrace();
	    	System.out.println("服务启动失败！");
	    }
	}

	
	/**
	 * 接收请求
	 * */
	public void receiveRequest() {
		ExecutorService pool = Executors.newFixedThreadPool(10);
		while (true) {
			try {
				Socket client = server.accept();
				System.out.println("用户"+client.getInetAddress().toString()+"建立连接" + client.toString());
				pool.submit(new Connection(client));  //使用线程处理每一个请求。
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
	}
	
	/**
	 * 停止服务
	 * 注：一般是不需要关闭服务的。
	 * */
	public void stop() {
		try {
			if (server != null) {
				server.close();
			}
		} catch (IOException e) {
			e.printStackTrace();
			System.out.println("服务器关闭失败！");
		}
	}
}

package com.dragon;

import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.Socket;
import java.nio.charset.Charset;
import java.nio.file.Files;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.Locale;
import java.util.TimeZone;

public class Connection implements Runnable {
	private static final String BLANK = " ";
	private static final String CRLF = "\r\n";
	private byte[] content;
	private byte[] header;
	private Socket client;
	
	public Connection(Socket client) {
		this.client = client;
	}
	
	@Override
	public void run() {
		//这里不对请求进行处理，只是收到请求之后会进行响应，而不管是什么请求
		//这里模拟服务器最原始的功能：请求、响应。
		this.response();
		System.out.println("线程执行结束了！");
	}
	
	/**
	 * 接收请求信息，这里只是一个简单的模拟，这里只能接收get请求。
	 * @throws IOException 
	 * */
	public String getRequestInfo(InputStream in) throws IOException {
		//只读取第一行，这是我们需要的全部内容
		StringBuilder requestLine = new StringBuilder(80);
		while (true) {
			int c = in.read();
			if (c == '\r' || c == '\n' || c == -1) break;
			requestLine.append((char)c);
		}
		return requestLine.toString();
	}
	
	/**
	 * 响应信息
	 * */
	public void response() {
		InputStream in = null;
		OutputStream out = null; 
		try {
			in = new BufferedInputStream(client.getInputStream());
			out = new BufferedOutputStream(client.getOutputStream());   //获取输出流
			String requestInfo = this.getRequestInfo(in);           //如果不读取客户端发来的数据，服务器就会出错。
			System.out.println(requestInfo);
		} catch (IOException e1) {
			e1.printStackTrace();
		}       //获取输入流
		
		
		//响应体数据
		File file = new File("D:/DragonFile/target/attitude.jpg");
		String contentType = null;  //文件的 MIME 类型
		try {
			content = Files.readAllBytes(file.toPath());           //使用 Files 工具类，一次性读取文件
			contentType = Files.probeContentType(file.toPath());   //获取文件的 MIME 类型
			long length = file.length();                           //获取文件字节长度
			header = this.getHeader(contentType, length);          // 填充响应头
 		} catch (IOException e) {
			e.printStackTrace();
		}
		
		try {
			out.write(header);        //写入Http报文头部部分
			out.write(content);       //写入Http报文数据部分
			out.flush();              //刷新输出流，确保缓冲区内数据已经发送完成
			
			System.out.println("报文总大小（字节）：" + (header.length + content.length));
		} catch (IOException e) {
			e.printStackTrace();
			System.out.println("客户断开连接或者发送失败！");
		} finally {
			//此处关闭 client 会导致程序出现问题，但是原因不清楚。
			try {
				if (client != null) {
					client.close();
				}
				System.out.println("请求结束了");
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
    }
	
	//响应头
	private byte[] getHeader(String contentType, long length) {
		return new StringBuilder()
				.append("HTTP/1.1").append(BLANK).append(200).append(BLANK).append("OK").append(CRLF)  // 响应头部
				.append("Server:"+"CrazyDragon").append(CRLF)
				.append("Date:").append(BLANK).append(this.getDate()).append(CRLF)
				.append("Content-Type:").append(BLANK).append(contentType).append(CRLF) //文件的 Content-Type 可通过Java获取。
				.append("Content-Length:").append(BLANK).append(length).append(CRLF).append(CRLF)
				.toString()
				.getBytes(Charset.forName("UTF-8"));
	}
	
	//获取时间
	private String getDate() {
		Date date = new Date();
		SimpleDateFormat format = new SimpleDateFormat("EEE, d MMM yyyy HH:mm:ss 'GMT'", Locale.US);
		format.setTimeZone(TimeZone.getTimeZone("GMT")); // 设置时区为GMT  
		return format.format(date);
	}
}

package com.dragon;

public class Test {
	public static void main(String[] args) {
		HttpServer httpServer = new HttpServer();
		httpServer.start();
	}
}