Gleipnir/jet
4
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

generator/*: allow passing DB directly

Browse source 
Closes #441
This commit is contained in:
TotallyGamerJet 2025-01-17 14:49:04 -05:00
parent 269bb4a3c9
commit e4ceb421de
3 changed files with 18 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

5
generator/sqlite/sqlite_generator.go
View file

@ -3,6 +3,7 @@ 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/sqlite"
@ -17,7 +18,11 @@ func GenerateDSN(dsn, destDir string, templates ...template.Template) error {
defer db.Close()
fmt.Println("Retrieving schema information...")
return GenerateDB(db, destDir, templates...)
}
// GenerateDB generates jet files using the provided *sql.DB
func GenerateDB(db *sql.DB, destDir string, templates ...template.Template) error {
generatorTemplate := template.Default(sqlite.Dialect)
if len(templates) > 0 {
generatorTemplate = templates[0]