ceph/teuthology/task/cfuse.py
2011-06-09 17:17:47 -07:00

108 lines
3.3 KiB
Python

import contextlib
import logging
import os
from teuthology import misc as teuthology
from orchestra import run
log = logging.getLogger(__name__)
@contextlib.contextmanager
def task(ctx, config):
"""
Mount/unmount a ``cfuse`` client.
The config is expected to be a list of clients to do this
operation on. This lets you e.g. set up one client with ``cfuse``
and another with ``kclient``.
tasks:
- ceph:
- cfuse: [client.0]
- interactive:
"""
log.info('Mounting cfuse clients...')
assert isinstance(config, list), \
"task fuse automatic configuration not supported yet, list all clients"
cfuse_daemons = {}
for role in config:
assert isinstance(role, basestring)
PREFIX = 'client.'
assert role.startswith(PREFIX)
id_ = role[len(PREFIX):]
(remote,) = ctx.cluster.only(role).remotes.iterkeys()
mnt = os.path.join('/tmp/cephtest', 'mnt.{id}'.format(id=id_))
remote.run(
args=[
'mkdir',
'--',
mnt,
],
)
proc = remote.run(
args=[
'/tmp/cephtest/binary/usr/local/bin/ceph-coverage',
'/tmp/cephtest/coverage',
'/tmp/cephtest/daemon-helper',
'/tmp/cephtest/binary/usr/local/bin/cfuse',
'-f',
'--name', 'client.{id}'.format(id=id_),
'-c', '/tmp/cephtest/ceph.conf',
# TODO cfuse doesn't understand dash dash '--',
mnt,
],
logger=log.getChild('cfuse.{id}'.format(id=id_)),
stdin=run.PIPE,
wait=False,
)
cfuse_daemons[id_] = proc
for role in config:
assert isinstance(role, basestring)
PREFIX = 'client.'
assert role.startswith(PREFIX)
id_ = role[len(PREFIX):]
(remote,) = ctx.cluster.only(role).remotes.iterkeys()
mnt = os.path.join('/tmp/cephtest', 'mnt.{id}'.format(id=id_))
teuthology.wait_until_fuse_mounted(
remote=remote,
fuse=cfuse_daemons[id_],
mountpoint=mnt,
)
try:
yield
finally:
log.info('Unmounting cfuse clients...')
for role in config:
assert isinstance(role, basestring)
PREFIX = 'client.'
assert role.startswith(PREFIX)
id_ = role[len(PREFIX):]
(remote,) = ctx.cluster.only(role).remotes.iterkeys()
mnt = os.path.join('/tmp/cephtest', 'mnt.{id}'.format(id=id_))
remote.run(
args=[
'fusermount',
'-u',
mnt,
],
)
run.wait(cfuse_daemons.itervalues())
for role in config:
assert isinstance(role, basestring)
PREFIX = 'client.'
assert role.startswith(PREFIX)
id_ = role[len(PREFIX):]
(remote,) = ctx.cluster.only(role).remotes.iterkeys()
mnt = os.path.join('/tmp/cephtest', 'mnt.{id}'.format(id=id_))
remote.run(
args=[
'rmdir',
'--',
mnt,
],
)