package com.whb.chat;
import java.awt.BorderLayout;
import java.awt.Button;
import java.awt.Frame;
import java.awt.GridLayout;
import java.awt.TextArea;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.event.WindowAdapter;
import java.awt.event.WindowEvent;
import java.io.*;
import java.net.*;
import java.util.*;
import javax.swing.JPanel;
public class ChatServer extends Frame {
Button OkBtn = new Button("启动服务器"), cancleBtn = new Button("关闭服务器"),
quiteBtn = new Button("退出");
TextArea ta = new TextArea();
JPanel mb1, mb2;
boolean started = false;
ServerSocket ss = null;
List<Client> clients = new ArrayList<>();
int CNo = 0;
public static boolean running = false;
public static void main(String[] args) {
ChatServer cs = new ChatServer();
}
public void start(boolean running) {
if (running) {
try {
ss = new ServerSocket(8888);
started = true;
} catch (BindException e) {
System.out.println("端口使用中...");
System.out.println("请关闭相关程序并重新启动服务器...");
System.exit(0);
} catch (IOException e) {
e.printStackTrace();
}
try {
while (started) {
Socket s = ss.accept();
Client c = new Client(s);
CNo++;
System.out.println("第" + CNo + "个客户端已连接");
System.out.println("共有" + CNo + "个客户端在线");
new Thread(c).start();
clients.add(c);
}
} catch (EOFException e) {
System.out.println("客户端断开连接了!");
} catch (IOException e) {
e.printStackTrace();
} finally {
try {
ss.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
}
class Client implements Runnable {
private Socket s;
private DataInputStream dis = null;
private DataOutputStream dos = null;
private boolean beconnected = false;
public Client(Socket s) {
// super();
this.s = s;
try {
dis = new DataInputStream(s.getInputStream());
dos = new DataOutputStream(s.getOutputStream());
beconnected = true;
} catch (IOException e) {
e.printStackTrace();
}
}
public void send(String str) {
try {
dos.writeUTF(str);
} catch (IOException e) {
clients.remove(this);
System.out.println("对方退出了,我从List里面去掉了!");
// e.printStackTrace();
}
}
@Override
public void run() {
while (running) {
try {
while (beconnected) {
String str = dis.readUTF();
System.out.println(str);
for (int i = 0; i < clients.size(); i++) {
Client c = clients.get(i);
c.send(str);
}
}
} catch (EOFException e) {
CNo--;
System.out.println("a Client closed!");
System.out.println("共有" + CNo + "个客户端在线");
} catch (IOException e) {
e.printStackTrace();
} finally {
try {
if (dis != null)
dis.close();
if (dos != null)
dos.close();
if (s != null)
s.close();
s = null;
} catch (IOException e1) {
e1.printStackTrace();
}
}
}
}
}
public ChatServer() {
mb1 = new JPanel();
mb2 = new JPanel();
mb2.setLayout(new GridLayout(1, 3));
mb1.add(ta);
mb2.add(OkBtn);
mb2.add(cancleBtn);
mb2.add(quiteBtn);
add(mb1, BorderLayout.NORTH);
add(mb2, BorderLayout.SOUTH);
OkBtn.addActionListener(new myListener());
cancleBtn.addActionListener(new myListener());
quiteBtn.addActionListener(new myListener());
this.addWindowListener(new WindowAdapter() {
@Override
public void windowClosing(WindowEvent e) {
running = false;
System.exit(0);
}
});
this.pack();
setVisible(true);
setBounds(400, 300, 480, 270);
start(running);
}
class myListener implements ActionListener {
@Override
public void actionPerformed(ActionEvent e) {
if (e.getSource() == OkBtn) {
running = true;
ta.setText(ta.getText() + "服务器启动成功!" + '\n');
start(running);
} else if (e.getSource() == cancleBtn) {
running = false;
start(running);
ta.setText(ta.getText() + "服务器已关闭!" + '\n');
} else if (e.getSource() == quiteBtn) {
running = false;
System.exit(0);
}
}
}
}