From aa5e95c177d9dee6f26b0051fb2a5342849528fb Mon Sep 17 00:00:00 2001
From: Yuchao Yan <yuchaoyan@microsoft.com>
Date: Thu, 10 Apr 2025 17:29:33 +0800
Subject: [PATCH] add count time logic

---
 tools/azure-sdk-tools/packaging_tools/sdk_generator.py | 3 +++
 tools/azure-sdk-tools/packaging_tools/sdk_package.py   | 6 ++++++
 2 files changed, 9 insertions(+)

diff --git a/tools/azure-sdk-tools/packaging_tools/sdk_generator.py b/tools/azure-sdk-tools/packaging_tools/sdk_generator.py
index d66ddc2e3d67..74b9229030ea 100644
--- a/tools/azure-sdk-tools/packaging_tools/sdk_generator.py
+++ b/tools/azure-sdk-tools/packaging_tools/sdk_generator.py
@@ -1,4 +1,5 @@
 import sys
+import time
 from typing import List, Dict, Any
 import argparse
 import json
@@ -223,6 +224,7 @@ def main(generate_input, generate_output):
     for readme_or_tsp in readme_and_tsp:
         _LOGGER.info(f"[CODEGEN]({readme_or_tsp})codegen begin")
         try:
+            code_generation_start_time = time.time()
             if "resource-manager" in readme_or_tsp:
                 relative_path_readme = str(Path(spec_folder, readme_or_tsp))
                 del_outdated_files(relative_path_readme)
@@ -244,6 +246,7 @@ def main(generate_input, generate_output):
             else:
                 del_outdated_generated_files(str(Path(spec_folder, readme_or_tsp)))
                 config = gen_typespec(readme_or_tsp, spec_folder, data["headSha"], data["repoHttpsUrl"])
+            _LOGGER.info(f"code generation cost time: {int(time.time() - code_generation_start_time)} seconds")
         except Exception as e:
             _LOGGER.error(f"fail to generate sdk for {readme_or_tsp}: {str(e)}")
             for hint_message in [
diff --git a/tools/azure-sdk-tools/packaging_tools/sdk_package.py b/tools/azure-sdk-tools/packaging_tools/sdk_package.py
index 7ff3c6900849..e288d8d72a08 100644
--- a/tools/azure-sdk-tools/packaging_tools/sdk_package.py
+++ b/tools/azure-sdk-tools/packaging_tools/sdk_package.py
@@ -1,4 +1,5 @@
 import sys
+import time
 import argparse
 import json
 import logging
@@ -45,12 +46,15 @@ def main(generate_input, generate_output):
             prefolder=prefolder,
             is_multiapi=package["isMultiapi"],
         )
+
+        changelog_generation_start_time = time.time()
         try:
             md_output = execute_func_with_timeout(change_log_func)
         except multiprocessing.TimeoutError:
             md_output = "change log generation was timeout!!!"
         except:
             md_output = "change log generation failed!!!"
+        _LOGGER.info(f"changelog generation cost time: {int(time.time() - changelog_generation_start_time)} seconds")
         package["changelog"] = {
             "content": md_output,
             "hasBreakingChange": "Breaking Changes" in md_output,
@@ -61,6 +65,7 @@ def main(generate_input, generate_output):
         _LOGGER.info(f"[PACKAGE]({package_name})[CHANGELOG]:{md_output}")
         # Generate api stub File
         folder_name = package["path"][0]
+        apiview_start_time = time.time()
         try:
             package_path = Path(sdk_folder, folder_name, package_name)
             check_call(
@@ -82,6 +87,7 @@ def main(generate_input, generate_output):
                     package["apiViewArtifact"] = str(Path(package_path, file))
         except Exception as e:
             _LOGGER.debug(f"Fail to generate ApiView token file for {package_name}: {e}")
+        _LOGGER.info(f"apiview generation cost time: {int(time.time() - apiview_start_time)} seconds")
 
         # check generated files and update package["version"]
         if package_name.startswith("azure-mgmt-"):