From f69cf7911ec70dc40e89eb5d0068922f589a77bf Mon Sep 17 00:00:00 2001
From: qowevisa <qowevisa@gmail.com>
Date: Sun, 9 Jun 2024 07:58:15 +0300
Subject: [PATCH] Changes in server

---
 cmd/server/main.go | 45 +++++++++++++++++++++++++++++++++++----------
 1 file changed, 35 insertions(+), 10 deletions(-)

diff --git a/cmd/server/main.go b/cmd/server/main.go
index 12e6ae0..d9991b5 100644
--- a/cmd/server/main.go
+++ b/cmd/server/main.go
@@ -10,21 +10,41 @@ import (
 	"net"
 	"os"
 	"os/signal"
+	"runtime"
+	"runtime/pprof"
 	"sync"
 	"syscall"
+	"time"
 
 	com "git.qowevisa.me/Qowevisa/gotell/communication"
 	"git.qowevisa.me/Qowevisa/gotell/env"
-	"git.qowevisa.me/Qowevisa/gotell/profilers"
 )
 
-func atEnd() {
-	profilers.GetMemoryProfiler()
+func captureHeapProfile() {
+	currentTime := time.Now().Format("2006_01_02T15_04")
+	filename := fmt.Sprintf("heap_%s.prof", currentTime)
+	f, err := os.Create(filename)
+	if err != nil {
+		log.Fatalf("could not create memory profile: %v", err)
+	}
+	defer f.Close()
+	runtime.GC()
+	if err := pprof.WriteHeapProfile(f); err != nil {
+		log.Fatalf("could not write memory profile: %v", err)
+	}
 }
 
 func main() {
-	cpuProfDefer := profilers.GetCPUProfiler()
-	defer cpuProfDefer()
+	f, err := os.Create("cpu.prof")
+	if err != nil {
+		panic(err)
+	}
+	defer f.Close()
+	if err := pprof.StartCPUProfile(f); err != nil {
+		panic(err)
+	}
+	defer pprof.StopCPUProfile()
+
 	userCenter.Init()
 	linkCenter.Init()
 	connCenter.Init()
@@ -55,20 +75,25 @@ func main() {
 	defer listener.Close()
 	log.Printf("server: listening on %s", service)
 
-	defer atEnd()
-
 	var wg sync.WaitGroup
+	quit := make(chan os.Signal, 1)
+	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
 
 	go func() {
-		c := make(chan os.Signal, 1)
-		signal.Notify(c, syscall.SIGINT, syscall.SIGTERM)
-		<-c
+		<-quit
 		log.Println("received shutdown signal")
 		listener.Close()
 		wg.Wait()
 		os.Exit(0)
 	}()
 
+	ticker := time.NewTicker(3 * time.Minute)
+	go func() {
+		for range ticker.C {
+			captureHeapProfile()
+		}
+	}()
+
 	for {
 		conn, err := listener.Accept()
 		if err != nil {