Add support for sqlite generated columns.
This commit is contained in:
parent
23cb5dcfbc
commit
71fb1c7cd1
3 changed files with 178 additions and 2 deletions
|
|
@ -5,6 +5,7 @@ import (
|
|||
"database/sql"
|
||||
"fmt"
|
||||
"github.com/go-jet/jet/v2/generator/metadata"
|
||||
"github.com/go-jet/jet/v2/internal/utils/semantic"
|
||||
"github.com/go-jet/jet/v2/qrm"
|
||||
"strings"
|
||||
)
|
||||
|
|
@ -42,16 +43,45 @@ func (p sqliteQuerySet) GetTablesMetaData(db *sql.DB, schemaName string, tableTy
|
|||
return tables, nil
|
||||
}
|
||||
|
||||
func getTableInfoQuery(db *sql.DB) (string, error) {
|
||||
var version string
|
||||
err := db.QueryRow("select sqlite_version();").Scan(&version)
|
||||
|
||||
if err != nil {
|
||||
return "", fmt.Errorf("failed to get sqlite version: %w", err)
|
||||
}
|
||||
|
||||
sqliteVersion, err := semantic.VersionFromString(version)
|
||||
|
||||
if err != nil {
|
||||
return "", fmt.Errorf("can't parse sqlite version: %w", err)
|
||||
}
|
||||
|
||||
// generated columns were added in version 3.26.0
|
||||
if sqliteVersion.Lt(semantic.Version{Major: 3, Minor: 26, Patch: 0}) {
|
||||
return `select * from pragma_table_info(?);`, nil
|
||||
}
|
||||
|
||||
return `select * from pragma_table_xinfo(?);`, nil
|
||||
}
|
||||
|
||||
func (p sqliteQuerySet) GetTableColumnsMetaData(db *sql.DB, schemaName string, tableName string) ([]metadata.Column, error) {
|
||||
query := fmt.Sprintf(`select * from pragma_table_info(?);`)
|
||||
|
||||
tableInfoQuery, err := getTableInfoQuery(db)
|
||||
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var columnInfos []struct {
|
||||
Name string
|
||||
Type string
|
||||
NotNull int32
|
||||
Pk int32
|
||||
Hidden int32
|
||||
}
|
||||
|
||||
_, err := qrm.Query(context.Background(), db, query, []interface{}{tableName}, &columnInfos)
|
||||
_, err = qrm.Query(context.Background(), db, tableInfoQuery, []interface{}{tableName}, &columnInfos)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to query '%s' column metadata: %w", tableName, err)
|
||||
}
|
||||
|
|
@ -60,11 +90,13 @@ func (p sqliteQuerySet) GetTableColumnsMetaData(db *sql.DB, schemaName string, t
|
|||
|
||||
for _, columnInfo := range columnInfos {
|
||||
columnType := getColumnType(columnInfo.Type)
|
||||
isGenerated := columnInfo.Hidden == 2 || columnInfo.Hidden == 3 // stored or virtual column
|
||||
|
||||
columns = append(columns, metadata.Column{
|
||||
Name: columnInfo.Name,
|
||||
IsPrimaryKey: columnInfo.Pk != 0,
|
||||
IsNullable: columnInfo.NotNull != 1,
|
||||
IsGenerated: isGenerated,
|
||||
DataType: metadata.DataType{
|
||||
Name: columnType,
|
||||
Kind: metadata.BaseType,
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue