用流解决商品进货问题（菜菜，大佬帮帮）

揽一池星河

每个商城都需要进货，而这些进货记录整理起来很不方便，本案例要求编写一个记录商城进货交易的程序，使用字节流将商场的进货信息记录在本地的csv文件中。程序具体要求如下：
当用户输入商品编号时，后台会根据商品编号查询到相应商品信息，并打印商品信息。接着让用户输入需要进货的商品数量，程序将原有的库存数量与输入的数量相加，将相加后的结果信息保存至本地的csv文件中。

Twilight6

用着笨方法试着写了下，这里 csv 文件结构是这样的：

1. Id,Goods,Inventory
   
2. 1,Banada,10
   
3. 2,Apple,10
   
4. 3,Durain,15
   
5. 4,Orange,10
   
6. 5,Strawberry,10

复制代码

GoodsInventory.java 文件：

1. import java.io.*;
   
2. import java.nio.charset.StandardCharsets;
   
3. import java.util.Map;
   
4. import java.util.Scanner;
   
5. import java.util.TreeMap;
   
6. 
   
7. public class GoodsInventory {
   
8.     private Map<Integer, Object[]> map = new TreeMap<>();
   
9.     private Object[] head;
   
10. 
    
11.     public boolean writeFile(String filePath, Map<Integer, Object[]> map) {
    
12.         FileOutputStream fos = null;
    
13.         try {
    
14.             fos = new FileOutputStream(filePath);
    
15.             fos.write(String.format("%s,%s,%s\n", head).getBytes(StandardCharsets.UTF_8));
    
16.             for (Integer id : map.keySet()){
    
17.                 fos.write(String.format("%s,%s,%s\n", map.get(id)).getBytes(StandardCharsets.UTF_8));
    
18.             }
    
19.             System.out.println("库存文件更新成功！");
    
20.         } catch (IOException e) {
    
21.             throw new RuntimeException(e);
    
22.         } finally {
    
23.             if (fos != null){
    
24.                 try {
    
25.                     fos.close();
    
26.                 } catch (IOException e) {
    
27.                     throw new RuntimeException(e);
    
28.                 }
    
29.             }
    
30.         }
    
31.         return false;
    
32.     }
    
33. 
    
34. 
    
35.     public boolean updateInventoryById(Integer id, Integer count){
    
36.         Object[] obj = map.get(id);
    
37.         if (obj == null){
    
38.             return false;
    
39.         }
    
40.         Integer temp = (Integer) obj[2] + count;
    
41.         if (temp < 0){
    
42.             System.out.println("库存不足！");
    
43.             return false;
    
44.         }
    
45.         obj[2] = (Integer) obj[2] + count;
    
46.         return true;
    
47.     }
    
48. 
    
49. 
    
50.     public Map<Integer, Object[]> readFile(String fileName){
    
51.         FileInputStream fis = null;
    
52.         BufferedReader br = null;
    
53.         try {
    
54.             fis = new FileInputStream(fileName);
    
55.             br = new BufferedReader(new InputStreamReader(fis));
    
56. 
    
57.             String data;
    
58.             String[] str;
    
59.             if ((data = br.readLine()) != null){
    
60.                 str = data.split(",");
    
61.                 head = str;
    
62.             }
    
63.             while ((data = br.readLine()) != null) {
    
64.                 str = data.split(",");
    
65.                 Object[] temp = {str[0], str[1], Integer.parseInt(str[2])};
    
66.                 map.put(Integer.parseInt(str[0]), temp);
    
67.             }
    
68.         } catch (Exception e) {
    
69.             throw new RuntimeException("文件未找到，请检查路径或文件名是否正确！");
    
70.         } finally {
    
71.             try {
    
72.                 if (br != null){
    
73.                     br.close();
    
74.                 } else if (fis != null)
    
75.                     fis.close();
    
76.             } catch (Exception e) {
    
77.                 throw new RuntimeException(e);
    
78.             }
    
79.         }
    
80.         return map;
    
81.     }
    
82. 
    
83.     public boolean printGoodsByIdData(Integer id){
    
84.         Object[] obj = map.get(id);
    
85.         if (obj == null){
    
86.             System.out.println("错误：没有找到此ID商品项！");
    
87.             System.out.print("请重新输入商品ID: ");
    
88.             return false;
    
89.         }
    
90.         System.out.printf("%s\t %s\t %s\n", head);
    
91.         System.out.printf("%s\t %s\t %s\n", map.get(id));
    
92.         return true;
    
93.     }
    
94. }

复制代码

GoodsInventoryTest.java 测试程序：

1. import java.util.Map;
   
2. import java.util.Scanner;
   
3. 
   
4. public class GoodsInventoryTest {
   
5. 
   
6.     public static void main(String[] args) {
   
7.         Scanner scan = new Scanner(System.in);
   
8. 
   
9.         String filePath = "./src/pers/uang/demo05/GoodsInventory.csv";
   
10.         GoodsInventory inventory = new GoodsInventory();
    
11.         Map<Integer , Object[]> map = inventory.readFile(filePath);
    
12.         int id = -1;
    
13.         boolean flag = false;
    
14.         boolean isNew = false;
    
15.         while (!flag){
    
16.             System.out.print("请输入商品ID: ");
    
17.             while (!flag) {
    
18.                 id = scan.nextInt();
    
19.                 scan.nextLine();
    
20.                 flag = inventory.printGoodsByIdData(id);
    
21.             }
    
22.             System.out.print("\n请输入需要更新的库存数量(负号表示库存减少):");
    
23.             int count = scan.nextInt();
    
24.             scan.nextLine();
    
25.             if (inventory.updateInventoryById(id, count)) {
    
26.                 isNew = true;
    
27.                 System.out.println("\n更新成功！");
    
28.             } else {
    
29.                 System.out.println("\n更新失败！");
    
30.             }
    
31.             System.out.print("还需要更新其它商品的库存记录吗(回车则继续, 任意输入退出系统): ");
    
32.             String isExit = scan.nextLine();
    
33.             if ("".equals(isExit)){
    
34.                 flag = false;
    
35.             }
    
36.         }
    
37. 
    
38.         // 写入文件中去
    
39.         if (isNew){
    
40.             inventory.writeFile(filePath, map);
    
41.         } else {
    
42.             System.out.println("库存暂未更新, 无需写入！");
    
43.         }
    
44.     }
    
45. }

复制代码