what
2221ca0ddb
- 新增 dialect/duckdb 方言，支持 DuckDB SQL 语法 - 新增 schema/dialect/duckdb DDL 操作支持 - dialect/sqlite3: 注册 sqlite3_with_if 驱动，连接时自动注册 IF() 函数 - engine: MakeConnection 对 sqlite3 自动使用带 IF 支持的驱动 - engine: 新增 DBConfig DuckDB 配置项及相关 Option 函数 - 统一各方言测试引用路径
93 lines
2.9 KiB
Go
93 lines
2.9 KiB
Go
package sqlite3
|
|
|
|
import (
|
|
"database/sql"
|
|
"time"
|
|
|
|
"git.fsdpf.net/go/db"
|
|
"git.fsdpf.net/go/db/exp"
|
|
gosqlite3 "github.com/mattn/go-sqlite3"
|
|
)
|
|
|
|
// DriverWithIF 是注册了 IF() 函数的 SQLite3 驱动名称。
|
|
// 使用该驱动名打开连接时，可在 SQL 中直接使用 IF(condition, trueVal, falseVal)。
|
|
const DriverWithIF = "sqlite3_with_if"
|
|
|
|
func DialectOptions() *db.SQLDialectOptions {
|
|
opts := db.DefaultDialectOptions()
|
|
|
|
opts.SupportsReturn = true
|
|
opts.SupportsOrderByOnUpdate = true
|
|
opts.SupportsLimitOnUpdate = true
|
|
opts.SupportsOrderByOnDelete = true
|
|
opts.SupportsLimitOnDelete = true
|
|
opts.SupportsConflictUpdateWhere = false
|
|
opts.SupportsInsertIgnoreSyntax = true
|
|
opts.SupportsConflictTarget = true
|
|
opts.SupportsMultipleUpdateTables = false
|
|
opts.WrapCompoundsInParens = false
|
|
opts.SupportsDistinctOn = false
|
|
opts.SupportsWindowFunction = false
|
|
opts.SupportsLateral = false
|
|
|
|
opts.PlaceHolderFragment = []byte("?")
|
|
opts.IncludePlaceholderNum = false
|
|
opts.QuoteRune = '`'
|
|
opts.DefaultValuesFragment = []byte("")
|
|
opts.True = []byte("1")
|
|
opts.False = []byte("0")
|
|
opts.TimeFormat = time.RFC3339Nano
|
|
opts.BooleanOperatorLookup = map[exp.BooleanOperation][]byte{
|
|
exp.EqOp: []byte("="),
|
|
exp.NeqOp: []byte("!="),
|
|
exp.GtOp: []byte(">"),
|
|
exp.GteOp: []byte(">="),
|
|
exp.LtOp: []byte("<"),
|
|
exp.LteOp: []byte("<="),
|
|
exp.InOp: []byte("IN"),
|
|
exp.NotInOp: []byte("NOT IN"),
|
|
exp.IsOp: []byte("IS"),
|
|
exp.IsNotOp: []byte("IS NOT"),
|
|
exp.LikeOp: []byte("LIKE"),
|
|
exp.NotLikeOp: []byte("NOT LIKE"),
|
|
exp.ILikeOp: []byte("LIKE"),
|
|
exp.NotILikeOp: []byte("NOT LIKE"),
|
|
exp.RegexpLikeOp: []byte("REGEXP"),
|
|
exp.RegexpNotLikeOp: []byte("NOT REGEXP"),
|
|
exp.RegexpILikeOp: []byte("REGEXP"),
|
|
exp.RegexpNotILikeOp: []byte("NOT REGEXP"),
|
|
}
|
|
opts.UseLiteralIsBools = false
|
|
opts.BitwiseOperatorLookup = map[exp.BitwiseOperation][]byte{
|
|
exp.BitwiseOrOp: []byte("|"),
|
|
exp.BitwiseAndOp: []byte("&"),
|
|
exp.BitwiseLeftShiftOp: []byte("<<"),
|
|
exp.BitwiseRightShiftOp: []byte(">>"),
|
|
}
|
|
opts.EscapedRunes = map[rune][]byte{
|
|
'\'': []byte("''"),
|
|
}
|
|
opts.InsertIgnoreClause = []byte("INSERT OR IGNORE INTO ")
|
|
opts.ConflictFragment = []byte(" ON CONFLICT ")
|
|
opts.ConflictDoUpdateFragment = []byte(" DO UPDATE SET ")
|
|
opts.ConflictDoNothingFragment = []byte(" DO NOTHING ")
|
|
opts.ForUpdateFragment = []byte("")
|
|
opts.OfFragment = []byte("")
|
|
opts.NowaitFragment = []byte("")
|
|
return opts
|
|
}
|
|
|
|
func init() {
|
|
sql.Register(DriverWithIF, &gosqlite3.SQLiteDriver{
|
|
ConnectHook: func(conn *gosqlite3.SQLiteConn) error {
|
|
return conn.RegisterFunc("IF", func(cond int64, trueVal, falseVal interface{}) interface{} {
|
|
if cond != 0 {
|
|
return trueVal
|
|
}
|
|
return falseVal
|
|
}, true)
|
|
},
|
|
})
|
|
db.RegisterDialect("sqlite3", DialectOptions())
|
|
}
|