31 lines
650 B
Go
31 lines
650 B
Go
|
package sqlite3
|
||
|
|
|
||
|
|
import "sync"
|
||
|
|
|
||
|
|
var (
|
||
|
|
// +checklocks:extRegistryMtx
|
||
|
|
extRegistry []func(*Conn) error
|
||
|
|
extRegistryMtx sync.RWMutex
|
||
|
|
)
|
||
|
|
|
||
|
|
// AutoExtension causes the entryPoint function to be invoked
|
||
|
|
// for each new database connection that is created.
|
||
|
|
//
|
||
|
|
// https://sqlite.org/c3ref/auto_extension.html
|
||
|
|
func AutoExtension(entryPoint func(*Conn) error) {
|
||
|
|
extRegistryMtx.Lock()
|
||
|
|
defer extRegistryMtx.Unlock()
|
||
|
|
extRegistry = append(extRegistry, entryPoint)
|
||
|
|
}
|
||
|
|
|
||
|
|
func initExtensions(c *Conn) error {
|
||
|
|
extRegistryMtx.RLock()
|
||
|
|
defer extRegistryMtx.RUnlock()
|
||
|
|
for _, f := range extRegistry {
|
||
|
|
if err := f(c); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|