tcpserver.java

美国卡耐基。梅隆大学SSD8课程练习3的答案

import java.io.*;
import java.net.*;
import java.util.*;

public class TCPServer {

	private Socket server;
	private DataInputStream inFromClient;
	private DataOutputStream outToClient;
	private byte[] in = new byte[1];
	private byte[] out = new byte[1];
	private int count;
	private int lost;
	
	TCPServer(){
		count = 0;
		lost = 0;
	}
	
	void setUpConnection(){
		try {
			ServerSocket socket = new ServerSocket(8000);
			server = socket.accept();
		} catch (IOException e) {
			// TODO Auto-generated catch block
			System.out.println("Socket connection error!");
		}
		try {
			inFromClient = new DataInputStream(server.getInputStream());
			outToClient = new DataOutputStream(server.getOutputStream());
		} catch (IOException e) {
			// TODO Auto-generated catch block
			System.out.println("IO stream error!");
		}
		
	}
	
	void receive(){
		try {
			while(count < 1000){
				inFromClient.read(in);
				if(in[0] != -1){
					count++;
					out[0] = in[0];
					outToClient.write(out);
				}else{
					lost++;
				}
			}
			System.out.println("The count received is: " + count);
			System.out.println("Lost " + lost + " packets");
		} catch (IOException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
	}
	
	void close(){
		try {
			inFromClient.close();
			outToClient.close();
			server.close();
		} catch (IOException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
	}
	
	public static void main(String[] args){
		TCPServer server = new TCPServer();
		server.setUpConnection();
		server.receive();
		server.close();
	}
}