bill-server-go/internal/infrastructure/fiber.go
2023-06-02 16:08:12 +08:00

88 lines
2.6 KiB
Go

package infrastructure
import (
"bill-go-fiber/internal/bill"
"bill-go-fiber/internal/label"
"bill-go-fiber/internal/misc"
"fmt"
"log"
_ "github.com/go-sql-driver/mysql"
"github.com/gofiber/fiber/v2"
"github.com/gofiber/fiber/v2/middleware/compress"
"github.com/gofiber/fiber/v2/middleware/cors"
"github.com/gofiber/fiber/v2/middleware/etag"
"github.com/gofiber/fiber/v2/middleware/favicon"
"github.com/gofiber/fiber/v2/middleware/limiter"
"github.com/gofiber/fiber/v2/middleware/logger"
"github.com/gofiber/fiber/v2/middleware/recover"
"github.com/gofiber/fiber/v2/middleware/requestid"
)
// Run our Fiber webserver.
func Run() {
// Try to connect to our database as the initial part.
//mariadb, err := ConnectToMariaDB()
sqlite, err := ConnectToSqlite()
if err != nil {
log.Fatal("Database connection error: $s", err)
}
// Creates a new Fiber instance.
app := fiber.New(fiber.Config{
AppName: "Docker MariaDB Clean Arch",
ServerHeader: "Fiber",
})
// Use global middlewares.
app.Use(cors.New())
app.Use(compress.New())
app.Use(etag.New())
app.Use(favicon.New())
app.Use(limiter.New(limiter.Config{
Max: 100,
LimitReached: func(c *fiber.Ctx) error {
return c.Status(fiber.StatusTooManyRequests).JSON(&fiber.Map{
"status": "fail",
"message": "You have requested too many in a single time-frame! Please wait another minute!",
})
},
}))
app.Use(logger.New())
app.Use(recover.New())
app.Use(requestid.New())
// Create repositories.
//cityRepository := city.NewCityRepository(mariadb)
//userRepository := user.NewUserRepository(mariadb)
billRepository := bill.NewBillRepository(sqlite)
labelRepository := label.NewLabelRepository(sqlite)
// Create all of our services.
//cityService := city.NewCityService(cityRepository)
//userService := user.NewUserService(userRepository)
billService := bill.NewBillService(billRepository)
labelService := label.NewLabelService(labelRepository)
// Prepare our endpoints for the API.
misc.NewMiscHandler(app.Group("/api/v1"))
//auth.NewAuthHandler(app.Group("/api/v1/auth"))
//city.NewCityHandler(app.Group("/api/v1/cities"), cityService)
//user.NewUserHandler(app.Group("/api/v1/users"), userService)
bill.NewBillHandler(app.Group("/api/v1/bill"), billService)
label.NewLabelHandler(app.Group("/api/v1/label"), labelService)
// Prepare an endpoint for 'Not Found'.
app.All("*", func(c *fiber.Ctx) error {
errorMessage := fmt.Sprintf("Route '%s' does not exist in this API!", c.OriginalURL())
return c.Status(fiber.StatusNotFound).JSON(&fiber.Map{
"status": "fail",
"message": errorMessage,
})
})
// Listen to port 8080.
log.Fatal(app.Listen(":8080"))
}