基于单个服务器处理多用户消息的一对一聊天系统

问题介绍

结合网络编程多线程实现基于星型结构的一对一聊天系统。

实现介绍

用户在控制台输入的格式要求：目标用户名 + 空格 + 消息内容
若目标用户存在，则服务器将消息内容传递给目标用户，反之服务器会给予“目标用户不存在的响应信息”。
目标用户控制台收到信息的格式为：发送方用户名: + 发送的消息内容

代码

客户端代码：

package netProgramTranscript;

import java.io.IOException;
import java.io.PrintWriter;
import java.net.Socket;

public class Client implements ScreenInputInterface {
    private SocketHandler socketHandler;

    public void start(String clientName){
        new Thread(new ScreenInputRunIns(this)).start();
        try {
            Socket clientSocket = new Socket("127.0.0.1",666);
            PrintWriter writer = new PrintWriter(clientSocket.getOutputStream());
            writer.println(clientName);
            writer.flush();
            socketHandler =   new SocketHandler(clientSocket,null);
            new Thread(socketHandler).start();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    @Override
    public void send(String msg) {
        socketHandler.sendMsg(msg);
    }

    /*    @Override
    public void send(String msg, SocketAddress address) {
        socketHandler.sendMsg(msg);
    }*/
}

服务端代码：

package netProgramTranscript;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Set;

public class Server implements ScreenInputInterface {
    private HashMap<String,SocketHandler> clients = new HashMap<>();
    private ServerSocket serverSocket;

    public void start(){
//        new Thread(new ScreenInputRunIns(this)).start();
        try {
            serverSocket = new ServerSocket(666);
            while (true){
                Socket receiveSocket = serverSocket.accept();
                String clientName = new BufferedReader(new InputStreamReader(receiveSocket.getInputStream())).readLine();
                SocketHandler socketHandler = new SocketHandler(receiveSocket,this);
                clients.put(clientName,socketHandler);
                System.out.println(clientName+"进入聊天系统");
                new Thread(socketHandler).start();
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    @Override
    public void send(String msg) {
    }

    public void send(Socket senderListenerSocket,String msg) throws IOException {
        String targetClient = msg.substring(0,msg.indexOf(" "));
        SocketHandler socketHandler = clients.get(targetClient);
        if (socketHandler!=null){
            Set<String> clientNames = clients.keySet();
            String senderName = null;
            for (String clientName:clientNames) {
                if (clients.get(clientName).getSocket()==senderListenerSocket){
                    senderName = clientName;
                    break;
                }
            }
            socketHandler.sendMsg(senderName+": "+msg.substring(msg.indexOf(" ")+1));
        }else {
            PrintWriter writer = new PrintWriter(senderListenerSocket.getOutputStream());
            writer.println("The target user does not exist!");
            writer.flush();
        }
    }

    /*    @Override
    public void send(String msg, SocketAddress address) {
        for (SocketHandler s:clients) {
            if (s.getSocket().getRemoteSocketAddress().equals(address)){
                continue;
            }
            s.sendMsg(msg);
        }
    }*/

    public static void main(String[] args) {
        new Server().start();
    }
}

屏幕输入信息接口：

package netProgramTranscript;

public interface ScreenInputInterface {
    void send(String msg);
//    void send(String msg, SocketAddress address);
}

用于各个端口输入信息的创建的线程对象：

package netProgramTranscript;

import java.util.Scanner;

public class ScreenInputRunIns implements Runnable{
    private ScreenInputInterface screenInputInterface;
    private Scanner input = new Scanner(System.in);

    public ScreenInputRunIns(ScreenInputInterface screenInputInterface) {
        this.screenInputInterface = screenInputInterface;
    }

    @Override
    public void run() {//屏幕实时输入
        while (true){
            String msg = input.nextLine();
//            screenInputInterface.send(msg,null);
            screenInputInterface.send(msg);
        }
    }
}

套接字处理器（用于传输数据：实际发送信息到输出流并实时监听各个端口的输入且负责显示）：

package netProgramTranscript;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.net.Socket;

public class SocketHandler implements Runnable{
    private Socket socket;
    private PrintWriter writer;
    private Server server;

    public SocketHandler(Socket socket, Server server) throws IOException {
        this.socket = socket;
        this.server = server;
        writer = new PrintWriter(this.socket.getOutputStream());
    }

    public Socket getSocket() {
        return socket;
    }

    public void sendMsg(String msg){
        writer.println(msg);
        writer.flush();
    }

    @Override
    public void run() {
        while(true){//监听输入流的输入
            try {
                BufferedReader reader = new BufferedReader(new InputStreamReader(socket.getInputStream()));
                String receiveMsg = null;
                if ((receiveMsg = reader.readLine())!=null&&!(receiveMsg.equals(""))){
                    if (server!=null){
//                        server.send(receiveMsg,socket.getRemoteSocketAddress());
                        server.send(socket,receiveMsg);
                    }else {
                        System.out.println(receiveMsg);
                    }
                }
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
}

负责启动的各个端口的客户端代码：

package netProgramTranscript;

import java.util.Scanner;

public class ClientLinked {
    public static void main(String[] args) {
        System.out.print("Enter the username: ");
        Scanner scanner = new Scanner(System.in);
        String celientName = scanner.next();
        new Client().start(celientName);
    }
}

package netProgramTranscript;

import java.util.Scanner;

public class ClientLinked1 {
    public static void main(String[] args) {
        System.out.print("Enter the username: ");
        Scanner scanner = new Scanner(System.in);
        String celientName = scanner.next();
        new Client().start(celientName);
    }
}

package netProgramTranscript;

import java.util.Scanner;

public class ClientLinked2 {
    public static void main(String[] args) {
        System.out.print("Enter the username: ");
        Scanner scanner = new Scanner(System.in);
        String celientName = scanner.next();
        new Client().start(celientName);
    }
}

多对多群聊实现参见：多对多的聊天系统（基于网络编程和多线程）