一个简单的Java服务器
| /** * */ package iotest.serversocket; import java.io.BufferedInputStream; import java.io.BufferedOutputStream; import java.io.ByteArrayOutputStream; import java.io.FileInputStream; import java.io.IOException; import java.io.InputStream; import java.io.OutputStream; import java.io.UnsupportedEncodingException; import java.net.ServerSocket; import java.net.Socket; /** * @author Brandon B. Lin * */ public class SingleFileHttpServer extends Thread { private byte[] content; private byte[] header; private int port = 80; public SingleFileHttpServer(String data， String encoding， String MIMEType， int port) throws UnsupportedEncodingException { this(data.getBytes(encoding)， encoding， MIMEType， port); } public SingleFileHttpServer(byte[] data， String encoding， String MIMEType， int port) throws UnsupportedEncodingException { this.content = data; this.port = port; createHeader(encoding， MIMEType); } private void createHeader(String encoding， String MIMEType) throws UnsupportedEncodingException { String header = "HTTP/1.0 200 OK " + "Server: OneFIle 1.0 " + "Content-length: " + content.length + " " + "Content-type: " + MIMEType + " "; this.header = header.getBytes(encoding); } @Override public void run() { try { ServerSocket server = new ServerSocket(port); log(server.getLocalPort()); while (true) { respond(server); } } catch (IOException exception) { System.err.println("Could not start server. Port Occupied!"); } } private void log(int port) { System.out.println("Accepting connection on port " + port); System.out.println("Data to be sent: "); try { System.out.write(content); } catch (IOException e) { e.printStackTrace(); } } private void respond(ServerSocket server) { Socket connection = null; try { connection = server.accept(); String request = readRequestFromSocket(connection); boolean writeHeader = (request.toString().indexOf("HTTP/") != -1); writeToSocket(connection， writeHeader); } catch (IOException e) { e.printStackTrace(); } finally { if (connection != null) try { connection.close(); } catch (IOException e) { e.printStackTrace(); } } } |
| private String readRequestFromSocket(Socket connection) throws IOException { InputStream in = new BufferedInputStream(connection.getInputStream()); StringBuffer request = new StringBuffer(80); while (true) { int readByte = in.read(); if (readByte == '' || readByte == ' ' || readByte == -1) break; request.append((char) readByte); } return request.toString(); } private void writeToSocket(Socket connection， boolean writeHeader) throws IOException { OutputStream out = new BufferedOutputStream( connection.getOutputStream()); if (writeHeader) { out.write(header); } out.write(content); out.flush(); } public static void main(String[] args) { String fileName = "index.html"; String contentType = "text/html"; String encoding = "ASCII"; int port = 80; byte[] data = readFileToByteArray(fileName); try { new SingleFileHttpServer(data， encoding， contentType， port).start(); } catch (UnsupportedEncodingException e) { e.printStackTrace(); } } private static byte[] readFileToByteArray(String fileName) { byte[] data = null; try { InputStream in = new FileInputStream(fileName); ByteArrayOutputStream out = new ByteArrayOutputStream(); int readByte; while ((readByte = in.read()) != -1) out.write(readByte); data = out.toByteArray(); in.close(); } catch (IOException e) { e.printStackTrace(); } return data; } } |
