You've already forked mariadb-columnstore-engine
mirror of
https://github.com/mariadb-corporation/mariadb-columnstore-engine.git
synced 2025-10-31 18:30:33 +03:00
a76e153a1d
Implements the initial upgrade capability across CMAPI and the CLI, including
repository setup, package operations, environment prechecks, and coordinated
cluster steps with progress reporting.
Details:
- CMAPI upgrade manager:
- Add `cmapi/cmapi_server/managers/upgrade/` modules:
- `repo.py`, `packages.py`, `preinstall.py`, `upgrade.py`, `utils.py` and `__init__.py`
- Extend endpoints and routing to expose upgrade operations and status:
- `cmapi_server/controllers/{endpoints.py, dispatcher.py, api_clients.py}`
- `cmapi_server/managers/{application.py, process.py}`
- Add improved constants and helpers for upgrade flow
- Backup/restore and safety:
- Add `cmapi_server/managers/backup_restore.py`
- Fix pre-upgrade backup regressions (due to `mcs_backup_manager.sh 3.17 changes`)
- Improve cluster version validation; add `ignore_missmatch` override
- CLI enhancements:
- Progress UI and richer feedback (`mcs_cluster_tool/tools_commands.py`, `README.md`, `mcs.1`)
- Add steps to start MDB and start MCS during/after upgrade
- Improved error surfacing for version validation
- Platform and packaging:
- Ubuntu and Rocky Linux support
- RHEL/DNF dry-run support
- Distro detection and platform-dependent logic hardened
- Logging improvements
- Updater service:
- Add `cmapi/updater/cmapi_updater.service.template` and `cmapi_updater.sh` to make CMAPI update itself
- Docs:
- Update mcs cli README and mcs.1 man file
- Add `cmapi/updater/README.md`
354 lines
10 KiB
Python
354 lines
10 KiB
Python
"""Typer application for restore Columnstore data."""
|
|
import logging
|
|
import sys
|
|
from typing import Optional
|
|
|
|
import typer
|
|
from typing_extensions import Annotated
|
|
|
|
from cmapi_server.process_dispatchers.base import BaseDispatcher
|
|
from cmapi_server.constants import MCS_BACKUP_MANAGER_SH
|
|
from mcs_cluster_tool.decorators import handle_output
|
|
from mcs_cluster_tool.helpers import cook_sh_arg
|
|
|
|
|
|
logger = logging.getLogger('mcs_cli')
|
|
# pylint: disable=unused-argument, too-many-arguments, too-many-locals
|
|
# pylint: disable=invalid-name, line-too-long
|
|
|
|
|
|
@handle_output
|
|
def restore(
|
|
l: Annotated[
|
|
str,
|
|
typer.Option(
|
|
'-l', '--load',
|
|
help='What date folder to load from the backup_location.'
|
|
)
|
|
] = '',
|
|
bl: Annotated[
|
|
str,
|
|
typer.Option(
|
|
'-bl', '--backup-location',
|
|
help=(
|
|
'Where the backup to load is found.\n'
|
|
'Example: /mnt/backups/'
|
|
)
|
|
)
|
|
] = '/tmp/backups/',
|
|
bd: Annotated[
|
|
str,
|
|
typer.Option(
|
|
'-bd', '--backup_destination',
|
|
help=(
|
|
'Is this backup on the same or remote server compared to '
|
|
'where this script is running.\n'
|
|
'Options: "Local" or "Remote"'
|
|
)
|
|
)
|
|
] = 'Local',
|
|
scp: Annotated[
|
|
str,
|
|
typer.Option(
|
|
'-scp', '--secure-copy-protocol',
|
|
help=(
|
|
'Used only if --backup-destination=Remote'
|
|
'The user/credentials that will be used to scp the backup files.'
|
|
'Example: "centos@10.14.51.62"'
|
|
)
|
|
)
|
|
] = '',
|
|
bb: Annotated[
|
|
str,
|
|
typer.Option(
|
|
'-bb', '--backup-bucket',
|
|
help=(
|
|
'Only used if --storage=S3\n'
|
|
'Name of the bucket to store the columnstore backups.\n'
|
|
'Example: "s3://my-cs-backups"'
|
|
)
|
|
)
|
|
] = '',
|
|
url: Annotated[
|
|
str,
|
|
typer.Option(
|
|
'-url', '--endpoint-url',
|
|
help=(
|
|
'Used by on premise S3 vendors.\n'
|
|
'Example: "http://127.0.0.1:8000"'
|
|
)
|
|
)
|
|
] = '',
|
|
s: Annotated[
|
|
str,
|
|
typer.Option(
|
|
'-s', '--storage',
|
|
help=(
|
|
'What storage topogoly is being used by Columnstore - found '
|
|
'in /etc/columnstore/storagemanager.cnf.\n'
|
|
'Options: "LocalStorage" or "S3"'
|
|
)
|
|
)
|
|
] = 'LocalStorage',
|
|
dbs: Annotated[
|
|
int,
|
|
typer.Option(
|
|
'-dbs', '--dbroots',
|
|
help='Number of database roots in the backup.'
|
|
)
|
|
] = 1,
|
|
pm: Annotated[
|
|
str,
|
|
typer.Option(
|
|
'-pm', '--nodeid',
|
|
help=(
|
|
'Forces the handling of the restore as this node as opposed '
|
|
'to whats detected on disk.'
|
|
)
|
|
)
|
|
] = '',
|
|
nb: Annotated[
|
|
str,
|
|
typer.Option(
|
|
'-nb', '--new-bucket',
|
|
help=(
|
|
'Defines the new bucket to copy the s3 data to from the '
|
|
'backup bucket. Use -nb if the new restored cluster should '
|
|
'use a different bucket than the backup bucket itself.'
|
|
)
|
|
)
|
|
] = '',
|
|
nr: Annotated[
|
|
str,
|
|
typer.Option(
|
|
'-nr', '--new-region',
|
|
help=(
|
|
'Defines the region of the new bucket to copy the s3 data to '
|
|
'from the backup bucket.'
|
|
)
|
|
)
|
|
] = '',
|
|
nk: Annotated[
|
|
str,
|
|
typer.Option(
|
|
'-nk', '--new-key',
|
|
help='Defines the aws key to connect to the new_bucket.'
|
|
)
|
|
] = '',
|
|
ns: Annotated[
|
|
str,
|
|
typer.Option(
|
|
'-ns', '--new-secret',
|
|
help=(
|
|
'Defines the aws secret of the aws key to connect to the '
|
|
'new_bucket.'
|
|
)
|
|
)
|
|
] = '',
|
|
P: Annotated[
|
|
int,
|
|
typer.Option(
|
|
'-P', '--parallel',
|
|
help=(
|
|
'Determines number of decompression and mdbstream threads. '
|
|
'Ignored if "-c/--compress" argument not set.'
|
|
)
|
|
)
|
|
] = 4,
|
|
ha: Annotated[
|
|
Optional[bool],
|
|
typer.Option(
|
|
'-ha', '--highavilability',
|
|
help=(
|
|
'Flag for high available systems (meaning shared storage '
|
|
'exists supporting the topology so that each node sees '
|
|
'all data)'
|
|
),
|
|
show_default=False
|
|
)
|
|
] = None,
|
|
cont: Annotated[
|
|
Optional[bool],
|
|
typer.Option(
|
|
'-cont', '--continue',
|
|
help=(
|
|
'This acknowledges data in your --new_bucket is ok to delete '
|
|
'when restoring S3. When set to true skips the enforcement '
|
|
'that new_bucket should be empty prior to starting a restore.'
|
|
),
|
|
show_default=False
|
|
)
|
|
] = None,
|
|
f: Annotated[
|
|
Optional[str],
|
|
typer.Option(
|
|
'-f', '--config-file',
|
|
help=(
|
|
'Path to backup configuration file to load variables from - '
|
|
'relative or full path accepted.'
|
|
),
|
|
show_default=False
|
|
)
|
|
] = None,
|
|
smdb: Annotated[
|
|
Optional[bool],
|
|
typer.Option(
|
|
'-smdb', '--skip-mariadb-backup',
|
|
help=(
|
|
'Skip restoring mariadb server via mariadb-backup - ideal for '
|
|
'only restoring columnstore.'
|
|
),
|
|
show_default=False
|
|
)
|
|
] = None,
|
|
sb: Annotated[
|
|
Optional[bool],
|
|
typer.Option(
|
|
'-sb', '--skip-bucket-data',
|
|
help=(
|
|
'Skip restoring columnstore data in the bucket - ideal if '
|
|
'looking to only restore mariadb server.'
|
|
)
|
|
)
|
|
] = None,
|
|
m: Annotated[
|
|
Optional[str],
|
|
typer.Option(
|
|
'-m', '--mode',
|
|
help=(
|
|
'Modes ["direct","indirect"] - direct backups run on the '
|
|
'columnstore nodes themselves. indirect run on another '
|
|
'machine that has read-only mounts associated with '
|
|
'columnstore/mariadb\n'
|
|
),
|
|
hidden=True,
|
|
show_default='direct'
|
|
)
|
|
] = None,
|
|
c: Annotated[
|
|
Optional[str],
|
|
typer.Option(
|
|
'-c', '--compress',
|
|
help=(
|
|
'Hint that the backup is compressed in X format. '
|
|
'Options: [ pigz ].'
|
|
),
|
|
show_default=False
|
|
)
|
|
] = None,
|
|
q: Annotated[
|
|
Optional[bool],
|
|
typer.Option(
|
|
'-q', '--quiet',
|
|
help='Silence verbose copy command outputs.',
|
|
show_default=False
|
|
)
|
|
] = None,
|
|
nv_ssl: Annotated[
|
|
Optional[bool],
|
|
typer.Option(
|
|
'-nv-ssl','--no-verify-ssl',
|
|
help='Skips verifying ssl certs, useful for onpremise s3 storage.',
|
|
show_default=False,
|
|
)
|
|
] = None,
|
|
li: Annotated[
|
|
Optional[bool],
|
|
typer.Option(
|
|
'-li', '--list',
|
|
help='List backups.',
|
|
show_default=False
|
|
)
|
|
] = None
|
|
):
|
|
"""Restore Columnstore (and/or MariaDB) data."""
|
|
|
|
# Local Storage Examples:
|
|
# ./$0 restore -s LocalStorage -bl /tmp/backups/ -bd Local -l 12-29-2021
|
|
# ./$0 restore -s LocalStorage -bl /tmp/backups/ -bd Remote -scp root@172.31.6.163 -l 12-29-2021
|
|
|
|
# S3 Storage Examples:
|
|
# ./$0 restore -s S3 -bb s3://my-cs-backups -l 12-29-2021
|
|
# ./$0 restore -s S3 -bb gs://on-premise-bucket -l 12-29-2021 -url http://127.0.0.1:8000
|
|
# ./$0 restore -s S3 -bb s3://my-cs-backups -l 08-16-2022 -nb s3://new-data-bucket -nr us-east-1 -nk AKIAxxxxxxx3FHCADF -ns GGGuxxxxxxxxxxnqa72csk5 -ha
|
|
arguments = []
|
|
for arg_name, value in locals().items():
|
|
sh_arg = cook_sh_arg(arg_name, value)
|
|
if sh_arg is None:
|
|
continue
|
|
arguments.append(sh_arg)
|
|
cmd = f'{MCS_BACKUP_MANAGER_SH} restore {" ".join(arguments)}'
|
|
success, _ = BaseDispatcher.exec_command(cmd, stdout=sys.stdout)
|
|
return {'success': success}
|
|
|
|
|
|
@handle_output
|
|
def dbrm_restore(
|
|
bl: Annotated[
|
|
str,
|
|
typer.Option(
|
|
'-bl', '--backup-location',
|
|
help='Path of where dbrm backups exist on disk.'
|
|
)
|
|
] = '/tmp/dbrm_backups',
|
|
l: Annotated[
|
|
str,
|
|
typer.Option(
|
|
'-l', '--load',
|
|
help='Name of the directory to restore from -bl'
|
|
)
|
|
] = '',
|
|
ns: Annotated[
|
|
Optional[bool],
|
|
typer.Option(
|
|
'-ns', '--no-start',
|
|
help=(
|
|
'Do not attempt columnstore startup post dbrm_restore.'
|
|
),
|
|
show_default=False
|
|
)
|
|
] = None,
|
|
sdbk: Annotated[
|
|
Optional[bool],
|
|
typer.Option(
|
|
'-sdbk', '--skip-dbrm-backup',
|
|
help=(
|
|
'Skip backing up dbrms before restoring.'
|
|
),
|
|
show_default=False
|
|
)
|
|
] = None,
|
|
ssm: Annotated[
|
|
Optional[bool],
|
|
typer.Option(
|
|
'-ssm', '--skip-storage-manager',
|
|
help='Skip backing up storagemanager directory.',
|
|
show_default=False
|
|
)
|
|
] = None,
|
|
li: Annotated[
|
|
bool,
|
|
typer.Option(
|
|
'-li', '--list',
|
|
help='List backups.',
|
|
show_default=False
|
|
)
|
|
] = None
|
|
):
|
|
"""Restore Columnstore DBRM data."""
|
|
|
|
# Default: ./$0 dbrm_restore --backup-location /tmp/dbrm_backups
|
|
|
|
# Examples:
|
|
# ./$0 dbrm_restore --backup-location /tmp/dbrm_backups --load dbrm_backup_20240318_172842
|
|
# ./$0 dbrm_restore --backup-location /tmp/dbrm_backups --load dbrm_backup_20240318_172842 --no-startdbrm_restore --path /tmp/dbrm_backups --directory dbrm_backup_20240318_172842 --no-start
|
|
arguments = []
|
|
for arg_name, value in locals().items():
|
|
sh_arg = cook_sh_arg(arg_name, value)
|
|
if sh_arg is None:
|
|
continue
|
|
arguments.append(sh_arg)
|
|
cmd = f'{MCS_BACKUP_MANAGER_SH} dbrm_restore {" ".join(arguments)}'
|
|
success, _ = BaseDispatcher.exec_command(cmd, stdout=sys.stdout)
|
|
return {'success': success}
|