diff options
Diffstat (limited to 'internal/db/bundb/trace.go')
-rw-r--r-- | internal/db/bundb/trace.go | 60 |
1 files changed, 0 insertions, 60 deletions
diff --git a/internal/db/bundb/trace.go b/internal/db/bundb/trace.go deleted file mode 100644 index 27b5e22ac..000000000 --- a/internal/db/bundb/trace.go +++ /dev/null @@ -1,60 +0,0 @@ -/* - GoToSocial - Copyright (C) 2021-2022 GoToSocial Authors admin@gotosocial.org - - This program is free software: you can redistribute it and/or modify - it under the terms of the GNU Affero General Public License as published by - the Free Software Foundation, either version 3 of the License, or - (at your option) any later version. - - This program is distributed in the hope that it will be useful, - but WITHOUT ANY WARRANTY; without even the implied warranty of - MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the - GNU Affero General Public License for more details. - - You should have received a copy of the GNU Affero General Public License - along with this program. If not, see <http://www.gnu.org/licenses/>. -*/ - -package bundb - -import ( - "context" - "time" - - "github.com/sirupsen/logrus" - "github.com/uptrace/bun" -) - -func newDebugQueryHook() bun.QueryHook { - return &debugQueryHook{} -} - -// debugQueryHook implements bun.QueryHook -type debugQueryHook struct { -} - -func (q *debugQueryHook) BeforeQuery(ctx context.Context, _ *bun.QueryEvent) context.Context { - // do nothing - return ctx -} - -// AfterQuery logs the time taken to query, the operation (select, update, etc), and the query itself as translated by bun. -func (q *debugQueryHook) AfterQuery(_ context.Context, event *bun.QueryEvent) { - dur := time.Since(event.StartTime).Round(time.Microsecond) - l := logrus.WithFields(logrus.Fields{ - "duration": dur, - "operation": event.Operation(), - }) - - if dur > 1*time.Second { - l.Warnf("SLOW DATABASE QUERY [%s] %s", dur, event.Query) - return - } - - if logrus.GetLevel() == logrus.TraceLevel { - l.Tracef("[%s] %s", dur, event.Query) - } else { - l.Debugf("[%s] %s", dur, event.Operation()) - } -} |