package chat;
import javafx.application.Application;
import javafx.application.Platform;
import javafx.geometry.Insets;
import javafx.scene.Scene;
import javafx.scene.control.Button;
import javafx.scene.control.TextArea;
import javafx.scene.layout.BorderPane;
import javafx.stage.Stage;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.InetSocketAddress;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.Iterator;
import java.util.List;
import java.util.Vector;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
public class ServerExample extends Application {
ExecutorService executorService;
ServerSocket serverSocket;
List connections = new Vector(); // ìŠ¤ë ˆë“œì— ì•ˆì „í•¨
void startServer(){
// 서버 시작 코드 (Executor Service ìƒì„±, ServerSocket ìƒì„± ë° í¬íŠ¸ ë°”ì¸ë”©, ì—°ê²° 수ë½)
executorService = Executors.newFixedThreadPool(Runtime.getRuntime().availableProcessors());
try{
serverSocket = new ServerSocket();
serverSocket.bind(new InetSocketAddress("localhost", 5001));
}catch(Exception e){
if(!serverSocket.isClosed()) { stopServer();}
return;
}
Runnable runnable = new Runnable() { // ìˆ˜ë½ ìž‘ì—…ì„ ìƒì„±
@Override
public void run() {
Platform.runLater(()-> {
// UI ì„¤ì •
displayText("[서버 시작]");
btnStartStop.setText("stop");
});
while(true){
try{
Socket socket = serverSocket.accept(); // ì—°ê²° 수ë½
String message = "[ì—°ê²° ìˆ˜ë½ : " + socket.getRemoteSocketAddress() + ": " + Thread.currentThread().getName() + "]";
Platform.runLater(()->displayText(message));
Client client = new Client(socket);
connections.add(client);
Platform.runLater(()-> displayText("[연결 개수 : " + connections.size() + " ]"));
}catch(Exception e){
if(!serverSocket.isClosed()) { stopServer();}
break;
}
}
}
};
executorService.submit(runnable); // ìŠ¤ë ˆë“œ í’€ì—ì„œ 처리
}
void stopServer(){
// 서버 종료 코드
try{
Iterator iterator = connections.iterator();
while(iterator.hasNext()){ // ëª¨ë“ socket 닫기
Client client = iterator.next();
client.socket.close();
iterator.remove();
}
if(serverSocket != null && !serverSocket.isClosed()){ // server socket 닫기
serverSocket.close();
}
if(executorService != null && !executorService.isShutdown()){ // ExecutorService 종료
executorService.shutdown();
}
Platform.runLater(() -> {
displayText("[서버 멈춤]");
btnStartStop.setText("start");
});
}catch(Exception e){ }
}
class Client {
// ë°ì´í„° í†µì‹ ì½”ë“œ
Socket socket;
Client(Socket socket){
this.socket = socket;
receive();
}
void receive(){
// ë°ì´í„° 받기 코드
Runnable runnable = new Runnable() { // 받기 ìž‘ì—… ìƒì„±
@Override
public void run() {
try{
while(true){
byte[] byteArr = new byte[100];
InputStream inputStream = socket.getInputStream();
int readByteCount = inputStream.read(byteArr); // ë°ì´í„° 받기
// í´ë¼ì´ì–¸íŠ¸ê°€ ì •ìƒì 으로 Socketì˜ close()를 í˜¸ì¶œí–ˆì„ ê²½ìš°
if(readByteCount == -1) { throw new IOException();}
String message = "[ìš”ì² ì²˜ë¦¬: " + socket.getRemoteSocketAddress() + ": " + Thread.currentThread().getName() + "]";
Platform.runLater(()->displayText(message));
String data = new String(byteArr, 0, readByteCount, "UTF-8");
for(Client client : connections){
client.send(data); // ëª¨ë“ í´ë¼ì´ì–¸íŠ¸ì—게 보냄
}
}
}catch(Exception e){
try{
connections.remove(Client.this);
String message = "[í´ë¼ì´ì–¸íŠ¸ í†µì‹ ì•ˆë¨: " + socket.getRemoteSocketAddress() + ": " + Thread.currentThread().getName() + " ]";
Platform.runLater(()->displayText(message));
socket.close();
}catch(IOException e2) {}
}
}
};
executorService.submit(runnable); // ìŠ¤ë ˆë“œí’€ì—ì„œ 처리
}
void send(String data){
// ë°ì´í„° ì „ì†¡ 코드
Runnable runnable = new Runnable() { // 보내기 ìž‘ì—… ìƒì„±
@Override
public void run() {
try{
byte[] byteArr = data.getBytes("UTF-8");
OutputStream outputStream = socket.getOutputStream();
outputStream.write(byteArr);
outputStream.flush();
}catch(Exception e){
try {
String message = "[í´ë¼ì´ì–¸íŠ¸ í†µì‹ ì•ˆë¨: " + socket.getRemoteSocketAddress() + ": " + Thread.currentThread().getName() + " ]";
Platform.runLater(() -> displayText(message));
connections.remove(Client.this);
socket.close();
}catch(IOException e2) {}
}
}
};
executorService.submit(runnable); // ìŠ¤ë ˆë“œí’€ì—ì„œ 처리
}
}
////////
// UI ìƒì„± 코드
TextArea txtDisplay;
Button btnStartStop;
@Override
public void start(Stage primaryStage) {
BorderPane root = new BorderPane();
root.setPrefSize(500, 300);
txtDisplay = new TextArea();
txtDisplay.setEditable(false);
BorderPane.setMargin(txtDisplay, new Insets(0,0,2,0));
root.setCenter(txtDisplay);
btnStartStop = new Button("start");
btnStartStop.setPrefHeight(30);
btnStartStop.setMaxWidth(Double.MAX_VALUE);
btnStartStop.setOnAction(e->{
if(btnStartStop.getText().equals("start")){
startServer();
}else if(btnStartStop.getText().equals("stop")){
stopServer();
}
});
root.setBottom(btnStartStop);
Scene scene = new Scene(root);
scene.getStylesheets().add(getClass().getResource("app.css").toString());
primaryStage.setScene(scene);
primaryStage.setTitle("Server");
primaryStage.setOnCloseRequest(event->stopServer());
primaryStage.show();
}
void displayText(String text){
txtDisplay.appendText(text + "\n");
}
public static void main(String[] args){
launch(args);
}
}