forked from mirrors/gotosocial
07727753b9
* Add whereNotEmptyAndNotNull * Add GetRemoteOlderThanDays * Add GetRemoteOlderThanDays * Add PruneRemote to Manager interface * Start implementing PruneRemote * add new attachment + status to tests * fix up and test GetRemoteOlderThan * fix bad import * PruneRemote: return number pruned * add Cached column to mediaattachment * update + test pruneRemote * update mediaTest * use Cached column * upstep bun to latest version * embed structs in mediaAttachment * migrate mediaAttachment to new format * don't default cached to true * select only remote media * update db dependencies * step bun back to last working version * update pruneRemote to use Cached field * fix storage path of test attachments * add recache logic to manager * fix trimmed aspect ratio * test prune and recache * return errwithcode * tidy up different paths for emoji vs attachment * fix incorrect thumbnail type being stored * expose TransportController to media processor * implement tee-ing recached content * add thoughts of dog to test fedi attachments * test get remote files * add comment on PruneRemote * add postData cleanup to recache * test thumbnail fetching * add incredible diagram * go mod tidy * buffer pipes for recache streaming * test for client stops reading after 1kb * add media-remote-cache-days to config * add cron package * wrap logrus so it's available to cron * start and stop cron jobs gracefully
113 lines
2 KiB
Go
113 lines
2 KiB
Go
package bun
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"strings"
|
|
"sync/atomic"
|
|
"time"
|
|
|
|
"github.com/uptrace/bun/schema"
|
|
)
|
|
|
|
type QueryEvent struct {
|
|
DB *DB
|
|
|
|
QueryAppender schema.QueryAppender // DEPRECATED: use IQuery instead
|
|
IQuery Query
|
|
Query string
|
|
QueryTemplate string
|
|
QueryArgs []interface{}
|
|
Model Model
|
|
|
|
StartTime time.Time
|
|
Result sql.Result
|
|
Err error
|
|
|
|
Stash map[interface{}]interface{}
|
|
}
|
|
|
|
func (e *QueryEvent) Operation() string {
|
|
if e.IQuery != nil {
|
|
return e.IQuery.Operation()
|
|
}
|
|
return queryOperation(e.Query)
|
|
}
|
|
|
|
func queryOperation(query string) string {
|
|
if idx := strings.IndexByte(query, ' '); idx > 0 {
|
|
query = query[:idx]
|
|
}
|
|
if len(query) > 16 {
|
|
query = query[:16]
|
|
}
|
|
return query
|
|
}
|
|
|
|
type QueryHook interface {
|
|
BeforeQuery(context.Context, *QueryEvent) context.Context
|
|
AfterQuery(context.Context, *QueryEvent)
|
|
}
|
|
|
|
func (db *DB) beforeQuery(
|
|
ctx context.Context,
|
|
iquery Query,
|
|
queryTemplate string,
|
|
queryArgs []interface{},
|
|
query string,
|
|
model Model,
|
|
) (context.Context, *QueryEvent) {
|
|
atomic.AddUint32(&db.stats.Queries, 1)
|
|
|
|
if len(db.queryHooks) == 0 {
|
|
return ctx, nil
|
|
}
|
|
|
|
event := &QueryEvent{
|
|
DB: db,
|
|
|
|
Model: model,
|
|
QueryAppender: iquery,
|
|
IQuery: iquery,
|
|
Query: query,
|
|
QueryTemplate: queryTemplate,
|
|
QueryArgs: queryArgs,
|
|
|
|
StartTime: time.Now(),
|
|
}
|
|
|
|
for _, hook := range db.queryHooks {
|
|
ctx = hook.BeforeQuery(ctx, event)
|
|
}
|
|
|
|
return ctx, event
|
|
}
|
|
|
|
func (db *DB) afterQuery(
|
|
ctx context.Context,
|
|
event *QueryEvent,
|
|
res sql.Result,
|
|
err error,
|
|
) {
|
|
switch err {
|
|
case nil, sql.ErrNoRows:
|
|
// nothing
|
|
default:
|
|
atomic.AddUint32(&db.stats.Errors, 1)
|
|
}
|
|
|
|
if event == nil {
|
|
return
|
|
}
|
|
|
|
event.Result = res
|
|
event.Err = err
|
|
|
|
db.afterQueryFromIndex(ctx, event, len(db.queryHooks)-1)
|
|
}
|
|
|
|
func (db *DB) afterQueryFromIndex(ctx context.Context, event *QueryEvent, hookIndex int) {
|
|
for ; hookIndex >= 0; hookIndex-- {
|
|
db.queryHooks[hookIndex].AfterQuery(ctx, event)
|
|
}
|
|
}
|