main.go

package main

import (
	"database/sql"
	"fmt"
	"net/http"
	"os"

	"github.com/CircleCI-Public/circleci-demo-go/service"
	_ "github.com/mattes/migrate/driver/postgres"
	"github.com/mattes/migrate/migrate"
)

func main() {
	db := SetupDB()
	server := service.NewServer(db)
	http.HandleFunc("/", server.ServeHTTP)
	http.ListenAndServe(":8080", nil)
}

func SetupDB() *service.Database {
	databaseUrl := os.Getenv("CONTACTS_DB_URL")
	if databaseUrl == "" {
		panic("CONTACTS_DB_URL must be set!")
	}

	sqlFiles := "./db/migrations"
	if sqlFilesEnv := os.Getenv("CONTACTS_DB_MIGRATIONS"); sqlFilesEnv != "" {
		sqlFiles = sqlFilesEnv
	}
	allErrors, ok := migrate.ResetSync(databaseUrl, sqlFiles)
	if !ok {
		panic(fmt.Sprintf("%+v", allErrors))
	}

	db, err := sql.Open("postgres", databaseUrl)
	if err != nil {
		panic(fmt.Sprintf("Unable to open DB connection: %+v", err))
	}

	return &service.Database{db}
}