1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
# Copyright (c) 2017-2018 VMware, Inc.
#
# 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.
import json
import logging
from rest_framework import status
from rest_framework.response import Response
from rest_framework.views import APIView
from vio.pub.exceptions import VimDriverVioException
from vio.pub.msapi import extsys
from vio.pub.vim.vimapi.nova import OperateHypervisor
from vio.pub.vim.vimapi.nova import OperateLimits
from cinderclient import client
logger = logging.getLogger(__name__)
class CapacityCheck(APIView):
def _check_capacity(self, hypervisor, requirement):
avail_vcpu = hypervisor.get("vcpus") - hypervisor.get("vcpus_used")
avail_mem = (hypervisor.get("memory_size") - hypervisor.get(
"memory_used"))/1024
avail_disk = hypervisor.get("local_disk_size") - hypervisor.get(
"local_disk_used")
req_vcpu = requirement.get("vCPU")
req_mem = requirement.get("Memory")
req_disk = requirement.get("Storage")
if avail_vcpu < req_vcpu:
return False
if avail_mem < req_mem:
return False
if avail_disk < req_disk:
return False
return True
def _check_nova_limits(self, limits, requirement):
avail_vcpu = (
limits.absolute.total_cores - limits.absolute.total_cores_used)
avail_mem = (
limits.absolute.total_ram - limits.absolute.total_ram_used)/1024
req_vcpu = requirement.get("vCPU")
req_mem = requirement.get("Memory")
if avail_vcpu >= 0 and avail_vcpu < req_vcpu:
return False
if avail_mem >= 0 and avail_mem < req_mem:
return False
return True
def _check_cinder_limits(self, limits, requirement):
absolute = limits['absolute']
avail_disk = (absolute.get(
"maxTotalVolumeGigabytes") - absolute.get("totalGigabytesUsed"))
req_disk = requirement.get("Storage")
if avail_disk < req_disk:
return False
return True
def post(self, request, vimid):
try:
requirement = json.loads(request.body)
except ValueError as ex:
return Response(data={'error': str(ex)},
status=status.HTTP_400_BAD_REQUEST)
try:
vim_info = extsys.get_vim_by_id(vimid)
except VimDriverVioException as e:
return Response(data={'error': str(e)}, status=e.status_code)
auth_info = {
"username": vim_info['userName'],
"password": vim_info['password'],
"url": vim_info['url'],
"project_name": vim_info['tenant']
}
# check nova limits
servers_op = OperateLimits.OperateLimits()
try:
nova_limits = servers_op.get_limits(auth_info, None)
except Exception as e:
logger.exception("get nova limits error %(e)s", {"e": e})
return Response(data={'error': str(e)}, status=e.status_code)
if not self._check_nova_limits(nova_limits, requirement):
return Response(
data={'result': False}, status=status.HTTP_200_OK)
# check cinder limits
cinder = client.Client(
"2", auth_info['username'], auth_info['password'],
auth_info['project_name'], auth_info['url'], insecure=True)
try:
limits = cinder.limits.get().to_dict()
except Exception as e:
logger.exception("get cinder limits error %(e)s", {"e": e})
return Response(data={'error': str(e)}, status=e.status_code)
if not self._check_cinder_limits(limits, requirement):
return Response(
data={'result': False}, status=status.HTTP_200_OK)
# check hypervisor resources
hypervisor_op = OperateHypervisor.OperateHypervisor()
try:
hypervisors = hypervisor_op.list_hypervisors(auth_info)
except Exception as e:
logger.exception("get hypervisors error %(e)s", {"e": e})
return Response(data={'error': str(e)}, status=e.status_code)
for hypervisor in hypervisors:
hyper = hypervisor_op.get_hypervisor(auth_info, hypervisor.id)
if self._check_capacity(hyper.to_dict(), requirement):
return Response(data={'result': True},
status=status.HTTP_200_OK)
return Response(data={'result': False}, status=status.HTTP_200_OK)
class CapacityCheckV1(CapacityCheck):
def post(self, request, cloud_owner, cloud_region):
return super(CapacityCheckV1, self).post(
request, cloud_owner + "_" + cloud_region)
|