better look of result

g-viet · Jul 2, 2019 · 2bb6e62 · 2bb6e62
1 parent 0dd72e3
commit 2bb6e62
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 9 deletions.
diff --git a/src/helper.ts b/src/helper.ts
@@ -18,7 +18,7 @@ const fetchStock = async (stockCode: string): Promise<any> => {
                 headers: {
                     "Content-Type": "application/json"
                 },
-                body: { "code": stockCode, "s": "1", "t": getCurrentTime() }
+                body: { code: stockCode, s: "1", t: getCurrentTime() }
             };
             request(options, (err: any, _res: any, body: any) => {
                 if (err) reject(err);

diff --git a/src/index.ts b/src/index.ts
@@ -19,10 +19,10 @@ program
         StockApi.gets(stocks);
     });
 program
-    .command('stream <stockCode>')
+    .command('stream <stocks...>')
     .description('get streaming stock price')
-    .action((stockCode: string) => {
-        StockApi.stream(stockCode);
+    .action((stocks: string[]) => {
+        StockApi.stream(stocks);
     });
 program
     .command('group [groupCode]')

diff --git a/src/stock_api.ts b/src/stock_api.ts
@@ -4,18 +4,25 @@ import { Stock } from './stock';
 const gets = (stockCodes: string[]) => {
     Stock.printfHeader();
     try {
-        return Promise.all(stockCodes.map((code) => {
-            return Helper.fetchStock(code).then(stock => stock && stock.printf());
-        }));
+        Promise.all(stockCodes.map((code) => {
+            return Helper.fetchStock(code);
+        })).then(stocks =>
+            stocks.map(stock => stock && stock.printf())
+        );
     } catch (err) {
         throw err;
     }
 }
 
-const stream = (stockCode: string) => {
+const stream = (stockCodes: string[]) => {
     Stock.printfHeader();
     setInterval(() => {
-        Helper.fetchStock(stockCode).then(stock => stock && stock.printf());
+        Promise.all(stockCodes.map((code) => {
+            return Helper.fetchStock(code);
+        })).then(stocks =>
+            stocks.map(stock => stock && stock.printf())
+        );
+        // Helper.fetchStock(stockCode).then(stock => stock && stock.printf());
     }, 2000);
 }