使用TCP同时加入多线程和容器实现群聊功能。
在服务端加入多线程,每当连接一个客户端就相当于为这个客户端开辟一个可以读信息和发送信息的服务。要实现群聊需要借助容器,有客户端连接时,把这个客户端放入容器,发送信息时遍历这个容器,向这个容器内的客户端逐个发送消息。
在客户端加入多线程,想要实现客户端的随时发消息和随时收消息可以加入一个发送线程和接收线程。发送线程通过IO流向服务端发送线程,接收线程通过IO流接收服务端发送过来的线程。
package Net.TCP;
import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.concurrent.CopyOnWriteArrayList;
/**
* 服务端
*
* @author zhaoy
*
*/
public class Test02_server{
private static CopyOnWriteArrayList<Channel> all = new CopyOnWriteArrayList<>();
public static void main(String[] args) throws Exception {
System.out.println("======服务端========");
//使用ServerSocket指定端口创建服务器
ServerSocket server = new ServerSocket(9999);
//阻塞式等待连接
while(true) {
Socket client = server.accept();
Channel c = new Channel(client);
System.out.println("有一个客户端连接");
all.add(c);
new Thread(c).start();
}
}
static class Channel implements Runnable{
private Socket client;
private DataInputStream dis;
private DataOutputStream dos;
public Channel(Socket client) {
this.client = client;
try {
dos = new DataOutputStream(client.getOutputStream());
dis = new DataInputStream(client.getInputStream());
} catch (IOException e) {
e.printStackTrace();
}
}
public String receive() {
String msg = "";
try {
msg = dis.readUTF();
} catch (IOException e) {
e.printStackTrace();
}
return msg;
}
public void send(String msg) {
System.out.println(msg);
try {
dos.writeUTF(msg);
} catch (IOException e) {
e.printStackTrace();
}
}
private void sendOthers(String msg,boolean isSys) {
for (Channel other : all) {
if(other==this) {
continue;
}
other.send(msg);
}
}
@Override
public void run() {
while(true) {
String msg = receive();
if(!msg.equals("")) {
sendOthers(msg, false);
}
}
}
}
}
package Net.TCP;
import java.net.Socket;
/**
* 客户端
*
* @author zhaoy
*
*/
public class Test02_client {
public static void main(String[] args) throws Exception {
System.out.println("------客户端 -------");
// 使用Socket 指定服务器的地址和端口
Socket client = new Socket("localhost", 9999);
// 发送消息
new Thread(new Test02_send(client)).start();
// 接收消息
new Thread(new Test02_receive(client)).start();
}
}
package Net.TCP;
import java.io.BufferedReader;
import java.io.DataOutputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.Socket;
/**
* 发送
*
* @author zhaoy
*
*/
public class Test02_send implements Runnable {
private BufferedReader console;
private DataOutputStream dos;
private Socket client;
public Test02_send(Socket client) {
this.client = client;
try {
console = new BufferedReader(new InputStreamReader(System.in));
dos = new DataOutputStream(client.getOutputStream());
} catch (IOException e) {
e.printStackTrace();
}
}
@Override
public void run() {
while (true) {
try {
String msg = console.readLine();
dos.writeUTF(msg);
dos.flush();
} catch (IOException e) {
e.printStackTrace();
}
}
}
}
package Net.TCP;
import java.io.DataInputStream;
import java.io.IOException;
import java.net.Socket;
/**
* 接收
*
* @author zhaoy
*
*/
public class Test02_receive implements Runnable{
private DataInputStream dis;
private Socket client;
public Test02_receive(Socket client) {
this.client = client;
try {
dis = new DataInputStream(client.getInputStream());
} catch (IOException e) {
e.printStackTrace();
}
}
@Override
public void run() {
while(true) {
String msg = "";
try {
msg = dis.readUTF();
System.out.println(msg);
} catch (IOException e) {
e.printStackTrace();
}
}
}
}