🟨 [Java] 물품 재고 관리 프로그램 만들기

자바

물품 재고 관리 프로그램 만들기

아직 클래스와 상속을 배우지 않은 상태에서 재고 관리 프로그램을 만들어 보도록 하자.

원래 DB를 연결해야 하지만 지금까지 배운 과정으로는 연결할 수 없기 때문에 그 과정은 조금 더 배우고 나면 관리자와 구매자를 나누고 DB를 연결해서 제대로된 프로그램을 만들어 다시 다루도록 하겠다.

 

코드를 하나씩 살펴보자.

static final int MAX = 100; // 배열 최대 크기

   static int[] productNumber = new int[MAX]; // 물품 번호
   static String[] productName = new String[MAX]; // 물품 이름
   static int[] productPrice = new int[MAX]; // 물품 가격
   static int[] productStock = new int[MAX]; // 물품 재고

   static int productCount = 0; // 총 물품 개수

   static int count = 0; // 처음 개수

   static Scanner sc = new Scanner(System.in);

먼저 static 변수로 물품 번호 / 물품 이름 / 물품 가격 / 재고를 선언해준다. 

여기서 count 는 물품 종류의 개수라고 보면 된다. 인덱스를 찾을 때 사용할 변수이다.

물품 등록 메서드

// 물품 등록
   private static void createProduct() {
      int idx = checkProductByProductNumber();
      if (idx == -1) {
         return;
      }

      System.out.print("물품 이름: ");
      productName[count] = sc.next();
      sc.nextLine();

      System.out.print("물품 가격: ");
      productPrice[count] = sc.nextInt();

      System.out.print("물품 재고: ");
      productStock[count] = sc.nextInt();

      System.out.println("등록이 완료되었습니다.");
      System.out.println();
      count++;
      productCount++;
   }

처음 count 가 0부터 시작이므로 처음 물품을 등록하게 되면 product_[0]에 제품이 등록된다. 

여기서 checkProductByProductNumber() 는 물품의 이름으로 물품이 존재하는지 구하는 메서드이다. 

따로 구현을 안해주어도 되지만 함수로 구현을 해주었다. 

   // 물품 번호로 물품이 이미 존재하는지 찾는 메서드
   private static int checkProductByProductNumber() {
      System.out.print("물품 번호: ");
      productNumber[count] = sc.nextInt();
      // 이미 존재하는지 확인
      for (int i = 0; i < count; i++) {
         if (productNumber[i] == productNumber[count]) {
            System.out.println("이미 존재하는 물품입니다.");
            System.out.println();
            return -1;
         }
      }
      return 0;
   }

물품 번호를 입력하고 그 제품 번호가 존재한다면 -1을 리턴하고 그 외의 경우는 0을 리턴한다. 

 

물품의 이름으로 물품의 인덱스를 구하는 메서드

   // 물품의 이름으로 물품의 인덱스를 구하는 메서드
   private static int getProductIndexByProductName() {
      System.out.print("물품의 이름을 입력하세요: ");
      String name = sc.next();
      int idx = -1;
      for (int i = 0; i < count; i++) {
         if (productName[i].equals(name)) {
            idx = i;
         }
      }
      return idx;
   }

만약 물품이 없다면 idx를 제대로 된 인덱스를 반환하고 그렇지 않다면 idx = -1 을 그대로 반환해준다. 

아래에서도 이 함수가 또 쓰이기 때문에 두 번 구현할 필요없이 이 함수를 가져다 쓰면 된다. 

효율성을 따져보면 좋은지는 모르겠지만 아직까지 효율성까지는 비교할 필요가 없다고 생각해서 이렇게 구현을 하였다. 

물품 전체 리스트 조회

   // 물품 전체 리스트 조회
   private static void readAllProduct() {
      for (int i = 0; i < count; i++) {
         System.out.println(productNumber[i] + "번 물품 " + productName[i] + "의 조회 결과");
         System.out.println("물품 번호: " + productNumber[i]);
         System.out.println("물품 이름: " + productName[i]);
         System.out.println("물품 가격: " + productPrice[i]);
         System.out.println("물품 재고: " + productStock[i]);
         System.out.println();
      }
   }

전체 물품을 탐색하면서 print 해준다. 범위는 i=0 부터 count-1 까지이다. (이는 배열의 인덱스가 마지막-1 까지 이므로)

특정 물품 재고 조회

   // 특정 물품 재고 조회
   private static void readProductStock() {
      System.out.print("재고 조회 입니다. 물품 이름을 입력해 주세요: ");
      String item = sc.next();
      int t = 0;
      for (int i = 0; i < count; i++) {
         if (productName[i].equals(item)) {
            System.out.println(item + "의 재고는 " + productStock[i] + "입니다.");
            System.out.println();
            t += 1;
         }
      }
      if (t == 0) {
         System.out.println("물품 이름을 잘 못 입력하였습니다.");
         System.out.println();
      }
   }

물품의 이름으로 재고를 조회하는 메서드이다. 모두 탐색을 해서 재고를 알려준다. 시간복잡도가 O(n)으로 더 효율적인 방법이 있겠지만 아직까지는 따지지 않고 구현을 할 것이다. 나중에 API를 만들게 되면 이러한 부분들을 신경써야 할 것이다.

여기서 t의 역할은 물품을 제대로 입력했는지 확인하는 역할이다. 물품을 모두 조회 했는데 입력한 물품 이름이 탐색되지 않았다면 이름을 잘못입력한 것이므로 t로 확인을 해주도록 한다. 

 

물품 수정

  // 물품 수정
   private static void updateProduct() {
      int idx = getProductIndexByProductName();

      if (idx == -1) {
         System.out.println("물품 이름을 잘 못 입력하였습니다.");
         return; // 함수 종료
      }

      System.out.println();
      System.out.println("---------UPDATE   PRODUCT INFO----------");
      System.out.print("현재 물품 번호(" + productNumber[idx] + ") :");
      productNumber[idx] = sc.nextInt();

      System.out.print("현재 물품 이름(" + productName[idx] + ") :");
      productName[idx] = sc.next();
      sc.nextLine();
      System.out.print("현재 물품 가격(" + productPrice[idx] + ") :");
      productPrice[idx] = sc.nextInt();

      System.out.print("현재 물품 재고(" + productStock[idx] + ") :");
      productStock[idx] = sc.nextInt();
      System.out.println();
   }

여기서 다시한번 인덱스가 사용된다. getProductIndexByProductName의 메서드를 사용해 인덱스를 가져온다. 

반환 값이 -1이면 물품 이름을 잘못입력한 것이고 그 외의 경우는 물품을 업데이트 해준다. 

여기서 또한 업데이트 방법이 여러가지가 있겠지만 나는 한번에 모두 물품을 수정하는 방법을 택했다. 

예를 들어 회원정보 수정에서 핸드폰번호만 바꾸는 경우, 이메일만 바꾸는 경우 ,, 이런 식으로 하나씩 API를 구성해도 되지만 한 번에 정보를 수정하고 넘겨주는 로직으로 만든 것이다. 

 

물품 삭제 

   // 물품 삭제
   private static void deleteProduct() {
      int idx = getProductIndexByProductName();

      if (idx == -1) {
         System.out.println("물품 이름을 잘 못 입력하였습니다.");
         return; // 함수 종료
      }

      for (int i = idx; i < count - 1; i++) {
         productNumber[i] = productNumber[i + 1];
         productName[i] = productName[i + 1];
         productPrice[i] = productPrice[i + 1];
         productStock[i] = productStock[i + 1];
      }
      count--;
      productCount--;
      System.out.println("물품 삭제가 완료되었습니다.");
   }

위에서 구현한 getProductIndexByProductName 메서드를 다시 한 번 사용한다. 이렇게 함수를 한 번 구현하고 두 번을 사용하게 되면 가독성도 그렇고 개발시에 편리함이 있다. 나는 함수로 기능을 구현해놓고 계속 쓰는 방식이 편한 것 같아서 사용하였다. 

물품의 삭제는 인덱스를 찾고 그 인덱스의 값에 다음 인덱스의 값을 넣어주는 것이다. 그러면 다음 인덱스가 앞으로 넘어오면서 값이 덮어 쓰이게 된다. 그리고 물품의 개수를 1개 줄여주면(count--) 뒤의 값이 그대로 있더라도 배열의 끝이 줄어들기 때문에 탐색을 할 때 맨 뒤까지 하지 않게된다. 

1 ,2, 3, 4, 5가 있다고 하자. 3번 물품을 삭제하려고 하면 1, 2, 4, 5가 되어야 한다. 

이는 1, 2번은 그대로 놔두고 3, 4, 5번만 바꿔주면 되는데 3, 4, 5에서 한칸씩 앞으로 가져오면 된다. 그러면 4, 5, 5 가 될 것이다. 마지막 5는 어차피 count--가 되므로 어떠한 값이 있더라도 탐색을 하지 않게 되는것이다. 그림으로 그려서 이해하면 더 쉬울 것이다. 하지만 귀찮아서

 

메인 함수

   public static void main(String[] args) {
      while (true) {
         System.out.printf("총 물품 개수 : %d\n", productCount);
         System.out.print("메뉴를 입력하세요.");
         System.out.println(
               "(I)nsert, (A)llProductRead, (S)tockCheck, (U)pdateProduct, (D)eleteProduct, (Q)uit");
         System.out.print("메뉴 입력: ");
         String menu = sc.next();
         menu = menu.toLowerCase();
         switch (menu.charAt(0)) {
         case 'i':
            System.out.println("물품 정보를 입력해 주세요.");
            if (count > MAX) {
               System.out.println("물품 등록 개수를 초과하였습니다.");
            } else {
               createProduct();
            }
            break;
         case 'a':
            // 전체 물품 조회
            System.out.println("전체 물품 조회입니다.");
            readAllProduct();
            break;
         case 's':
            // 물품 재고 조회
            readProductStock();
            break;
         case 'u':
            // 물품 이름을 기준으로 업데이트
            System.out.println("물품을 업데이트 합니다.");
            updateProduct();
            break;
         case 'd':
            System.out.println("물품을 삭제 합니다.");
            deleteProduct();
            break;
         case 'q':
            System.out.println("프로그램을 종료합니다.");
            sc.close(); // Scanner 객체를 닫아줍니다.
            System.exit(0); // 프로그램을 종료시킵니다.
            break;
         default:
            System.out.println("메뉴를 잘못 입력하였습니다.");

         }
      }
   }

while 문에서 무한루프를 돌게 된다. q가 입력되면 프로그램을 종료하고 어떠한 입력값이 들어오냐에 따라 나누어 진다. 

문자열을 입력받아 첫번째 글자가 어떤 글자냐에 따라 case문에 들어가 실행 후 종료된다. 

 

전체 코드

package Assignment2;

import java.util.Scanner;

public class ProducStockManagementSystem {
   static final int MAX = 100; // 배열 최대 크기

   static int[] productNumber = new int[MAX]; // 물품 번호
   static String[] productName = new String[MAX]; // 물품 이름
   static int[] productPrice = new int[MAX]; // 물품 가격
   static int[] productStock = new int[MAX]; // 물품 재고

   static int productCount = 0; // 총 물품 개수

   static int count = 0; // 처음 개수

   static Scanner sc = new Scanner(System.in);

   // 물품의 이름으로 물품의 인덱스를 구하는 메서드
   private static int getProductIndexByProductName() {
      System.out.print("물품의 이름을 입력하세요: ");
      String name = sc.next();
      int idx = -1;
      for (int i = 0; i < count; i++) {
         if (productName[i].equals(name)) {
            idx = i;
         }
      }
      return idx;
   }

   // 물품 번호로 물품이 이미 존재하는지 찾는 메서드
   private static int checkProductByProductNumber() {
      System.out.print("물품 번호: ");
      productNumber[count] = sc.nextInt();
      // 이미 존재하는지 확인
      for (int i = 0; i < count; i++) {
         if (productNumber[i] == productNumber[count]) {
            System.out.println("이미 존재하는 물품입니다.");
            System.out.println();
            return -1;
         }
      }
      return 0;
   }

   // 물품 등록
   private static void createProduct() {
      int idx = checkProductByProductNumber();
      if (idx == -1) {
         return;
      }

      System.out.print("물품 이름: ");
      productName[count] = sc.next();
      sc.nextLine();

      System.out.print("물품 가격: ");
      productPrice[count] = sc.nextInt();

      System.out.print("물품 재고: ");
      productStock[count] = sc.nextInt();

      System.out.println("등록이 완료되었습니다.");
      System.out.println();
      count++;
      productCount++;
   }

   // 물품 전체 리스트 조회
   private static void readAllProduct() {
      for (int i = 0; i < count; i++) {
         System.out.println(productNumber[i] + "번 물품 " + productName[i] + "의 조회 결과");
         System.out.println("물품 번호: " + productNumber[i]);
         System.out.println("물품 이름: " + productName[i]);
         System.out.println("물품 가격: " + productPrice[i]);
         System.out.println("물품 재고: " + productStock[i]);
         System.out.println();
      }
   }

   // 특정 물품 재고 조회
   private static void readProductStock() {
      System.out.print("재고 조회 입니다. 물품 이름을 입력해 주세요: ");
      String item = sc.next();
      int t = 0;
      for (int i = 0; i < count; i++) {
         if (productName[i].equals(item)) {
            System.out.println(item + "의 재고는 " + productStock[i] + "입니다.");
            System.out.println();
            t += 1;
         }
      }
      if (t == 0) {
         System.out.println("물품 이름을 잘 못 입력하였습니다.");
         System.out.println();
      }
   }

   // 물품 수정
   private static void updateProduct() {
      int idx = getProductIndexByProductName();

      if (idx == -1) {
         System.out.println("물품 이름을 잘 못 입력하였습니다.");
         return; // 함수 종료
      }

      System.out.println();
      System.out.println("---------UPDATE   PRODUCT INFO----------");
      System.out.print("현재 물품 번호(" + productNumber[idx] + ") :");
      productNumber[idx] = sc.nextInt();

      System.out.print("현재 물품 이름(" + productName[idx] + ") :");
      productName[idx] = sc.next();
      sc.nextLine();
      System.out.print("현재 물품 가격(" + productPrice[idx] + ") :");
      productPrice[idx] = sc.nextInt();

      System.out.print("현재 물품 재고(" + productStock[idx] + ") :");
      productStock[idx] = sc.nextInt();
      System.out.println();
   }

   // 물품 삭제
   private static void deleteProduct() {
      int idx = getProductIndexByProductName();

      if (idx == -1) {
         System.out.println("물품 이름을 잘 못 입력하였습니다.");
         return; // 함수 종료
      }

      for (int i = idx; i < count - 1; i++) {
         productNumber[i] = productNumber[i + 1];
         productName[i] = productName[i + 1];
         productPrice[i] = productPrice[i + 1];
         productStock[i] = productStock[i + 1];
      }
      count--;
      productCount--;
      System.out.println("물품 삭제가 완료되었습니다.");
   }


   public static void main(String[] args) {
      while (true) {
         System.out.printf("총 물품 개수 : %d\n", productCount);
         System.out.print("메뉴를 입력하세요.");
         System.out.println(
               "(I)nsert, (A)llProductRead, (S)tockCheck, (U)pdateProduct, (D)eleteProduct, (Q)uit");
         System.out.print("메뉴 입력: ");
         String menu = sc.next();
         menu = menu.toLowerCase();
         switch (menu.charAt(0)) {
         case 'i':
            System.out.println("물품 정보를 입력해 주세요.");
            if (count > MAX) {
               System.out.println("물품 등록 개수를 초과하였습니다.");
            } else {
               createProduct();
            }
            break;
         case 'a':
            // 전체 물품 조회
            System.out.println("전체 물품 조회입니다.");
            readAllProduct();
            break;
         case 's':
            // 물품 재고 조회
            readProductStock();
            break;
         case 'u':
            // 물품 이름을 기준으로 업데이트
            System.out.println("물품을 업데이트 합니다.");
            updateProduct();
            break;
         case 'd':
            System.out.println("물품을 삭제 합니다.");
            deleteProduct();
            break;
         case 'q':
            System.out.println("프로그램을 종료합니다.");
            sc.close(); // Scanner 객체를 닫아줍니다.
            System.exit(0); // 프로그램을 종료시킵니다.
            break;
         default:
            System.out.println("메뉴를 잘못 입력하였습니다.");

         }
      }
   }

}

짧은 시간안에 간단하게 짜다보니 오류가 있을수도 있습니다. 문제점 또는 개선방안이 있다면 댓글로 부탁드립니다.