gitea/routers/web/admin/diagnosis.go

// Copyright 2023 The Gitea Authors.
// SPDX-License-Identifier: MIT

package admin

import (
	"archive/zip"
	"fmt"
	"runtime/pprof"
	"time"

	"code.gitea.io/gitea/modules/httplib"
	"code.gitea.io/gitea/modules/tailmsg"
	"code.gitea.io/gitea/modules/util"
	"code.gitea.io/gitea/services/context"
)

func MonitorDiagnosis(ctx *context.Context) {
	seconds := min(max(ctx.FormInt64("seconds"), 1), 300)

	httplib.ServeSetHeaders(ctx.Resp, &httplib.ServeHeaderOptions{
		ContentType: "application/zip",
		Disposition: "attachment",
		Filename:    fmt.Sprintf("gitea-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
	}

	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"))
	}
}