class _ElastiCacheService(_AwsService): service_name = 'ElastiCache' def connect(self): if self.conn is None: logger.debug("Connecting to %s", self.service_name) self.conn = ElastiCacheConnection() logger.info("Connected to %s", self.service_name) def find_usage(self): """ Determine the current usage for each limit of this service, and update corresponding Limit via :py:meth:`~.AwsLimit._add_current_usage`. """ logger.debug("Checking usage for service %s", self.service_name) self.connect() for lim in self.limits.values(): lim._reset_usage() self._find_usage_nodes() self._find_usage_subnet_groups() self._find_usage_parameter_groups() self._find_usage_security_groups() self._have_usage = True logger.debug("Done checking usage.") def _find_usage_nodes(self): """find usage for cache nodes""" nodes = 0 clusters = self.conn.describe_cache_clusters(show_cache_node_info=True)[ 'DescribeCacheClustersResponse']['DescribeCacheClustersResult'][ 'CacheClusters'] for cluster in clusters: nodes += len(cluster['CacheNodes']) self.limits['Nodes per Cluster']._add_current_usage( len(cluster['CacheNodes']), aws_type='AWS::ElastiCache::CacheCluster', resource_id=cluster['CacheClusterId'], ) self.limits['Clusters']._add_current_usage( len(clusters), aws_type='AWS::ElastiCache::CacheCluster' ) self.limits['Nodes']._add_current_usage( nodes, aws_type='AWS::ElastiCache::CacheNode' ) def _find_usage_subnet_groups(self): """find usage for elasticache subnet groups""" groups = self.conn.describe_cache_subnet_groups()[ 'DescribeCacheSubnetGroupsResponse'][ 'DescribeCacheSubnetGroupsResult'][ 'CacheSubnetGroups'] self.limits['Subnet Groups']._add_current_usage( len(groups), aws_type='AWS::ElastiCache::SubnetGroup' ) def _find_usage_parameter_groups(self): """find usage for elasticache parameter groups""" groups = self.conn.describe_cache_parameter_groups()[ 'DescribeCacheParameterGroupsResponse'][ 'DescribeCacheParameterGroupsResult'][ 'CacheParameterGroups'] self.limits['Parameter Groups']._add_current_usage( len(groups), aws_type='AWS::ElastiCache::ParameterGroup' ) def _find_usage_security_groups(self): """find usage for elasticache security groups""" groups = self.conn.describe_cache_security_groups()[ 'DescribeCacheSecurityGroupsResponse'][ 'DescribeCacheSecurityGroupsResult'][ 'CacheSecurityGroups'] self.limits['Security Groups']._add_current_usage( len(groups), aws_type='WS::ElastiCache::SecurityGroup' ) def get_limits(self): """ Return all known limits for this service, as a dict of their names to :py:class:`~.AwsLimit` objects. :returns: dict of limit names to :py:class:`~.AwsLimit` objects :rtype: dict """ if self.limits != {}: return self.limits limits = {} limits['Nodes'] = AwsLimit( 'Nodes', self, 50, self.warning_threshold, self.critical_threshold, limit_type='AWS::ElastiCache::CacheNode', ) limits['Clusters'] = AwsLimit( 'Clusters', self, 50, self.warning_threshold, self.critical_threshold, limit_type='AWS::ElastiCache::CacheCluster', ) limits['Nodes per Cluster'] = AwsLimit( 'Nodes per Cluster', self, 20, self.warning_threshold, self.critical_threshold, limit_type='AWS::ElastiCache::CacheNode', ) limits['Subnet Groups'] = AwsLimit( 'Subnet Groups', self, 50, self.warning_threshold, self.critical_threshold, limit_type='AWS::ElastiCache::SubnetGroup', ) limits['Parameter Groups'] = AwsLimit( 'Parameter Groups', self, 20, self.warning_threshold, self.critical_threshold, limit_type='AWS::ElastiCache::ParameterGroup', ) limits['Security Groups'] = AwsLimit( 'Security Groups', self, 50, self.warning_threshold, self.critical_threshold, limit_type='WS::ElastiCache::SecurityGroup', ) self.limits = limits return limits def required_iam_permissions(self): """ Return a list of IAM Actions required for this Service to function properly. All Actions will be shown with an Effect of "Allow" and a Resource of "*". :returns: list of IAM Action strings :rtype: list """ return [ "elasticache:DescribeCacheClusters", "elasticache:DescribeCacheParameterGroups", "elasticache:DescribeCacheSecurityGroups", "elasticache:DescribeCacheSubnetGroups", ]
class _ElastiCacheService(_AwsService): service_name = 'ElastiCache' def connect(self): """Connect to API if not already connected; set self.conn.""" if self.conn is not None: return elif self.region: self.conn = self.connect_via(boto.elasticache.connect_to_region) else: self.conn = ElastiCacheConnection() def find_usage(self): """ Determine the current usage for each limit of this service, and update corresponding Limit via :py:meth:`~.AwsLimit._add_current_usage`. """ logger.debug("Checking usage for service %s", self.service_name) self.connect() for lim in self.limits.values(): lim._reset_usage() self._find_usage_nodes() self._find_usage_subnet_groups() self._find_usage_parameter_groups() self._find_usage_security_groups() self._have_usage = True logger.debug("Done checking usage.") def _find_usage_nodes(self): """find usage for cache nodes""" nodes = 0 clusters = self.conn.describe_cache_clusters(show_cache_node_info=True)[ 'DescribeCacheClustersResponse']['DescribeCacheClustersResult'][ 'CacheClusters'] for cluster in clusters: nodes += len(cluster['CacheNodes']) self.limits['Nodes per Cluster']._add_current_usage( len(cluster['CacheNodes']), aws_type='AWS::ElastiCache::CacheCluster', resource_id=cluster['CacheClusterId'], ) self.limits['Clusters']._add_current_usage( len(clusters), aws_type='AWS::ElastiCache::CacheCluster' ) self.limits['Nodes']._add_current_usage( nodes, aws_type='AWS::ElastiCache::CacheNode' ) def _find_usage_subnet_groups(self): """find usage for elasticache subnet groups""" groups = self.conn.describe_cache_subnet_groups()[ 'DescribeCacheSubnetGroupsResponse'][ 'DescribeCacheSubnetGroupsResult'][ 'CacheSubnetGroups'] self.limits['Subnet Groups']._add_current_usage( len(groups), aws_type='AWS::ElastiCache::SubnetGroup' ) def _find_usage_parameter_groups(self): """find usage for elasticache parameter groups""" groups = self.conn.describe_cache_parameter_groups()[ 'DescribeCacheParameterGroupsResponse'][ 'DescribeCacheParameterGroupsResult'][ 'CacheParameterGroups'] self.limits['Parameter Groups']._add_current_usage( len(groups), aws_type='AWS::ElastiCache::ParameterGroup' ) def _find_usage_security_groups(self): """find usage for elasticache security groups""" try: # If EC2-Classic isn't available (e.g., a new account) # this method will fail with: # Code: "InvalidParameterValue" # Message: "Use of cache security groups is not permitted in # this API version for your account." # Type: "Sender" groups = self.conn.describe_cache_security_groups()[ 'DescribeCacheSecurityGroupsResponse'][ 'DescribeCacheSecurityGroupsResult'][ 'CacheSecurityGroups'] except BotoServerError: logger.debug("caught BotoServerError checking ElastiCache security " "groups (account without EC2-Classic?)") groups = [] self.limits['Security Groups']._add_current_usage( len(groups), aws_type='WS::ElastiCache::SecurityGroup' ) def get_limits(self): """ Return all known limits for this service, as a dict of their names to :py:class:`~.AwsLimit` objects. :returns: dict of limit names to :py:class:`~.AwsLimit` objects :rtype: dict """ if self.limits != {}: return self.limits limits = {} limits['Nodes'] = AwsLimit( 'Nodes', self, 50, self.warning_threshold, self.critical_threshold, limit_type='AWS::ElastiCache::CacheNode', ) limits['Clusters'] = AwsLimit( 'Clusters', self, 50, self.warning_threshold, self.critical_threshold, limit_type='AWS::ElastiCache::CacheCluster', ) limits['Nodes per Cluster'] = AwsLimit( 'Nodes per Cluster', self, 20, self.warning_threshold, self.critical_threshold, limit_type='AWS::ElastiCache::CacheNode', ) limits['Subnet Groups'] = AwsLimit( 'Subnet Groups', self, 50, self.warning_threshold, self.critical_threshold, limit_type='AWS::ElastiCache::SubnetGroup', ) limits['Parameter Groups'] = AwsLimit( 'Parameter Groups', self, 20, self.warning_threshold, self.critical_threshold, limit_type='AWS::ElastiCache::ParameterGroup', ) limits['Security Groups'] = AwsLimit( 'Security Groups', self, 50, self.warning_threshold, self.critical_threshold, limit_type='WS::ElastiCache::SecurityGroup', ) self.limits = limits return limits def required_iam_permissions(self): """ Return a list of IAM Actions required for this Service to function properly. All Actions will be shown with an Effect of "Allow" and a Resource of "*". :returns: list of IAM Action strings :rtype: list """ return [ "elasticache:DescribeCacheClusters", "elasticache:DescribeCacheParameterGroups", "elasticache:DescribeCacheSecurityGroups", "elasticache:DescribeCacheSubnetGroups", ]