fix profileData initialisation

mycrEEpy · Aug 18, 2023 · 7c4aead · 7c4aead
1 parent 5336909
commit 7c4aead
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/internal/pgopher/collector.go b/internal/pgopher/collector.go
@@ -85,7 +85,7 @@ func (p profileCollector) Run() {
 	case "kubernetes":
 		name := fmt.Sprintf("pgopher-profile-%s", p.target.Name)
 
-		profileData := make([]byte, 0, buf.Len())
+		profileData := make([]byte, base64.StdEncoding.EncodedLen(buf.Len()))
 
 		base64.StdEncoding.Encode(profileData, buf.Bytes())
 

diff --git a/internal/pgopher/http.go b/internal/pgopher/http.go
@@ -84,7 +84,7 @@ func (s *Server) handleProfile(ctx echo.Context) error {
 		defer file.Close()
 		defer os.Remove(file.Name())
 
-		profileData := make([]byte, 0, len(resp.Data["profile"]))
+		profileData := make([]byte, base64.StdEncoding.DecodedLen(len(resp.Data["profile"])))
 
 		_, err = base64.StdEncoding.Decode(profileData, resp.Data["profile"])
 		if err != nil {