服务器端代码:
package JavaDay5_20.NetworkProgramming.Demo1;
import org.junit.jupiter.api.Test;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.ArrayList;
import java.util.List;
/**
* @author myvina@qq.com
* @date 18-5-20 下午4:10
*/
public class MyServer implements Runnable {
private ServerSocket server = null;
private List<Socket> sockets = null;
public MyServer(int port) {
try {
server = new ServerSocket(port);
} catch (IOException e) {
e.printStackTrace();
}
this.sockets = new ArrayList<>();
}
@Override
public void run() {
try {
while(true) {
Socket socket = server.accept();
String message = socket.getInetAddress().getHostAddress().toString();
System.out.println(message + "连接上了");
sendMessageToAllUser(message + "已加入聊天室");
sockets.add(socket);
new readThread(socket).start();
}
} catch (IOException e) {
e.printStackTrace();
}
}
private void sendMessageToAllUser(String message) {
for(Socket socket : sockets) {
if(socket != null && socket.isConnected()) {
try {
OutputStream os = socket.getOutputStream();
os.write(message.getBytes());
os.flush();
} catch (IOException e) {
e.printStackTrace();
}
}
}
}
private class readThread extends Thread {
InputStream is = null;
public readThread(Socket socket) {
try {
is = socket.getInputStream();
} catch (IOException e) {
e.printStackTrace();
}
}
@Override
public void run() {
try {
int len = 0;
byte[] buf = new byte[1024];
while((len = is.read(buf)) != -1) {
sendMessageToAllUser(new String(buf, 0, len));
}
} catch (IOException e) {
e.printStackTrace();
}
}
}
public static void main(String[] args) {
new Thread(new MyServer(9999)).start();
}
}
客户端1代码没变:
package JavaDay5_20.NetworkProgramming.Demo1;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.Socket;
import java.net.UnknownHostException;
import java.util.Scanner;
/**
* @author myvina@qq.com
* @date 18-5-20 下午4:13
*/
public class MyClient implements Runnable {
Socket socket = null;
public MyClient(String host, int port) {
try {
socket = new Socket(host, port);
System.out.println("成功连接服务器...");
} catch (UnknownHostException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
}
}
@Override
public void run() {
new SendMessage().start();
try {
InputStream is = socket.getInputStream();
byte[] buf = new byte[1024];
int len = 0;
while((len = is.read(buf)) != -1) {
System.out.println(new String(buf, 0, len));
}
} catch (IOException e) {
e.printStackTrace();
}
}
private class SendMessage extends Thread {
@Override
public void run() {
Scanner scanner = null;
OutputStream os = null;
try {
scanner = new Scanner(System.in);
os = socket.getOutputStream();
String in = "";
do {
in = scanner.next();
os.write(in.getBytes());
os.flush();
} while (!in.equals("quit"));
} catch (IOException e) {
e.printStackTrace();
}
scanner.close();
try {
os.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
public static void main(String[] args) {
new Thread(new MyClient("127.0.0.1",9999)).start();
}
}
客户端2代码:
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.Socket;
import java.net.UnknownHostException;
import java.util.Scanner;
/**
* @author myvina@qq.com
* @date 18-5-19 下午4:33
*/
public class MyClient implements Runnable {
Socket socket = null;
public MyClient(String host, int port) {
try {
socket = new Socket(host, port);
System.out.println("成功连接服务器...");
} catch (UnknownHostException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
}
}
@Override
public void run() {
new SendMessage().start();
try {
InputStream is = socket.getInputStream();
byte[] buf = new byte[1024];
int len = 0;
while((len = is.read(buf)) != -1) {
System.out.println(new String(buf, 0, len));
}
} catch (IOException e) {
e.printStackTrace();
}
}
private class SendMessage extends Thread {
@Override
public void run() {
Scanner scanner = null;
OutputStream os = null;
try {
scanner = new Scanner(System.in);
os = socket.getOutputStream();
String in = "";
do {
in = scanner.next();
os.write(in.getBytes());
os.flush();
} while (!in.equals("quit"));
} catch (IOException e) {
e.printStackTrace();
}
scanner.close();
try {
os.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
public static void main(String[] args) {
new Thread(new MyClient("10.12.92.39",9999)).start();
}
}
"10.12.92.39"为我现在本机IP地址(使用ifconfig命令获得)
运行结果如下:
总结:通过这次多人聊天室程序的编写,对socket编程理解更深了,希望以后能更深地理解!