package main
|
|
import (
|
"context"
|
"fmt"
|
"log"
|
"model-engine/config"
|
"model-engine/cron"
|
"model-engine/db"
|
"model-engine/pkg/logger"
|
"os"
|
"os/signal"
|
"syscall"
|
)
|
|
func Init() {
|
|
//配置初始化
|
config.Init()
|
|
// 日志初始化
|
logger.Init(*config.LogConf)
|
defer logger.Sync()
|
|
//数据库初始化
|
err := db.Init()
|
if err != nil {
|
log.Fatalf("Error connecting to the database: %v", err)
|
}
|
|
//定时任务开启
|
if err = cron.Run(); err != nil {
|
log.Fatalf("Error Run cron: %v", err)
|
}
|
}
|
|
// GracefulShutdown listens for OS signals and triggers a cleanup function before exiting.
|
func GracefulShutdown(ctx context.Context, cleanup func()) {
|
// Create a channel to receive the signal.
|
sigChan := make(chan os.Signal, 1)
|
signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM)
|
|
// Wait for a signal or the provided context to be canceled.
|
select {
|
case sig := <-sigChan:
|
fmt.Printf("Received signal: %v\n", sig)
|
case <-ctx.Done():
|
fmt.Println("Context was canceled.")
|
}
|
|
// Perform cleanup operations.
|
if cleanup != nil {
|
cleanup()
|
}
|
|
// Stop listening for signals.
|
signal.Stop(sigChan)
|
|
// Exit the program.
|
os.Exit(0)
|
}
|
|
// Cleanup handles cleanup tasks before exiting.
|
func Cleanup() {
|
fmt.Println("Cleaning up resources...")
|
// Your cleanup code here, such as closing database connections, stopping goroutines, etc.
|
}
|
|
func main() {
|
// Initialize the application.
|
Init()
|
|
// Create a background context that can be used to cancel the graceful shutdown.
|
ctx, cancel := context.WithCancel(context.Background())
|
defer cancel()
|
|
// Start a goroutine to call GracefulShutdown with a cleanup function.
|
go GracefulShutdown(ctx, Cleanup)
|
|
fmt.Println("Application is running. Press Ctrl+C to exit.")
|
|
// Block the main goroutine indefinitely.
|
select {}
|
}
|
