基于ServerSocket的多人聊天室模拟实现

package chatroom;

import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.ArrayList;
import java.util.List;
/**
 * 利用ServerSocket实现的多人聊天
 * @author 风潇潇
 *
 */
public class Server {
//所有封装后的客户端通道的集合
List<MyTunnel> chatRoom = new ArrayList<MyTunnel>();
public static void main(String[] args) throws IOException {
new Server().start();
}
public void start() throws IOException{
//以8888端口创建服务端
ServerSocket sever = new ServerSocket(8888);
//依据接收到的请求，为每个client创建通道并开启单独县城
while(true){
Socket client = sever.accept();
MyTunnel mt = new MyTunnel(client);
chatRoom.add(mt);
new Thread(mt).start();

}
}
/**
 * 将客户端收发信息封装为内部类，便于使用
 * @author 风潇潇
 *
 */
private class MyTunnel implements Runnable{
private DataOutputStream dos;
private DataInputStream dis;
boolean isRunning=true;
private String name ;
//初始化，创建输入流输出流同client建立联系
public MyTunnel(Socket client) {
try {
dos = new DataOutputStream(client.getOutputStream());
dis = new DataInputStream(client.getInputStream());
} catch (IOException e) {
isRunning=false;
CloseUtil.Close.closeAll(dos,dis);
}
}
//用户名设置
public void setName(String name){
this.name=name;
}
//从用户处收取消息
private String getMsg(){
String msg =null;
try {
msg=dis.readUTF();
} catch (IOException e) {
isRunning=false;
CloseUtil.Close.closeAll(dos,dis);
chatRoom.remove(this);
}
return msg;
}
//向用户发送消息
public void send(String msg){
if(msg==null){
return;
}
try {
dos.writeUTF(msg);
dos.flush();
} catch (IOException e) {
isRunning=false;
CloseUtil.Close.closeAll(dos,dis);
chatRoom.remove(this);
}
}
//聊天室和私聊
public void sendAll(){
String msg = this.getMsg();
if(msg!=null&&msg.startsWith("@")&&msg.contains("：")){
String name = msg.substring(1, msg.indexOf('：'));
String content = msg.substring(msg.indexOf('：')+1);
for(MyTunnel mt:chatRoom){
if(mt.name.equals(name)){
mt.send(this.name+"的悄悄话："+content);
}
else{
this.send(name+"用户不存在");
}
}
}else{
for(MyTunnel mt:chatRoom){
if(mt==this){
continue;
}
mt.send(this.name+":"+msg);
}
}
}
//初始化用户，接受用户端用户名，并发送欢迎消息
public void iniClient(){
String name = this.getMsg();
this.setName(name);
for(MyTunnel mt:chatRoom){
if(mt==this){
continue;
}
mt.send("欢迎"+name+"进入聊天室");
}
}
@Override
public void run() {
iniClient();
while(isRunning){
sendAll();
}
}

}

}

package chatroom;

import java.io.BufferedReader;
import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.Socket;
import java.net.UnknownHostException;

public class Client {
//用户名设置与线程启动
public static void main(String[] args) throws UnknownHostException, IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
String name=null;
do{
System.out.println("欢迎使用请输入你的用户名：");
name = br.readLine();
}while(null==name||name.equals(""));
br.close();
Socket client = new Socket("localhost",8888);
new Thread(new Input(client)).start();;
new Thread(new Output(client,name)).start();;
}
}
//接受消息
class Input implements Runnable{
private DataInputStream dis;
private boolean isRunning = true;
public Input(Socket client) {
try {
dis=new DataInputStream(client.getInputStream());
} catch (IOException e) {
CloseUtil.Close.closeAll(dis);
isRunning=false;
}
}
//从服务器接收消息
private String getMsg(){
String msg=null;
try {
msg=dis.readUTF();
} catch (IOException e) {
CloseUtil.Close.closeAll(dis);
isRunning=false;
}
return msg;
}
//将消息打印至控制台
public void print(){
String msg = this.getMsg();
if(msg!=null){
System.out.println(msg);
}

}
@Override
public void run() {
while(isRunning){
print();
}
}

}

//发送消息
class Output implements Runnable{
private DataOutputStream dos;
private BufferedReader br;
private boolean isRunning = true;
public Output(Socket client,String name) {
try {
this.dos = new DataOutputStream(client.getOutputStream());
this.br = new BufferedReader(new InputStreamReader(System.in));
this.dos.writeUTF(name);
} catch (IOException e) {
CloseUtil.Close.closeAll(dos,br);
isRunning=false;
}

}
//从控制台读入消息
private String getMsg(){
String msg = null;
try {
msg = br.readLine();
} catch (IOException e) {
CloseUtil.Close.closeAll(dos,br);
isRunning=false;
}
return msg;
}
//将消息发送至服务器
public void send(){
String msg = getMsg();
try {
if(msg!=null){
dos.writeUTF(msg);
dos.flush();
}
} catch (IOException e) {
CloseUtil.Close.closeAll(dos,br);
isRunning=false;
}
}

@Override
public void run() {
while(isRunning){
send();
}

}

}

package CloseUtil;

import java.io.Closeable;
import java.io.IOException;

public class Close {
public static void closeAll(Closeable... io) {
for(Closeable c:io){
try {
if(c!=null){
c.close();
}
} catch (IOException e) {
}
}
}
}