TCP/UDP实现相同聊天功能的代码

注意：此代码完全是以入门水平为标准，内部有许多值得优化的地方，但作为了解TCP/UDP的功能还是可以的。
实际代码一定要进行优化，至少要启用线程。

UDP的实现部分 原文： http://www.java2000.net/p10504

接收代码

1. package chat;
3. import java.net.DatagramPacket;
4. import java.net.DatagramSocket;
5. import java.net.InetAddress;
7. /**
8.  * UDP 接收数据的代码。
9.  * 
10.  * @author 赵学庆 www.java2000.net
11.  */
12. public class UDPReceiveTest {
13.   void display() {
14.     byte buf[] = new byte[256];
15.     DatagramPacket sPacket;
16.     DatagramPacket rPacket;
17.     DatagramSocket socket;
18.     int count;
19.     byte[] res = "OK".getBytes();
20.     try {
21.       socket = new DatagramSocket(8888);
22.       while (true) {
23.         rPacket = new DatagramPacket(buf, buf.length);
24.         // 接收数据报
25.         socket.receive(rPacket);
26.         // 获取发送方的地址
27.         InetAddress addr = rPacket.getAddress();
28.         // 获取发送方的端口
29.         int port = rPacket.getPort();
30.         count = rPacket.getLength();
31.         System.out.println("接收到来自 " + addr.getHostAddress() + " 数据:");
32.         System.out.println(new String(buf, 0, count));
33.         // 发送数据报
34.         sPacket = new DatagramPacket(res, res.length, addr, port);
35.         socket.send(sPacket);
36.       }
37.     } catch (Exception e) {
38.       System.out.println(e);
39.     }
40.   }
42.   public static void main(String[] args) {
43.     UDPReceiveTest app = new UDPReceiveTest();
44.     app.display();
45.   }
46. }

发送代码

1. package chat;
3. import java.net.DatagramPacket;
4. import java.net.DatagramSocket;
5. import java.net.InetAddress;
7. /**
8.  * UDP 发送数据的代码。
9.  * 
10.  * @author 赵学庆 www.java2000.net
11.  */
12. public class UDPSendTest {
13.   public static void main(String[] args) {
14.     UDPSendTest app = new UDPSendTest();
15.     app.display();
16.   }
18.   void display() {
19.     byte buf[] = new byte[256];
20.     DatagramPacket sPacket;
21.     DatagramPacket rPacket;
22.     DatagramSocket socket;
23.     int count;
24.     byte[] host = new byte[128];
25.     try {
26.       socket = new DatagramSocket();
27.       // 输入目的方的主机名
28.       System.out.print("请输入目的方的主机名: ");
29.       count = System.in.read(host);
30.       InetAddress addr = InetAddress.getByName(new String(host, 0, count - 2));
31.       while (true) {
32.         System.out.print("请输入要发送的数据: ");
33.         count = System.in.read(buf);
34.         // 发送数据报
35.         sPacket = new DatagramPacket(buf, count - 2, addr, 8888);
36.         socket.send(sPacket);
37.         // 接收数据报
38.         rPacket = new DatagramPacket(buf, buf.length);
39.         socket.receive(rPacket);
40.         System.out.print("返回信息: ");
41.         String str = new String(rPacket.getData(), 0, rPacket.getLength());
42.         System.out.println(str);
43.       }
44.     } catch (Exception e) {
45.       System.out.println(e);
46.     }
47.   }
48. }

TCP的实现部分 原文： http://www.java2000.net/p10506
接收代码

1. package chat;
3. import java.io.BufferedReader;
4. import java.io.InputStream;
5. import java.io.InputStreamReader;
6. import java.io.PrintWriter;
7. import java.net.ServerSocket;
8. import java.net.Socket;
10. /**
11.  * TCP 接收数据的代码。
12.  * 
13.  * @author 赵学庆 www.java2000.net
14.  */
15. public class TCPReceiveTest {
16.   void display() {
17.     ServerSocket socket;
18.     try {
19.       socket = new ServerSocket(8888);
20.       while (true) {
21.         Socket s = socket.accept();
22.         InputStream is = s.getInputStream();
23.         BufferedReader reader = new BufferedReader(new InputStreamReader(is));
24.         PrintWriter out = new PrintWriter(s.getOutputStream());
25.         String line = reader.readLine();
26.         System.out.println("接收到来自 " + s.getRemoteSocketAddress() + " 数据:");
27.         System.out.println(line);
28.         out.print("OK");
29.         out.flush();
30.         out.close();
31.         reader.close();
32.         s.close();
33.       }
34.     } catch (Exception e) {
35.       System.out.println(e);
36.     }
37.   }
39.   public static void main(String[] args) {
40.     TCPReceiveTest app = new TCPReceiveTest();
41.     app.display();
42.   }
43. }

发送代码

1. package chat;
3. import java.io.BufferedReader;
4. import java.io.InputStream;
5. import java.io.InputStreamReader;
6. import java.io.PrintWriter;
7. import java.net.InetAddress;
8. import java.net.Socket;
10. /**
11.  * TCP 发送数据的代码。
12.  * 
13.  * @author 赵学庆 www.java2000.net
14.  */
15. public class TCPSendTest {
16.   public static void main(String[] args) {
17.     TCPSendTest app = new TCPSendTest();
18.     app.display();
19.   }
21.   void display() {
22.     Socket s;
23.     byte buf[] = new byte[256];
24.     int count;
25.     byte[] host = new byte[128];
26.     try {
27.       // 输入目的方的主机名
28.       System.out.print("请输入目的方的主机名: ");
29.       count = System.in.read(host);
30.       InetAddress addr = InetAddress.getByName(new String(host, 0, count - 2));
31.       while (true) {
32.         System.out.print("请输入要发送的数据: ");
33.         count = System.in.read(buf);
34.         // 发送数据报
35.         s = new Socket(addr.getHostAddress(), 8888);
36.         InputStream is = s.getInputStream();
37.         BufferedReader reader = new BufferedReader(new InputStreamReader(is));
38.         PrintWriter out = new PrintWriter(s.getOutputStream());
39.         out.println(new String(buf, 0, count - 2));
40.         out.flush();
41.         String line = reader.readLine();
42.         System.out.print("返回信息: ");
43.         System.out.println(line);
44.         reader.close();
45.         out.close();
46.         s.close();
47.       }
48.     } catch (Exception e) {
49.       System.out.println(e);
50.     }
51.   }
52. }