add api for clear buffer pool watermark (opencomputeproject#1804)

sonic-mgmt testcase need a rpc api to clear buffer pool watermark, to read more accurate watermark.
JaiOCP · Jun 26, 2023 · bbf7aeb · bbf7aeb
1 parent af6f9da
commit bbf7aeb
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/test/saithrift/src/switch_sai.thrift b/test/saithrift/src/switch_sai.thrift
@@ -393,6 +393,9 @@ service switch_sai_rpc {
     list<i64> sai_thrift_get_buffer_pool_stats(
                         1: sai_thrift_object_id_t buffer_pool_id,
                         2: list<sai_thrift_buffer_pool_stat_counter_t> counter_ids);
+    sai_thrift_status_t sai_thrift_clear_buffer_pool_stats(
+                        1: sai_thrift_object_id_t buffer_pool_id,
+                        2: list<sai_thrift_buffer_pool_stat_counter_t> counter_ids);
     sai_thrift_status_t sai_thrift_set_priority_group_attribute(1: sai_thrift_object_id_t pg_id,
                                                                 2: sai_thrift_attribute_t thrift_attr)
     list<i64> sai_thrift_get_pg_stats(

diff --git a/test/saithrift/src/switch_sai_rpc_server.cpp b/test/saithrift/src/switch_sai_rpc_server.cpp
@@ -3460,6 +3460,27 @@ class switch_sai_rpcHandler : virtual public switch_sai_rpcIf {
       }
   }
 
+  sai_thrift_status_t sai_thrift_clear_buffer_pool_stats(const sai_thrift_object_id_t buffer_pool_id,
+                                        const std::vector<sai_thrift_buffer_pool_stat_counter_t> &thrift_counter_ids) {
+      printf("sai_thrift_clear_buffer_pool_stats\n");
+
+      sai_status_t status = SAI_STATUS_SUCCESS;
+      sai_buffer_api_t *buffer_api;
+      status = sai_api_query(SAI_API_BUFFER, (void **) &buffer_api);
+      if (status != SAI_STATUS_SUCCESS) {
+          SAI_THRIFT_LOG_ERR("Failed to query buffer_api, status: %d", status);
+          return status;
+      }
+
+      status = buffer_api->clear_buffer_pool_stats((sai_object_id_t)buffer_pool_id,
+                                                 (uint32_t)thrift_counter_ids.size(),
+                                                 (const sai_stat_id_t *)thrift_counter_ids.data());
+      if (status != SAI_STATUS_SUCCESS) {
+          SAI_THRIFT_LOG_ERR("Failed to clear_buffer_pool_stats, status: %d", status);
+          return status;
+      }
+  }
+
   sai_thrift_status_t sai_thrift_set_priority_group_attribute(const sai_thrift_object_id_t pg_id, const sai_thrift_attribute_t& thrift_attr) {
       printf("sai_thrift_set_priority_group_attribute\n");
       sai_status_t status = SAI_STATUS_SUCCESS;