main.go

package main

import (
	"database/sql"
	"fmt"
	"log"
	"os"

	"github.com/ZeeshanTamboli/slack-clone-services/database"

	"github.com/ZeeshanTamboli/slack-clone-services/api/handlers"

	"github.com/joho/godotenv"

	_ "github.com/lib/pq"
)

const (
	dbhost = "DBHOST"
	dbport = "DBPORT"
	dbuser = "DBUSER"
	dbpass = "DBPASS"
	dbname = "DBNAME"
)

func init() {
	if err := godotenv.Load(); err != nil {
		log.Fatal("Env files could not be loaded. Err: ", err)
	}
}

func main() {
	initDb()
	defer database.DBCon.Close() // This will close the db if the server fails to start and exits this main func
	handlers.InitializeRoutes()
}

func initDb() {
	config := dbConfig()
	var err error
	psqlInfo := fmt.Sprintf("host=%s port=%s user=%s "+"password=%s dbname=%s sslmode=disable", config[dbhost], config[dbport], config[dbuser], config[dbpass], config[dbname])

	database.DBCon, err = sql.Open("postgres", psqlInfo)
	if err != nil {
		panic(err)
	}

	err = database.DBCon.Ping()
	if err != nil {
		panic(err)
	}
	fmt.Println("Successfully connected")
}

func dbConfig() map[string]string {
	conf := make(map[string]string)
	host, ok := os.LookupEnv(dbhost)
	if !ok {
		panic("DBHOST environment variable required but not set")
	}

	port, ok := os.LookupEnv(dbport)
	if !ok {
		panic("DBPORT environment variable required but not set")
	}

	user, ok := os.LookupEnv(dbuser)
	if !ok {
		panic("DBUSER environment variable required but not set")
	}

	password, ok := os.LookupEnv(dbpass)
	if !ok {
		panic("DBPASS environment variable required but not set")
	}

	name, ok := os.LookupEnv(dbname)
	if !ok {
		panic("DBNAME environment variable required but not set")
	}

	conf[dbhost] = host
	conf[dbport] = port
	conf[dbuser] = user
	conf[dbpass] = password
	conf[dbname] = name

	return conf
}