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
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
|
import os
import uuid
import logging
from contextlib import contextmanager
import cephfs
from .template import GroupTemplate
from ..fs_util import listdir
from ..exception import VolumeException
log = logging.getLogger(__name__)
class Trash(GroupTemplate):
GROUP_NAME = "_deleting"
def __init__(self, fs, vol_spec):
self.fs = fs
self.vol_spec = vol_spec
self.groupname = Trash.GROUP_NAME
@property
def path(self):
return os.path.join(self.vol_spec.base_dir.encode('utf-8'), self.groupname.encode('utf-8'))
@property
def unique_trash_path(self):
"""
return a unique trash directory entry path
"""
return os.path.join(self.path, str(uuid.uuid4()).encode('utf-8'))
def _get_single_dir_entry(self, exclude_list=[]):
exclude_list.extend((b".", b".."))
try:
with self.fs.opendir(self.path) as d:
entry = self.fs.readdir(d)
while entry:
if entry.d_name not in exclude_list:
return entry.d_name
entry = self.fs.readdir(d)
return None
except cephfs.Error as e:
raise VolumeException(-e.args[0], e.args[1])
def get_trash_entry(self, exclude_list):
"""
get a trash entry excluding entries provided.
:praram exclude_list: entries to exclude
:return: trash entry
"""
return self._get_single_dir_entry(exclude_list)
def purge(self, trashpath, should_cancel):
"""
purge a trash entry.
:praram trash_entry: the trash entry to purge
:praram should_cancel: callback to check if the purge should be aborted
:return: None
"""
def rmtree(root_path):
log.debug("rmtree {0}".format(root_path))
try:
with self.fs.opendir(root_path) as dir_handle:
d = self.fs.readdir(dir_handle)
while d and not should_cancel():
if d.d_name not in (b".", b".."):
d_full = os.path.join(root_path, d.d_name)
if d.is_dir():
rmtree(d_full)
else:
self.fs.unlink(d_full)
d = self.fs.readdir(dir_handle)
except cephfs.ObjectNotFound:
return
except cephfs.Error as e:
raise VolumeException(-e.args[0], e.args[1])
# remove the directory only if we were not asked to cancel
# (else we would fail to remove this anyway)
if not should_cancel():
self.fs.rmdir(root_path)
# catch any unlink errors
try:
rmtree(trashpath)
except cephfs.Error as e:
raise VolumeException(-e.args[0], e.args[1])
def dump(self, path):
"""
move an filesystem entity to trash can.
:praram path: the filesystem path to be moved
:return: None
"""
try:
self.fs.rename(path, self.unique_trash_path)
except cephfs.Error as e:
raise VolumeException(-e.args[0], e.args[1])
def link(self, path, bname):
pth = os.path.join(self.path, bname)
try:
self.fs.symlink(path, pth)
except cephfs.Error as e:
raise VolumeException(-e.args[0], e.args[1])
def delink(self, bname):
pth = os.path.join(self.path, bname)
try:
self.fs.unlink(pth)
except cephfs.Error as e:
raise VolumeException(-e.args[0], e.args[1])
def create_trashcan(fs, vol_spec):
"""
create a trash can.
:param fs: ceph filesystem handle
:param vol_spec: volume specification
:return: None
"""
trashcan = Trash(fs, vol_spec)
try:
fs.mkdirs(trashcan.path, 0o700)
except cephfs.Error as e:
raise VolumeException(-e.args[0], e.args[1])
@contextmanager
def open_trashcan(fs, vol_spec):
"""
open a trash can. This API is to be used as a context manager.
:param fs: ceph filesystem handle
:param vol_spec: volume specification
:return: yields a trash can object (subclass of GroupTemplate)
"""
trashcan = Trash(fs, vol_spec)
try:
fs.stat(trashcan.path)
except cephfs.Error as e:
raise VolumeException(-e.args[0], e.args[1])
yield trashcan
|
