summaryrefslogtreecommitdiff
path: root/vendor/github.com/uptrace/bun/dialect/sqlitedialect/dialect.go
diff options
context:
space:
mode:
authorLibravatar kim <89579420+NyaaaWhatsUpDoc@users.noreply.github.com>2021-08-29 15:41:41 +0100
committerLibravatar GitHub <noreply@github.com>2021-08-29 16:41:41 +0200
commited462245730bd7832019bd43e0bc1c9d1c055e8e (patch)
tree1caad78ea6aabf5ea93c93a8ade97176b4889500 /vendor/github.com/uptrace/bun/dialect/sqlitedialect/dialect.go
parentMention fixup (#167) (diff)
downloadgotosocial-ed462245730bd7832019bd43e0bc1c9d1c055e8e.tar.xz
Add SQLite support, fix un-thread-safe DB caches, small performance f… (#172)
* Add SQLite support, fix un-thread-safe DB caches, small performance fixes Signed-off-by: kim (grufwub) <grufwub@gmail.com> * add SQLite licenses to README Signed-off-by: kim (grufwub) <grufwub@gmail.com> * appease the linter, and fix my dumbass-ery Signed-off-by: kim (grufwub) <grufwub@gmail.com> * make requested changes Signed-off-by: kim (grufwub) <grufwub@gmail.com> * add back comment Signed-off-by: kim (grufwub) <grufwub@gmail.com>
Diffstat (limited to 'vendor/github.com/uptrace/bun/dialect/sqlitedialect/dialect.go')
-rw-r--r--vendor/github.com/uptrace/bun/dialect/sqlitedialect/dialect.go94
1 files changed, 94 insertions, 0 deletions
diff --git a/vendor/github.com/uptrace/bun/dialect/sqlitedialect/dialect.go b/vendor/github.com/uptrace/bun/dialect/sqlitedialect/dialect.go
new file mode 100644
index 000000000..33d25f2d1
--- /dev/null
+++ b/vendor/github.com/uptrace/bun/dialect/sqlitedialect/dialect.go
@@ -0,0 +1,94 @@
+package sqlitedialect
+
+import (
+ "database/sql"
+ "reflect"
+ "sync"
+
+ "github.com/uptrace/bun/dialect"
+ "github.com/uptrace/bun/dialect/feature"
+ "github.com/uptrace/bun/dialect/sqltype"
+ "github.com/uptrace/bun/schema"
+)
+
+type Dialect struct {
+ tables *schema.Tables
+ features feature.Feature
+
+ appenderMap sync.Map
+ scannerMap sync.Map
+}
+
+func New() *Dialect {
+ d := new(Dialect)
+ d.tables = schema.NewTables(d)
+ d.features = feature.Returning | feature.InsertTableAlias | feature.DeleteTableAlias
+ return d
+}
+
+func (d *Dialect) Init(*sql.DB) {}
+
+func (d *Dialect) Name() dialect.Name {
+ return dialect.SQLite
+}
+
+func (d *Dialect) Features() feature.Feature {
+ return d.features
+}
+
+func (d *Dialect) Tables() *schema.Tables {
+ return d.tables
+}
+
+func (d *Dialect) OnTable(table *schema.Table) {
+ for _, field := range table.FieldMap {
+ d.onField(field)
+ }
+}
+
+func (d *Dialect) onField(field *schema.Field) {
+ // INTEGER PRIMARY KEY is an alias for the ROWID.
+ // It is safe to convert all ints to INTEGER, because SQLite types don't have size.
+ switch field.DiscoveredSQLType {
+ case sqltype.SmallInt, sqltype.BigInt:
+ field.DiscoveredSQLType = sqltype.Integer
+ }
+}
+
+func (d *Dialect) IdentQuote() byte {
+ return '"'
+}
+
+func (d *Dialect) Append(fmter schema.Formatter, b []byte, v interface{}) []byte {
+ return schema.Append(fmter, b, v, nil)
+}
+
+func (d *Dialect) Appender(typ reflect.Type) schema.AppenderFunc {
+ if v, ok := d.appenderMap.Load(typ); ok {
+ return v.(schema.AppenderFunc)
+ }
+
+ fn := schema.Appender(typ, nil)
+
+ if v, ok := d.appenderMap.LoadOrStore(typ, fn); ok {
+ return v.(schema.AppenderFunc)
+ }
+ return fn
+}
+
+func (d *Dialect) FieldAppender(field *schema.Field) schema.AppenderFunc {
+ return schema.FieldAppender(d, field)
+}
+
+func (d *Dialect) Scanner(typ reflect.Type) schema.ScannerFunc {
+ if v, ok := d.scannerMap.Load(typ); ok {
+ return v.(schema.ScannerFunc)
+ }
+
+ fn := scanner(typ)
+
+ if v, ok := d.scannerMap.LoadOrStore(typ, fn); ok {
+ return v.(schema.ScannerFunc)
+ }
+ return fn
+}