Skip to content
This repository has been archived by the owner on Oct 15, 2024. It is now read-only.

Commit

Permalink
Add support for Redshift Serverless namespaces, snapshots and workgro…
Browse files Browse the repository at this point in the history
…ups. (#1194)
  • Loading branch information
mdirkse authored Feb 23, 2024
1 parent d6b1374 commit 566bb86
Show file tree
Hide file tree
Showing 3 changed files with 206 additions and 0 deletions.
68 changes: 68 additions & 0 deletions resources/redshiftserverless-namespaces.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,68 @@
package resources

import (
"github.com/aws/aws-sdk-go/aws"
"github.com/aws/aws-sdk-go/aws/session"
"github.com/aws/aws-sdk-go/service/redshiftserverless"
"github.com/rebuy-de/aws-nuke/v2/pkg/types"
)

type RedshiftServerlessNamespace struct {
svc *redshiftserverless.RedshiftServerless
namespace *redshiftserverless.Namespace
}

func init() {
register("RedshiftServerlessNamespace", ListRedshiftServerlessNamespaces)
}

func ListRedshiftServerlessNamespaces(sess *session.Session) ([]Resource, error) {
svc := redshiftserverless.New(sess)
resources := []Resource{}

params := &redshiftserverless.ListNamespacesInput{
MaxResults: aws.Int64(100),
}

for {
output, err := svc.ListNamespaces(params)
if err != nil {
return nil, err
}

for _, namespace := range output.Namespaces {
resources = append(resources, &RedshiftServerlessNamespace{
svc: svc,
namespace: namespace,
})
}

if output.NextToken == nil {
break
}

params.NextToken = output.NextToken
}

return resources, nil
}

func (n *RedshiftServerlessNamespace) Properties() types.Properties {
properties := types.NewProperties().
Set("CreationDate", n.namespace.CreationDate).
Set("NamespaceName", n.namespace.NamespaceName)

return properties
}

func (n *RedshiftServerlessNamespace) Remove() error {
_, err := n.svc.DeleteNamespace(&redshiftserverless.DeleteNamespaceInput{
NamespaceName: n.namespace.NamespaceName,
})

return err
}

func (n *RedshiftServerlessNamespace) String() string {
return *n.namespace.NamespaceName
}
69 changes: 69 additions & 0 deletions resources/redshiftserverless-snapshots.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,69 @@
package resources

import (
"github.com/aws/aws-sdk-go/aws"
"github.com/aws/aws-sdk-go/aws/session"
"github.com/aws/aws-sdk-go/service/redshiftserverless"
"github.com/rebuy-de/aws-nuke/v2/pkg/types"
)

type RedshiftServerlessSnapshot struct {
svc *redshiftserverless.RedshiftServerless
snapshot *redshiftserverless.Snapshot
}

func init() {
register("RedshiftServerlessSnapshot", ListRedshiftServerlessSnapshots)
}

func ListRedshiftServerlessSnapshots(sess *session.Session) ([]Resource, error) {
svc := redshiftserverless.New(sess)
resources := []Resource{}

params := &redshiftserverless.ListSnapshotsInput{
MaxResults: aws.Int64(100),
}

for {
output, err := svc.ListSnapshots(params)
if err != nil {
return nil, err
}

for _, snapshot := range output.Snapshots {
resources = append(resources, &RedshiftServerlessSnapshot{
svc: svc,
snapshot: snapshot,
})
}

if output.NextToken == nil {
break
}

params.NextToken = output.NextToken
}

return resources, nil
}

func (s *RedshiftServerlessSnapshot) Properties() types.Properties {
properties := types.NewProperties().
Set("CreateTime", s.snapshot.SnapshotCreateTime).
Set("Namespace", s.snapshot.NamespaceName).
Set("SnapshotName", s.snapshot.SnapshotName)

return properties
}

func (s *RedshiftServerlessSnapshot) Remove() error {
_, err := s.svc.DeleteSnapshot(&redshiftserverless.DeleteSnapshotInput{
SnapshotName: s.snapshot.SnapshotName,
})

return err
}

func (s *RedshiftServerlessSnapshot) String() string {
return *s.snapshot.SnapshotName
}
69 changes: 69 additions & 0 deletions resources/redshiftserverless-workgroups.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,69 @@
package resources

import (
"github.com/aws/aws-sdk-go/aws"
"github.com/aws/aws-sdk-go/aws/session"
"github.com/aws/aws-sdk-go/service/redshiftserverless"
"github.com/rebuy-de/aws-nuke/v2/pkg/types"
)

type RedshiftServerlessWorkgroup struct {
svc *redshiftserverless.RedshiftServerless
workgroup *redshiftserverless.Workgroup
}

func init() {
register("RedshiftServerlessWorkgroup", ListRedshiftServerlessWorkgroups)
}

func ListRedshiftServerlessWorkgroups(sess *session.Session) ([]Resource, error) {
svc := redshiftserverless.New(sess)
resources := []Resource{}

params := &redshiftserverless.ListWorkgroupsInput{
MaxResults: aws.Int64(100),
}

for {
output, err := svc.ListWorkgroups(params)
if err != nil {
return nil, err
}

for _, workgroup := range output.Workgroups {
resources = append(resources, &RedshiftServerlessWorkgroup{
svc: svc,
workgroup: workgroup,
})
}

if output.NextToken == nil {
break
}

params.NextToken = output.NextToken
}

return resources, nil
}

func (w *RedshiftServerlessWorkgroup) Properties() types.Properties {
properties := types.NewProperties().
Set("CreationDate", w.workgroup.CreationDate).
Set("Namespace", w.workgroup.NamespaceName).
Set("WorkgroupName", w.workgroup.WorkgroupName)

return properties
}

func (w *RedshiftServerlessWorkgroup) Remove() error {
_, err := w.svc.DeleteWorkgroup(&redshiftserverless.DeleteWorkgroupInput{
WorkgroupName: w.workgroup.WorkgroupName,
})

return err
}

func (w *RedshiftServerlessWorkgroup) String() string {
return *w.workgroup.WorkgroupName
}

0 comments on commit 566bb86

Please sign in to comment.