taskqueueworker: parameterize sql persistent

golangid · May 22, 2023 · 614b389 · 614b389
1 parent efbb3d0
commit 614b389
Show file tree

Hide file tree

Showing 5 changed files with 216 additions and 204 deletions.
diff --git a/cmd/candi/template_cmd.go b/cmd/candi/template_cmd.go
@@ -52,8 +52,8 @@ import (
 
 	"github.com/pressly/goose/v3"
 
-	{{if eq .SQLDriver "postgres"}}_ "github.com/lib/pq"{{else if eq .SQLDriver "mysql"}}_ "github.com/go-sql-driver/mysql"{{end}}
-	"gorm.io/driver/{{.SQLDriver}}"
+	{{if eq .SQLDriver "postgres"}}_ "github.com/lib/pq"{{else if eq .SQLDriver "mysql"}}_ "github.com/go-sql-driver/mysql"{{else if eq .SQLDriver "sqlite3"}}_ "github.com/mattn/go-sqlite3"{{end}}
+	{{if eq .SQLDriver "sqlite3"}}"gorm.io/driver/sqlite"{{else}}"gorm.io/driver/{{.SQLDriver}}"{{end}}
 	"gorm.io/gorm"
 )
 
@@ -90,9 +90,9 @@ func main() {
 		}
 
 		if migrateTables := migrations.GetMigrateTables(); len(migrateTables) > 0 {
-			gormWrite, err := gorm.Open({{ .SQLDriver }}.New({{ .SQLDriver }}.Config{
+			gormWrite, err := gorm.Open({{if eq .SQLDriver "sqlite3"}}sqlite.Dialector{Conn: sqlDeps.WriteDB()}{{else}}{{ .SQLDriver }}.New({{ .SQLDriver }}.Config{
 				Conn: sqlDeps.WriteDB(),
-			}), &gorm.Config{
+			}){{end}}, &gorm.Config{
 				SkipDefaultTransaction:                   true,
 				DisableForeignKeyConstraintWhenMigrating: true,
 			})

diff --git a/cmd/candi/template_repository.go b/cmd/candi/template_repository.go
@@ -44,7 +44,7 @@ import (
 
 	{{ if .IsMonorepo }}"monorepo/globalshared"{{else}}"{{$.PackagePrefix}}/pkg/shared"{{end}}
 
-	"gorm.io/driver/{{.SQLDriver}}"
+	{{if eq .SQLDriver "sqlite3"}}"gorm.io/driver/sqlite"{{else}}"gorm.io/driver/{{.SQLDriver}}"{{end}}
 	"gorm.io/gorm"{{end}}` + `
 )
 
@@ -70,20 +70,20 @@ var (
 
 // setSharedRepoSQL set the global singleton "RepoSQL" implementation
 func setSharedRepoSQL(readDB, writeDB *sql.DB) {
-	{{if .SQLUseGORM}}gormRead, err := gorm.Open({{.SQLDriver}}.New({{.SQLDriver}}.Config{
+	{{if .SQLUseGORM}}gormRead, err := gorm.Open({{if eq .SQLDriver "sqlite3"}}sqlite.Dialector{Conn: readDB}{{else}}{{ .SQLDriver }}.New({{ .SQLDriver }}.Config{
 		Conn: readDB,
-	}), &gorm.Config{})
+	}){{end}}, &gorm.Config{})
 	if err != nil {
 		panic(err)
 	}
 
-	gormWrite, err := gorm.Open({{.SQLDriver}}.New({{.SQLDriver}}.Config{
+	gormWrite, err := gorm.Open({{if eq .SQLDriver "sqlite3"}}sqlite.Dialector{Conn: writeDB}{{else}}{{ .SQLDriver }}.New({{ .SQLDriver }}.Config{
 		Conn: writeDB,
-	}), &gorm.Config{SkipDefaultTransaction: true})
+	}){{end}}, &gorm.Config{SkipDefaultTransaction: true})
 	if err != nil {
 		panic(err)
 	}
-	
+
 	{{ if .IsMonorepo }}global{{end}}shared.AddGormCallbacks(gormRead)
 	{{ if .IsMonorepo }}global{{end}}shared.AddGormCallbacks(gormWrite){{end}}