rds.go

/******************************************************************************
Cloud Resource Counter
File: rds.go

Summary: Provides a count of all RDS instances.
******************************************************************************/

package main

import (
	"github.com/aws/aws-sdk-go/service/rds"

	color "github.com/logrusorgru/aurora"
)

// RDSInstances retrieves the count of all RDS Instances either for all regions
// (allRegions is true) or the region associated with the session. This method
// gives status back to the user via the supplied ActivityMonitor instance.
func RDSInstances(sf ServiceFactory, am ActivityMonitor, allRegions bool) int {
	// Indicate activity
	am.StartAction("Retrieving RDS instance counts")

	// Should we get the counts for all regions?
	instanceCount := 0
	if allRegions {
		// Get the list of all enabled regions for this account
		regionsSlice := GetEC2Regions(sf.GetEC2InstanceService(""), am)

		// Loop through all of the regions
		for _, regionName := range regionsSlice {
			// Get the RDS instance counts for a specific region
			instanceCount += rdsInstancesForSingleRegion(sf.GetRDSInstanceService(regionName), am)
		}
	} else {
		// Get the RDS instance counts for the region selected by this session
		instanceCount = rdsInstancesForSingleRegion(sf.GetRDSInstanceService(""), am)
	}

	// Indicate end of activity
	am.EndAction("OK (%d)", color.Bold(instanceCount))

	return instanceCount
}

func rdsInstancesForSingleRegion(rdsis *RDSInstanceService, am ActivityMonitor) int {
	// Construct our input to find all RDS instances
	input := &rds.DescribeDBInstancesInput{}

	// Indicate activity
	am.Message(".")

	// Invoke our service
	instanceCount := 0
	err := rdsis.InspectInstances(input, func(page *rds.DescribeDBInstancesOutput, lastPage bool) bool {
		// Loop through the DB Instances...
		for _, dbi := range page.DBInstances {
			if dbi.DBInstanceStatus != nil && *dbi.DBInstanceStatus == "available" {
				instanceCount++
			}
		}

		return true
	})

	// Check for error
	am.CheckError(err)

	return instanceCount
}