forked from sonic-net/sonic-buildimage
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[db migrator] Introduce the DB migration infrastructure (sonic-net#519)
* [db migrator] Introduce the DB migration infrastructure - Add infrastructure to migrate DB contents from a lower version schema to a higher version schema, eventually reach latest version. - if config load_minigraph is executed, the db schema version is always the latest. Signed-off-by: Ying Xie <ying.xie@microsoft.com> * [db migrator] address issues found in test
- Loading branch information
Showing
3 changed files
with
151 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,138 @@ | ||
#!/usr/bin/env python | ||
|
||
import sys | ||
import argparse | ||
import syslog | ||
from swsssdk import ConfigDBConnector | ||
|
||
|
||
SYSLOG_IDENTIFIER = 'db_migrator' | ||
|
||
|
||
def log_info(msg): | ||
syslog.openlog(SYSLOG_IDENTIFIER) | ||
syslog.syslog(syslog.LOG_INFO, msg) | ||
syslog.closelog() | ||
|
||
|
||
def log_error(msg): | ||
syslog.openlog(SYSLOG_IDENTIFIER) | ||
syslog.syslog(syslog.LOG_ERR, msg) | ||
syslog.closelog() | ||
|
||
|
||
class DBMigrator(): | ||
def __init__(self): | ||
""" | ||
Version string format: | ||
version_<major>_<minor>_<build> | ||
major: starting from 1, sequentially incrementing in master | ||
branch. | ||
minor: in github branches, minor version stays in 0. This minor | ||
version creates space for private branches derived from | ||
github public branches. These private branches shall use | ||
none-zero values. | ||
build: sequentially increase within a minor version domain. | ||
""" | ||
self.CURRENT_VERSION = 'version_1_0_1' | ||
|
||
self.TABLE_NAME = 'VERSIONS' | ||
self.TABLE_KEY = 'DATABASE' | ||
self.TABLE_FIELD = 'VERSION' | ||
self.configDB = ConfigDBConnector() | ||
self.configDB.db_connect('CONFIG_DB') | ||
|
||
|
||
def migrate_pfc_wd_table(self): | ||
# Migrate all data entries from table PFC_WD_TABLE to PFC_WD | ||
data = self.configDB.get_table('PFC_WD_TABLE') | ||
for key in data.keys(): | ||
self.configDB.set_entry('PFC_WD', key, data[key]) | ||
self.configDB.delete_table('PFC_WD_TABLE') | ||
|
||
|
||
def version_unknown(self): | ||
""" | ||
version_unknown tracks all SONiC versions that doesn't have a version | ||
string defined in config_DB. | ||
Nothing can be assumped when migrating from this version to the next | ||
version. | ||
Any migration operation needs to test if the DB is in expected format | ||
before migrating date to the next version. | ||
""" | ||
|
||
log_info('Handling version_unknown') | ||
|
||
# NOTE: Uncomment next 3 lines of code when the migration code is in | ||
# place. Note that returning specific string is intentional, | ||
# here we only intended to migrade to DB version 1.0.1. | ||
# If new DB version is added in the future, the incremental | ||
# upgrade will take care of the subsequent migrations. | ||
# self.migrate_pfc_wd_table() | ||
# self.set_version('version_1_0_1') | ||
# return 'version_1_0_1' | ||
|
||
|
||
def version_1_0_1(self): | ||
""" | ||
Current latest version. Nothing to do here. | ||
""" | ||
log_info('Handling version_1_0_1') | ||
|
||
return None | ||
|
||
|
||
def get_version(self): | ||
version = self.configDB.get_entry(self.TABLE_NAME, self.TABLE_KEY) | ||
if version and version[self.TABLE_FIELD]: | ||
return version[self.TABLE_FIELD] | ||
|
||
return 'version_unknown' | ||
|
||
|
||
def set_version(self, version=None): | ||
if not version: | ||
version = self.CURRENT_VERSION | ||
log_info('Setting version to ' + version) | ||
entry = { self.TABLE_FIELD : version } | ||
self.configDB.set_entry(self.TABLE_NAME, self.TABLE_KEY, entry) | ||
|
||
|
||
def migrate(self): | ||
version = self.get_version() | ||
log_info('Upgrading from version ' + version) | ||
while version: | ||
next_version = getattr(self, version)() | ||
if next_version == version: | ||
raise Exception('Version migrate from %s stuck in same version' % version) | ||
version = next_version | ||
|
||
|
||
def main(): | ||
try: | ||
parser = argparse.ArgumentParser() | ||
|
||
parser.add_argument('-o', | ||
dest='operation', | ||
metavar='operation (migrate, set_version, get_version)', | ||
type = str, | ||
required = False, | ||
choices=['migrate', 'set_version', 'get_version'], | ||
help = 'operation to perform [default: get_version]', | ||
default='get_version') | ||
args = parser.parse_args() | ||
operation = args.operation | ||
|
||
dbmgtr = DBMigrator() | ||
result = getattr(dbmgtr, operation)() | ||
if result: | ||
print(str(result)) | ||
|
||
except Exception as e: | ||
log_error('Caught excetion: ' + str(e)) | ||
parser.print_help() | ||
sys.exit(1) | ||
|
||
|
||
if __name__ == "__main__": | ||
main() |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters