feat: added db schema to postgres template (#245)

* feat: added db schema to postgres template
Melkeydev · Jun 12, 2024 · 1750ccf · 1750ccf
1 parent 60a726e
commit 1750ccf
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/cmd/template/dbdriver/files/env/postgres.tmpl b/cmd/template/dbdriver/files/env/postgres.tmpl
@@ -3,3 +3,4 @@ DB_PORT=5432
 DB_DATABASE=blueprint
 DB_USERNAME=melkey
 DB_PASSWORD=password1234
+DB_SCHEMA=public
diff --git a/cmd/template/dbdriver/files/service/postgres.tmpl b/cmd/template/dbdriver/files/service/postgres.tmpl
@@ -34,6 +34,7 @@ var (
 	username = os.Getenv("DB_USERNAME")
 	port     = os.Getenv("DB_PORT")
 	host     = os.Getenv("DB_HOST")
+        schema   = os.Getenv("DB_SCHEMA")
 	dbInstance *service
 )
 
@@ -42,7 +43,7 @@ func New() Service {
 	if dbInstance != nil {
 		return dbInstance
 	}
-	connStr := fmt.Sprintf("postgres://%s:%s@%s:%s/%s?sslmode=disable", username, password, host, port, database)
+	connStr := fmt.Sprintf("postgres://%s:%s@%s:%s/%s?sslmode=disable&search_path=%s", username, password, host, port, database, schema)
 	db, err := sql.Open("pgx", connStr)
 	if err != nil {
 		log.Fatal(err)