update with new packages

hellojebus · Jun 8, 2018 · a5e849d · a5e849d
1 parent e4f7fdf
commit a5e849d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/main.go b/main.go
@@ -4,6 +4,7 @@ import (
 	"os"
 	"log"
 	"net/http"
+	"github.com/hellojebus/go-envoz-api/db"
 )
 
 func main(){
@@ -13,8 +14,8 @@ func main(){
 	router := NewRouter()
 
 	//Setup database
-	DB = SetupDB()
-	defer DB.Close()
+	db.DB = db.SetupDB()
+	defer db.DB.Close()
 
 	//create http server
 	log.Fatal(http.ListenAndServe(":"+port, router))