summaryrefslogtreecommitdiff
path: root/vendor/github.com/ncruces/go-sqlite3/func.go
diff options
context:
space:
mode:
authorLibravatar Terin Stock <terinjokes@gmail.com>2025-03-09 17:47:56 +0100
committerLibravatar Terin Stock <terinjokes@gmail.com>2025-03-10 01:59:49 +0100
commit3ac1ee16f377d31a0fb80c8dae28b6239ac4229e (patch)
treef61faa581feaaeaba2542b9f2b8234a590684413 /vendor/github.com/ncruces/go-sqlite3/func.go
parent[chore] update URLs to forked source (diff)
downloadgotosocial-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.go233
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)
- }
-}