diff options
author | 2025-03-09 17:47:56 +0100 | |
---|---|---|
committer | 2025-03-10 01:59:49 +0100 | |
commit | 3ac1ee16f377d31a0fb80c8dae28b6239ac4229e (patch) | |
tree | f61faa581feaaeaba2542b9f2b8234a590684413 /vendor/github.com/ncruces/go-sqlite3/func.go | |
parent | [chore] update URLs to forked source (diff) | |
download | gotosocial-3ac1ee16f377d31a0fb80c8dae28b6239ac4229e.tar.xz |
[chore] remove vendor
Diffstat (limited to 'vendor/github.com/ncruces/go-sqlite3/func.go')
-rw-r--r-- | vendor/github.com/ncruces/go-sqlite3/func.go | 233 |
1 files changed, 0 insertions, 233 deletions
diff --git a/vendor/github.com/ncruces/go-sqlite3/func.go b/vendor/github.com/ncruces/go-sqlite3/func.go deleted file mode 100644 index 6b69368b4..000000000 --- a/vendor/github.com/ncruces/go-sqlite3/func.go +++ /dev/null @@ -1,233 +0,0 @@ -package sqlite3 - -import ( - "context" - "sync" - - "github.com/tetratelabs/wazero/api" - - "github.com/ncruces/go-sqlite3/internal/util" -) - -// CollationNeeded registers a callback to be invoked -// whenever an unknown collation sequence is required. -// -// https://sqlite.org/c3ref/collation_needed.html -func (c *Conn) CollationNeeded(cb func(db *Conn, name string)) error { - var enable int32 - if cb != nil { - enable = 1 - } - rc := res_t(c.call("sqlite3_collation_needed_go", stk_t(c.handle), stk_t(enable))) - if err := c.error(rc); err != nil { - return err - } - c.collation = cb - return nil -} - -// AnyCollationNeeded uses [Conn.CollationNeeded] to register -// a fake collating function for any unknown collating sequence. -// The fake collating function works like BINARY. -// -// This can be used to load schemas that contain -// one or more unknown collating sequences. -func (c Conn) AnyCollationNeeded() error { - rc := res_t(c.call("sqlite3_anycollseq_init", stk_t(c.handle), 0, 0)) - if err := c.error(rc); err != nil { - return err - } - c.collation = nil - return nil -} - -// CreateCollation defines a new collating sequence. -// -// https://sqlite.org/c3ref/create_collation.html -func (c *Conn) CreateCollation(name string, fn func(a, b []byte) int) error { - var funcPtr ptr_t - defer c.arena.mark()() - namePtr := c.arena.string(name) - if fn != nil { - funcPtr = util.AddHandle(c.ctx, fn) - } - rc := res_t(c.call("sqlite3_create_collation_go", - stk_t(c.handle), stk_t(namePtr), stk_t(funcPtr))) - return c.error(rc) -} - -// CreateFunction defines a new scalar SQL function. -// -// https://sqlite.org/c3ref/create_function.html -func (c *Conn) CreateFunction(name string, nArg int, flag FunctionFlag, fn ScalarFunction) error { - var funcPtr ptr_t - defer c.arena.mark()() - namePtr := c.arena.string(name) - if fn != nil { - funcPtr = util.AddHandle(c.ctx, fn) - } - rc := res_t(c.call("sqlite3_create_function_go", - stk_t(c.handle), stk_t(namePtr), stk_t(nArg), - stk_t(flag), stk_t(funcPtr))) - return c.error(rc) -} - -// ScalarFunction is the type of a scalar SQL function. -// Implementations must not retain arg. -type ScalarFunction func(ctx Context, arg ...Value) - -// CreateWindowFunction defines a new aggregate or aggregate window SQL function. -// If fn returns a [WindowFunction], then an aggregate window function is created. -// If fn returns an [io.Closer], it will be called to free resources. -// -// https://sqlite.org/c3ref/create_function.html -func (c *Conn) CreateWindowFunction(name string, nArg int, flag FunctionFlag, fn func() AggregateFunction) error { - var funcPtr ptr_t - defer c.arena.mark()() - namePtr := c.arena.string(name) - if fn != nil { - funcPtr = util.AddHandle(c.ctx, fn) - } - call := "sqlite3_create_aggregate_function_go" - if _, ok := fn().(WindowFunction); ok { - call = "sqlite3_create_window_function_go" - } - rc := res_t(c.call(call, - stk_t(c.handle), stk_t(namePtr), stk_t(nArg), - stk_t(flag), stk_t(funcPtr))) - return c.error(rc) -} - -// AggregateFunction is the interface an aggregate function should implement. -// -// https://sqlite.org/appfunc.html -type AggregateFunction interface { - // Step is invoked to add a row to the current window. - // The function arguments, if any, corresponding to the row being added, are passed to Step. - // Implementations must not retain arg. - Step(ctx Context, arg ...Value) - - // Value is invoked to return the current (or final) value of the aggregate. - Value(ctx Context) -} - -// WindowFunction is the interface an aggregate window function should implement. -// -// https://sqlite.org/windowfunctions.html -type WindowFunction interface { - AggregateFunction - - // Inverse is invoked to remove the oldest presently aggregated result of Step from the current window. - // The function arguments, if any, are those passed to Step for the row being removed. - // Implementations must not retain arg. - Inverse(ctx Context, arg ...Value) -} - -// OverloadFunction overloads a function for a virtual table. -// -// https://sqlite.org/c3ref/overload_function.html -func (c *Conn) OverloadFunction(name string, nArg int) error { - defer c.arena.mark()() - namePtr := c.arena.string(name) - rc := res_t(c.call("sqlite3_overload_function", - stk_t(c.handle), stk_t(namePtr), stk_t(nArg))) - return c.error(rc) -} - -func destroyCallback(ctx context.Context, mod api.Module, pApp ptr_t) { - util.DelHandle(ctx, pApp) -} - -func collationCallback(ctx context.Context, mod api.Module, pArg, pDB ptr_t, eTextRep uint32, zName ptr_t) { - if c, ok := ctx.Value(connKey{}).(*Conn); ok && c.handle == pDB && c.collation != nil { - name := util.ReadString(mod, zName, _MAX_NAME) - c.collation(c, name) - } -} - -func compareCallback(ctx context.Context, mod api.Module, pApp ptr_t, nKey1 int32, pKey1 ptr_t, nKey2 int32, pKey2 ptr_t) uint32 { - fn := util.GetHandle(ctx, pApp).(func(a, b []byte) int) - return uint32(fn(util.View(mod, pKey1, int64(nKey1)), util.View(mod, pKey2, int64(nKey2)))) -} - -func funcCallback(ctx context.Context, mod api.Module, pCtx, pApp ptr_t, nArg int32, pArg ptr_t) { - args := getFuncArgs() - defer putFuncArgs(args) - db := ctx.Value(connKey{}).(*Conn) - fn := util.GetHandle(db.ctx, pApp).(ScalarFunction) - callbackArgs(db, args[:nArg], pArg) - fn(Context{db, pCtx}, args[:nArg]...) -} - -func stepCallback(ctx context.Context, mod api.Module, pCtx, pAgg, pApp ptr_t, nArg int32, pArg ptr_t) { - args := getFuncArgs() - defer putFuncArgs(args) - db := ctx.Value(connKey{}).(*Conn) - callbackArgs(db, args[:nArg], pArg) - fn, _ := callbackAggregate(db, pAgg, pApp) - fn.Step(Context{db, pCtx}, args[:nArg]...) -} - -func finalCallback(ctx context.Context, mod api.Module, pCtx, pAgg, pApp ptr_t) { - db := ctx.Value(connKey{}).(*Conn) - fn, handle := callbackAggregate(db, pAgg, pApp) - fn.Value(Context{db, pCtx}) - if err := util.DelHandle(ctx, handle); err != nil { - Context{db, pCtx}.ResultError(err) - return // notest - } -} - -func valueCallback(ctx context.Context, mod api.Module, pCtx, pAgg ptr_t) { - db := ctx.Value(connKey{}).(*Conn) - fn := util.GetHandle(db.ctx, pAgg).(AggregateFunction) - fn.Value(Context{db, pCtx}) -} - -func inverseCallback(ctx context.Context, mod api.Module, pCtx, pAgg ptr_t, nArg int32, pArg ptr_t) { - args := getFuncArgs() - defer putFuncArgs(args) - db := ctx.Value(connKey{}).(*Conn) - callbackArgs(db, args[:nArg], pArg) - fn := util.GetHandle(db.ctx, pAgg).(WindowFunction) - fn.Inverse(Context{db, pCtx}, args[:nArg]...) -} - -func callbackAggregate(db *Conn, pAgg, pApp ptr_t) (AggregateFunction, ptr_t) { - if pApp == 0 { - handle := util.Read32[ptr_t](db.mod, pAgg) - return util.GetHandle(db.ctx, handle).(AggregateFunction), handle - } - - // We need to create the aggregate. - fn := util.GetHandle(db.ctx, pApp).(func() AggregateFunction)() - if pAgg != 0 { - handle := util.AddHandle(db.ctx, fn) - util.Write32(db.mod, pAgg, handle) - return fn, handle - } - return fn, 0 -} - -func callbackArgs(db *Conn, arg []Value, pArg ptr_t) { - for i := range arg { - arg[i] = Value{ - c: db, - handle: util.Read32[ptr_t](db.mod, pArg+ptr_t(i)*ptrlen), - } - } -} - -var funcArgsPool sync.Pool - -func putFuncArgs(p *[_MAX_FUNCTION_ARG]Value) { - funcArgsPool.Put(p) -} - -func getFuncArgs() *[_MAX_FUNCTION_ARG]Value { - if p := funcArgsPool.Get(); p == nil { - return new([_MAX_FUNCTION_ARG]Value) - } else { - return p.(*[_MAX_FUNCTION_ARG]Value) - } -} |