mirror of
https://codeberg.org/forgejo/forgejo.git
synced 2025-04-24 20:47:30 +00:00
358863999e
Some checks are pending
/ release (push) Waiting to run
testing / backend-checks (push) Waiting to run
testing / frontend-checks (push) Waiting to run
testing / test-unit (push) Blocked by required conditions
testing / test-e2e (push) Blocked by required conditions
testing / test-remote-cacher (redis) (push) Blocked by required conditions
testing / test-remote-cacher (valkey) (push) Blocked by required conditions
testing / test-remote-cacher (garnet) (push) Blocked by required conditions
testing / test-remote-cacher (redict) (push) Blocked by required conditions
testing / test-mysql (push) Blocked by required conditions
testing / test-pgsql (push) Blocked by required conditions
testing / test-sqlite (push) Blocked by required conditions
testing / security-check (push) Blocked by required conditions
- 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 forgejo/forgejo#6470 ## 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>
89 lines
2.3 KiB
Go
89 lines
2.3 KiB
Go
// Copyright 2023 The Gitea Authors.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package admin
|
|
|
|
import (
|
|
"archive/zip"
|
|
"bytes"
|
|
"fmt"
|
|
"io"
|
|
"runtime"
|
|
"runtime/pprof"
|
|
"runtime/trace"
|
|
"time"
|
|
|
|
"forgejo.org/modules/httplib"
|
|
"forgejo.org/services/context"
|
|
)
|
|
|
|
func MonitorDiagnosis(ctx *context.Context) {
|
|
seconds := ctx.FormInt64("seconds")
|
|
seconds = max(5, min(300, seconds))
|
|
|
|
httplib.ServeSetHeaders(ctx.Resp, &httplib.ServeHeaderOptions{
|
|
ContentType: "application/zip",
|
|
Disposition: "attachment",
|
|
Filename: fmt.Sprintf("forgejo-diagnosis-%s.zip", time.Now().Format("20060102-150405")),
|
|
})
|
|
|
|
zipWriter := zip.NewWriter(ctx.Resp)
|
|
defer zipWriter.Close()
|
|
|
|
f, err := zipWriter.CreateHeader(&zip.FileHeader{Name: "goroutine-before.txt", Method: zip.Deflate, Modified: time.Now()})
|
|
if err != nil {
|
|
ctx.ServerError("Failed to create zip file", err)
|
|
return
|
|
}
|
|
_ = pprof.Lookup("goroutine").WriteTo(f, 1)
|
|
|
|
f, err = zipWriter.CreateHeader(&zip.FileHeader{Name: "cpu-profile.dat", Method: zip.Deflate, Modified: time.Now()})
|
|
if err != nil {
|
|
ctx.ServerError("Failed to create zip file", err)
|
|
return
|
|
}
|
|
|
|
if err := pprof.StartCPUProfile(f); err != nil {
|
|
_, _ = f.Write([]byte(err.Error()))
|
|
}
|
|
|
|
traceBuf := &bytes.Buffer{}
|
|
if err := trace.Start(traceBuf); err != nil {
|
|
_, _ = traceBuf.Write([]byte(err.Error()))
|
|
}
|
|
|
|
select {
|
|
case <-time.After(time.Duration(seconds) * time.Second):
|
|
case <-ctx.Done():
|
|
}
|
|
pprof.StopCPUProfile()
|
|
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()})
|
|
if err != nil {
|
|
ctx.ServerError("Failed to create zip file", err)
|
|
return
|
|
}
|
|
_ = pprof.Lookup("goroutine").WriteTo(f, 1)
|
|
|
|
f, err = zipWriter.CreateHeader(&zip.FileHeader{Name: "heap.dat", Method: zip.Deflate, Modified: time.Now()})
|
|
if err != nil {
|
|
ctx.ServerError("Failed to create zip file", err)
|
|
return
|
|
}
|
|
// To avoid showing memory that actually can be cleaned, run the garbage
|
|
// collector.
|
|
runtime.GC()
|
|
_ = pprof.Lookup("heap").WriteTo(f, 0)
|
|
}
|