Java聊天室——实现多人聊天、私聊、群聊

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
package project.java;
//客户端
//输入、输出流分别为处于一个线程
import java.io.IOException;
import java.io.PrintStream;
import java.net.Socket;
import java.util.Scanner;
class OutToServer2 implements Runnable{
private Socket client;
public OutToServer2(Socket client) {
super();
this.client = client;
}
@Override
public void run() {
PrintStream printStream;
try {
printStream = new PrintStream(client.getOutputStream(),true);
Scanner scanner = new Scanner(System.in);
scanner.useDelimiter("n");
while(true) {
System.out.println("请输入：");
while(scanner.hasNext()) {
String string = scanner.nextLine();
printStream.println(string);
if(string.equals("拜拜")) {
System.out.println("客户端退出");
printStream.close();
scanner.close();
//client.close();
break;
}
}
}
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
}
class ReadFromServer2 implements Runnable{
private Socket client;
public ReadFromServer2(Socket client) {
super();
this.client = client;
}
public void run() {
try {
Scanner scanner = new Scanner(client.getInputStream());
scanner.useDelimiter("n");
while(scanner.hasNext()) {
System.out.println("服务器端输入："+scanner.nextLine());
}
} catch (IOException e) {
e.printStackTrace();
}
}
}
public class SingelClient2 {
public static void main(String[] args) {
String serverName = "127.0.0.1";
int port = 6666;
Socket client;
try {
client = new Socket(serverName,port);
Thread ouThread = new Thread(new OutToServer2(client));
Thread inThread = new Thread(new ReadFromServer2(client));
ouThread.start();
inThread.start();
} catch (Exception e) {
e.printStackTrace();
}
}
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
package project.java;
import java.io.IOException;
import java.io.PrintStream;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Scanner;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
//服务器端
/**
* Map<String,Socket>
* 用户注册：username:yyy
* 群聊：G:hello
* 私聊：P:yyy-hhh
*/
class
ExcuteClientServer1 implements Runnable{
private Socket client;
private Map<String,Socket> clientMap;
public ExcuteClientServer1(Socket client, Map<String, Socket> clientMap) {
super();
this.client = client;
this.clientMap = clientMap;
}
public void run() {
try {
//拿到客户端输入流，读取用户信息
Scanner scanner = new Scanner(client.getInputStream());
String string = null;
while(true){
if(scanner.hasNext()) {
string = scanner.nextLine();
Pattern pattern = Pattern.compile("rn|r|n");
Matcher matcher = pattern.matcher(string);
string = matcher.replaceAll("");
//用户注册
if(string.startsWith("user")) {
//获取用户名
String useName = string.split("\:")[1];
userRegist(useName, client);
continue;
}
//群聊
else if(string.startsWith("G")) {
String message = string.split("\:")[1];
gropChat(message);
continue;
}
//私聊
else if(string.startsWith("P")) {
String temp = string.split("\:")[1];
//取得用户名
String useName = temp.split("\-")[0];
//取得消息内容
String message = temp.split("\-")[1];
privateChat(useName, message);
continue;
}
//用户退出
else if(string.contains("拜拜")) {
//先根据Socket知道用户名
String useName = getUseName(client);
System.out.println("用户名为"+useName +"的用户下线了！！！");
clientMap.remove(useName);
continue;
}
}
}
}catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
//获取key值（即由端口号找到用户名）
public String getUseName(Socket socket) {
String useName = null;
for(String getKey : clientMap.keySet()) {
if(clientMap.get(getKey).equals(socket)) {
useName = getKey;
}
}
return useName;
}
//注册实现
public void userRegist(String useName,Socket client) {
System.out.println("用户姓名为：" + useName);
System.out.println("用户socket为：" + client);
System.out.println("用户名为"+ useName +"的用户上线了！");
System.out.println("当前用户数为："+ (clientMap.size()+1) +"人");
clientMap.put(useName, client);
}
//群聊实现
public void gropChat(String message) {
Iterator<Entry<String, Socket>> iterable = clientMap.entrySet().iterator();
for(Map.Entry<String, Socket> stringSocketEntry:clientMap.entrySet()) {
try {
Socket socket = stringSocketEntry.getValue();
PrintStream printStream = new PrintStream(socket.getOutputStream(),true);
System.out.println("用户说"+message);
printStream.println(message);
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
}
//私聊实现
public void privateChat(String useName,String message) {
//根据对应的useName找到对应的Socket
Socket privateSocket = clientMap.get(useName);
try {
PrintStream printStream = new PrintStream(privateSocket.getOutputStream());
printStream.println("用户名为"+getUseName(client)+"的用户对你说："+message);
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
}
public class SingleServer2 {
public static void main(String[] args) throws IOException {
ServerSocket serverSocket = new ServerSocket(6666);
//存取用户信息（用户名和Socket）
Map<String, Socket> map = new HashMap<String, Socket>();
//线程池，线程大小为20
ExecutorService executorService =Executors.newFixedThreadPool(20);
System.out.println("等待客户连接中...");
try {
for(int i = 0;i < 20;i ++) {
Socket socket = serverSocket.accept();
System.out.println("有新的用户连接："+socket.getInetAddress()+socket.getPort());
executorService.execute(new ExcuteClientServer1(socket,map));
}
executorService.shutdown();
serverSocket.close();
} catch (Exception e) {
}
}
}