pass db connection to routes

Author: isoppp

_httptest/status.http

@@ -0,0 +1,3 @@
+###
+GET http://localhost:8081/api/status
+Accept: application/json

_httptest/v1.http

@@ -0,0 +1,3 @@
+###
+GET http://localhost:8081/api/v1
+Accept: application/json

cmd/api/main.go

@@ -1,21 +1,30 @@
 package main
 
 import (
+	"database/sql"
 	"fmt"
 	"log"
 	"net/http"
 	"sandbox-go-api-sqlboiler-rest-auth/internal/config"
 	"sandbox-go-api-sqlboiler-rest-auth/internal/routes"
+
+	_ "github.com/lib/pq"
 )
 
 func main() {
-	cfg, err := config.NewEnvConfig()
+	cfg, err := config.NewConfig()
 	if err != nil {
 		log.Fatalln(err)
 		return
 	}
+	connStr := "host=localhost port=5433 dbname=sandbox user=postgres password=postgres sslmode=disable"
+	db, err := sql.Open("postgres", connStr)
+	if err != nil {
+		log.Fatalln("cannot open the database")
+	}
+	defer db.Close()
 
-	r := routes.NewRouter()
+	r := routes.NewRouter(db)
 
 	log.Printf("server is running on port %v", cfg.Port)
 	err = http.ListenAndServe(fmt.Sprintf(":%s", cfg.Port), r)

internal/config/config.go

@@ -2,12 +2,12 @@ package config
 
 import "github.com/caarlos0/env/v6"
 
-type EnvConfig struct {
+type Config struct {
 	Port string `env:"PORT,required"`
 }
 
-func NewEnvConfig() (*EnvConfig, error) {
-	config := EnvConfig{}
+func NewConfig() (*Config, error) {
+	config := Config{}
 	if err := env.Parse(&config); err != nil {
 		return &config, err
 	}

internal/routes/no_versioning.go

@@ -0,0 +1,20 @@
+package routes
+
+import (
+	"database/sql"
+	"log"
+	"net/http"
+
+	"github.com/go-chi/chi/v5"
+)
+
+func noVersioningRoutes(db *sql.DB) chi.Router {
+	r := chi.NewRouter()
+	r.Get("/status", func(w http.ResponseWriter, r *http.Request) {
+		_, err := w.Write([]byte("working"))
+		if err != nil {
+			log.Fatalln(err)
+		}
+	})
+	return r
+}

internal/routes/routes.go

@@ -1,7 +1,7 @@
 package routes
 
 import (
-	"net/http"
+	"database/sql"
 	"time"
 
 	"github.com/go-chi/render"
@@ -10,15 +10,7 @@ import (
 	"github.com/go-chi/chi/v5/middleware"
 )
 
-type Health struct {
-	OK bool `json:"ok"`
-}
-
-func (h *Health) Render(_ http.ResponseWriter, _ *http.Request) error {
-	return nil
-}
-
-func NewRouter() *chi.Mux {
+func NewRouter(db *sql.DB) *chi.Mux {
 	r := chi.NewRouter()
 	r.Use(middleware.RequestID)
 	r.Use(middleware.RealIP)
@@ -27,14 +19,9 @@ func NewRouter() *chi.Mux {
 	r.Use(middleware.Timeout(60 * time.Second))
 	r.Use(render.SetContentType(render.ContentTypeJSON))
 
-	r.Route("/api/", func(r chi.Router) {
-		r.Get("/health", func(w http.ResponseWriter, r *http.Request) {
-			err := render.Render(w, r, &Health{OK: true})
-			if err != nil {
-				http.Error(w, http.StatusText(500), 500)
-				return
-			}
-		})
+	r.Route("/api", func(r chi.Router) {
+		r.Mount("/", noVersioningRoutes(db))
+		r.Mount("/v1", v1Routes(db))
 	})
 	return r
 }

internal/routes/v1.go

@@ -0,0 +1,20 @@
+package routes
+
+import (
+	"database/sql"
+	"log"
+	"net/http"
+
+	"github.com/go-chi/chi/v5"
+)
+
+func v1Routes(db *sql.DB) chi.Router {
+	r := chi.NewRouter()
+	r.Get("/", func(w http.ResponseWriter, r *http.Request) {
+		_, err := w.Write([]byte("v1 index"))
+		if err != nil {
+			log.Fatalln(err)
+		}
+	})
+	return r
+}