1
0
Fork 0
mirror of https://codeberg.org/forgejo/forgejo.git synced 2025-04-30 21:28:29 +00:00

fix: delay-write trace.dat for forgejo diagnosis ()

- Delay the writing of `trace.dat` to the forgejo diagnosis zip file. It's not possible to write `cpu-profile.dat` and `trace.dat` at the same time. This caused the implemention to prematurely close `cpu-profile.dat` and leave it as an empty file.
- Regression of 

## Testing
1. Go to `/admin/monitor`
2. Click on "Download diagnosis report"
3. Open the zip file and check that all files are non-empty.

Reviewed-on: https://codeberg.org/forgejo/forgejo/pulls/7597
Reviewed-by: Earl Warren <earl-warren@noreply.codeberg.org>
Co-authored-by: Gusted <postmaster@gusted.xyz>
Co-committed-by: Gusted <postmaster@gusted.xyz>
This commit is contained in:
Gusted 2025-04-22 08:14:41 +00:00 committed by Earl Warren
parent 603d655ec4
commit 358863999e

View file

@ -5,7 +5,9 @@ package admin
import ( import (
"archive/zip" "archive/zip"
"bytes"
"fmt" "fmt"
"io"
"runtime" "runtime"
"runtime/pprof" "runtime/pprof"
"runtime/trace" "runtime/trace"
@ -45,14 +47,9 @@ func MonitorDiagnosis(ctx *context.Context) {
_, _ = f.Write([]byte(err.Error())) _, _ = f.Write([]byte(err.Error()))
} }
f, err = zipWriter.CreateHeader(&zip.FileHeader{Name: "trace.dat", Method: zip.Deflate, Modified: time.Now()}) traceBuf := &bytes.Buffer{}
if err != nil { if err := trace.Start(traceBuf); err != nil {
ctx.ServerError("Failed to create zip file", err) _, _ = traceBuf.Write([]byte(err.Error()))
return
}
if err := trace.Start(f); err != nil {
_, _ = f.Write([]byte(err.Error()))
} }
select { select {
@ -62,6 +59,17 @@ func MonitorDiagnosis(ctx *context.Context) {
pprof.StopCPUProfile() pprof.StopCPUProfile()
trace.Stop() trace.Stop()
f, err = zipWriter.CreateHeader(&zip.FileHeader{Name: "trace.dat", Method: zip.Deflate, Modified: time.Now()})
if err != nil {
ctx.ServerError("Failed to create zip file", err)
return
}
if _, err := io.Copy(f, traceBuf); err != nil {
ctx.ServerError("Failed to create zip file", err)
return
}
f, err = zipWriter.CreateHeader(&zip.FileHeader{Name: "goroutine-after.txt", Method: zip.Deflate, Modified: time.Now()}) f, err = zipWriter.CreateHeader(&zip.FileHeader{Name: "goroutine-after.txt", Method: zip.Deflate, Modified: time.Now()})
if err != nil { if err != nil {
ctx.ServerError("Failed to create zip file", err) ctx.ServerError("Failed to create zip file", err)