@cloud-carbon-footprint/aws/src/lib/RDS.ts

The core logic to get cloud usage data and estimate energy and carbon emissions from Amazon Web Services.

/*
 * © 2021 Thoughtworks, Inc.
 */

import moment from 'moment'
import { reduceBy, concat } from 'ramda'
import {
  ICloudService,
  FootprintEstimate,
  Cost,
  CloudConstantsEmissionsFactors,
  CloudConstants,
} from '@cloud-carbon-footprint/core'
import RDSComputeService from './RDSCompute'
import RDSStorage from './RDSStorage'

export default class RDS implements ICloudService {
  serviceName = 'RDS'

  constructor(
    private rdsComputeService: RDSComputeService,
    private rdsStorageService: RDSStorage,
  ) {}

  async getEstimates(
    start: Date,
    end: Date,
    region: string,
    emissionsFactors: CloudConstantsEmissionsFactors,
    constants: CloudConstants,
  ): Promise<FootprintEstimate[]> {
    const rdsComputeEstimates = this.rdsComputeService.getEstimates(
      start,
      end,
      region,
      emissionsFactors,
      constants,
    )
    const rdsStorageEstimates = this.rdsStorageService.getEstimates(
      start,
      end,
      region,
      emissionsFactors,
      constants,
    )
    const resolvedEstimates: FootprintEstimate[][] = await Promise.all([
      rdsComputeEstimates,
      rdsStorageEstimates,
    ])
    const combinedEstimates: FootprintEstimate[] = resolvedEstimates.flat()

    interface CalculatedFootprintEstimate {
      timestamp: Date
      co2e: number
      kilowattHours: number
    }

    const result: { [key: number]: CalculatedFootprintEstimate } = {}

    combinedEstimates.forEach((estimate) => {
      const timestamp: number = estimate.timestamp.getTime()
      if (result[timestamp]) {
        result[timestamp].co2e += estimate.co2e
        result[timestamp].kilowattHours += estimate.kilowattHours
      } else {
        result[timestamp] = {
          timestamp: estimate.timestamp,
          co2e: estimate.co2e,
          kilowattHours: estimate.kilowattHours,
        }
      }
    })

    return Object.values(result)
  }

  async getCosts(start: Date, end: Date, region: string): Promise<Cost[]> {
    const rdsComputeCosts = await this.rdsComputeService.getCosts(
      start,
      end,
      region,
    )
    const rdsStorageCosts = await this.rdsStorageService.getCosts(
      start,
      end,
      region,
    )

    const rdsCosts = concat(rdsComputeCosts, rdsStorageCosts)

    const groupingFn = (cost: Cost) => {
      return moment(cost.timestamp).utc().format('YYYY-MM-DD')
    }

    const accumulatingFn = (accumulator: Cost, cost: Cost): Cost => {
      accumulator.timestamp =
        accumulator.timestamp ||
        new Date(moment(cost.timestamp).utc().format('YYYY-MM-DD'))
      accumulator.amount += cost.amount
      accumulator.currency = cost.currency || 'USD'
      return accumulator
    }

    return Object.values(
      reduceBy(
        accumulatingFn,
        { amount: 0, currency: undefined, timestamp: undefined },
        groupingFn,
        rdsCosts,
      ),
    )
  }
}