Merge branch 'check-ceph-snap-revert' into '5.1.0'

<fix>[cephprimarystorage]: check cap before revert snap

See merge request zstackio/zstack-utility!4656

Author: gitlab

cephprimarystorage/cephprimarystorage/cephagent.py

@@ -67,7 +67,7 @@ def __init__(self, success=True, error=None):
         self.error = error if error else ''
         self.totalCapacity = None
         self.availableCapacity = None
-        self.poolCapacities = None
+        self.poolCapacities = None  # type: list[CephPoolCapacity]
         self.type = None
 
     def set_err(self, err):
@@ -765,13 +765,27 @@ def delete_pool(self, req):
     def rollback_snapshot(self, req):
         cmd = jsonobject.loads(req[http.REQUEST_BODY])
         spath = self._normalize_install_path(cmd.snapshotPath)
+        rsp = RollbackSnapshotRsp()
+        self._set_capacity_to_response(rsp)
+
+        self.validate_snapshot_rollback(spath, rsp)
+
         driver = self.get_driver(cmd)
         driver.rollback_snapshot(cmd)
-        rsp = RollbackSnapshotRsp()
         rsp.size = self._get_file_size(spath)
-        self._set_capacity_to_response(rsp)
         return jsonobject.dumps(rsp)
 
+    def validate_snapshot_rollback(self, spath, rsp):
+        asize = self._get_file_actual_size(spath)
+        if asize is None:
+            return
+
+        pool_name = spath.split("/")[0]
+        if pool_name in rsp.poolCapacities:
+            cap = rsp.poolCapacities[pool_name]
+            if cap.usedCapacity + asize > cap.totalCapacity * 0.9:
+                raise Exception("In the worst case, The rollback operation will exceed the 90%% capacity of the pool[%s]" % pool_name)
+
     @staticmethod
     def _wrap_shareable_cmd(cmd, cmd_string):
         if cmd.shareable: