logikonline
12f4ea03a8
Some checks failed
Build and Release / Create Release (push) Successful in 0s
Trigger Vault Plugin Rebuild / Trigger Vault Rebuild (push) Successful in 0s
Build and Release / Integration Tests (PostgreSQL) (push) Successful in 2m48s
Build and Release / Lint (push) Failing after 5m2s
Build and Release / Build Binaries (amd64, windows, windows-latest) (push) Has been skipped
Build and Release / Build Binaries (amd64, darwin, linux-latest) (push) Has been skipped
Build and Release / Build Binaries (amd64, linux, linux-latest) (push) Has been skipped
Build and Release / Build Binaries (arm64, darwin, linux-latest) (push) Has been skipped
Build and Release / Build Binaries (arm64, linux, linux-latest) (push) Has been skipped
Build and Release / Unit Tests (push) Successful in 5m37s
Go's semantic import versioning requires v2+ modules to include the major version in the module path. This enables using proper version tags (v3.x.x) instead of pseudo-versions. Updated module path: code.gitcaddy.com/server/v3
48 lines
1.5 KiB
Go
48 lines
1.5 KiB
Go
// Copyright 2024 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package db
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"code.gitcaddy.com/server/v3/modules/gtprof"
|
|
"code.gitcaddy.com/server/v3/modules/log"
|
|
"code.gitcaddy.com/server/v3/modules/setting"
|
|
|
|
"xorm.io/xorm/contexts"
|
|
)
|
|
|
|
type EngineHook struct {
|
|
Threshold time.Duration
|
|
Logger log.Logger
|
|
}
|
|
|
|
var _ contexts.Hook = (*EngineHook)(nil)
|
|
|
|
func (*EngineHook) BeforeProcess(c *contexts.ContextHook) (context.Context, error) {
|
|
ctx, _ := gtprof.GetTracer().Start(c.Ctx, gtprof.TraceSpanDatabase)
|
|
return ctx, nil
|
|
}
|
|
|
|
func (h *EngineHook) AfterProcess(c *contexts.ContextHook) error {
|
|
span := gtprof.GetContextSpan(c.Ctx)
|
|
if span != nil {
|
|
// Do not record SQL parameters here:
|
|
// * It shouldn't expose the parameters because they contain sensitive information, end users need to report the trace details safely.
|
|
// * Some parameters contain quite long texts, waste memory and are difficult to display.
|
|
span.SetAttributeString(gtprof.TraceAttrDbSQL, c.SQL)
|
|
span.End()
|
|
} else {
|
|
setting.PanicInDevOrTesting("span in database engine hook is nil")
|
|
}
|
|
if c.ExecuteTime >= h.Threshold {
|
|
// 8 is the amount of skips passed to runtime.Caller, so that in the log the correct function
|
|
// is being displayed (the function that ultimately wants to execute the query in the code)
|
|
// instead of the function of the slow query hook being called.
|
|
h.Logger.Log(8, &log.Event{Level: log.WARN}, "[Slow SQL Query] %s %v - %v", c.SQL, c.Args, c.ExecuteTime)
|
|
}
|
|
return nil
|
|
}
|