图书管理系统

public class Teat {
    //创建图书集合，添加测试数据
    /*
           键 ；图书类型        值 ；该类型下的图书集合
     */
    static HashMap<String, ArrayList<Book>> bookMap = new HashMap<>();
    public static void main(String[] args) {
        //创建名著集合
        ArrayList<Book> mz = new ArrayList<>();
        mz.add(new Book("西游记", 13, "吴承恩"));
        mz.add(new Book("水浒传", 15, "施耐庵"));
        mz.add(new Book("三国演义", 19, "罗贯中"));
        mz.add(new Book("红楼梦", 22, "曹雪芹"));
        bookMap.put("名著", mz);//添加到图书类型里面去
        //创建it书籍
        ArrayList<Book> it = new ArrayList<>();
        it.add(new Book("Java入门到精通", 22, "张三"));
        it.add(new Book("Php入门到精通", 22, "李四"));
        it.add(new Book("C++入门到精通", 22, "王五"));
        bookMap.put("编程", it);
        System.out.println("-----欢迎进入图书管理系统-----");
        int use = 123;
        int use2 = 456;
        Scanner sc = new Scanner(System.in);
        System.out.println("请输入用户名");
        int useName = sc.nextInt();
        System.out.println("请输入用户密码");
        int useEntry = sc.nextInt();
        while (true) {
            if ((useName == use) && (use2 == useEntry)) {
                System.out.println("1,查看所有书籍");
                System.out.println("2,添加书籍");
                System.out.println("3,删除书籍");
                System.out.println("4,修改书籍");
                System.out.println("5,退出");
                System.out.println("请输入你的选择");
            } else {
                System.out.println("用户名或密码错误，请重新输入");
            }

            int i = sc.nextInt();
            switch (i) {
                case 1:
                    showBook();
                    break;
                case 2:
                    addBook();
                    break;
                case 3:
                    delBook();
                    break;
                case 4:
                    reviseBook();
                    break;
                case 5:
                    System.out.println("欢迎使用");
                    return;
            }
        }
    }
    //查看图书的方法
    public static void showBook() {
        System.out.println("类型\t\t书名\t\t\t价格\t\t\t作者");
        //遍历bookMap集合
        Set<Map.Entry<String, ArrayList<Book>>> entries = bookMap.entrySet();
        for (Map.Entry<String, ArrayList<Book>> entry : entries) {
            String type = entry.getKey();// 键 图书类型
            ArrayList<Book> bookList = entry.getValue();//值；某种类型下的图书集合
            System.out.println(type);
            //遍历某种类型下的图书
            for (Book b : bookList) {
                String name = b.getName();
                double price = b.getPrice();
                String author = b.getAuthor();
                System.out.println("\t\t" + name + "\t\t\t" + price + "\t\t\t" + author);
            }
        }
    }
    //添加图书方法
    //键盘录入
    /*
     首次添加图书类型， 创建ArrayList集合 存储该类型图书
    把录入的图书对象存进集合ArrayList
       类型已存在，根据类型获取图书集合ArrayList
       把录入的图书对象存进集合ArrayList
     */
    public static void addBook() {
        //键盘录入图书类型
        Scanner sc = new Scanner(System.in);
        System.out.println("请输入要添加书籍的类型");
        String type = sc.next();
        ArrayList<Book> list = bookMap.get(type);
        if (list == null) {
            //首次添加的图书类型，创建ArrayList<>集合存进去
            list = new ArrayList<>();
        }
        //录入图书信息
        System.out.println("请输入要添加的书名");
        String name = sc.next();
        //如果书名已经存在不要添加
        // 遍历集合
        for (Book b : list) {
            if (b.getName().equals(name)) {
                System.out.println("书名已经存在，请重新输入");
                return;
            }
        }
        System.out.println("请输入添加书的价格");
        double price = sc.nextDouble();
        System.out.println("请输入添加书的作者");
        String author = sc.next();
        Book b = new Book(name, price, author);
        //把录入的图书队象存进ArrayList集合
        list.add(b);
        //将类型和图书集合添加回bookMap
        bookMap.put(type, list);
        System.out.println("添加" + name + "成功");
    }
    //删除图书
    /*
    录入信息  类型不存在  结束方法
            类型 存在   录入书名  书名不存在  结束方法
                                   存在   根据类型或取该类型图书集合 遍历集合  找到要删除的书  调用remove方法进行删除
     */
    public static void delBook() {
        //录入图书类型
        Scanner sc = new Scanner(System.in);
        System.out.println("请输入要删除书籍的类型");
        String type = sc.next();
        if (bookMap.containsKey(type) == false) {
            System.out.println("类型不存在，删除失败");
            return;
        }
        System.out.println("请输入要删除的书名");
        String name = sc.next();
        //获取该类型图书
        ArrayList<Book> list = bookMap.get(type);
        for (int i = 0; i < list.size(); i++) {
            Book b = list.get(i);
            if (b.getName().equals(name)) {
                list.remove(i);
                System.out.println("删除成功");
                return;
            }
        }
        System.out.println("删除失败，书名不存在");
    }
    //修改图书
    public static void reviseBook() {
        Scanner sc = new Scanner(System.in);
        System.out.println("请输入你要修改的书籍类型");
        String type = sc.next();
        if (bookMap.containsKey(type) == false) {
            System.out.println("类型不存在，修改失败");
            return;
        }
        System.out.println("请输入修改的书名");
        String name = sc.next();
        ArrayList<Book> list = bookMap.get(type);
        for (int i = 0; i < list.size(); i++) {
            Book book = list.get(i);
            if (book.getName().equals(name)) {
                //执行修改
                System.out.println("请输入新书名");
                String newBook = sc.next();
                System.out.println("请输入新的价格");
                double newPrice = sc.nextDouble();
                System.out.println("请输入新的作者");
                String newAuthor = sc.next();
                book.setName(newBook);
                book.setPrice(newPrice);
                book.setAuthor(newAuthor);
                System.out.println("修改成功");
            }
        }
        System.out.println("书名修改失败");
    }
}