v

core/database/database.go

@@ -4,20 +4,30 @@ import (
 	"database/sql"
 	"log"
 
-	"../../config"
 	_ "github.com/go-sql-driver/mysql"
 )
 
+//DB : Conexão com db
 var DB *sql.DB
 
-func Init() error {
+//Config : struct do banco de dados
+type Config struct {
+	Host   string
+	Port   string
+	User   string
+	Pass   string
+	Schema string
+}
+
+//Init : iniciação do metodo de conexão
+func Init(config Config) error {
 
 	//user:pass@tcp(host:port)/schema?parseTime=true
-	var dataSource = config.DATABASE_USER + ":" +
-		config.DATABASE_PASS + "@tcp(" +
-		config.DATABASE_HOST + ":" +
-		config.DATABASE_PORT + ")/" +
-		config.DATABASE_SCHEMA + "?parseTime=true"
+	var dataSource = config.User + ":" +
+		config.Pass + "@tcp(" +
+		config.Host + ":" +
+		config.Port + ")/" +
+		config.Schema + "?parseTime=true"
 
 	var err error
 

main.go

@@ -13,8 +13,15 @@ import (
 )
 
 func main() {
+	dbConfig := database.Config{
+		config.DATABASE_HOST,
+		config.DATABASE_PORT,
+		config.DATABASE_USER,
+		config.DATABASE_PASS,
+		config.DATABASE_SCHEMA,
+	}
 
-	database.Init()
+	database.Init(dbConfig)
 
 	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
 		res.SetResponseWriter(w)