From a4a83e4b549e858d4f8c821591bfb74706d846e8 Mon Sep 17 00:00:00 2001 From: Vidhu Kant Sharma Date: Sat, 28 May 2022 14:45:09 +0530 Subject: Moved the blog to new repository --- src/db/db.go | 49 +++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 49 insertions(+) create mode 100644 src/db/db.go (limited to 'src/db') diff --git a/src/db/db.go b/src/db/db.go new file mode 100644 index 0000000..7ca96ff --- /dev/null +++ b/src/db/db.go @@ -0,0 +1,49 @@ +/* + * vidhublog: Vidhu Kant's Blog + * Copyright (C) 2022 Vidhu Kant Sharma + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program. If not, see . + */ + +package db + +import ( + "database/sql" + _ "github.com/go-sql-driver/mysql" + "github.com/joho/godotenv" + "os" +) + +// ConnectDB opens a connection to the database +// TODO: properly handle all errors +func ConnectDB() *sql.DB { + // load ENV + err := godotenv.Load() + if err != nil { + panic(err.Error()) + } + + // read ENV + username := os.Getenv("DB_USERNAME") + password := os.Getenv("DB_PASSWORD") + database := os.Getenv("DB_DATABASE") + + // connect to database + db, err := sql.Open("mysql", username + ":" + password + "@/" + database) + if err != nil { + panic(err.Error()) + } + + return db +} -- cgit v1.2.3