game_demo/main.go

56 lines
1.1 KiB
Go

package main
import (
_ "game_demo/docs"
"context"
"fmt"
"game_demo/pkg/setting"
"game_demo/routers"
ginSwagger "github.com/swaggo/gin-swagger"
"github.com/swaggo/gin-swagger/swaggerFiles"
"log"
"net/http"
"os"
"os/signal"
"time"
)
// @title GameDemo API
// @version 1.0
// @description This is a sample server Petstore server.
// @termsOfService http://swagger.io/terms/
// @BasePath /api
func main() {
router := routers.InitRouter()
router.GET("/swagger/*any", ginSwagger.WrapHandler(swaggerFiles.Handler))
s := &http.Server{
Addr: fmt.Sprintf(":%d", setting.HTTPPort),
Handler: router,
ReadTimeout: setting.ReadTimeout,
WriteTimeout: setting.WriteTimeout,
MaxHeaderBytes: 1 << 20,
}
go func() {
if err := s.ListenAndServe(); err != nil {
log.Printf("Listen: %s\n", err)
}
}()
quit := make(chan os.Signal)
signal.Notify(quit, os.Interrupt)
<- quit
log.Printf("Shutdown Server")
ctx, cancel := context.WithTimeout(context.Background(), 5 * time.Second)
defer cancel()
if err := s.Shutdown(ctx); err != nil {
log.Fatal("Server Shutdown:", err)
}
log.Printf("Server exiting")
}