chore(volumes): modify the volume size

Author: lijianguo

rootfs/api/models/volume.py

@@ -42,6 +42,16 @@ def delete(self, *args, **kwargs):
         # Delete from DB
         return super(Volume, self).delete(*args, **kwargs)
 
+    @staticmethod
+    def _get_size(size):
+        """ Format volume limit value """
+        if size[-2:-1].isalpha() and size[-1].isalpha():
+            size = size[:-1]
+
+        if size[-1].isalpha():
+            size = size.upper() + "i"
+        return size
+
     def attach(self):
         try:
             self._scheduler.pvc.get(self.app.id, self.name)
@@ -52,7 +62,7 @@ def attach(self):
             logger.info(e)
             try:
                 kwargs = {
-                    "size": self.size,
+                    "size": self._get_size(self.size),
                     "storage_class": settings.DRYCC_APP_STORAGE_CLASS,
                 }
                 self._scheduler.pvc.create(self.app.id, self.name, **kwargs)

rootfs/api/serializers.py

@@ -666,7 +666,7 @@ def validate_size(data):
             raise serializers.ValidationError(
                 "Volume size limit format: <number><unit> or <number><unit>/<number><unit>, "
                 "where unit = B, K, M or G")
-        return data
+        return data.upper()
 
     @staticmethod
     def validate_path(data):

rootfs/scheduler/resources/pod.py

@@ -281,7 +281,7 @@ def _get_termination_grace_period(kwargs):
         return timeout_global if timeout_local is None else int(timeout_local)
 
     @staticmethod
-    def _get_memory(size, ):
+    def _get_memory(size):
         """ Format memory limit value """
         if size[-2:-1].isalpha() and size[-1].isalpha():
             size = size[:-1]