[Feat] 주식주문 거래 발생 시 관련 데이터 업데이트 기능 추가

- 매도/매수 발생 시 보유 현금 및 최대 매도가능 주식 수 갱신되도록 기능 구현 Issues #17
codestates-seb · Sep 13, 2023 · 95a63df · 95a63df
1 parent 015e94d
commit 95a63df
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 5 deletions.
diff --git a/client/src/components/StockOrderSection/StockOrder.tsx b/client/src/components/StockOrderSection/StockOrder.tsx
@@ -73,8 +73,8 @@ const StockOrder = ({ corpName }: { corpName: string }) => {
 
   // 주문 실패 케이스 1) 개장시간  2) 가격/거래량 설정
   // 🔴 3시 30분 이후 작업 위해 closingTime 조건 해제
-  // const orderFailureCase01 = nonBusinessDay;
-  const orderFailureCase01 = nonBusinessDay || closingTime;
+  const orderFailureCase01 = nonBusinessDay;
+  // const orderFailureCase01 = nonBusinessDay || closingTime;
   const orderFailureCase02 = orderPrice === 0 || orderVolume === 0;
 
   return (

diff --git a/client/src/hooks/useGetStockOrderRecord.ts b/client/src/hooks/useGetStockOrderRecord.ts
@@ -7,7 +7,7 @@ const url = "http://ec2-13-125-246-160.ap-northeast-2.compute.amazonaws.com:8080
 
 const useGetStockOrderRecord = () => {
   const isLogin = useSelector((state: StateProps) => state.login);
-  const { data, isLoading, isError } = useQuery("record", getOrderRecord, {
+  const { data, isLoading, isError } = useQuery("orderRecord", getOrderRecord, {
     enabled: isLogin === 1,
   });
 

diff --git a/client/src/hooks/useTradeStock.ts b/client/src/hooks/useTradeStock.ts
@@ -1,6 +1,6 @@
 import { useSelector } from "react-redux";
 import { StateProps } from "../models/stateProps";
-import { useMutation } from "react-query";
+import { useMutation, useQueryClient } from "react-query";
 import axios from "axios";
 
 const useTradeStock = () => {
@@ -9,7 +9,14 @@ const useTradeStock = () => {
   const orderPrice = useSelector((state: StateProps) => state.stockOrderPrice);
   const orderVolume = useSelector((state: StateProps) => state.stockOrderVolume);
 
-  const orderRequest = useMutation(() => postOrderRequest(orderType, companyId, orderPrice, orderVolume));
+  const queryClient = useQueryClient();
+  const orderRequest = useMutation(() => postOrderRequest(orderType, companyId, orderPrice, orderVolume), {
+    onSuccess: () => {
+      queryClient.invalidateQueries("cash");
+      queryClient.invalidateQueries("holdingStock");
+      queryClient.invalidateQueries("orderRecord");
+    },
+  });
   return orderRequest;
 };