refactor: Remove unnecessary whitespace and add TruncateClosedPnL fun…

…ction
rluisr · Feb 17, 2024 · 6701ebe · 6701ebe
1 parent 46a9181
commit 6701ebe
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 0 deletions.
diff --git a/pkg/adapter/gateway/tv_repository.go b/pkg/adapter/gateway/tv_repository.go
@@ -55,3 +55,7 @@ func (r *TVRepository) GetUniqueSymbol() ([]string, error) {
 	err := r.RODB.Model(&domain.Order{}).Distinct("symbol").Find(&symbols).Error
 	return symbols, err
 }
+
+func (r *TVRepository) TruncateClosedPnL() error {
+	return r.RWDB.Exec("TRUNCATE closed_pnl;").Error
+}
diff --git a/pkg/usecase/interfaces/repositories.go b/pkg/usecase/interfaces/repositories.go
@@ -30,6 +30,7 @@ type TVRepository interface {
 	SaveClosedPnL(closedPnL []*domain.ClosedPnL) error
 	UpdateOrder(*domain.Order) error
 	GetUniqueSymbol() ([]string, error)
+	TruncateClosedPnL() error
 }
 
 type BybitRepository interface {

diff --git a/pkg/usecase/tv_interactor.go b/pkg/usecase/tv_interactor.go
@@ -87,6 +87,12 @@ func (i *TVInteractor) CreateOrder(c *gin.Context) (domain.TVOrderResponse, erro
 }
 
 func (i *TVInteractor) FetchPL() error {
+	// TODO TRUNCATE せずに増分更新する
+	err := i.TVRepository.TruncateClosedPnL()
+	if err != nil {
+		return err
+	}
+
 	symbols, err := i.TVRepository.GetUniqueSymbol()
 	if err != nil {
 		return err