자바를 자바 fp

서버단 제작

일단 TreeSet을 사용하고 내부 서버를 연결하는 것부터 제작해 보자

특정 문자들이 있는지 확인
https://highcode.tistory.com/6

package cse3040fp;

import java.io.BufferedReader;
import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.FileOutputStream;
import java.io.FileReader;
import java.io.IOException;
import java.io.OutputStream;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.TreeMap;

@SuppressWarnings("serial")
class chMap extends TreeMap<String, List<String>> {

    @Override
    public String toString() {
        String ans = "";
        List<Map.Entry<String, List<String>>> list = new ArrayList<>(super.entrySet());
        Collections.sort(list,new Comparator<Map.Entry<String, List<String>>>() {
            @Override
            public int compare(Map.Entry<String, List<String>> o1,Map.Entry<String, List<String>> o2) {
                return o1.getKey().toLowerCase().compareTo(o2.getKey().toLowerCase());
            }
        });
        for(Map.Entry<String, List<String>> entry : list) {
            System.out.println(entry.getKey() + " : " + entry.getValue().get(0)+ " : " + entry.getValue().get(1));
        }
        return ans;
    }
}

public class Server {
    private HashMap<String, DataOutputStream> clients;
    private static Map<String, List<String>> BookList;

    Server() {
        clients = new HashMap<>();
        Collections.synchronizedMap(clients);
    }

    public void start(int port) {
        // 네트워크를 연결할 serverSocket과
        // 데이터를 받아올 socket
        ServerSocket serverSocket = null;
        Socket socket = null;

        try {
            // 서버단과 함께 입력된 포트 번호를 기준으로 새로운 소켓 제작
            serverSocket = new ServerSocket(port);

            // Client가 올 때마다 소켓을 연결해줌
            while (true) {
                socket = serverSocket.accept();
                ServerReceiver thread = new ServerReceiver(socket);
                thread.start();
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public static void readBooks(String path) {
        try {
            BufferedReader br = new BufferedReader(new FileReader(path));
            while (true) {
                String Line = br.readLine();
                if (Line == null)
                    break;
                String[] sLine = Line.split("\t");
                List<String> temp = new ArrayList<>();
                temp.add(sLine[1]);
                temp.add(sLine[2]);
                BookList.put(sLine[0], temp);
            }
            br.close();
        } catch (IOException e) {
            System.out.println("Input file not found.");
            System.exit(0);
        } catch (Exception e) {
            e.printStackTrace();
        }
        System.out.println(BookList);
    }

    public static void main(String args[]) {
        // 성분을 제대로 주었을 경우에만 작동
        if (args.length != 1) {
            System.out.println("Please give the port number as an argument.");
            System.exit(0);
        } else {
            //프로그램이 시작되면 라이브러리를 제작함
            BookList =new chMap();
            readBooks("books.txt");
            new Server().start(Integer.parseInt(args[0]));
        }
    }

    synchronized void writeFile() {
        try {
            @SuppressWarnings("resource")
            OutputStream out = new FileOutputStream("books.txt");
            String ans="";

            List<Map.Entry<String, List<String>>> list = new ArrayList<>(BookList.entrySet());
            Collections.sort(list,new Comparator<Map.Entry<String, List<String>>>() {
                @Override
                public int compare(Map.Entry<String, List<String>> o1,Map.Entry<String, List<String>> o2) {
                    return o1.getKey().toLowerCase().compareTo(o2.getKey().toLowerCase());
                }
            });
            for(Map.Entry<String, List<String>> entry : list) {
                ans = entry.getKey() + "\t" + entry.getValue().get(0) + "\t" + entry.getValue().get(1) + "\t\n";
                out.write(ans.getBytes());
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    synchronized boolean addBook(String title, String author) {
        Iterator<Map.Entry<String, List<String>>> it = BookList.entrySet().iterator();
        while (it.hasNext()) {
            Map.Entry<String, List<String>> entry = it.next();
            if (entry.getKey().toString().toLowerCase().equals(title.toLowerCase())) {
                return false;
            }
        }

        List<String> temp = new ArrayList<String>();
        temp.add(author);temp.add("-");
        try {
            BookList.put(title, temp);
            writeFile();
        } catch (Exception e) {
            e.printStackTrace();
        }

        return true;
    }

    synchronized String borrowBook(String title, String borrower) {
        Iterator<Map.Entry<String, List<String>>> it = BookList.entrySet().iterator();
        while (it.hasNext()) {
            Map.Entry<String, List<String>> entry = it.next();
            String borrowBook = entry.getKey().toString();
            System.out.println(borrowBook.toLowerCase());
            if (borrowBook.toLowerCase().equals(title.toLowerCase()) && entry.getValue().get(1).equals("-")) {
                BookList.get(borrowBook).set(1, borrower);
                writeFile();
                return borrowBook;
            }
        }
        return "";
    }

    synchronized String returnBook(String title, String returner) {
        Iterator<Map.Entry<String, List<String>>> it = BookList.entrySet().iterator();
        while (it.hasNext()) {
            Map.Entry<String, List<String>> entry = it.next();
            String borrowBook = entry.getKey().toString();
            if (borrowBook.toLowerCase().equals(title.toLowerCase()) && !BookList.get(borrowBook).get(1).equals("-")) {
                BookList.get(borrowBook).set(1, "-");
                writeFile();
                return borrowBook;
            }
        }
        return "";
    }

    String infoBook(String name) {
        String ans = "";
        Iterator<Map.Entry<String, List<String>>> it = BookList.entrySet().iterator();
        if (name.equals("-"))
            return "";
        while (it.hasNext()) {
            Map.Entry<String, List<String>> entry = it.next();
            String borrowBook = entry.getKey().toString();
            if (BookList.get(borrowBook).get(1).equals(name)) {
                ans += "\t" + borrowBook + ", " + BookList.get(borrowBook).get(0);
            }
        }
        return ans;
    }

    String searchBook(String word) {
        String ans = "";
        Iterator<Map.Entry<String, List<String>>> it = BookList.entrySet().iterator();
        while (it.hasNext()) {
            Map.Entry<String, List<String>> entry = it.next();
            String borrowBook = entry.getKey().toString();
            if (BookList.get(borrowBook).get(0).toLowerCase().indexOf(word.toLowerCase()) >= 0
                    || borrowBook.toLowerCase().indexOf(word.toLowerCase()) >= 0) {
                ans += "\t" + borrowBook + ", " + BookList.get(borrowBook).get(0);
            }
        }
        return ans;
    }

    //////////////////////////////////////
    ///// Server Main Function ///////////
    /////////////////////////////////////
    class ServerReceiver extends Thread {
        Socket socket;
        DataInputStream in;
        DataOutputStream out;

        // 크라이언트를 받아들이고 Stream을 연결
        ServerReceiver(Socket socket) {
            this.socket = socket;
            try {
                in = new DataInputStream(socket.getInputStream());
                out = new DataOutputStream(socket.getOutputStream());
            } catch (IOException e) {
                e.printStackTrace();
            }
        }

        public void run() {
            String name = "";
            String Line, order[];
            String bookname = "";
            try {
                name = in.readUTF();
                clients.put(name, out);
                System.out.println(name + " login");

                while (in != null) {
                    Line = in.readUTF();
                    if (Line == null)
                        break;
                    order = Line.split("\t");
                    switch (order[0]) {
                    case "add":
                        if (addBook(order[1], order[2])) {
                            System.out.println(name + "added new Book : " + order[1] + ", " + order[2]);
                            System.out.println(BookList);
                            out.writeUTF("addS");
                        } else {
                            System.out.println(order[0] + "adding is Blocked");
                            out.writeUTF("addF");
                        }
                        break;
                    case "borrow":
                        if ((bookname = borrowBook(order[1], name)) != "") {
                            System.out.println(name + " borrowed Book : " + order[1]);
                            System.out.println(BookList);
                            out.writeUTF("borrowS\t" + bookname);
                        } else {
                            System.out.println(name + " can't borrow book : " + order[1]);
                            out.writeUTF("borrowF");
                        }
                        break;
                    case "return":
                        if ((bookname = returnBook(order[1], name)) != "") {
                            System.out.println(name + " returned Book : " + order[1]);
                            System.out.println(BookList);
                            out.writeUTF("returnS\t" + bookname);
                        } else {
                            System.out.println(name + " cant's return book: " + order[1]);
                            out.writeUTF("returnF");
                        }
                        break;
                    case "info":
                        System.out.println(BookList);
                        System.out.println(name + " looking for borrowed books");
                        out.writeUTF("info" + infoBook(name));
                        break;
                    case "search":
                        System.out.println(BookList);
                        System.out.println(name + " looking for specific books");
                        out.writeUTF("search" + searchBook(order[1]));
                        break;
                    }
                }
            } catch (IOException e) {
                e.printStackTrace();
            } finally {
                System.out.println(name + " logout");
                clients.remove(name);
            }
        }

    }
}

Client 단

package cse3040fp;

import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.IOException;
import java.net.ConnectException;
import java.net.Socket;
import java.util.Scanner;



public class Client {

    static class ClientSender extends Thread {
        Socket socket;
        DataOutputStream out;
        String name;

        ClientSender(Socket socket, String name) {
            this.socket = socket;
            try {
                out = new DataOutputStream(socket.getOutputStream());
                this.name = name;
            } catch (Exception e) {
                e.printStackTrace();
            }
        }

        public void run() {
            @SuppressWarnings("resource")
            Scanner scanner = new Scanner(System.in);
            String title, author, order, sender = "";

            // 총 6가지 경우에 대해서 입력을 받게 됨
            try {
                if (out != null) {
                    out.writeUTF(name);
                    sleep(100);
                }
                while (out != null) {
                    System.out.print("\n"+name + ">> ");
                    order = scanner.nextLine().trim();
                    switch (order) {
                    case "add":
                        System.out.print("add-title> ");
                        title = scanner.nextLine().trim();
                        System.out.print("add-author> ");
                        author = scanner.nextLine().trim();
                        if (title.equals(" ") || title.equals("") || author.equals("") || author.equals(" ")) {
                        } else {
                            sender = "add" + "\t" + title + "\t" + author;
                            try {
                                out.writeUTF(sender);
                                sleep(100);
                            } catch (IOException e) {
                                e.printStackTrace();
                            }
                        }
                        break;
                    case "borrow":
                        System.out.print("borrow-title> ");
                        title = scanner.nextLine().trim();
                        if (title.equals(" ") || title.equals("")) {
                        } else {
                            sender = "borrow" + "\t" + title;
                            try {
                                out.writeUTF(sender);
                                sleep(100);
                            } catch (IOException e) {
                                e.printStackTrace();
                            }
                        }
                        break;
                    case "return":
                        System.out.print("return-title> ");
                        title = scanner.nextLine().trim();
                        if (title.equals(" ") || title.equals("")) {
                        } else {
                            sender = "return" + "\t" + title;
                            try {
                                out.writeUTF(sender);
                                sleep(100);
                            } catch (IOException e) {
                                e.printStackTrace();
                            }
                        }
                        break;
                    case "info":
                        sender = "info"+"\t"+name;
                        try {
                            out.writeUTF(sender);
                            sleep(100);
                        } catch (IOException e) {
                            e.printStackTrace();
                        }
                        break;
                    case "search":
                        while(true) {
                            System.out.print("search-string> ");
                            title=scanner.nextLine().trim();
                            if(title.length()<3) {
                                System.out.println("Search string must be longer than 2 characters.");
                            }else if(title.equals("") || title.equals(" ")) {}
                            else {
                                sender="search"+"\t"+title;
                                try {
                                    out.writeUTF(sender);
                                    sleep(100);
                                } catch (IOException e) {
                                    e.printStackTrace();
                                }
                                break;
                            }
                        }
                        break;
                    default:
                        System.out.println("[available commands]\n" + "add: add a new book to the list of books.\n"
                                + "borrow: borrow a book from the library.\n"
                                + "return: return a book to the library.\n"
                                + "info: show list of books I am currently borrowing.\n" + "search: search for books.");
                        break;
                    }
                }
            } catch (IOException e) {
                e.printStackTrace();
            } catch (InterruptedException e) {
                e.printStackTrace();
            }

        }
    }

    static class ClientReceiver extends Thread {
        Socket socket;
        DataInputStream in;

        ClientReceiver(Socket socket) {
            this.socket = socket;
            try {
                in = new DataInputStream(socket.getInputStream());
            } catch (IOException e) {
                e.printStackTrace();
            }
        }

        public void run() {
            String Line;
            String order[];
            //총 8가지 경우에 대해서 출력을 맡게 됨
            while (in != null) {
                try {
                    Line=in.readUTF();
                    order=Line.split("\t");
                    switch(order[0]) {
                    case "addS":
                        System.out.println("A new book added to the list.");
                        break;
                    case "addF":
                        System.out.println("The book already exists in the list.");
                        break;
                    case "borrowS":
                        System.out.println("You borrowed a book. - "+order[1]);
                        break;
                    case "borrowF":
                        System.out.println("The book is not available.");
                        break;
                    case "returnS":
                        System.out.println("You returned a book. - "+order[1]);
                        break;
                    case "returnF":
                        System.out.println("You did not borrow the book.");
                        break;
                    case "info":
                        System.out.println("You are currently borrowing "+(order.length-1)+ " books:");
                        for (int i = 1 ; i<order.length;++i) {
                            System.out.println((i)+". "+order[i]);
                        }
                        break;
                    case "search":
                        System.out.println("Your search matched "+(order.length-1)+ "  results.");
                        for (int i = 1 ; i<order.length;++i) {
                            System.out.println((i)+". "+order[i]);
                        }
                        break;
                    }
                } catch (IOException e) {
                    e.printStackTrace();
                    System.exit(0);
                }
            }
        }
    }

    public static void main(String[] args) {
        if (args.length != 2) {
            System.out.println("Please give the IP address and port number as arguments.");
            System.exit(0);
        }

        try {
            Socket socket = new Socket(args[0], Integer.parseInt(args[1]));
            String name = "";
            while (true) {
                @SuppressWarnings("resource")
                Scanner scan = new Scanner(System.in);
                System.out.print("Enter UserID>> ");
                name = scan.nextLine();
                if (name.contains(" ") || !name.matches("^[0-9|a-z|A-Z]*$") || name.equals("")) {
                    System.out.println("UserID must be a single word with lowercase alphabets and numbers.\n");
                } else
                    break;
            }
            Thread sender = new Thread(new ClientSender(socket, name));
            Thread receiver = new Thread(new ClientReceiver(socket));
            sender.start();
            receiver.start();
        } catch (ConnectException ce) {
            ce.printStackTrace();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}