From 9f79b00d6af0224056028332ab926dbcd32d6136 Mon Sep 17 00:00:00 2001
From: Nikitin Aleksandr <nikitinav@atomsbt.ru>
Date: Fri, 4 Apr 2025 11:08:34 +0300
Subject: [PATCH] =?UTF-8?q?=D1=81=D0=B4=D0=B5=D0=BB=D0=B0=D0=BB=20PprofNat?=
 =?UTF-8?q?s1()?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../sync_exchange_connect.go                  | 19 ++++++++++++++++++-
 .../sync_exchange_connect_test.go             |  7 +++++++
 2 files changed, 25 insertions(+), 1 deletion(-)

diff --git a/sync_exchange_connect/sync_exchange_connect.go b/sync_exchange_connect/sync_exchange_connect.go
index 583d6cf6..284776e2 100644
--- a/sync_exchange_connect/sync_exchange_connect.go
+++ b/sync_exchange_connect/sync_exchange_connect.go
@@ -11,6 +11,7 @@ import (
 	"github.com/ManyakRus/starter/stopapp"
 	"gitlab.aescorp.ru/dsp_dev/claim/common/sync_exchange"
 	"gitlab.aescorp.ru/dsp_dev/claim/common/sync_exchange/sync_types"
+	"os"
 	"runtime/pprof"
 	"sync"
 	"time"
@@ -175,7 +176,9 @@ loop:
 // PprofNats1 - профилирование памяти отправляет в NATS 1 раз
 func PprofNats1() error {
 	var err error
-	topicHeapProfile := serviceName + ".heap_profile"
+
+	TextTest := TextTestOrEmpty()
+	topicHeapProfile := serviceName + TextTest + ".heap_profile"
 	var buf bytes.Buffer
 	err = pprof.WriteHeapProfile(&buf)
 	if err != nil {
@@ -194,3 +197,17 @@ func PprofNats1() error {
 
 	return err
 }
+
+// TextTestOrEmpty - возвращает "_test" или ""
+func TextTestOrEmpty() string {
+	Otvet := "_test"
+	stage := os.Getenv("STAGE")
+	switch stage {
+	case "dev":
+		Otvet = ""
+	case "prod":
+		Otvet = ""
+	}
+
+	return Otvet
+}
diff --git a/sync_exchange_connect/sync_exchange_connect_test.go b/sync_exchange_connect/sync_exchange_connect_test.go
index df003972..4e0dc983 100644
--- a/sync_exchange_connect/sync_exchange_connect_test.go
+++ b/sync_exchange_connect/sync_exchange_connect_test.go
@@ -57,3 +57,10 @@ func TestPprofNats1(t *testing.T) {
 
 	PprofNats1()
 }
+
+func TestTextTestOrEmpty(t *testing.T) {
+	Otvet := TextTestOrEmpty()
+	if Otvet != "" && Otvet != "_test" {
+		t.Errorf("TextTestOrEmpty() error: Otvet=%v", Otvet)
+	}
+}