Ejemplo n.º 1
0
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",
        ]
Ejemplo n.º 2
0
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",
        ]