Skip to content

Commit

Permalink
show commands for system ready and application extension package app …
Browse files Browse the repository at this point in the history
…ready support
  • Loading branch information
sg893052 committed Apr 30, 2022
1 parent fdb79b8 commit bdbfe85
Show file tree
Hide file tree
Showing 8 changed files with 223 additions and 1 deletion.
119 changes: 119 additions & 0 deletions scripts/sysreadyshow
Original file line number Diff line number Diff line change
@@ -0,0 +1,119 @@
#!/usr/bin/env python3

"""
Script to show system ready status.
"""

import os
import sys
import argparse
from tabulate import tabulate
from natsort import natsorted

# mock the redis for unit test purposes #
try:
if os.environ["UTILITIES_UNIT_TESTING"] == "1":
modules_path = os.path.join(os.path.dirname(__file__), "..")
test_path = os.path.join(modules_path, "tests")
sys.path.insert(0, modules_path)
sys.path.insert(0, test_path)
import mock_tables.dbconnector #lgtm [py/unused-import]
except KeyError:
pass

from swsscommon.swsscommon import SonicV2Connector

header = ['Service-Name', 'Service-Status', 'App-Ready-Status', 'Down-Reason']
header_detail = ['Service-Name', 'Service-Status', 'App-Ready-Status', 'Down-Reason', 'AppStatus-UpdateTime']

ALL_TABLE_NAME = 'ALL_SERVICE_STATUS'
SERVICE_STATUS = 'service_status'
APP_READY_STATUS = 'app_ready_status'
FAIL_REASON = 'fail_reason'
UPDATE_TIME = 'update_time'

class SysreadyShow(object):
def __init__(self):
self.db = SonicV2Connector(host="127.0.0.1")
self.db.connect(self.db.STATE_DB)

def show(self,type):
TABLE_NAME = ALL_TABLE_NAME
SYSREADY_TABLE = "SYSTEM_READY|SYSTEM_STATE"

keys = self.db.keys(self.db.STATE_DB, TABLE_NAME + '*')
if not keys:
print('No info\n')
return

table = []
for key in natsorted(keys):
key_list = key.split('|')
if len(key_list) != 2: # error data in DB, log it and ignore
print('Warn: Invalid key in table {}: {}'.format(TABLE_NAME, key))
continue

name = key_list[1]
data_dict = self.db.get_all(self.db.STATE_DB, key)
try:
service_status = data_dict[SERVICE_STATUS]
app_ready_status = data_dict[APP_READY_STATUS]
fail_reason = data_dict[FAIL_REASON]
update_time = data_dict[UPDATE_TIME]
except ValueError as e:
print('Error in data_dict')

if type == "alldetail":
table.append((name, service_status, app_ready_status, fail_reason, update_time))
header_info = header_detail
else:
table.append((name, service_status, app_ready_status, fail_reason))
header_info = header

sysready_state = self.db.get(self.db.STATE_DB, SYSREADY_TABLE, "Status")
if sysready_state == "UP":
print("System is ready\n")
else:
print("System is not ready - one or more services are not up\n")


if type == "allbrief":
return


if table:
print(tabulate(table, header_info, tablefmt='simple', stralign='left'))
else:
print('No sysready status data available\n')


def main():
parser = argparse.ArgumentParser(description='Display the System Ready status',
formatter_class=argparse.RawTextHelpFormatter,
epilog="""
Examples:
sysreadyshow --all
sysreadyshow --allbrief
sysreadyshow --alldetail
""")

parser.add_argument('-a', '--all', action='store_true', help='all service status', default=True)
parser.add_argument('-b', '--allbrief', action='store_true', help='all service status brief', default=False)
parser.add_argument('-d', '--alldetail', action='store_true', help='all service status detail', default=False)
args = parser.parse_args()

try:
sysready = SysreadyShow()
if args.alldetail:
sysready.show("alldetail")
elif args.allbrief:
sysready.show("allbrief")
else:
sysready.show("all")
except Exception as e:
print(str(e), file=sys.stderr)
sys.exit(1)


if __name__ == "__main__":
main()
3 changes: 2 additions & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -147,7 +147,8 @@
'scripts/null_route_helper',
'scripts/coredump_gen_handler.py',
'scripts/techsupport_cleanup.py',
'scripts/check_db_integrity.py'
'scripts/check_db_integrity.py',
'scripts/sysreadyshow'
],
entry_points={
'console_scripts': [
Expand Down
31 changes: 31 additions & 0 deletions show/system_health.py
Original file line number Diff line number Diff line change
Expand Up @@ -198,3 +198,34 @@ def monitor_list():
entry.append(element[1]['type'])
table.append(entry)
click.echo(tabulate(table, header))


@system_health.group('sysready-status',invoke_without_command=True)
@click.pass_context
def sysready_status(ctx):
"""Show system-health system ready status"""

if ctx.invoked_subcommand is None:
try:
cmd = "sysreadyshow --all"
clicommon.run_command(cmd, display_cmd=False)
except Exception as e:
click.echo("Exception: {}".format(str(e)))


@sysready_status.command('brief')
def sysready_status_brief():
try:
cmd = "sysreadyshow --allbrief"
clicommon.run_command(cmd, display_cmd=False)
except Exception as e:
click.echo("Exception: {}".format(str(e)))


@sysready_status.command('detail')
def sysready_status_detail():
try:
cmd = "sysreadyshow --alldetail"
clicommon.run_command(cmd, display_cmd=False)
except Exception as e:
click.echo("Exception: {}".format(str(e)))
1 change: 1 addition & 0 deletions sonic_package_manager/manifest.py
Original file line number Diff line number Diff line change
Expand Up @@ -177,6 +177,7 @@ def unmarshal(self, value):
ManifestField('asic-service', DefaultMarshaller(bool), False),
ManifestField('host-service', DefaultMarshaller(bool), True),
ManifestField('delayed', DefaultMarshaller(bool), False),
ManifestField('check_up_status', DefaultMarshaller(bool), False),
ManifestRoot('warm-shutdown', [
ManifestArray('after', DefaultMarshaller(str)),
ManifestArray('before', DefaultMarshaller(str)),
Expand Down
1 change: 1 addition & 0 deletions sonic_package_manager/service_creator/feature.py
Original file line number Diff line number Diff line change
Expand Up @@ -143,4 +143,5 @@ def get_non_configurable_feature_entries(manifest) -> Dict[str, str]:
'has_per_asic_scope': str(manifest['service']['asic-service']),
'has_global_scope': str(manifest['service']['host-service']),
'has_timer': str(manifest['service']['delayed']),
'check_up_status': str(manifest['service']['check_up_status']),
}
27 changes: 27 additions & 0 deletions tests/mock_tables/state_db.json
Original file line number Diff line number Diff line change
Expand Up @@ -822,5 +822,32 @@
"admin_status": "up",
"mtu": "9100",
"speed": "1000"
},
"ALL_SERVICE_STATUS|mgmt-framework": {
"app_ready_status": "OK",
"fail_reason": "-",
"service_status": "OK",
"update_time": "-"
},
"ALL_SERVICE_STATUS|swss": {
"app_ready_status": "OK",
"fail_reason": "-",
"service_status": "OK",
"update_time": "-"
},
"ALL_SERVICE_STATUS|bgp": {
"app_ready_status": "Down",
"fail_reason": "Inactive",
"service_status": "Down",
"update_time": "-"
},
"ALL_SERVICE_STATUS|pmon": {
"app_ready_status": "OK",
"fail_reason": "-",
"service_status": "OK",
"update_time": "-"
},
"SYSTEM_READY|SYSTEM_STATE": {
"Status":"DOWN"
}
}
5 changes: 5 additions & 0 deletions tests/sonic_package_manager/test_service_creator.py
Original file line number Diff line number Diff line change
Expand Up @@ -215,6 +215,7 @@ def test_feature_registration(mock_sonic_db, manifest):
'has_per_asic_scope': 'False',
'has_global_scope': 'True',
'has_timer': 'False',
'check_up_status': 'False',
})


Expand All @@ -227,6 +228,7 @@ def test_feature_update(mock_sonic_db, manifest):
'has_per_asic_scope': 'False',
'has_global_scope': 'True',
'has_timer': 'False',
'check_up_status': 'False',
}
mock_connector = Mock()
mock_connector.get_entry = Mock(return_value=curr_feature_config)
Expand All @@ -249,6 +251,7 @@ def test_feature_update(mock_sonic_db, manifest):
'has_per_asic_scope': 'False',
'has_global_scope': 'True',
'has_timer': 'True',
'check_up_status': 'False',
}),
], any_order=True)

Expand All @@ -268,6 +271,7 @@ def test_feature_registration_with_timer(mock_sonic_db, manifest):
'has_per_asic_scope': 'False',
'has_global_scope': 'True',
'has_timer': 'True',
'check_up_status': 'False',
})


Expand All @@ -285,4 +289,5 @@ def test_feature_registration_with_non_default_owner(mock_sonic_db, manifest):
'has_per_asic_scope': 'False',
'has_global_scope': 'True',
'has_timer': 'False',
'check_up_status': 'False',
})
37 changes: 37 additions & 0 deletions tests/system_health_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -306,6 +306,43 @@ def test_health_detail(self):
"""
assert result.output == expected

def test_health_systemready(self):
runner = CliRunner()
result = runner.invoke(show.cli.commands["system-health"].commands["sysready-status"])
click.echo(result.output)
print("myresult:{}".format(result.output))
expected = """\
System is not ready - one or more services are not up
Service-Name Service-Status App-Ready-Status Down-Reason
-------------- ---------------- ------------------ -------------
bgp Down Down Inactive
mgmt-framework OK OK -
pmon OK OK -
swss OK OK -
"""
assert result.output == expected
result = runner.invoke(show.cli.commands["system-health"].commands["sysready-status"],["brief"])
click.echo(result.output)
print("myresult:{}".format(result.output))
expected = """\
System is not ready - one or more services are not up
"""
assert result.output == expected
result = runner.invoke(show.cli.commands["system-health"].commands["sysready-status"],["detail"])
click.echo(result.output)
print("myresult:{}".format(result.output))
expected = """\
System is not ready - one or more services are not up
Service-Name Service-Status App-Ready-Status Down-Reason AppStatus-UpdateTime
-------------- ---------------- ------------------ ------------- ----------------------
bgp Down Down Inactive -
mgmt-framework OK OK - -
pmon OK OK - -
swss OK OK - -
"""

@classmethod
def teardown_class(cls):
print("TEARDOWN")
Expand Down

0 comments on commit bdbfe85

Please sign in to comment.