80 lines
1.9 KiB
Go
80 lines
1.9 KiB
Go
package sqlite
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"fmt"
|
|
"github.com/go-jet/jet/v2/generator/metadata"
|
|
"github.com/go-jet/jet/v2/internal/utils/throw"
|
|
"github.com/go-jet/jet/v2/qrm"
|
|
"strings"
|
|
)
|
|
|
|
// sqliteQuerySet is dialect query set for SQLite
|
|
type sqliteQuerySet struct{}
|
|
|
|
func (p sqliteQuerySet) GetTablesMetaData(db *sql.DB, schemaName string, tableType metadata.TableType) []metadata.Table {
|
|
query := `
|
|
SELECT name as "table.name"
|
|
FROM sqlite_master
|
|
WHERE type=? AND name != 'sqlite_sequence'
|
|
ORDER BY name;
|
|
`
|
|
sqlTableType := "table"
|
|
|
|
if tableType == metadata.ViewTable {
|
|
sqlTableType = "view"
|
|
}
|
|
|
|
var tables []metadata.Table
|
|
|
|
_, err := qrm.Query(context.Background(), db, query, []interface{}{sqlTableType}, &tables)
|
|
throw.OnError(err)
|
|
|
|
for i := range tables {
|
|
tables[i].Columns = p.GetTableColumnsMetaData(db, schemaName, tables[i].Name)
|
|
}
|
|
|
|
return tables
|
|
}
|
|
|
|
func (p sqliteQuerySet) GetTableColumnsMetaData(db *sql.DB, schemaName string, tableName string) []metadata.Column {
|
|
query := fmt.Sprintf(`select * from pragma_table_info(?);`)
|
|
var columnInfos []struct {
|
|
Name string
|
|
Type string
|
|
NotNull int32
|
|
Pk int32
|
|
}
|
|
|
|
_, err := qrm.Query(context.Background(), db, query, []interface{}{tableName}, &columnInfos)
|
|
throw.OnError(err)
|
|
|
|
var columns []metadata.Column
|
|
|
|
for _, columnInfo := range columnInfos {
|
|
columnType := getColumnType(columnInfo.Type)
|
|
|
|
columns = append(columns, metadata.Column{
|
|
Name: columnInfo.Name,
|
|
IsPrimaryKey: columnInfo.Pk != 0,
|
|
IsNullable: columnInfo.NotNull != 1,
|
|
DataType: metadata.DataType{
|
|
Name: columnType,
|
|
Kind: metadata.BaseType,
|
|
IsUnsigned: false,
|
|
},
|
|
})
|
|
}
|
|
|
|
return columns
|
|
}
|
|
|
|
// will convert VARCHAR(10) -> VARCHAR, etc...
|
|
func getColumnType(columnType string) string {
|
|
return strings.TrimSpace(strings.Split(columnType, "(")[0])
|
|
}
|
|
|
|
func (p sqliteQuerySet) GetEnumsMetaData(db *sql.DB, schemaName string) []metadata.Enum {
|
|
return nil
|
|
}
|