jmiller
bab29a83fa
Sweep all .go imports + go.mod, plus module-path refs in .golangci.yml, Dockerfile, swagger templates, locale example, and repo/wiki URLs (MokoGitea-Fork -> MokoGIT). Part of the MokoGIT rebrand. Claude-Session: https://claude.ai/code/session_01Bqe7fAuHQeiLueYfeHFrHw
77 lines
2.4 KiB
Go
77 lines
2.4 KiB
Go
// Copyright 2023 The Gitea Authors.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package admin
|
|
|
|
import (
|
|
"archive/zip"
|
|
"fmt"
|
|
"runtime/pprof"
|
|
"time"
|
|
|
|
"code.mokoconsulting.tech/MokoConsulting/MokoGIT/modules/httplib"
|
|
"code.mokoconsulting.tech/MokoConsulting/MokoGIT/modules/tailmsg"
|
|
"code.mokoconsulting.tech/MokoConsulting/MokoGIT/modules/util"
|
|
"code.mokoconsulting.tech/MokoConsulting/MokoGIT/services/context"
|
|
)
|
|
|
|
func MonitorDiagnosis(ctx *context.Context) {
|
|
seconds := min(max(ctx.FormInt64("seconds"), 1), 300)
|
|
|
|
httplib.ServeSetHeaders(ctx.Resp, httplib.ServeHeaderOptions{
|
|
ContentType: "application/zip",
|
|
Filename: fmt.Sprintf("gitea-diagnosis-%s.zip", time.Now().Format("20060102-150405")),
|
|
ContentDisposition: httplib.ContentDispositionAttachment,
|
|
})
|
|
|
|
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
|
|
}
|
|
|
|
err = pprof.StartCPUProfile(f)
|
|
if err == nil {
|
|
time.Sleep(time.Duration(seconds) * time.Second)
|
|
pprof.StopCPUProfile()
|
|
} else {
|
|
_, _ = f.Write([]byte(err.Error()))
|
|
}
|
|
|
|
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
|
|
}
|
|
_ = pprof.Lookup("heap").WriteTo(f, 0)
|
|
|
|
f, err = zipWriter.CreateHeader(&zip.FileHeader{Name: "perftrace.txt", Method: zip.Deflate, Modified: time.Now()})
|
|
if err != nil {
|
|
ctx.ServerError("Failed to create zip file", err)
|
|
return
|
|
}
|
|
for _, record := range tailmsg.GetManager().GetTraceRecorder().GetRecords() {
|
|
_, _ = f.Write(util.UnsafeStringToBytes(record.Time.Format(time.RFC3339)))
|
|
_, _ = f.Write([]byte(" "))
|
|
_, _ = f.Write(util.UnsafeStringToBytes((record.Content)))
|
|
_, _ = f.Write([]byte("\n\n"))
|
|
}
|
|
}
|