Java-TCP/IP编程-02 * 客户端： 主线程：从控制台录入消息，发送到服务器 子线程：接收服务器转发来的消息

Client端

package com.qf.demo02;


import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.net.Socket;


/*
 * 客户端：
 主线程：从控制台录入消息，发送到服务器
 子线程：接收服务器转发来的消息
 */
public class Client {
public static void main(String[] args) {
try {
Socket s = new Socket("127.0.0.1", 6666);


new ClientThread(s).start();


BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(
s.getOutputStream()));
BufferedReader input = new BufferedReader(new InputStreamReader(
System.in));
while (true) {
String sClient = input.readLine();


bw.write(sClient);
bw.newLine();
bw.flush();


if ("over".equals(sClient))
break;
}
s.shutdownOutput();
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
}


class ClientThread extends Thread {
private Socket socket;


public ClientThread(Socket socket) {
this.socket = socket;
}


@Override
public void run() {
try {
BufferedReader br = new BufferedReader(new InputStreamReader(
socket.getInputStream()));
while (true) {
String sServer = br.readLine();
if (sServer == null)
break;
System.out.println("server:" + sServer);
}
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
}

Server端

package com.qf.demo02;


import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.Collection;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;


public class Server {
public static void main(String[] args) {
try {
ServerSocket ss = new ServerSocket(6666);
Map<String, Socket> map = new HashMap<>();


while (true) {
Socket s = ss.accept();


new ServerThread(s, map).start();
}
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
}


class ServerThread extends Thread {
private Socket socket;
private Map<String, Socket> map;


public ServerThread(Socket socket, Map<String, Socket> map) {
super();
this.socket = socket;
this.map = map;
}


@Override
public void run() {
try {
BufferedReader br = new BufferedReader(new InputStreamReader(
socket.getInputStream()));
while (true) {
String sClient = br.readLine();


if (sClient.startsWith("#")) {
// 用户注册 #zhangsan
// 截取用户名，将用户名与Socket作为键值对存储到Map集合
String name = sClient.substring(1);
map.put(name, socket);


// 将连接的客户端的用户名进行群发
// 组织所有用户名的字符串，keySet
StringBuffer sb = new StringBuffer();
Iterator<String> it = map.keySet().iterator();
while (it.hasNext()) {
sb.append(it.next() + " ");
}
// 群发，values
Collection<Socket> col = map.values();
for (Socket s : col) {
BufferedWriter bw = new BufferedWriter(
new OutputStreamWriter(s.getOutputStream()));
bw.write(sb.toString());
bw.newLine();
bw.flush();
}
} else if (sClient.startsWith("@")) {
// 聊天
/*
* @zhangsan:nihao 截取用户名和消息 根据用户名从Map集合中获得Socket，发送消息
*/
int index = sClient.indexOf(":");
String name = sClient.substring(1, index);
String info = sClient.substring(index + 1);


Socket s = map.get(name);
BufferedWriter bw = new BufferedWriter(
new OutputStreamWriter(s.getOutputStream()));
bw.write(info);
bw.newLine();
bw.flush();
}


}
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
}