test: pprof

openimsdk · Jul 28, 2023 · db69115 · db69115
1 parent f74c8b9
commit db69115
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 0 deletions.
diff --git a/cmd/api/admin_api/main.go b/cmd/api/admin_api/main.go
@@ -19,6 +19,7 @@ import (
 	mw2 "github.com/OpenIMSDK/chat/pkg/common/mw"
 	"math/rand"
 	"net"
+	"net/http"
 	"strconv"
 	"time"
 
@@ -31,9 +32,13 @@ import (
 	"google.golang.org/grpc/credentials/insecure"
 
 	"github.com/gin-gonic/gin"
+	_ "net/http/pprof"
 )
 
 func main() {
+	go func() {
+		_ = http.ListenAndServe(":6061", nil)
+	}()
 	rand.Seed(time.Now().UnixNano())
 	if err := config.InitConfig(); err != nil {
 		panic(err)

diff --git a/cmd/api/chat_api/main.go b/cmd/api/chat_api/main.go
@@ -19,6 +19,7 @@ import (
 	mw2 "github.com/OpenIMSDK/chat/pkg/common/mw"
 	"math/rand"
 	"net"
+	"net/http"
 	"strconv"
 	"time"
 
@@ -31,9 +32,13 @@ import (
 	"google.golang.org/grpc/credentials/insecure"
 
 	"github.com/gin-gonic/gin"
+	_ "net/http/pprof"
 )
 
 func main() {
+	go func() {
+		_ = http.ListenAndServe(":6062", nil)
+	}()
 	rand.Seed(time.Now().UnixNano())
 	err := config.InitConfig()
 	if err != nil {

diff --git a/cmd/rpc/admin/main.go b/cmd/rpc/admin/main.go
@@ -17,12 +17,17 @@ package main
 import (
 	"github.com/OpenIMSDK/chat/pkg/common/chatrpcstart"
 	"github.com/OpenIMSDK/tools/log"
+	"net/http"
 
 	"github.com/OpenIMSDK/chat/internal/rpc/admin"
 	"github.com/OpenIMSDK/chat/pkg/common/config"
+	_ "net/http/pprof"
 )
 
 func main() {
+	go func() {
+		_ = http.ListenAndServe(":6063", nil)
+	}()
 	if err := config.InitConfig(); err != nil {
 		panic(err)
 	}

diff --git a/cmd/rpc/chat/main.go b/cmd/rpc/chat/main.go
@@ -17,12 +17,17 @@ package main
 import (
 	"github.com/OpenIMSDK/chat/pkg/common/chatrpcstart"
 	"github.com/OpenIMSDK/tools/log"
+	"net/http"
 
 	"github.com/OpenIMSDK/chat/internal/rpc/chat"
 	"github.com/OpenIMSDK/chat/pkg/common/config"
+	_ "net/http/pprof"
 )
 
 func main() {
+	go func() {
+		_ = http.ListenAndServe(":6064", nil)
+	}()
 	if err := config.InitConfig(); err != nil {
 		panic(err)
 	}