mirror of
https://github.com/ceph/ceph
synced 2024-12-13 23:17:07 +00:00
98 lines
3.1 KiB
Python
98 lines
3.1 KiB
Python
import contextlib
|
|
import logging
|
|
import os
|
|
|
|
from teuthology import misc as teuthology
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
def get_clients(ctx, roles):
|
|
for role in roles:
|
|
assert isinstance(role, basestring)
|
|
PREFIX = 'client.'
|
|
assert role.startswith(PREFIX)
|
|
id_ = role[len(PREFIX):]
|
|
(remote,) = ctx.cluster.only(role).remotes.iterkeys()
|
|
yield (id_, remote)
|
|
|
|
|
|
@contextlib.contextmanager
|
|
def task(ctx, config):
|
|
"""
|
|
Mount/unmount a ``kernel`` client.
|
|
|
|
The config is optional and defaults to mounting on all clients. If
|
|
a config is given, it 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]
|
|
- kclient: [client.1]
|
|
- interactive:
|
|
"""
|
|
log.info('Mounting kernel clients...')
|
|
assert config is None or isinstance(config, list), \
|
|
"task kclient got invalid config"
|
|
|
|
if config is None:
|
|
config = ['client.{id}'.format(id=id_)
|
|
for id_ in teuthology.all_roles_of_type(ctx.cluster, 'client')]
|
|
clients = list(get_clients(ctx=ctx, roles=config))
|
|
|
|
for id_, remote in clients:
|
|
log.debug('Mounting client client.{id}...'.format(id=id_))
|
|
remotes_and_roles = ctx.cluster.remotes.items()
|
|
roles = [roles for (remote, roles) in remotes_and_roles]
|
|
ips = [host for (host, port) in (remote.ssh.get_transport().getpeername() for (remote, roles) in remotes_and_roles)]
|
|
mons = teuthology.get_mons(roles, ips).values()
|
|
mnt = os.path.join('/tmp/cephtest', 'mnt.{id}'.format(id=id_))
|
|
secret = '/tmp/cephtest/data/client.{id}.secret'.format(id=id_)
|
|
teuthology.write_secret_file(remote, 'client.{id}'.format(id=id_), secret)
|
|
|
|
remote.run(
|
|
args=[
|
|
'mkdir',
|
|
'--',
|
|
mnt,
|
|
],
|
|
)
|
|
|
|
remote.run(
|
|
args=[
|
|
'sudo',
|
|
'/tmp/cephtest/binary/usr/local/bin/ceph-coverage',
|
|
'/tmp/cephtest/archive/coverage',
|
|
'/tmp/cephtest/binary/usr/local/sbin/mount.ceph',
|
|
'{mons}:/'.format(mons=','.join(mons)),
|
|
mnt,
|
|
'-v',
|
|
'-o',
|
|
'name={id},secretfile={secret}'.format(id=id_,
|
|
secret=secret),
|
|
],
|
|
)
|
|
|
|
try:
|
|
yield
|
|
finally:
|
|
log.info('Unmounting kernel clients...')
|
|
for id_, remote in clients:
|
|
log.debug('Unmounting client client.{id}...'.format(id=id_))
|
|
mnt = os.path.join('/tmp/cephtest', 'mnt.{id}'.format(id=id_))
|
|
remote.run(
|
|
args=[
|
|
'sudo',
|
|
'umount',
|
|
mnt,
|
|
],
|
|
)
|
|
remote.run(
|
|
args=[
|
|
'rmdir',
|
|
'--',
|
|
mnt,
|
|
],
|
|
)
|