Merge "Add capacity_check API to vio plugin"

5 files changed, 133 insertions, 2 deletions

diff --git a/vio/requirements.txt b/vio/requirements.txt
index 8d5a51c..9c2a4b4 100644
--- a/vio/requirements.txt
+++ b/vio/requirements.txt
@@ -14,6 +14,7 @@ httplib2==0.9.2
 
 # for call openstack api
 openstacksdk==0.9.14
+python-cinderclient==3.5.0
 
 # for unit test
 django-nose>=1.4.0
diff --git a/vio/vio/pub/vim/vimapi/nova/OperateHypervisor.py b/vio/vio/pub/vim/vimapi/nova/OperateHypervisor.py
index a9ad572..1249162 100644
--- a/vio/vio/pub/vim/vimapi/nova/OperateHypervisor.py
+++ b/vio/vio/pub/vim/vimapi/nova/OperateHypervisor.py
@@ -24,7 +24,7 @@ class OperateHypervisor(OperateNova):
     def get_hypervisor(self, data, hypervisor, **kwargs):
         try:
             return self.request('get_hypervisor', data,
-                                project_id=data['project_id'],
+                                project_id=data.get('project_id'),
                                 hypervisor=hypervisor,
                                 **kwargs)
         except exceptions.ResourceNotFound:
@@ -33,7 +33,7 @@ class OperateHypervisor(OperateNova):
     def list_hypervisors(self, data, **query):
         try:
             return self.request('list_hypervisors', data,
-                                project_id=data['project_id'],
+                                project_id=data.get('project_id'),
                                 **query)
         except exceptions.ResourceNotFound:
             return None
diff --git a/vio/vio/swagger/urls.py b/vio/vio/swagger/urls.py
index 89a5974..2dcae22 100644
--- a/vio/vio/swagger/urls.py
+++ b/vio/vio/swagger/urls.py
@@ -48,6 +48,9 @@ from vio.swagger.views.proxyplugin.dns.views import DesignateVersionLink
 from vio.swagger.views.registry.views import Registry
 from vio.swagger.views.registry.views import UnRegistry
 
+# Capacity Check
+from vio.swagger.views.capacity.views import CapacityCheck
+
 # Extensions
 from vio.swagger.views.extensions.views import Extensions
 
@@ -229,6 +232,10 @@ urlpatterns = [
     url(r'^api/multicloud-vio/v0/(?P<vimid>[0-9a-z-A-Z\-\_]+)$',
         UnRegistry.as_view()),
 
+    # CapacityCheck
+    url(r'^api/multicloud-vio/v0/(?P<vimid>[0-9a-z-A-Z\-\_]+)/capacity_check$',
+        CapacityCheck.as_view()),
+
     #   proxy
     url(r'^api/multicloud-vio/v0/(?P<vimid>[0-9a-z-A-Z\-\_]+)/identity/v3',
         TokenView.as_view()),
diff --git a/vio/vio/swagger/views/capacity/__init__.py b/vio/vio/swagger/views/capacity/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/vio/vio/swagger/views/capacity/__init__.py
diff --git a/vio/vio/swagger/views/capacity/views.py b/vio/vio/swagger/views/capacity/views.py
new file mode 100644
index 0000000..ce5d854
--- /dev/null
+++ b/vio/vio/swagger/views/capacity/views.py
@@ -0,0 +1,123 @@
+# 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 json.JSONDecodeError 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)