Add cluster parameter groups.
This commit is contained in:
parent
2418c83252
commit
5bbcc4505f
4 changed files with 195 additions and 8 deletions
|
|
@ -38,6 +38,13 @@ class ClusterSecurityGroupNotFoundError(RedshiftClientError):
|
|||
"Security group {0} not found.".format(group_identifier))
|
||||
|
||||
|
||||
class ClusterParameterGroupNotFoundError(RedshiftClientError):
|
||||
def __init__(self, group_identifier):
|
||||
super(ClusterParameterGroupNotFoundError, self).__init__(
|
||||
'ClusterParameterGroupNotFound',
|
||||
"Parameter group {0} not found.".format(group_identifier))
|
||||
|
||||
|
||||
class InvalidSubnetError(RedshiftClientError):
|
||||
def __init__(self, subnet_identifier):
|
||||
super(InvalidSubnetError, self).__init__(
|
||||
|
|
|
|||
|
|
@ -5,6 +5,7 @@ from moto.core import BaseBackend
|
|||
from moto.ec2 import ec2_backends
|
||||
from .exceptions import (
|
||||
ClusterNotFoundError,
|
||||
ClusterParameterGroupNotFoundError,
|
||||
ClusterSecurityGroupNotFoundError,
|
||||
ClusterSubnetGroupNotFoundError,
|
||||
InvalidSubnetError,
|
||||
|
|
@ -27,7 +28,6 @@ class Cluster(object):
|
|||
self.db_name = db_name if db_name else "dev"
|
||||
self.vpc_security_group_ids = vpc_security_group_ids
|
||||
self.cluster_subnet_group_name = cluster_subnet_group_name
|
||||
self.cluster_parameter_group_name = cluster_parameter_group_name
|
||||
self.publicly_accessible = publicly_accessible
|
||||
self.encrypted = encrypted
|
||||
|
||||
|
|
@ -37,6 +37,11 @@ class Cluster(object):
|
|||
self.automated_snapshot_retention_period = automated_snapshot_retention_period if automated_snapshot_retention_period else 1
|
||||
self.preferred_maintenance_window = preferred_maintenance_window if preferred_maintenance_window else "Mon:03:00-Mon:03:30"
|
||||
|
||||
if cluster_parameter_group_name:
|
||||
self.cluster_parameter_group_name = [cluster_parameter_group_name]
|
||||
else:
|
||||
self.cluster_parameter_group_name = ['default.redshift-1.0']
|
||||
|
||||
if cluster_security_groups:
|
||||
self.cluster_security_groups = cluster_security_groups
|
||||
else:
|
||||
|
|
@ -72,6 +77,14 @@ class Cluster(object):
|
|||
if security_group.id in self.vpc_security_group_ids
|
||||
]
|
||||
|
||||
@property
|
||||
def parameter_groups(self):
|
||||
return [
|
||||
parameter_group for parameter_group
|
||||
in self.redshift_backend.describe_cluster_parameter_groups()
|
||||
if parameter_group.cluster_parameter_group_name in self.cluster_parameter_group_name
|
||||
]
|
||||
|
||||
def to_json(self):
|
||||
return {
|
||||
"MasterUsername": self.master_username,
|
||||
|
|
@ -90,7 +103,10 @@ class Cluster(object):
|
|||
"Encrypted": self.encrypted,
|
||||
"DBName": self.db_name,
|
||||
"PreferredMaintenanceWindow": self.preferred_maintenance_window,
|
||||
"ClusterParameterGroups": [],
|
||||
"ClusterParameterGroups": [{
|
||||
"ParameterApplyStatus": "in-sync",
|
||||
"ParameterGroupName": group.cluster_parameter_group_name,
|
||||
} for group in self.parameter_groups],
|
||||
"ClusterSecurityGroups": [{
|
||||
"Status": "active",
|
||||
"ClusterSecurityGroupName": group.cluster_security_group_name,
|
||||
|
|
@ -150,6 +166,21 @@ class SecurityGroup(object):
|
|||
}
|
||||
|
||||
|
||||
class ParameterGroup(object):
|
||||
|
||||
def __init__(self, cluster_parameter_group_name, group_family, description):
|
||||
self.cluster_parameter_group_name = cluster_parameter_group_name
|
||||
self.group_family = group_family
|
||||
self.description = description
|
||||
|
||||
def to_json(self):
|
||||
return {
|
||||
"ParameterGroupFamily": self.group_family,
|
||||
"Description": self.description,
|
||||
"ParameterGroupName": self.cluster_parameter_group_name,
|
||||
}
|
||||
|
||||
|
||||
class RedshiftBackend(BaseBackend):
|
||||
|
||||
def __init__(self, ec2_backend):
|
||||
|
|
@ -158,6 +189,13 @@ class RedshiftBackend(BaseBackend):
|
|||
self.security_groups = {
|
||||
"Default": SecurityGroup("Default", "Default Redshift Security Group")
|
||||
}
|
||||
self.parameter_groups = {
|
||||
"default.redshift-1.0": ParameterGroup(
|
||||
"default.redshift-1.0",
|
||||
"redshift-1.0",
|
||||
"Default Redshift parameter group",
|
||||
)
|
||||
}
|
||||
self.ec2_backend = ec2_backend
|
||||
|
||||
def reset(self):
|
||||
|
|
@ -239,6 +277,27 @@ class RedshiftBackend(BaseBackend):
|
|||
return self.security_groups.pop(security_group_identifier)
|
||||
raise ClusterSecurityGroupNotFoundError(security_group_identifier)
|
||||
|
||||
def create_cluster_parameter_group(self, cluster_parameter_group_name,
|
||||
group_family, description):
|
||||
parameter_group = ParameterGroup(cluster_parameter_group_name, group_family, description)
|
||||
self.parameter_groups[cluster_parameter_group_name] = parameter_group
|
||||
|
||||
return parameter_group
|
||||
|
||||
def describe_cluster_parameter_groups(self, parameter_group_name=None):
|
||||
parameter_groups = self.parameter_groups.values()
|
||||
if parameter_group_name:
|
||||
if parameter_group_name in self.parameter_groups:
|
||||
return [self.parameter_groups[parameter_group_name]]
|
||||
else:
|
||||
raise ClusterParameterGroupNotFoundError(parameter_group_name)
|
||||
return parameter_groups
|
||||
|
||||
def delete_cluster_parameter_group(self, parameter_group_name):
|
||||
if parameter_group_name in self.parameter_groups:
|
||||
return self.parameter_groups.pop(parameter_group_name)
|
||||
raise ClusterParameterGroupNotFoundError(parameter_group_name)
|
||||
|
||||
|
||||
redshift_backends = {}
|
||||
for region in boto.redshift.regions():
|
||||
|
|
|
|||
|
|
@ -82,7 +82,6 @@ class RedshiftResponse(BaseResponse):
|
|||
"publicly_accessible": self._get_param("PubliclyAccessible"),
|
||||
"encrypted": self._get_param("Encrypted"),
|
||||
}
|
||||
|
||||
cluster = self.redshift_backend.modify_cluster(**cluster_kwargs)
|
||||
|
||||
return json.dumps({
|
||||
|
|
@ -206,3 +205,52 @@ class RedshiftResponse(BaseResponse):
|
|||
}
|
||||
}
|
||||
})
|
||||
|
||||
def create_cluster_parameter_group(self):
|
||||
cluster_parameter_group_name = self._get_param('ParameterGroupName')
|
||||
group_family = self._get_param('ParameterGroupFamily')
|
||||
description = self._get_param('Description')
|
||||
|
||||
parameter_group = self.redshift_backend.create_cluster_parameter_group(
|
||||
cluster_parameter_group_name,
|
||||
group_family,
|
||||
description,
|
||||
)
|
||||
|
||||
return json.dumps({
|
||||
"CreateClusterParameterGroupResponse": {
|
||||
"CreateClusterParameterGroupResult": {
|
||||
"ClusterParameterGroup": parameter_group.to_json(),
|
||||
},
|
||||
"ResponseMetadata": {
|
||||
"RequestId": "384ac68d-3775-11df-8963-01868b7c937a",
|
||||
}
|
||||
}
|
||||
})
|
||||
|
||||
def describe_cluster_parameter_groups(self):
|
||||
cluster_parameter_group_name = self._get_param("ParameterGroupName")
|
||||
parameter_groups = self.redshift_backend.describe_cluster_parameter_groups(cluster_parameter_group_name)
|
||||
|
||||
return json.dumps({
|
||||
"DescribeClusterParameterGroupsResponse": {
|
||||
"DescribeClusterParameterGroupsResult": {
|
||||
"ParameterGroups": [parameter_group.to_json() for parameter_group in parameter_groups]
|
||||
},
|
||||
"ResponseMetadata": {
|
||||
"RequestId": "384ac68d-3775-11df-8963-01868b7c937a",
|
||||
}
|
||||
}
|
||||
})
|
||||
|
||||
def delete_cluster_parameter_group(self):
|
||||
cluster_parameter_group_name = self._get_param("ParameterGroupName")
|
||||
self.redshift_backend.delete_cluster_parameter_group(cluster_parameter_group_name)
|
||||
|
||||
return json.dumps({
|
||||
"DeleteClusterParameterGroupResponse": {
|
||||
"ResponseMetadata": {
|
||||
"RequestId": "384ac68d-3775-11df-8963-01868b7c937a",
|
||||
}
|
||||
}
|
||||
})
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue