java之TCP(Socket,serverSocket)实例

Posted 2020-10-01 yutingliuyl

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.ServerSocket;
import java.net.Socket;
import java.net.UnknownHostException;

import org.junit.Test;

public class TestTCP {

	
	@Test
	public void rend() throws UnknownHostException, IOException{
		
		Socket socket = new Socket("127.0.0.1",9999);
		
		int i = (int) (Math.random()*1000);
		
		String strMessage = "send:"+i;
		
		System.out.println(strMessage);
		socket.getOutputStream().write(strMessage.getBytes());
		socket.shutdownOutput();
		
		BufferedReader buffRead = new BufferedReader(new InputStreamReader(socket.getInputStream()));
		String str = buffRead.readLine();
		System.out.println(str);
		
		buffRead.close();
		socket.close();
		System.out.println("socket.close()");
	}
	
	@Test
	public void revice() throws IOException{
		ServerSocket serverSocket = new ServerSocket(9999);
		boolean flag = true;
		
		BufferedReader buffRead = null;
		Socket socket = null;
		while(flag){
			socket = serverSocket.accept();
		    buffRead = new BufferedReader(new InputStreamReader(socket.getInputStream()));
			String str = buffRead.readLine();
			System.out.println(str);
			
			socket.shutdownInput();
			
			socket.getOutputStream().write(("revice:"+str).getBytes());
			
			
			socket.close();
			System.out.println("socket.close()");
		}
		
		buffRead.close();
		
		serverSocket.close();
		System.out.println("serverSocket close!");
		
	}
	

}