【实例简介】
【实例截图】
【核心代码】
package com.csu.server;
import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import com.csu.bean.User;
import com.csu.constant.ContentFlag;
import com.csu.tool.FormatDate;
import com.csu.tool.StreamTool;
import com.csu.tool.XmlParser;
public class ChatServer {
private ExecutorService executorService;
private int port;
private boolean quit = false;
private ServerSocket server;
private List taskList = new ArrayList();
public ChatServer(int port) {
this.port = port;
executorService = Executors.newFixedThreadPool(Runtime.getRuntime().availableProcessors() * 50);
}
public void quit(){
this.quit = true;
try {
for(SocketTask tast : taskList){
tast.input.close();
}
server.close();
} catch (IOException e) {
e.printStackTrace();
}
}
public void start() throws Exception {
server = new ServerSocket(port);
new Thread(new Runnable() {
public void run() {
while (!quit) {
try {
System.out.println("等待用户的socket请求");
Socket socket = server.accept();
SocketTask newTask = new SocketTask(socket);
taskList.add(newTask);
executorService.execute(newTask);
System.out.println("启动一个线程开始处理socket请求");
} catch (Exception e) {
System.out.println("服务器终止!关闭所有线程");
}
}
}
}).start();
}
private final class SocketTask implements Runnable{
private Socket s;
private DataInputStream input;
private DataOutputStream output;
private User curUser;
public SocketTask(Socket socket) {
s = socket;
try {
input = new DataInputStream(s.getInputStream());
output = new DataOutputStream(s.getOutputStream());
} catch (IOException e) {
e.printStackTrace();
}
}
public void sendMsg(String msg, byte[] datas){
try {
if(null!= msg){
output.writeUTF(msg);
}
if(null!= datas){
output.writeInt(datas.length);
output.write(datas,0,datas.length);
}
output.flush();
} catch (IOException e) {
e.printStackTrace();
}
}
@Override
public void run() {
try {
while(true){
String msgCtn = input.readUTF();
if(msgCtn.startsWith(ContentFlag.REGOSTER_FLAG)){
String name = msgCtn.substring(ContentFlag.REGOSTER_FLAG.length()).trim();
String userId = XmlParser.saveUserInfo(name, input);
System.out.println("生成的用户id" userId);
this.sendMsg(userId, null);
taskList.remove(this);
break;
}else if(msgCtn.startsWith(ContentFlag.ONLINE_FLAG)){
String loginId = msgCtn.substring(ContentFlag.ONLINE_FLAG.length()).trim();
System.out.println("当前登录用户的ID:" loginId);
curUser = XmlParser.queryUserById(Integer.parseInt(loginId));
int imgNums = taskList.size() - 1;
System.out.println("当前在线的人数:" taskList.size());
this.output.writeInt(imgNums);
if(imgNums > 0){
for(SocketTask task : taskList){
if(task != this){
long userId = task.curUser.getId();
File img = new File("image\\" XmlParser.queryUserById(userId).getImg());
FileInputStream flleInput = new FileInputStream(img);
byte data[] = StreamTool.readStream(flleInput);
this.sendMsg(String.valueOf(userId), data);
}
}
}
File imgFile = new File("image\\" curUser.getImg());
FileInputStream flleInput = new FileInputStream(imgFile);
byte datas[] = StreamTool.readStream(flleInput);
String send_person = curUser.getName();
String send_ctn = "进入聊天室!";
String send_date = FormatDate.getCurDate();
StringBuilder json = new StringBuilder();
json.append("[{");
json.append("id:").append(loginId)
.append(",send_person:\"").append(send_person)
.append("\",send_ctn:\"").append(send_ctn)
.append("\",send_date:\"").append(send_date);
json.append("\"}]");
System.out.println("json:" json);
for(SocketTask tast : taskList){
System.out.println("循环向客户端发送消息");
tast.sendMsg(ContentFlag.ONLINE_FLAG this.curUser.getId(), null);
tast.sendMsg(json.toString(), datas);
}
flleInput.close();
}else if(msgCtn.startsWith(ContentFlag.OFFLINE_FLAG)){
taskList.remove(this);
String id = msgCtn.substring(ContentFlag.OFFLINE_FLAG.length()).trim();
StringBuilder json = new StringBuilder();
json.append("[{");
json.append("id:").append(this.curUser.getId())
.append(",send_person:\"").append(this.curUser.getName())
.append("\",send_ctn:\"").append("退出聊天室!")
.append("\",send_date:\"").append(FormatDate.getCurDate());
json.append("\"}]");
for(SocketTask tast : taskList){
if(tast!= this){
tast.sendMsg(ContentFlag.OFFLINE_FLAG id, null);
tast.sendMsg(json.toString(), null);
}
}
System.out.println("用户" curUser.getName() "退出!,关闭线程" Thread.currentThread().getName());
break;
}else if(msgCtn.startsWith(ContentFlag.RECORD_FLAG)){
String filename = msgCtn.substring(ContentFlag.RECORD_FLAG.length());
long recordTime = input.readLong();
byte datas[] = StreamTool.readStream(input);
StringBuilder json = new StringBuilder();
json.append("[{");
json.append("id:").append(this.curUser.getId())
.append(",send_person:\"").append(this.curUser.getName())
.append("\",send_ctn:\"").append(recordTime/1000 "\'")
.append("\",send_date:\"").append(FormatDate.getCurDate())
.append("\",recordTime:\"").append(recordTime);
json.append("\"}]");
System.out.println("json:" json);
for(SocketTask tast : taskList){
tast.sendMsg(ContentFlag.RECORD_FLAG filename, null);
tast.sendMsg(json.toString(), datas);
}
}else{
StringBuilder json = new StringBuilder();
json.append("[{");
json.append("id:").append(this.curUser.getId())
.append(",send_person:\"").append(this.curUser.getName())
.append("\",send_ctn:\"").append(msgCtn)
.append("\",send_date:\"").append(FormatDate.getCurDate());
json.append("\"}]");
for(SocketTask tast : taskList){
tast.sendMsg(json.toString(), null);
}
}
}
} catch (Exception e) {
taskList.remove(this);
System.out.println("关闭线程" Thread.currentThread().getName());
} finally {
try {
if (null != input)
input.close();
if (null != output)
output.close();
if (null != s)
s.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
}
}