144 lines
5.6 KiB
Python
144 lines
5.6 KiB
Python
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
"""Placement API handlers for usage information."""
|
|
|
|
import collections
|
|
|
|
from oslo_serialization import jsonutils
|
|
from oslo_utils import encodeutils
|
|
from oslo_utils import timeutils
|
|
import webob
|
|
|
|
from placement import exception
|
|
from placement import microversion
|
|
from placement.objects import resource_provider as rp_obj
|
|
from placement.objects import usage as usage_obj
|
|
from placement.policies import usage as policies
|
|
from placement.schemas import usage as schema
|
|
from placement import util
|
|
from placement import wsgi_wrapper
|
|
|
|
|
|
def _serialize_usages(resource_provider, usage):
|
|
usage_dict = {resource.resource_class: resource.usage
|
|
for resource in usage}
|
|
return {'resource_provider_generation': resource_provider.generation,
|
|
'usages': usage_dict}
|
|
|
|
|
|
@wsgi_wrapper.PlacementWsgify
|
|
@util.check_accept('application/json')
|
|
def list_usages(req):
|
|
"""GET a dictionary of resource provider usage by resource class.
|
|
|
|
If the resource provider does not exist return a 404.
|
|
|
|
On success return a 200 with an application/json representation of
|
|
the usage dictionary.
|
|
"""
|
|
context = req.environ['placement.context']
|
|
context.can(policies.PROVIDER_USAGES)
|
|
uuid = util.wsgi_path_item(req.environ, 'uuid')
|
|
want_version = req.environ[microversion.MICROVERSION_ENVIRON]
|
|
|
|
# Resource provider object needed for two things: If it is
|
|
# NotFound we'll get a 404 here, which needs to happen because
|
|
# get_all_by_resource_provider_uuid can return an empty list.
|
|
# It is also needed for the generation, used in the outgoing
|
|
# representation.
|
|
try:
|
|
resource_provider = rp_obj.ResourceProvider.get_by_uuid(
|
|
context, uuid)
|
|
except exception.NotFound as exc:
|
|
raise webob.exc.HTTPNotFound(
|
|
"No resource provider with uuid %(uuid)s found: %(error)s" %
|
|
{'uuid': uuid, 'error': exc})
|
|
|
|
usage = usage_obj.get_all_by_resource_provider_uuid(context, uuid)
|
|
|
|
response = req.response
|
|
response.body = encodeutils.to_utf8(jsonutils.dumps(
|
|
_serialize_usages(resource_provider, usage)))
|
|
req.response.content_type = 'application/json'
|
|
if want_version.matches((1, 15)):
|
|
req.response.cache_control = 'no-cache'
|
|
# While it would be possible to generate a last-modified time
|
|
# based on the collection of allocations that result in a usage
|
|
# value (with some spelunking in the SQL) that doesn't align with
|
|
# the question that is being asked in a request for usages: What
|
|
# is the usage, now? So the last-modified time is set to utcnow.
|
|
req.response.last_modified = timeutils.utcnow(with_timezone=True)
|
|
return req.response
|
|
|
|
|
|
@wsgi_wrapper.PlacementWsgify
|
|
@microversion.version_handler('1.9')
|
|
@util.check_accept('application/json')
|
|
def get_total_usages(req):
|
|
"""GET the sum of usages for a project or a project/user.
|
|
|
|
On success return a 200 and an application/json body representing the
|
|
sum/total of usages.
|
|
Return 404 Not Found if the wanted microversion does not match.
|
|
"""
|
|
project_id = req.GET.get('project_id')
|
|
user_id = req.GET.get('user_id')
|
|
consumer_type = req.GET.get('consumer_type')
|
|
|
|
context = req.environ['placement.context']
|
|
context.can(
|
|
policies.TOTAL_USAGES,
|
|
target={'project_id': project_id})
|
|
want_version = req.environ[microversion.MICROVERSION_ENVIRON]
|
|
|
|
want_schema = schema.GET_USAGES_SCHEMA_1_9
|
|
show_consumer_type = want_version.matches((1, 38))
|
|
if show_consumer_type:
|
|
want_schema = schema.GET_USAGES_SCHEMA_V1_38
|
|
util.validate_query_params(req, want_schema)
|
|
|
|
if show_consumer_type:
|
|
usages = usage_obj.get_by_consumer_type(
|
|
context, project_id, user_id=user_id, consumer_type=consumer_type)
|
|
else:
|
|
usages = usage_obj.get_all_by_project_user(context, project_id,
|
|
user_id=user_id)
|
|
|
|
response = req.response
|
|
if show_consumer_type:
|
|
usage = collections.defaultdict(dict)
|
|
for resource in usages:
|
|
ct = resource.consumer_type
|
|
rc = resource.resource_class
|
|
cc = resource.consumer_count
|
|
used = resource.usage
|
|
usage[ct][rc] = used
|
|
usage[ct]['consumer_count'] = cc
|
|
usages_dict = {
|
|
'usages': usage
|
|
}
|
|
else:
|
|
usages_dict = {'usages': {resource.resource_class: resource.usage
|
|
for resource in usages}}
|
|
|
|
response.body = encodeutils.to_utf8(jsonutils.dumps(usages_dict))
|
|
req.response.content_type = 'application/json'
|
|
if want_version.matches((1, 15)):
|
|
req.response.cache_control = 'no-cache'
|
|
# While it would be possible to generate a last-modified time
|
|
# based on the collection of allocations that result in a usage
|
|
# value (with some spelunking in the SQL) that doesn't align with
|
|
# the question that is being asked in a request for usages: What
|
|
# is the usage, now? So the last-modified time is set to utcnow.
|
|
req.response.last_modified = timeutils.utcnow(with_timezone=True)
|
|
return req.response
|