From c0c8849371203f2d71a3931bb8022e6c34e9e3f4 Mon Sep 17 00:00:00 2001
From: Kexiang Wang <kx.wang@hotmail.com>
Date: Fri, 26 Jul 2024 17:10:53 -0400
Subject: [PATCH] feat(risedev): allow dbname of pg in risedev

---
 src/risedevtool/src/risedev_env.rs | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/risedevtool/src/risedev_env.rs b/src/risedevtool/src/risedev_env.rs
index bff4062f72097..c40acf3170708 100644
--- a/src/risedevtool/src/risedev_env.rs
+++ b/src/risedevtool/src/risedev_env.rs
@@ -110,11 +110,13 @@ pub fn generate_risedev_env(services: &Vec<ServiceConfig>) -> String {
                 let port = &c.port;
                 let user = &c.user;
                 let password = &c.password;
+                let database = &c.database;
                 // These envs are used by `postgres` cli.
                 writeln!(env, r#"PGHOST="{host}""#,).unwrap();
                 writeln!(env, r#"PGPORT="{port}""#,).unwrap();
                 writeln!(env, r#"PGUSER="{user}""#,).unwrap();
                 writeln!(env, r#"PGPASSWORD="{password}""#,).unwrap();
+                writeln!(env, r#"PGDATABASE="{database}""#,).unwrap();
             }
             _ => {}
         }