Init
This commit is contained in:
104
cmd/history-api/main.go
Normal file
104
cmd/history-api/main.go
Normal file
@@ -0,0 +1,104 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
// _ "history-api/docs"
|
||||
"history-api/internal/gen/sqlc"
|
||||
"history-api/pkg/cache"
|
||||
"history-api/pkg/config"
|
||||
_ "history-api/pkg/log"
|
||||
|
||||
"fmt"
|
||||
"history-api/pkg/database"
|
||||
"os/signal"
|
||||
"syscall"
|
||||
"time"
|
||||
|
||||
"github.com/rs/zerolog/log"
|
||||
)
|
||||
|
||||
func gracefulShutdown(fiberServer *FiberServer, done chan bool) {
|
||||
// Create context that listens for the interrupt signal from the OS.
|
||||
ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
|
||||
defer stop()
|
||||
|
||||
// Listen for the interrupt signal.
|
||||
<-ctx.Done()
|
||||
|
||||
log.Info().Msg("shutting down gracefully, press Ctrl+C again to force")
|
||||
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||||
defer cancel()
|
||||
|
||||
if err := fiberServer.App.ShutdownWithContext(ctx); err != nil {
|
||||
log.Info().Msgf("Server forced to shutdown with error: %v", err)
|
||||
}
|
||||
|
||||
log.Info().Msg("Server exiting")
|
||||
|
||||
done <- true
|
||||
}
|
||||
|
||||
//export StartServer
|
||||
func StartServer() {
|
||||
err := config.LoadEnv()
|
||||
if err != nil {
|
||||
log.Error().Msg(err.Error())
|
||||
panic(err)
|
||||
}
|
||||
pool, err := database.Connect()
|
||||
if err != nil {
|
||||
log.Error().Msg(err.Error())
|
||||
panic(err)
|
||||
}
|
||||
defer pool.Close()
|
||||
queries := sqlc.New(pool)
|
||||
|
||||
err = cache.Connect()
|
||||
if err != nil {
|
||||
log.Error().Msg(err.Error())
|
||||
panic(err)
|
||||
}
|
||||
|
||||
serverIp, _ := config.GetConfig("SERVER_IP")
|
||||
if serverIp == "" {
|
||||
serverIp = "127.0.0.1"
|
||||
}
|
||||
|
||||
httpPort, _ := config.GetConfig("SERVER_PORT")
|
||||
if httpPort == "" {
|
||||
httpPort = "3434"
|
||||
}
|
||||
|
||||
serverHttp := NewHttpServer()
|
||||
serverHttp.RegisterFiberRoutes()
|
||||
Singleton = serverHttp
|
||||
|
||||
done := make(chan bool, 1)
|
||||
|
||||
err = serverHttp.App.Listen(fmt.Sprintf("%s:%s", serverIp, httpPort))
|
||||
if err != nil {
|
||||
log.Error().Msgf("Error: app failed to start on port %s, %v", httpPort, err)
|
||||
panic(err)
|
||||
}
|
||||
|
||||
// Run graceful shutdown in a separate goroutine
|
||||
go gracefulShutdown(serverHttp, done)
|
||||
|
||||
// Wait for the graceful shutdown to complete
|
||||
<-done
|
||||
log.Info().Msg("Graceful shutdown complete.")
|
||||
}
|
||||
|
||||
// @title Firefly Manager API
|
||||
// @version 1.0
|
||||
// @description API to update Firefly Manager data
|
||||
// @host localhost:3344
|
||||
// @BasePath /
|
||||
|
||||
// @securityDefinitions.apikey BearerAuth
|
||||
// @in header
|
||||
// @name Authorization
|
||||
func main() {
|
||||
StartServer()
|
||||
}
|
||||
55
cmd/history-api/server.go
Normal file
55
cmd/history-api/server.go
Normal file
@@ -0,0 +1,55 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
// "history-api/internal/routes"
|
||||
// "history-api/internal/services"
|
||||
|
||||
swagger "github.com/gofiber/contrib/v3/swaggerui"
|
||||
"github.com/gofiber/fiber/v3"
|
||||
"github.com/gofiber/fiber/v3/middleware/cors"
|
||||
"github.com/gofiber/fiber/v3/middleware/logger"
|
||||
)
|
||||
|
||||
var (
|
||||
Singleton *FiberServer
|
||||
)
|
||||
|
||||
type FiberServer struct {
|
||||
App *fiber.App
|
||||
}
|
||||
|
||||
func NewHttpServer() *FiberServer {
|
||||
server := &FiberServer{
|
||||
App: fiber.New(fiber.Config{
|
||||
ServerHeader: "http-server",
|
||||
AppName: "http-server",
|
||||
}),
|
||||
}
|
||||
cfg := swagger.Config{
|
||||
BasePath: "/",
|
||||
FilePath: "./docs/swagger.json",
|
||||
Path: "swagger",
|
||||
Title: "Swagger API Docs",
|
||||
}
|
||||
|
||||
server.App.Use(swagger.New(cfg))
|
||||
server.App.Use(logger.New())
|
||||
return server
|
||||
}
|
||||
|
||||
func (s *FiberServer) RegisterFiberRoutes() {
|
||||
// Apply CORS middleware
|
||||
s.App.Use(cors.New(cors.Config{
|
||||
AllowOrigins: []string{"*"},
|
||||
AllowMethods: []string{"GET", "POST", "PUT", "DELETE", "OPTIONS", "PATCH"},
|
||||
AllowHeaders: []string{"Accept", "Authorization", "Content-Type", "Origin"},
|
||||
AllowCredentials: false,
|
||||
MaxAge: 300,
|
||||
}))
|
||||
|
||||
// routes.UserRoutes(s.App)
|
||||
// routes.AuthRoutes(s.App)
|
||||
// routes.MediaRoute(s.App)
|
||||
// routes.NotFoundRoute(s.App)
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user