182e411a64
Never set the primary key flag as part of the column definition. When using sqlite3, this prevents mixed int/other type composite primary key (because the int column always wins and get the index for itself). Rather, let's just set the primary key at the end, whether it's composite or not. Note that the int key will automatically be AUTOINCREMENT if its type is "integer". For that reason, always use that type, rather than "bigint", as bigint maps to "integer" eventually anyway.
96 lines
2.4 KiB
Go
96 lines
2.4 KiB
Go
package gorm
|
|
|
|
import (
|
|
"fmt"
|
|
"reflect"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
type sqlite3 struct {
|
|
commonDialect
|
|
}
|
|
|
|
func init() {
|
|
RegisterDialect("sqlite3", &sqlite3{})
|
|
}
|
|
|
|
func (sqlite3) GetName() string {
|
|
return "sqlite3"
|
|
}
|
|
|
|
// Get Data Type for Sqlite Dialect
|
|
func (s *sqlite3) DataTypeOf(field *StructField) string {
|
|
var dataValue, sqlType, size, additionalType = ParseFieldStructForDialect(field, s)
|
|
|
|
if sqlType == "" {
|
|
switch dataValue.Kind() {
|
|
case reflect.Bool:
|
|
sqlType = "bool"
|
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uintptr, reflect.Int64, reflect.Uint64:
|
|
sqlType = "integer"
|
|
case reflect.Float32, reflect.Float64:
|
|
sqlType = "real"
|
|
case reflect.String:
|
|
if size > 0 && size < 65532 {
|
|
sqlType = fmt.Sprintf("varchar(%d)", size)
|
|
} else {
|
|
sqlType = "text"
|
|
}
|
|
case reflect.Struct:
|
|
if _, ok := dataValue.Interface().(time.Time); ok {
|
|
sqlType = "datetime"
|
|
}
|
|
default:
|
|
if _, ok := dataValue.Interface().([]byte); ok {
|
|
sqlType = "blob"
|
|
}
|
|
}
|
|
}
|
|
|
|
if sqlType == "" {
|
|
panic(fmt.Sprintf("invalid sql type %s (%s) for sqlite3", dataValue.Type().Name(), dataValue.Kind().String()))
|
|
}
|
|
|
|
if strings.TrimSpace(additionalType) == "" {
|
|
return sqlType
|
|
}
|
|
return fmt.Sprintf("%v %v", sqlType, additionalType)
|
|
}
|
|
|
|
func (s sqlite3) HasIndex(tableName string, indexName string) bool {
|
|
var count int
|
|
s.db.QueryRow(fmt.Sprintf("SELECT count(*) FROM sqlite_master WHERE tbl_name = ? AND sql LIKE '%%INDEX %v ON%%'", indexName), tableName).Scan(&count)
|
|
return count > 0
|
|
}
|
|
|
|
func (s sqlite3) HasTable(tableName string) bool {
|
|
var count int
|
|
s.db.QueryRow("SELECT count(*) FROM sqlite_master WHERE type='table' AND name=?", tableName).Scan(&count)
|
|
return count > 0
|
|
}
|
|
|
|
func (s sqlite3) HasColumn(tableName string, columnName string) bool {
|
|
var count int
|
|
s.db.QueryRow(fmt.Sprintf("SELECT count(*) FROM sqlite_master WHERE tbl_name = ? AND (sql LIKE '%%\"%v\" %%' OR sql LIKE '%%%v %%');\n", columnName, columnName), tableName).Scan(&count)
|
|
return count > 0
|
|
}
|
|
|
|
func (s sqlite3) CurrentDatabase() (name string) {
|
|
var (
|
|
ifaces = make([]interface{}, 3)
|
|
pointers = make([]*string, 3)
|
|
i int
|
|
)
|
|
for i = 0; i < 3; i++ {
|
|
ifaces[i] = &pointers[i]
|
|
}
|
|
if err := s.db.QueryRow("PRAGMA database_list").Scan(ifaces...); err != nil {
|
|
return
|
|
}
|
|
if pointers[1] != nil {
|
|
name = *pointers[1]
|
|
}
|
|
return
|
|
}
|