58 lines
1.4 KiB
Go
58 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"github.com/gin-gonic/gin"
|
|
"go-gin-api/app/config"
|
|
"go-gin-api/app/route"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"time"
|
|
)
|
|
|
|
func main() {
|
|
gin.SetMode(config.AppMode)
|
|
engine := gin.New()
|
|
|
|
// 设置路由
|
|
route.SetupRouter(engine)
|
|
|
|
server := &http.Server{
|
|
Addr : config.AppPort,
|
|
Handler : engine,
|
|
ReadTimeout : config.AppReadTimeout * time.Second,
|
|
WriteTimeout : config.AppWriteTimeout * time.Second,
|
|
}
|
|
|
|
fmt.Println("|-----------------------------------|")
|
|
fmt.Println("| go-gin-api |")
|
|
fmt.Println("|-----------------------------------|")
|
|
fmt.Println("| Go Http Server Start Successful |")
|
|
fmt.Println("| Port" + config.AppPort + " Pid:" + fmt.Sprintf("%d", os.Getpid()) + " |")
|
|
fmt.Println("|-----------------------------------|")
|
|
fmt.Println("")
|
|
|
|
go func() {
|
|
if err := server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
|
log.Fatalf("HTTP server listen: %s\n", err)
|
|
}
|
|
}()
|
|
|
|
// 等待中断信号以优雅地关闭服务器（设置 5 秒的超时时间）
|
|
signalChan := make(chan os.Signal)
|
|
signal.Notify(signalChan, os.Interrupt)
|
|
sig := <-signalChan
|
|
log.Println("Get Signal:", sig)
|
|
log.Println("Shutdown Server ...")
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5 * time.Second)
|
|
defer cancel()
|
|
if err := server.Shutdown(ctx); err != nil {
|
|
log.Fatal("Server Shutdown:", err)
|
|
}
|
|
log.Println("Server exiting")
|
|
}
|