Handle division by zero

3 years ago · a30c58cd02
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());