mirror of
https://github.com/ceph/ceph
synced 2024-12-25 21:03:31 +00:00
79dfe2cbef
Python modules are not allowed to contain hyphens in the name. Using hyphens only works here because we're using low level __import__. Change run_tasks to replace '-' with '_' when reading configs, and rename the task modules to have valid python module names. Aside from general hygiene, the motivation to do this is to allow one task to build on code in another task by importing it. Signed-off-by: John Spray <john.spray@inktank.com>
138 lines
3.7 KiB
Python
138 lines
3.7 KiB
Python
"""
|
|
Mount cifs clients. Unmount when finished.
|
|
"""
|
|
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 cifs 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.
|
|
|
|
Example that starts smbd and mounts cifs on all nodes::
|
|
|
|
tasks:
|
|
- ceph:
|
|
- samba:
|
|
- cifs-mount:
|
|
- interactive:
|
|
|
|
Example that splits smbd and cifs:
|
|
|
|
tasks:
|
|
- ceph:
|
|
- samba: [samba.0]
|
|
- cifs-mount: [client.0]
|
|
- ceph-fuse: [client.1]
|
|
- interactive:
|
|
|
|
Example that specifies the share name:
|
|
|
|
tasks:
|
|
- ceph:
|
|
- ceph-fuse:
|
|
- samba:
|
|
samba.0:
|
|
cephfuse: "{testdir}/mnt.0"
|
|
- cifs-mount:
|
|
client.0:
|
|
share: cephfuse
|
|
|
|
:param ctx: Context
|
|
:param config: Configuration
|
|
"""
|
|
log.info('Mounting cifs clients...')
|
|
|
|
if config is None:
|
|
config = dict(('client.{id}'.format(id=id_), None)
|
|
for id_ in teuthology.all_roles_of_type(ctx.cluster, 'client'))
|
|
elif isinstance(config, list):
|
|
config = dict((name, None) for name in config)
|
|
|
|
clients = list(teuthology.get_clients(ctx=ctx, roles=config.keys()))
|
|
|
|
from teuthology.task.samba import get_sambas
|
|
samba_roles = ['samba.{id_}'.format(id_=id_) for id_ in teuthology.all_roles_of_type(ctx.cluster, 'samba')]
|
|
sambas = list(get_sambas(ctx=ctx, roles=samba_roles))
|
|
(ip, _) = sambas[0][1].ssh.get_transport().getpeername()
|
|
log.info('samba ip: {ip}'.format(ip=ip))
|
|
|
|
for id_, remote in clients:
|
|
mnt = os.path.join(teuthology.get_testdir(ctx), 'mnt.{id}'.format(id=id_))
|
|
log.info('Mounting cifs client.{id} at {remote} {mnt}...'.format(
|
|
id=id_, remote=remote,mnt=mnt))
|
|
|
|
remote.run(
|
|
args=[
|
|
'mkdir',
|
|
'--',
|
|
mnt,
|
|
],
|
|
)
|
|
|
|
rolestr = 'client.{id_}'.format(id_=id_)
|
|
unc = "ceph"
|
|
log.info("config: {c}".format(c=config))
|
|
if config[rolestr] is not None and 'share' in config[rolestr]:
|
|
unc = config[rolestr]['share']
|
|
|
|
remote.run(
|
|
args=[
|
|
'sudo',
|
|
'mount',
|
|
'-t',
|
|
'cifs',
|
|
'//{sambaip}/{unc}'.format(sambaip=ip, unc=unc),
|
|
'-o',
|
|
'username=ubuntu,password=ubuntu',
|
|
mnt,
|
|
],
|
|
)
|
|
|
|
remote.run(
|
|
args=[
|
|
'sudo',
|
|
'chown',
|
|
'ubuntu:ubuntu',
|
|
'{m}/'.format(m=mnt),
|
|
],
|
|
)
|
|
|
|
try:
|
|
yield
|
|
finally:
|
|
log.info('Unmounting cifs clients...')
|
|
for id_, remote in clients:
|
|
remote.run(
|
|
args=[
|
|
'sudo',
|
|
'umount',
|
|
mnt,
|
|
],
|
|
)
|
|
for id_, remote in clients:
|
|
while True:
|
|
try:
|
|
remote.run(
|
|
args=[
|
|
'rmdir', '--', mnt,
|
|
run.Raw('2>&1'),
|
|
run.Raw('|'),
|
|
'grep', 'Device or resource busy',
|
|
],
|
|
)
|
|
import time
|
|
time.sleep(1)
|
|
except Exception:
|
|
break
|