- ensure there quota limits before creating a new server - ensure a server is in desired state (ACTIVE or SHUTOFF) Change-Id: Id91123fd4ca4114d92bc7b836257e64505b10404changes/13/771813/22
Federico Ressi
2 years ago
parent
cba8d571bf
commit
9aaa10c9d7
@ -0,0 +1,60 @@
|
||||
# Copyright 2021 Red Hat
|
||||
#
|
||||
# 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.
|
||||
from __future__ import absolute_import
|
||||
|
||||
import typing
|
||||
|
||||
from keystoneclient import base
|
||||
from keystoneclient.v2_0 import tenants as tenants_v2
|
||||
from keystoneclient.v2_0 import users as users_v2
|
||||
from keystoneclient.v3 import projects as projects_v3
|
||||
from keystoneclient.v3 import users as users_v3
|
||||
|
||||
|
||||
KeystoneResourceType = typing.Union[str, base.Resource]
|
||||
ProjectType = typing.Union[str, tenants_v2.Tenant, projects_v3.Project]
|
||||
UserType = typing.Union[str, users_v2.User, users_v3.User]
|
||||
|
||||
|
||||
def get_project_id(
|
||||
project: typing.Optional[ProjectType] = None,
|
||||
session=None) -> str:
|
||||
if project is not None:
|
||||
return get_keystone_resource_id(project)
|
||||
if session is not None:
|
||||
return session.auth.auth_ref.project_id
|
||||
raise ValueError("'project' and 'session' can't be None ata the same "
|
||||
"time.")
|
||||
|
||||
|
||||
def get_user_id(
|
||||
user: typing.Optional[UserType] = None,
|
||||
session=None) -> str:
|
||||
if user is not None:
|
||||
return get_keystone_resource_id(user)
|
||||
if session is not None:
|
||||
return session.auth.auth_ref.user_id
|
||||
raise ValueError("'project' and 'session' can't be None ata the same "
|
||||
"time.")
|
||||
|
||||
|
||||
def get_keystone_resource_id(resource: KeystoneResourceType) -> str:
|
||||
if isinstance(resource, str):
|
||||
return resource
|
||||
|
||||
if isinstance(resource, base.Resource):
|
||||
return resource.id
|
||||
|
||||
raise TypeError(f"Object {resource} is not a valid Keystone resource "
|
||||
"type")
|
||||
@ -0,0 +1,87 @@
|
||||
# Copyright 2021 Red Hat
|
||||
#
|
||||
# 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.
|
||||
from __future__ import absolute_import
|
||||
|
||||
from oslo_log import log
|
||||
|
||||
from tobiko.openstack import keystone
|
||||
from tobiko.openstack.nova import _client
|
||||
|
||||
|
||||
LOG = log.getLogger(__name__)
|
||||
|
||||
|
||||
def get_nova_quota_set(project: keystone.ProjectType = None,
|
||||
user: keystone.UserType = None,
|
||||
client: _client.NovaClientType = None,
|
||||
**params):
|
||||
client = _client.nova_client(client)
|
||||
project_id = keystone.get_project_id(project=project,
|
||||
session=client.client.session)
|
||||
user_id = user and keystone.get_user_id(user=user) or None
|
||||
return client.quotas.get(project_id, user_id=user_id, **params)
|
||||
|
||||
|
||||
def set_nova_quota_set(project: keystone.ProjectType = None,
|
||||
user: keystone.UserType = None,
|
||||
client: _client.NovaClientType = None,
|
||||
**params):
|
||||
client = _client.nova_client(client)
|
||||
project_id = keystone.get_project_id(project=project,
|
||||
session=client.client.session)
|
||||
user_id = user and keystone.get_user_id(user=user) or None
|
||||
return client.quotas.update(project_id, user_id=user_id, **params)
|
||||
|
||||
|
||||
def ensure_nova_quota_limits(project: keystone.ProjectType = None,
|
||||
user: keystone.UserType = None,
|
||||
client: _client.NovaClientType = None,
|
||||
**required):
|
||||
client = _client.nova_client(client)
|
||||
project = keystone.get_project_id(project=project,
|
||||
session=client.client.session)
|
||||
user = user and keystone.get_user_id(user=user) or None
|
||||
if user:
|
||||
# Must increase project limits before user ones
|
||||
ensure_nova_quota_limits(project=project, client=client,
|
||||
**required)
|
||||
|
||||
quota_set = get_nova_quota_set(project=project, user=user,
|
||||
client=client, detail=True)
|
||||
actual_limits = {}
|
||||
increment_limits = {}
|
||||
for name, needed in required.items():
|
||||
quota = getattr(quota_set, name)
|
||||
limit: int = quota['limit']
|
||||
if limit > 0:
|
||||
in_use: int = max(0, quota['in_use']) + max(0, quota['reserved'])
|
||||
if in_use + needed >= limit:
|
||||
actual_limits[name] = limit
|
||||
increment_limits[name] = max(10, limit * 2)
|
||||
|
||||
if increment_limits:
|
||||
LOG.info(f"Increment Nova quota limits (project={project}, "
|
||||
f"user={user}): {actual_limits} -> {increment_limits}...")
|
||||
set_nova_quota_set(project=project, user=user, client=client,
|
||||
**increment_limits)
|
||||
quota_set = get_nova_quota_set(project=project, user=user,
|
||||
client=client, detail=True)
|
||||
new_limits = {
|
||||
name: getattr(quota_set, name)['limit']
|
||||
for name in increment_limits.keys()}
|
||||
|
||||
LOG.info(f"Nova quota limit increased (project={project}, "
|
||||
f"user={user}): {actual_limits} -> {new_limits}...")
|
||||
|
||||
return quota_set
|
||||
Loading…
Reference in new issue