diff --git a/05-example-web-application/api-golang/main.go b/05-example-web-application/api-golang/main.go index aabc62e..e6a5a4b 100644 --- a/05-example-web-application/api-golang/main.go +++ b/05-example-web-application/api-golang/main.go @@ -1,6 +1,7 @@ package main import ( + "io/ioutil" "log" "os" "time" @@ -11,7 +12,16 @@ import ( ) func init() { - errDB := database.InitDB(os.Getenv("DATABASE_URL")) + databaseUrl := os.Getenv("DATABASE_URL") + if databaseUrl == "" { + content, err := ioutil.ReadFile(os.Getenv("DATABASE_URL_FILE")) + if err != nil { + log.Fatal(err) + } + databaseUrl = string(content) + } + + errDB := database.InitDB(databaseUrl) if errDB != nil { log.Fatalf("⛔ Unable to connect to database: %v\n", errDB) } else { diff --git a/05-example-web-application/api-node/src/db.js b/05-example-web-application/api-node/src/db.js index ff726e5..50ba813 100644 --- a/05-example-web-application/api-node/src/db.js +++ b/05-example-web-application/api-node/src/db.js @@ -1,7 +1,13 @@ +const fs = require('fs'); + const { Pool } = require('pg'); +databaseUrl = + process.env.DATABASE_URL || + fs.readFileSync(process.env.DATABASE_URL_FILE, 'utf8'); + const pool = new Pool({ - connectionString: process.env.DATABASE_URL, + connectionString: databaseUrl, }); // the pool will emit an error on behalf of any idle clients