From ee0797406088d3367332a6959a57bdb57a529ae3 Mon Sep 17 00:00:00 2001 From: Leyla Becker Date: Sat, 29 Mar 2025 19:29:17 -0500 Subject: [PATCH] set up connection to DB --- src/database/database.go | 17 +++++++---------- 1 file changed, 7 insertions(+), 10 deletions(-) diff --git a/src/database/database.go b/src/database/database.go index 18f695b..9a9735c 100644 --- a/src/database/database.go +++ b/src/database/database.go @@ -10,20 +10,17 @@ import ( func main() { // urlExample := "postgres://username:password@localhost:5432/database_name" - conn, err := pgx.Connect(context.Background(), os.Getenv("PGHOST")) + host := os.Getenv("PGHOST") + database := os.Getenv("DB_NAME") + + connectString := + fmt.Sprintf("postgres:///?host=%s&database=%s", host, database) + + conn, err := pgx.Connect(context.Background(), connectString) if err != nil { fmt.Fprintf(os.Stderr, "Unable to connect to database: %v\n", err) os.Exit(1) } defer conn.Close(context.Background()) - var name string - var weight int64 - err = conn.QueryRow(context.Background(), "select name, weight from widgets where id=$1", 42).Scan(&name, &weight) - if err != nil { - fmt.Fprintf(os.Stderr, "QueryRow failed: %v\n", err) - os.Exit(1) - } - - fmt.Println(name, weight) }