summaryrefslogtreecommitdiffstats
path: root/src/pybind/mgr/volumes/fs/operations/pin_util.py
blob: 9ea79e546e267e18390379a3a7657945c79275cb (plain)
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
import os
import errno

import cephfs

from ..exception import VolumeException
from distutils.util import strtobool

_pin_value = {
    "export": lambda x: int(x),
    "distributed": lambda x: int(strtobool(x)),
    "random": lambda x: float(x),
}
_pin_xattr = {
    "export": "ceph.dir.pin",
    "distributed": "ceph.dir.pin.distributed",
    "random": "ceph.dir.pin.random",
}

def pin(fs, path, pin_type, pin_setting):
    """
    Set a pin on a directory.
    """
    assert pin_type in _pin_xattr

    try:
        pin_setting = _pin_value[pin_type](pin_setting)
    except ValueError as e:
        raise VolumeException(-errno.EINVAL, f"pin value wrong type: {pin_setting}")

    try:
        fs.setxattr(path, _pin_xattr[pin_type], str(pin_setting).encode('utf-8'), 0)
    except cephfs.Error as e:
        raise VolumeException(-e.args[0], e.args[1])