32 lines
846 B
Go
32 lines
846 B
Go
package sqlite
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
"github.com/go-jet/jet/v2/generator/metadata"
|
|
"github.com/go-jet/jet/v2/generator/template"
|
|
"github.com/go-jet/jet/v2/internal/utils"
|
|
"github.com/go-jet/jet/v2/internal/utils/throw"
|
|
"github.com/go-jet/jet/v2/sqlite"
|
|
)
|
|
|
|
// GenerateDSN generates jet files using dsn connection string
|
|
func GenerateDSN(dsn, destDir string, templates ...template.Template) (err error) {
|
|
defer utils.ErrorCatch(&err)
|
|
|
|
db, err := sql.Open("sqlite3", dsn)
|
|
throw.OnError(err)
|
|
defer utils.DBClose(db)
|
|
|
|
fmt.Println("Retrieving schema information...")
|
|
|
|
generatorTemplate := template.Default(sqlite.Dialect)
|
|
if len(templates) > 0 {
|
|
generatorTemplate = templates[0]
|
|
}
|
|
|
|
schemaMetadata := metadata.GetSchema(db, &sqliteQuerySet{}, "")
|
|
|
|
template.ProcessSchema(destDir, schemaMetadata, generatorTemplate)
|
|
return
|
|
}
|