Menu.java

import java.util.Scanner;

public class Menu {
    public static void main(String args[]) {
        try {
            // step1 load the driver class
            Class.forName("com.mysql.jdbc.Driver");

            Scanner sc = new Scanner(System.in);
            // int pid, Qty, update, t=0;
            char ch;
            do {
                System.out.print("User choice:\n1.Admin\n2.Customer\nEnter choice: ");
                int c1 = sc.nextInt();
                if (c1 == 1) {
                    Admin oba = new Admin();
                    System.out.print(
                            "Admin functions:\n1.Add a new product\n2.Update inventory\n3.Change price\n4.Delete a product\n5.View all orders for a certain product\nEnter choice: ");
                    int c2 = sc.nextInt();

                    switch (c2) {
                        case 1:
                            oba.Add_Product();
                            break;
                        case 2:
                            oba.Update_Inventory();
                            break;
                        case 3:
                            oba.Change_Price();
                            break;
                        case 4:
                            oba.Delete_Product();
                            break;
                        case 5:
                            oba.View_Orders();
                            break;
                        default:
                            System.out.println("Invalid choice!");
                    }
                } else if (c1 == 2) {
                    Customers obc = new Customers();
                    System.out.println("Are you a returning customer?");
                    ch = sc.next().charAt(0);
                    if (ch == 'y')
                        obc.Order(1);

                    else
                        obc.Order(0);
                }
                System.out.println("Do you want to continue: ");
                ch = sc.next().charAt(0);
                System.out.println();
            } while (ch == 'y');
            sc.close();
        } catch (Exception e) {
            System.out.println(e);
        }
    }

}