54 lines
1.3 KiB
Go
54 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"embed"
|
|
"errors"
|
|
"log"
|
|
"net/http"
|
|
"os/signal"
|
|
"syscall"
|
|
"time"
|
|
|
|
"oneimg/backend/app"
|
|
"oneimg/backend/routes"
|
|
"oneimg/backend/utils/watermark"
|
|
)
|
|
|
|
//go:embed frontend/dist/**
|
|
var fs embed.FS
|
|
|
|
//go:embed frontend/src/assets/fonts/**
|
|
var fontFs embed.FS
|
|
|
|
func main() {
|
|
system := app.Init()
|
|
r := routes.SetupRoutes(fs)
|
|
watermark.Init(fontFs)
|
|
|
|
port := system.Config.Port
|
|
log.Printf("应用初始化完成，监听 :%s", port)
|
|
server := &http.Server{
|
|
Addr: ":" + port,
|
|
Handler: r,
|
|
ReadHeaderTimeout: 10 * time.Second,
|
|
IdleTimeout: 90 * time.Second,
|
|
MaxHeaderBytes: 1 << 20,
|
|
}
|
|
// No global write/read body timeout: large uploads and external storage
|
|
// streams use request-scoped limits, not an arbitrary response deadline.
|
|
ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
|
|
defer stop()
|
|
go func() {
|
|
<-ctx.Done()
|
|
shutdownCtx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
|
|
defer cancel()
|
|
if err := server.Shutdown(shutdownCtx); err != nil {
|
|
log.Printf("HTTP graceful shutdown failed: %v", err)
|
|
}
|
|
}()
|
|
if err := server.ListenAndServe(); err != nil && !errors.Is(err, http.ErrServerClosed) {
|
|
log.Fatal("服务启动失败:", err)
|
|
}
|
|
}
|