33 lines
718 B
Go
33 lines
718 B
Go
|
package app
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"go.mongodb.org/mongo-driver/mongo"
|
||
|
|
"go.uber.org/zap"
|
||
|
|
"os"
|
||
|
|
"os/signal"
|
||
|
|
"syscall"
|
||
|
|
"verification/internal/server"
|
||
|
|
)
|
||
|
|
|
||
|
|
func gracefulShutdown(ctx context.Context, logger *zap.Logger, httpSrv *server.HTTP, mongoClient *mongo.Client) {
|
||
|
|
interrupt := make(chan os.Signal, 1)
|
||
|
|
signal.Notify(interrupt, os.Interrupt, syscall.SIGTERM)
|
||
|
|
killSignal := <-interrupt
|
||
|
|
switch killSignal {
|
||
|
|
case os.Interrupt:
|
||
|
|
logger.Info("AppInterrupted")
|
||
|
|
case syscall.SIGTERM:
|
||
|
|
logger.Info("AppTerminated")
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := httpSrv.Stop(); err != nil {
|
||
|
|
logger.Error("HttpServerShutdown", zap.Error(err))
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := mongoClient.Disconnect(ctx); err != nil {
|
||
|
|
logger.Error("BoltDB", zap.Error(err))
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|