Adds more repository methods and configs. Refs #3

pkg/configs/db.go

@@ -1,15 +1,11 @@
 package configs
 
 import (
-	"log"
 	"os"
-
-	"gorm.io/driver/mysql"
-	"gorm.io/gorm"
 )
 
 const (
-	dsnKey = "DATABASE_DSN"
+	dsnKey         = "DATABASE_DSN"
 	statusQueryKey = "DATABASE_STATUS_QUERY"
 )
 
@@ -29,21 +25,7 @@ func GetStatusQuery() string {
 	return statusQuery
 }
 
-// GetDB returns a GORM object connected to database, if everything goes well.
-func GetDB() *gorm.DB {
-	log.Printf("Connecting to database with DSN: %s", dsn)
-	db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
-	if err != nil {
-		panic(err)
-	}
-	return db
-}
-
-func loadDBConfig() {
+func init() {
 	dsn = os.Getenv(dsnKey)
 	statusQuery = os.Getenv(statusQueryKey)
-}
-
-func init() {
-	loadDBConfig()
 }

pkg/repositories/mysql_repository.go

@@ -5,6 +5,7 @@ import (
 
 	_ "github.com/go-sql-driver/mysql"
 
+	"github.com/ariel17/be-challenge-arios/pkg/configs"
 	"github.com/ariel17/be-challenge-arios/pkg/models"
 )
 
@@ -16,8 +17,8 @@ type mysqlRepository struct {
 	db *sql.DB
 }
 
-func (m *mysqlRepository) Connect(dsn string) error {
-	db, err := sql.Open("mysql", dsn)
+func (m *mysqlRepository) Connect() error {
+	db, err := sql.Open("mysql", configs.GetDSN())
 	if err != nil {
 		return err
 	}
@@ -32,6 +33,11 @@ func (m *mysqlRepository) Close() error {
 	return nil
 }
 
+func (m *mysqlRepository) GetStatus() error {
+	_, err := m.db.Query(configs.GetStatusQuery())
+	return err
+}
+
 func (m *mysqlRepository) AddPerson(person models.Person) error {
 	query := "INSERT INTO `persons` (`id`, `name`, `date_of_birth`, `nationality`) VALUES (?, ?, ?, ?)"
 	_, err := m.db.Exec(query, person.ID, person.Name, person.DateOfBirth, person.Nationality)

pkg/repositories/repositories.go

@@ -3,8 +3,9 @@ package repositories
 import "github.com/ariel17/be-challenge-arios/pkg/models"
 
 type Repository interface {
-	Connect(dsn string) error
+	Connect() error
 	Close() error
+	GetStatus() error
 
 	AddPerson(person models.Person) error
 	AddTeam(team models.Team) error