#33 [refactor] : 업데이트 기능 추가 & 소수 자리수 늘림

CSID-DGU · Jun 17, 2024 · 5000085 · 5000085
1 parent dec8fe7
commit 5000085
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 4 deletions.
diff --git a/backend/src/main/java/org/dgu/backend/domain/TradingOption.java b/backend/src/main/java/org/dgu/backend/domain/TradingOption.java
@@ -40,7 +40,7 @@ public class TradingOption extends BaseEntity {
     @Column(name = "trading_unit_price", nullable = false)
     private Long tradingUnitPrice;
 
-    @Column(name = "coin_count", nullable = false, scale = 10)
+    @Column(name = "coin_count", nullable = false, precision = 30, scale = 15)
     private BigDecimal coinCount;
 
     @Column(name = "trading_count", nullable = false)
@@ -56,24 +56,40 @@ public class TradingOption extends BaseEntity {
     private LocalDateTime endDate;
 
     @Builder
-    public TradingOption(User user, Portfolio portfolio, Long initialCapital, Long currentCapital, Long tradingUnitPrice, int tradingCount, int buyingCount, LocalDateTime startDate, LocalDateTime endDate) {
+    public TradingOption(User user, Portfolio portfolio, Long initialCapital, Long currentCapital, Long tradingUnitPrice, int tradingCount, LocalDateTime startDate, LocalDateTime endDate) {
         this.user = user;
         this.portfolio = portfolio;
         this.initialCapital = initialCapital;
         this.currentCapital = currentCapital;
         this.tradingUnitPrice = tradingUnitPrice;
         this.coinCount = BigDecimal.ZERO;
         this.tradingCount = tradingCount;
-        this.buyingCount = buyingCount;
+        this.buyingCount = 0;
         this.startDate = startDate;
         this.endDate = endDate;
     }
 
+    public void updateCurrentCapital(Long currentCapital) {
+        this.currentCapital = currentCapital;
+    }
+
+    public void updateInitialCapital(Long initialCapital) {
+        this.initialCapital = initialCapital;
+    }
+
     public void updateAvgPrice(Double avgPrice) {
         this.avgPrice = avgPrice;
     }
 
     public void updateCoinCount(BigDecimal coinCount) {
         this.coinCount = coinCount;
     }
+
+    public void resetBuyingCount() {
+        this.buyingCount= 0;
+    }
+
+    public void plusBuyingCount() {
+        this.buyingCount++;
+    }
 }
diff --git a/backend/src/main/java/org/dgu/backend/dto/TradingDto.java b/backend/src/main/java/org/dgu/backend/dto/TradingDto.java
@@ -34,7 +34,6 @@ public TradingOption to(User user, Portfolio portfolio, PortfolioOption portfoli
                     .currentCapital(fund)
                     .tradingUnitPrice(fund / portfolioOption.getTradingUnit())
                     .tradingCount(portfolioOption.getTradingUnit())
-                    .buyingCount(0)
                     .startDate(startDate)
                     .endDate(endDate)
                     .build();