package iris_util

import (
	stdContext "context"
	"gitee.com/xlizy/common-go/base/response"
	"gitee.com/xlizy/common-go/utils/middleware"
	"github.com/kataras/iris/v12"
	"github.com/kataras/iris/v12/context"
	"os"
	"os/signal"
	"syscall"
	"time"
)

var _app *iris.Application

func init() {
	// GracefulShutdown
	signals := make(chan os.Signal, 1)
	signal.Notify(signals, syscall.SIGINT, syscall.SIGTERM)
	go func() {
		<-signals
		gracefulShutdown()
	}()
}
func GetDefaultApp() *iris.Application {
	_app = iris.New()
	_app.UseRouter(middleware.GobalRecover, middleware.Default)
	_app.Get("/actuator/health", func(ctx *context.Context) {
		ctx.JSON(response.Succ())
	})
	return _app
}

// gracefulShutdown 优雅关闭
func gracefulShutdown() {
	ctx, cancel := stdContext.WithTimeout(stdContext.Background(), 15*time.Second)
	defer cancel()
	_ = _app.Shutdown(ctx)
}