(22.10.22.)Java 프로그래밍: HashSet을 이용한 CRUD

Book 이라는 클래스를 만들어 속성은 저자, 제목, 출판일, 가격 4가지로 하고 동일한 도서 정보가 중복되지 않도록 키보드에서 정보를 컬렉션에 저장해보고,  저자, 제목, 출판일 속성이 모두 일치하는 도서를 동일 도서로 간주한다. 이를 통해 CRUD 기능을 만들어보자!

 

Book Class 코드

import java.util.Objects;
 
public class Book 
{
    private String author;
    private String title;
    private String pubDate;
    private int price;
    
    public Book() {}
    
    public Book(String author, String title, String pubDate, int price) {
        this.author = author;
        this.title = title;
        this.pubDate = pubDate;
        this.price = price;
    }
 
    @Override
    public boolean equals(Object obj) {
        Book other = (Book) obj;
        return this.author.equals(other.author) &&
                this.title.equals(other.title) &&
                this.pubDate.equals(other.pubDate) ? true : false;
    }
    @Override
    public int hashCode() {
        return Objects.hash(author, title, pubDate);
    }
    public String getAuthor() {
        return author;
    }
    public void setAuthor(String author) {
        this.author = author;
    }
    public String getTitle() {
        return title;
    }
    public void setTitle(String title) {
        this.title = title;
    }
    public String getPubDate() {
        return pubDate;
    }
    public void setPubDate(String pubDate) {
        this.pubDate = pubDate;
    }
    public int getPrice() {
        return price;
    }
    public void setPrice(int price) {
        this.price = price;
    }
}

Main Class 코드

import java.util.ArrayList;
import java.util.HashSet;
import java.util.Iterator;
import java.util.Scanner;
import com.ezen.java.oop.Book;
 
public class SetCRUD 
{
    static enum Menu { LIST, ADD, FIND, UPDATE, DELETE, EXIT};
    
    public static void main(String[] args) 
    {
        HashSet<Book> set = new HashSet<>();
        Menu menu;
        boolean go = true;
        while(go){
            menu = menu();
            switch(menu)
            {
            case LIST:  list(set);                 break;
            case ADD:   add(set);                break;
            case FIND:  printBook(find(set));     break;
            case UPDATE:update(set);             break;
            case DELETE:delete(set);            break;
            case EXIT:  go=false;                 break;
            default: System.out.println("메뉴입력 오류");
            }
        }
        System.out.println("프로그램 종료");
    }
    
    private static Menu menu(){
        Scanner kbd = new Scanner(System.in);
        System.out.print("목록(s), 추가(a), 검색(f), 수정(u), 삭제(d), 종료(x):");
        String m = kbd.nextLine().strip();
        Menu menu = null;
        switch(m)
        {
            case "s": menu = Menu.LIST;     break;
            case "a": menu = Menu.ADD;         break;
            case "f": menu = Menu.FIND;     break;
            case "u": menu = Menu.UPDATE;     break;
            case "d": menu = Menu.DELETE;     break;
            case "x": menu = Menu.EXIT;     break;
            default : menu = null; 
        }
        return menu;
    }
    
    private static void add(HashSet<Book> set){
        Scanner kbd = new Scanner(System.in);
        
        System.out.print("저자:");
        String author = kbd.nextLine().strip();
        System.out.print("제목:");
        String title = kbd.nextLine().strip();
        System.out.print("출판일:");
        String pubDate = kbd.nextLine().strip();
        System.out.print("가격:");
        int price = kbd.nextInt();
        kbd.nextLine();
 
        set.add(new Book(author,title,pubDate,price));
    }
    
    private static void list(HashSet<Book> set){
        System.out.println();
        Iterator<Book> iter = set.iterator();
        while(iter.hasNext())
        {
            printBook(iter.next());
        }
        System.out.println();
    }
    
    private static Book getKeyword(){
        Scanner kbd = new Scanner(System.in);
        System.out.print("저자:");
        String author = kbd.nextLine().strip();
        System.out.print("제목:");
        String title = kbd.nextLine().strip();
        System.out.print("출판일:");
        String pubDate = kbd.nextLine().strip();
        
        Book book = new Book();
        book.setAuthor(author);
        book.setTitle(title);
        book.setPubDate(pubDate);
        return book;
    }
    
    private static Book find(HashSet<Book> set){
        Book key = getKeyword();
        if(set.contains(key))
        {
            ArrayList<Book> list = new ArrayList<>(set);
            return list.get(list.indexOf(key));
        }else {
            System.out.println("검색실패");
        }
        return null;
    }
    
    private static void printBook(Book book){
        String author = book.getAuthor();
        String title = book.getTitle();
        String pubDate = book.getPubDate();
        int price = book.getPrice();
        
        System.out.printf("%s\t%s\t%s\t%d \n", author,title,pubDate,price);
    }
    
    private static void update(HashSet<Book> set){
        Book found = find(set);
        if(found!=null) {
            System.out.println("수정대상 발견");
            System.out.print("새 가격:");
            Scanner kbd = new Scanner(System.in);
            int newPrice = kbd.nextInt();
            found.setPrice(newPrice);
            System.out.println("가격 수정 성공");
        }else {
            System.out.println("수정 대상 검색 실패");
        }
    }
    
    private static void delete(HashSet<Book> set){
        Book found = find(set);
        if(found!=null) {
            
            if(set.remove(found))
            {
                System.out.println("삭제 성공");
            }
        }else {
            System.out.println("삭제 대상 검색 실패");
        }
    }
}

 

메소드 별로 나누어 본다면 먼저 메뉴 열거형 변수형 만들어서 메뉴를 만들어 준다. 메뉴 메소드에선 키보드에서  입력을 받으면 메인메소드에 입력값을 보내고 호출된 메뉴 메소드에 맞는 메소드로 가서 기능이 작동하게 된다. 메인 메소드에선  메뉴 메소드를 호출하여 상응하는 메소드를 불러오게 되는 역할을 한다.

Book클래스의 Hashset 자료구조를 static으로 해주지 않았기 때문에 메뉴에서 호출할 메소드 들은 Book클래스의 Hashset 자료구조의 변수를 사용하기 위해서 모두 파라미터로 받아야한다.

---

add() 메소드

private static void add(HashSet<Book> set)
    {
        Scanner kbd = new Scanner(System.in);
        
        System.out.print("저자:");
        String author = kbd.nextLine().strip();
        System.out.print("제목:");
        String title = kbd.nextLine().strip();
        System.out.print("출판일:");
        String pubDate = kbd.nextLine().strip();
        System.out.print("가격:");
        int price = kbd.nextInt();
        kbd.nextLine();
 
        set.add(new Book(author,title,pubDate,price));
    }

list() 메소드

1 2 3 4 5 6 7 8 9 10

private static void list(HashSet<Book> set)     {         System.out.println();         Iterator<Book> iter = set.iterator();         while(iter.hasNext())         {             printBook(iter.next());         }         System.out.println();     } Colored by Color Scripter

cs

add 메소드로 인해 추가된 Book의 정보를 볼 수 있는 메소드이다. 

HashSet 자료구조는 순서가 없으므로 반복문을 돌릴 수 없기 때문에 Iterator 반복자를 써야한다.

순서를 나열 할 수 있게 된 Book 객체를 printBook 메소드에 넣어주면 printBook에서 정해둔 양식에 맞게 Book 객체의 정보를 볼 수 있게 된다.

 

printBook() 메소드

private static void printBook(Book book)
    {
        String author = book.getAuthor();
        String title = book.getTitle();
        String pubDate = book.getPubDate();
        int price = book.getPrice();
        
        System.out.printf("%s\t%s\t%s\t%d \n", author,title,pubDate,price);
    }

Book 객체 정보를 출력할 때 출력 양식을 정해둔 메소드이다. 출력용으로 따로 구분하여 만들어 놓았다.

---

getKeyword()메소드

private static Book getKeyword()
    {
        Scanner kbd = new Scanner(System.in);
        System.out.print("저자:");
        String author = kbd.nextLine().strip();
        System.out.print("제목:");
        String title = kbd.nextLine().strip();
        System.out.print("출판일:");
        String pubDate = kbd.nextLine().strip();
        
        Book book = new Book();
        book.setAuthor(author);
        book.setTitle(title);
        book.setPubDate(pubDate);
        return book;
    }

책의 저자, 제목, 출판일이 같은 것으로 찾기로 했으므로 따로 키워드 메소드를 만들어서 새로운 Book 객체를 만들어 키보드에서 입력 받은 저자, 제목, 출판일을 저장하여 이 Book객체와 같은 Book객체가 있는지 찾을 것이다.

---

find()메소드

private static Book find(HashSet<Book> set)
    {
        Book key = getKeyword();
        if(set.contains(key))
        {
            ArrayList<Book> list = new ArrayList<>(set);
            return list.get(list.indexOf(key));
        }else {
            System.out.println("검색실패");
        }
        return null;
    }

getKeyword()메소드를 호출하여 받아온 Book 객체를 key라는 변수에 저장해 준다. HashSet<Book> set에 key의 내용과 같은 것이 있는지 찾기위해 contains 기능을 쓴다. 이때 HashSet 자료구조는 순서가 없기때문에 HashSet를 ArrayList 자료구조로 바꿔주어 찾은 인데스 번호의 Book 객체 정보를 반환해주도록 한다. 리턴 된 정보는 printBook 메소드로 가서 값을 찾은 정보를 콘솔에 출력해주게 된다.

---

update()메소드

1 2 3 4 5 6 7 8 9 10 11 12 13 14

private static void update(HashSet<Book> set)     {         Book found = find(set);         if(found!=null) {             System.out.println("수정대상 발견");             System.out.print("새 가격:");             Scanner kbd = new Scanner(System.in);             int newPrice = kbd.nextInt();             found.setPrice(newPrice);             System.out.println("가격 수정 성공");         }else {             System.out.println("수정 대상 검색 실패");         }     } Colored by Color Scripter

cs

수정해주기 위해선 먼저 수정할 Book객체가 있는지 찾아야한다. 그래서 찾는 과정은 find()메소드 과정까지 같다. 찾기 위해 키보드로 값을 입력 받은 find() 메소드의 리턴 값을 found라는 이름의 새로운 Book객체에 저장하여 준다. find() 메소드에선 찾으려는 Book 객체 또는 null를 리턴하므로 null이 아니면 수정대상이 발견했다 하고 새롭게 수정할 가격 값을 키보드에서 입력받고 새로운 가격 값을 Book 클래스의 setPrice() 메소드를 통해서 found Book객체에 저장한다.

---

delete()메소드

private static void delete(HashSet<Book> set)
    {
        Book found = find(set);
        if(found!=null) {
            
            if(set.remove(found))
            {
                System.out.println("삭제 성공");
            }
        }else {
            System.out.println("삭제 대상 검색 실패");
        }
    }