From df937a778157a366239353ae4bbd3b444b951137 Mon Sep 17 00:00:00 2001
From: wufarong <wufarong@huawei.com>
Date: Wed, 18 Jun 2025 17:50:40 +0800
Subject: [PATCH] =?UTF-8?q?=E4=B8=9C=E6=B9=96=E5=A2=9E=E5=8A=A0traceId?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: wufarong <wufarong@huawei.com>
Change-Id: I3665c47331b14cfdcc5f4b3b1f627aa64a68f27b
---
 frameworks/core/src/ans_notification.cpp                       | 2 ++
 .../advanced_notification_publish.cpp                          | 3 +++
 2 files changed, 5 insertions(+)

diff --git a/frameworks/core/src/ans_notification.cpp b/frameworks/core/src/ans_notification.cpp
index fc132733f..48eca4b6f 100644
--- a/frameworks/core/src/ans_notification.cpp
+++ b/frameworks/core/src/ans_notification.cpp
@@ -26,6 +26,7 @@
 #include "notification_local_live_view_subscriber.h"
 #include "system_ability_definition.h"
 #include "unique_fd.h"
+#include "hitrace_util.h"
 
 #include <memory>
 #include <thread>
@@ -280,6 +281,7 @@ ErrCode AnsNotification::PublishNotification(const std::string &label, const Not
 ErrCode AnsNotification::PublishNotificationForIndirectProxy(const NotificationRequest &request)
 {
     HITRACE_METER_NAME(HITRACE_TAG_NOTIFICATION, __PRETTY_FUNCTION__);
+    TraceChainUtil traceChain = TraceChainUtil();
     ANS_LOGI("PublishNotificationForIndirectProxy,notificationId:%{public}u", request.GetNotificationId());
 
     if (request.GetContent() == nullptr || request.GetNotificationType() == NotificationContent::Type::NONE) {
diff --git a/services/ans/src/advanced_notification_manager/advanced_notification_publish.cpp b/services/ans/src/advanced_notification_manager/advanced_notification_publish.cpp
index 6ae602d31..695043268 100644
--- a/services/ans/src/advanced_notification_manager/advanced_notification_publish.cpp
+++ b/services/ans/src/advanced_notification_manager/advanced_notification_publish.cpp
@@ -162,6 +162,8 @@ ErrCode AdvancedNotificationService::PublishNotificationForIndirectProxyWithMaxC
 ErrCode AdvancedNotificationService::PublishNotificationForIndirectProxy(const sptr<NotificationRequest> &request)
 {
     HITRACE_METER_NAME(HITRACE_TAG_NOTIFICATION, __PRETTY_FUNCTION__);
+    TraceChainUtil traceChain = TraceChainUtil();
+    OHOS::HiviewDFX::HiTraceId traceId = OHOS::HiviewDFX::HiTraceChain::GetId();
     ANS_LOGD("%{public}s", __FUNCTION__);
 
     HaMetaMessage message = HaMetaMessage(EventSceneId::SCENE_9, EventBranchId::BRANCH_0);
@@ -171,6 +173,7 @@ ErrCode AdvancedNotificationService::PublishNotificationForIndirectProxy(const s
         NotificationAnalyticsUtil::ReportPublishFailedEvent(request, message);
         return ERR_ANS_INVALID_PARAM;
     }
+    SetChainIdToExtraInfo(request, traceId);
     ErrCode result = PrePublishRequest(request);
     if (result != ERR_OK) {
         return result;
-- 
Gitee