mirror of
https://github.com/hanzoai/base.git
synced 2026-10-21 03:41:50 +00:00
zeekay
85387a93ab
SQLDialect interface abstracts SQLite vs PostgreSQL SQL differences: - JSON ops (json_extract vs ->>, json_each vs jsonb_array_elements) - Date formatting (strftime vs to_char) - Schema introspection (PRAGMA vs information_schema) - ID generation (randomblob vs gen_random_bytes) Adds PostgresDBConnect via pgx driver and DataDSN/AuxDSN config fields for connecting Base to Neon serverless PostgreSQL instead of SQLite.
104 lines
2.4 KiB
Go
104 lines
2.4 KiB
Go
package core
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
)
|
|
|
|
// SQLiteDialect implements SQLDialect for SQLite databases.
|
|
type SQLiteDialect struct{}
|
|
|
|
func (d *SQLiteDialect) DriverName() string {
|
|
return "sqlite3"
|
|
}
|
|
|
|
func (d *SQLiteDialect) Placeholder(_ int) string {
|
|
return "?"
|
|
}
|
|
|
|
func (d *SQLiteDialect) AutoincrementType() string {
|
|
return "INTEGER PRIMARY KEY AUTOINCREMENT"
|
|
}
|
|
|
|
func (d *SQLiteDialect) JSONExtract(column, path string) string {
|
|
// Ensure path starts with $.
|
|
if !strings.HasPrefix(path, "$.") && !strings.HasPrefix(path, "$[") {
|
|
path = "$." + path
|
|
}
|
|
return fmt.Sprintf("json_extract(%s, '%s')", column, path)
|
|
}
|
|
|
|
func (d *SQLiteDialect) JSONEach(column string) string {
|
|
return fmt.Sprintf("json_each(%s)", column)
|
|
}
|
|
|
|
func (d *SQLiteDialect) JSONValid(column string) string {
|
|
return fmt.Sprintf("json_valid(%s)", column)
|
|
}
|
|
|
|
func (d *SQLiteDialect) JSONArray(values ...string) string {
|
|
return fmt.Sprintf("json_array(%s)", strings.Join(values, ", "))
|
|
}
|
|
|
|
func (d *SQLiteDialect) Strftime(format, column string) string {
|
|
return fmt.Sprintf("strftime('%s', %s)", format, column)
|
|
}
|
|
|
|
func (d *SQLiteDialect) Now() string {
|
|
return "datetime('now')"
|
|
}
|
|
|
|
func (d *SQLiteDialect) RandomID(byteLen int) string {
|
|
return fmt.Sprintf("lower(hex(randomblob(%d)))", byteLen)
|
|
}
|
|
|
|
func (d *SQLiteDialect) Concat(parts ...string) string {
|
|
return strings.Join(parts, " || ")
|
|
}
|
|
|
|
func (d *SQLiteDialect) GroupConcat(column, separator string) string {
|
|
return fmt.Sprintf("group_concat(%s, '%s')", column, separator)
|
|
}
|
|
|
|
func (d *SQLiteDialect) TableColumns(table string) string {
|
|
return fmt.Sprintf("PRAGMA table_info('%s')", table)
|
|
}
|
|
|
|
func (d *SQLiteDialect) HasTable(table string) string {
|
|
return fmt.Sprintf(
|
|
"SELECT 1 FROM sqlite_master WHERE type='table' AND name='%s' LIMIT 1",
|
|
table,
|
|
)
|
|
}
|
|
|
|
func (d *SQLiteDialect) ListTables() string {
|
|
return "SELECT name FROM sqlite_master WHERE type='table' ORDER BY name"
|
|
}
|
|
|
|
func (d *SQLiteDialect) ListViews() string {
|
|
return "SELECT name FROM sqlite_master WHERE type='view' ORDER BY name"
|
|
}
|
|
|
|
func (d *SQLiteDialect) NormalizeColumnType(sqliteType string) string {
|
|
return sqliteType
|
|
}
|
|
|
|
func (d *SQLiteDialect) Vacuum() string {
|
|
return "VACUUM"
|
|
}
|
|
|
|
func (d *SQLiteDialect) BooleanValue(b bool) string {
|
|
if b {
|
|
return "1"
|
|
}
|
|
return "0"
|
|
}
|
|
|
|
func (d *SQLiteDialect) SupportsReturning() bool {
|
|
return false
|
|
}
|
|
|
|
func (d *SQLiteDialect) CastToText(column string) string {
|
|
return fmt.Sprintf("CAST(%s AS TEXT)", column)
|
|
}
|