HealthCheckController.ts

import { Controller, Get } from '@nestjs/common';
import HealthCheckResponse from '../domain/HealthCheckResponse';
import STS from 'aws-sdk/clients/sts';
import { ArcUnhealthyException } from '../error/ArchUnhealthyException';

@Controller('/health')
export default class HealthCheckController {

  private readonly secureTokenService: STS;

  constructor(secureTokenService: STS) {
    this.secureTokenService = secureTokenService;
  }

  @Get()
  async getHealth(): Promise<HealthCheckResponse> {
    try {
      const getCallerIdentityResponse = await this.secureTokenService.getCallerIdentity().promise()
      return {
        status: 'healthy',
        arn: getCallerIdentityResponse.Arn
      }
    } catch (e) {
      throw new ArcUnhealthyException({
        status: 'unhealthy',
        msg: e
      })
    }
  }
}