added the config commands for memory statistics

config/memory_statistics.py

@@ -0,0 +1,106 @@
+import click
+import sys
+from swsscommon.swsscommon import ConfigDBConnector
+
+
+# Simulate the AbbreviationGroup from utilities_common.cli
+class AbbreviationGroup(click.Group):
+    def get_command(self, ctx, cmd_name):
+        # Fallback to default command if abbreviation not found
+        return super().get_command(ctx, cmd_name)
+
+
+#
+# 'memory-statistics' group ('sudo config memory-statistics ...')
+#
+@click.group(cls=AbbreviationGroup, name="memory-statistics")
+def memory_statistics():
+    """Configure the Memory Statistics feature"""
+    pass
+
+
+def check_memory_statistics_table_existence(memory_statistics_table):
+    """Checks whether the 'MEMORY_STATISTICS' table is configured in Config DB."""
+    if not memory_statistics_table:
+        click.echo("Unable to retrieve 'MEMORY_STATISTICS' table from Config DB.")
+        sys.exit(1)
+
+    if "memory_statistics" not in memory_statistics_table:
+        click.echo("Unable to retrieve key 'memory_statistics' from MEMORY_STATISTICS table.")
+        sys.exit(2)
+
+
+#
+# 'enable' command ('sudo config memory-statistics enable')
+#
+@memory_statistics.command(name="enable", short_help="Enable the Memory Statistics feature")
+def memory_statistics_enable():
+    """Enable the Memory Statistics feature"""
+    db = ConfigDBConnector()
+    db.connect()
+
+    memory_statistics_table = db.get_table("MEMORY_STATISTICS")
+    check_memory_statistics_table_existence(memory_statistics_table)
+
+    # Set enabled to true and disabled to false
+    db.mod_entry("MEMORY_STATISTICS", "memory_statistics", {"enabled": "true", "disabled": "false"})
+    click.echo("Memory Statistics feature enabled.")
+    click.echo("Save SONiC configuration using 'config save' to persist the changes.")
+
+
+#
+# 'disable' command ('sudo config memory-statistics disable')
+#
+@memory_statistics.command(name="disable", short_help="Disable the Memory Statistics feature")
+def memory_statistics_disable():
+    """Disable the Memory Statistics feature"""
+    db = ConfigDBConnector()
+    db.connect()
+
+    memory_statistics_table = db.get_table("MEMORY_STATISTICS")
+    check_memory_statistics_table_existence(memory_statistics_table)
+
+    # Set enabled to false and disabled to true
+    db.mod_entry("MEMORY_STATISTICS", "memory_statistics", {"enabled": "false", "disabled": "true"})
+    click.echo("Memory Statistics feature disabled.")
+    click.echo("Save SONiC configuration using 'config save' to persist the changes.")
+
+
+#
+# 'retention-period' command ('sudo config memory-statistics retention-period ...')
+#
+@memory_statistics.command(name="retention-period", short_help="Configure the retention period for Memory Statistics")
+@click.argument('retention_period', metavar='<retention_period>', required=True, type=int)
+def memory_statistics_retention_period(retention_period):
+    """Set the retention period for Memory Statistics"""
+    db = ConfigDBConnector()
+    db.connect()
+
+    memory_statistics_table = db.get_table("MEMORY_STATISTICS")
+    check_memory_statistics_table_existence(memory_statistics_table)
+
+    db.mod_entry("MEMORY_STATISTICS", "memory_statistics", {"retention_time": retention_period})
+    click.echo(f"Memory Statistics retention period set to {retention_period} days.")
+    click.echo("Save SONiC configuration using 'config save' to persist the changes.")
+
+
+#
+# 'sampling-interval' command ('sudo config memory-statistics sampling-interval ...')
+#
+@memory_statistics.command(name="sampling-interval", short_help="Configure the sampling interval for Memory Statistics")
+@click.argument('sampling_interval', metavar='<sampling_interval>', required=True, type=int)
+def memory_statistics_sampling_interval(sampling_interval):
+    """Set the sampling interval for Memory Statistics"""
+    db = ConfigDBConnector()
+    db.connect()
+
+    memory_statistics_table = db.get_table("MEMORY_STATISTICS")
+    check_memory_statistics_table_existence(memory_statistics_table)
+
+    db.mod_entry("MEMORY_STATISTICS", "memory_statistics", {"sampling_interval": sampling_interval})
+    click.echo(f"Memory Statistics sampling interval set to {sampling_interval} minutes.")
+    click.echo("Save SONiC configuration using 'config save' to persist the changes.")
+
+
+if __name__ == "__main__":
+    memory_statistics()

show/memory_statistics.py

@@ -0,0 +1,100 @@
+import click
+from tabulate import tabulate
+
+import utilities_common.cli as clicommon
+from swsscommon.swsscommon import ConfigDBConnector
+
+
+#
+# 'memory-statistics' group (show memory-statistics ...)
+#
+@click.group(cls=clicommon.AliasedGroup, name="memory-statistics")
+def memory_statistics():
+    """Show memory statistics configuration and logs"""
+    pass
+
+
+def get_memory_statistics_config(field_name):
+    """Fetches the configuration of memory_statistics from `CONFIG_DB`.
+
+    Args:
+      field_name: A string containing the field name in the sub-table of 'memory_statistics'.
+
+    Returns:
+      field_value: If field name was found, then returns the corresponding value.
+                   Otherwise, returns "Unknown".
+    """
+    field_value = "Unknown"
+    config_db = ConfigDBConnector()
+    config_db.connect()
+    memory_statistics_table = config_db.get_table("MEMORY_STATISTICS")
+    if (memory_statistics_table and
+            "memory_statistics" in memory_statistics_table and
+            field_name in memory_statistics_table["config"]):
+        field_value = memory_statistics_table["memory_statistics"][field_name]
+
+    return field_value
+
+
+@memory_statistics.command(name="memory_statitics", short_help="Show the configuration of memory statistics")
+def config():
+    admin_mode = "Disabled"
+    admin_enabled = get_memory_statistics_config("enabled")
+    if admin_enabled == "true":
+        admin_mode = "Enabled"
+
+    click.echo("Memory Statistics administrative mode: {}".format(admin_mode))
+
+    retention_time = get_memory_statistics_config("retention_time")
+    click.echo("Memory Statistics retention time (days): {}".format(retention_time))
+
+    sampling_interval = get_memory_statistics_config("sampling_interval")
+    click.echo("Memory Statistics sampling interval (minutes): {}".format(sampling_interval))
+
+
+def fetch_memory_statistics(starting_time=None, ending_time=None, select=None):
+    """Fetch memory statistics from the database.
+
+    Args:
+        starting_time: The starting time for filtering the statistics.
+        ending_time: The ending time for filtering the statistics.
+        additional_options: Any additional options for filtering or formatting.
+
+    Returns:
+        A list of memory statistics entries.
+    """
+    config_db = ConfigDBConnector()
+    config_db.connect()
+
+    memory_statistics_table = config_db.get_table("MEMORY_STATISTICS")
+    filtered_statistics = []
+
+    for key, entry in memory_statistics_table.items():
+        # Add filtering logic here based on starting_time, ending_time, and select
+        if (not starting_time or entry.get("time") >= starting_time) and \
+           (not ending_time or entry.get("time") <= ending_time):
+            # Implement additional filtering based on select if needed
+            filtered_statistics.append(entry)
+
+    return filtered_statistics
+
+
+@memory_statistics.command(name="logs", short_help="Show memory statistics logs with optional filtering")
+@click.argument('starting_time', required=False)
+@click.argument('ending_time', required=False)
+@click.argument('additional_options', required=False, nargs=-1)
+def show_memory_statistics_logs(starting_time, ending_time, select):
+    """Show memory statistics logs with optional filtering by time and select."""
+
+    # Fetch memory statistics
+    memory_statistics = fetch_memory_statistics(starting_time, ending_time, select)
+
+    if not memory_statistics:
+        click.echo("No memory statistics available for the given parameters.")
+        return
+
+    # Display the memory statistics
+    headers = ["Time", "Statistic", "Value"]  # Adjust according to the actual fields
+    table_data = [[entry.get("time"), entry.get("statistic"), entry.get("value")] for entry in memory_statistics]
+
+    click.echo(tabulate(table_data, headers=headers, tablefmt="grid"))