一个简单的socket例子

一、简单说明

以下的这个例子是从一本书上看来的。主要分为客户端与服务端，当一个客户端请求服务端时间时，服务端就开一个线程，并且返回当前时间给客户端。数据读取用的是readLine()，遇到过一个坑，解决了好一会儿。以下上代码：

二、代码片段

1、服务端TimeServer.java

package com.yxf.demo.service.socket;

import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;

/**
 * 服务端
 *@author yao xingfen
 *@date 2018年2月23日下午3:15:06
 */
public class TimeServer {
	public static void main(String[] args) {
		int port = 8089;
		if(null != args && args.length > 0){
			try {
				port = Integer.parseInt(args[0]);
			} catch (NumberFormatException e) {
			}
		}
		ServerSocket serverSocket = null;
		try {
			serverSocket = new ServerSocket(port);
			Socket socket  = null;
			while(true){
				socket = serverSocket.accept();
				new TimeServerHandler(socket).run();
			}
		} catch (IOException e) {
			e.printStackTrace();
		}finally {
			if(serverSocket != null){
				try {
					serverSocket.close();
					System.out.println("close serverSocket");
				} catch (IOException e) {
					e.printStackTrace();
				}
			}
		}
		
	}
}

2、服务端socket线程类

package com.yxf.demo.service.socket;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.net.Socket;
import java.util.Date;

/**
 *服务端读写线程
 *@author yao xingfen
 *@date 2018年2月23日下午3:25:33
 */
public class TimeServerHandler implements Runnable{

	/**
	 * 服务端socket
	 */
	private Socket socket;
	
	public TimeServerHandler(Socket socket){
		this.socket = socket;
	}
	@Override
	public void run() {
		BufferedReader in = null;
		PrintWriter out = null;
		try {
			in  = new BufferedReader(new InputStreamReader(this.socket.getInputStream()));
			out = new PrintWriter(this.socket.getOutputStream());
			String body = null;
			String currentTime = null;
			while(true){
				body = in.readLine();
				if(body == null){
					break;
				}
				System.out.println("SERVER:receive client message is:" + body);
				currentTime = "NOW".equalsIgnoreCase(body)?new Date(System.currentTimeMillis()).toString():"BAD QUERY";
				System.out.println("SERVER:server response is "+currentTime);
				out.println(currentTime);
				out.flush();
			}
		} catch (IOException e) {
			if(null != in){
				try {
					in.close();
				} catch (IOException e1) {
					e1.printStackTrace();
				}
			}
			if(null != out){
				out.close();
			}
			if(null != socket){
				try {
					socket.close();
				} catch (IOException e1) {
					e1.printStackTrace();
				}
			}
		}
		
		
	}

}

3、客户端

package com.yxf.demo.service.socket;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.net.Socket;

/**
 *时间客户端
 *@author yao xingfen
 *@date 2018年2月23日下午3:49:56
 */
public class TimeClient {
	public static void main(String[] args) {
		int port = 8089;
		String host = "127.0.0.1";
		if(null != args && args.length > 0){
			try {
				port = Integer.parseInt(args[0]);
				System.out.println("client port is :"+port);
			} catch (NumberFormatException e) {
			}
		}
		Socket socket = null;
		BufferedReader in = null;
		PrintWriter out = null;
		try {
			socket = new Socket(host, port);
			in = new BufferedReader(new InputStreamReader(socket.getInputStream()));
			out = new PrintWriter(socket.getOutputStream(),true);
			out.println("NOW");
			out.flush(); 
			System.out.println("CLIENT:send order to server success");
			String response = null;
			response = in.readLine();
			System.out.println("CLIENT:now is:" + response);
		} catch (Exception e) {
			e.printStackTrace();
		}finally {
			if(null != socket){
				try {
					socket.close();
				} catch (IOException e) {
					e.printStackTrace();
				}
			}
			if(null != in){
				try {
					in.close();
				} catch (IOException e) {
					e.printStackTrace();
				}
			}
			if(null != out){
				out.close();
			}
		}
	}
}

三、补充说明

*关于端口号的选择

0~1023的端口号为系统所保留，例如http服务的端口号为80,telnet服务的端口号为21,ftp服务的端口号为23, 所以我们在选择端口号时，最好选择一个大于1023的数以防止发生冲突

*概念性的东西

bio与nio，我们的socket就是bio阻塞性io，当有客户端连接时，服务端才会继续往下走，不然就一直阻塞在accept();当客户端有返回数据时才会继续往下读取数据，不然就会阻塞在readLine();比如以上例子中，将客户端代码改为先写后读，程序就会阻塞在readLine()的地方，这个方法只会在有数据来的时候才执行。以上的例子中使用的是一个客户端连接就会在服务端开一个线程，这样当连接数增多时，服务端的压力会越来越大。好的优化办法可以使用线程池。但这也是只针对连接数相对较少的情况。而nio是非阻塞io，比如我们的netty，我所理解的是服务端会不断地去问有没有数据，有数据返回数据，没有数据也会返回相应的信息。（以上只是我的个人见解，如有问题请指正，灰常感谢）