[소켓 프로그래밍] 스트림을 이용한 채팅 서비스를 제공하는 서버 만들기

 

클라이언트 메시지 처리 스레드
	소켓

	생성자(소켓)

	run 메소드 구현
		무한 반복
			메시지 받아서 출력

서버 클래스
	포트 번호
	서버 소켓
	클라이언트 소켓들을 저장할 수 있는 변수

	메인
		서버 실행
		무한 반복
			클라이언트 접속 허용
			변수에 클라이언트 소켓 추가
			클라이언트 메시지 처리 스레드 생성
			스레드 실행


메시지 송신 스레드
	메시지를 받는 코드를 무한 반복
메시지 수신 스레드
	메시지를 보내는 코드를 무한 반복

클라이언트 클래스
	메인
		서버 주소
		서버 포트 번호
		소켓

		메시지 송신 스레드 생성
		메시지 송신 스레드 실행

		메시지 수신 스레드 생성
		메시지 수신 스레드 실행

		무한 반복

 

소켓 통신 프로그램: OSI 7계층에서 4계층까지 알아서 해주는 프로그램
3계층 주소 : IP주소
4계층 주소: 포트 번호

---

서버 코드

Server.java

package day07.project;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class MultiThreadServer {
    static List<Socket> socketList = new ArrayList<Socket>();
    static Map<String, Socket> idSocket = new HashMap<>();

    public static void main(String[] args) {
        int port = 3424;

        try
        {
            ServerSocket serverSocket = new ServerSocket(port);

            while (true)
            {
                Socket socket = serverSocket.accept();
                socketList.add(socket);

                InputStream is = socket.getInputStream();
                InputStreamReader isr = new InputStreamReader(is);
                BufferedReader bir = new BufferedReader(isr);

                //아이디 읽어 오기
                String id = bir.readLine();
                System.out.println(id + "님이 입장하셨습니다.");

                //사용자 아이디 저장
                idSocket.put(id,socket);

                Thread mpt = new MessageProcessThread(socket);
                mpt.start();
            }
        }
        catch (IOException e)
        {
            throw new RuntimeException(e);
        }


    }
}

 

MessageProcessThread.java

package day07.project;

import javax.sound.midi.MidiSystem;
import java.io.*;
import java.net.Socket;
import java.net.SocketAddress;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class MessageProcessThread extends Thread {
    Socket socket;

    public MessageProcessThread(Socket socket) {
        this.socket = socket;
    }

    @Override
    public void run() {

        try {
            InputStream is = socket.getInputStream();
            InputStreamReader isr = new InputStreamReader(is);
            BufferedReader bir = new BufferedReader(isr);

            while (true) {
                String message = bir.readLine();
                System.out.println(message);

                //현재 접속 중인 사용자 목록을 볼 수 있는 기능 구현
                if (message.contains("사용자 목록")) {
                    for (int i = 0; i < MultiThreadServer.socketList.size(); i++) {
                        Socket client = MultiThreadServer.socketList.get(i);
                        OutputStream os = client.getOutputStream();
                        OutputStreamWriter osw = new OutputStreamWriter(os);
                        BufferedWriter bw = new BufferedWriter(osw);
                        for (String s : MultiThreadServer.idSocket.keySet())
                            bw.write(s + "\n");
                        bw.flush();
                    }

                }
                //1대 1 채팅 구현
                else if (message.startsWith("To:")) {

                    String user = message.split(":")[1];
                    Socket s = MultiThreadServer.idSocket.get(user);

                    String msg = message.split(":")[2];
                    OutputStream os = s.getOutputStream();
                    OutputStreamWriter osw = new OutputStreamWriter(os);
                    BufferedWriter bw = new BufferedWriter(osw);
                    bw.write("ONETOONE: ");
                    bw.write(msg + "\n");
                    bw.flush();
                } else {
                    for (int i = 0; i < MultiThreadServer.socketList.size(); i++) {

                        Socket client = MultiThreadServer.socketList.get(i);
                        OutputStream os = client.getOutputStream();
                        OutputStreamWriter osw = new OutputStreamWriter(os);
                        BufferedWriter bw = new BufferedWriter(osw);
                        bw.write(message + "\n");
                        bw.flush();
                    }
                }
            }
        }
        catch(IOException e){
            throw new RuntimeException(e);
        }
    }
}

 

---

클라이언트 코드

Client.java

package day07;

import day07.ReceiveMsgThread;
import day07.SendMsgThread;

import java.io.*;
import java.net.Socket;
import java.util.Scanner;

public class Client {
    public static void main(String[] args) {
        String serverIp = "192.168.0.126";
        int serverPort = 3424;
        Socket cs;
        try
        {
            cs= new Socket(serverIp, serverPort);
            Thread sendMsgThread = new SendMsgThread(cs);
            sendMsgThread.start();

            Thread receiveMsgThread = new ReceiveMsgThread(cs);
            receiveMsgThread.start();

            while(true)
            {

            }
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }
}

 

ReceiveMessageThread.java

package day07;

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

public class ReceiveMsgThread extends Thread{
    Socket socket;

    public ReceiveMsgThread(Socket socket) {
        this.socket = socket;
    }

    @Override
    public void run() {

        try
        {
            InputStream  is = socket.getInputStream();
            InputStreamReader isr = new InputStreamReader(is);
            BufferedReader bir = new BufferedReader(isr);
            while (true)
            {
                String data = bir.readLine();
                System.out.println(data);
            }

        }
        catch (IOException e)
        {
            throw new RuntimeException(e);
        }

    }
}

 

SendMessageThread.java

package day07;

import java.io.BufferedWriter;
import java.io.IOException;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.net.Socket;
import java.util.Scanner;

public class SendMsgThread extends Thread{
    Socket socket;

    public SendMsgThread(Socket socket) {
        this.socket = socket;
    }

    @Override
    public void run() {

        try {
            Scanner sc = new Scanner(System.in);
            OutputStream os = socket.getOutputStream();
            OutputStreamWriter osw = new OutputStreamWriter(os);
            BufferedWriter bw = new BufferedWriter(osw);

            System.out.print("아이디를 입력하세요 : ");
            String id = sc.nextLine();
            bw.write(id + "\n");
            bw.flush();

            while(true) {
                System.out.print("내용을 입력하세요 : ");
                String str = sc.nextLine();

                bw.write(str + "\n");
                bw.flush();

            }
        } catch (IOException e) {
            throw new RuntimeException(e);
        }

    }
}

---

실행 결과 

Server

Client 끼리 통신