Handle division by zero (#1398)

2 years ago · 5c6cc4fed5
1 changed files with 4 additions and 2 deletions
--- a/apps/api/src/app/portfolio/portfolio-calculator.ts
+++ b/apps/api/src/app/portfolio/portfolio-calculator.ts
@ -1036,10 +1036,12 @@ export class PortfolioCalculator {
      const transactionInvestment =
        order.type === 'BUY'
          ? order.quantity.mul(order.unitPrice).mul(this.getFactor(order.type))
-          : totalInvestment
+          : totalUnits.gt(0)
+          ? totalInvestment
              .div(totalUnits)
              .mul(order.quantity)
-              .mul(this.getFactor(order.type));
+              .mul(this.getFactor(order.type))
+          : new Big(0);

      if (PortfolioCalculator.ENABLE_LOGGING) {
        console.log('totalInvestment', totalInvestment.toNumber());