83 lines
2.2 KiB
Go
83 lines
2.2 KiB
Go
package repository
|
|
|
|
import (
|
|
"github.com/aws/aws-sdk-go/aws/session"
|
|
"github.com/aws/aws-sdk-go/service/rds"
|
|
"github.com/aws/aws-sdk-go/service/rds/rdsiface"
|
|
"github.com/cloudskiff/driftctl/pkg/remote/cache"
|
|
)
|
|
|
|
type RDSRepository interface {
|
|
ListAllDBInstances() ([]*rds.DBInstance, error)
|
|
ListAllDBSubnetGroups() ([]*rds.DBSubnetGroup, error)
|
|
ListAllDBClusters() ([]*rds.DBCluster, error)
|
|
}
|
|
|
|
type rdsRepository struct {
|
|
client rdsiface.RDSAPI
|
|
cache cache.Cache
|
|
}
|
|
|
|
func NewRDSRepository(session *session.Session, c cache.Cache) *rdsRepository {
|
|
return &rdsRepository{
|
|
rds.New(session),
|
|
c,
|
|
}
|
|
}
|
|
|
|
func (r *rdsRepository) ListAllDBInstances() ([]*rds.DBInstance, error) {
|
|
if v := r.cache.Get("rdsListAllDBInstances"); v != nil {
|
|
return v.([]*rds.DBInstance), nil
|
|
}
|
|
|
|
var result []*rds.DBInstance
|
|
input := &rds.DescribeDBInstancesInput{}
|
|
err := r.client.DescribeDBInstancesPages(input, func(res *rds.DescribeDBInstancesOutput, lastPage bool) bool {
|
|
result = append(result, res.DBInstances...)
|
|
return !lastPage
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
r.cache.Put("rdsListAllDBInstances", result)
|
|
return result, nil
|
|
}
|
|
|
|
func (r *rdsRepository) ListAllDBSubnetGroups() ([]*rds.DBSubnetGroup, error) {
|
|
if v := r.cache.Get("rdsListAllDBSubnetGroups"); v != nil {
|
|
return v.([]*rds.DBSubnetGroup), nil
|
|
}
|
|
|
|
var subnetGroups []*rds.DBSubnetGroup
|
|
input := rds.DescribeDBSubnetGroupsInput{}
|
|
err := r.client.DescribeDBSubnetGroupsPages(&input,
|
|
func(resp *rds.DescribeDBSubnetGroupsOutput, lastPage bool) bool {
|
|
subnetGroups = append(subnetGroups, resp.DBSubnetGroups...)
|
|
return !lastPage
|
|
},
|
|
)
|
|
|
|
r.cache.Put("rdsListAllDBSubnetGroups", subnetGroups)
|
|
return subnetGroups, err
|
|
}
|
|
|
|
func (r *rdsRepository) ListAllDBClusters() ([]*rds.DBCluster, error) {
|
|
cacheKey := "rdsListAllDBClusters"
|
|
if v := r.cache.Get(cacheKey); v != nil {
|
|
return v.([]*rds.DBCluster), nil
|
|
}
|
|
|
|
var clusters []*rds.DBCluster
|
|
input := rds.DescribeDBClustersInput{}
|
|
err := r.client.DescribeDBClustersPages(&input,
|
|
func(resp *rds.DescribeDBClustersOutput, lastPage bool) bool {
|
|
clusters = append(clusters, resp.DBClusters...)
|
|
return !lastPage
|
|
},
|
|
)
|
|
|
|
r.cache.Put(cacheKey, clusters)
|
|
return clusters, err
|
|
}
|