mirror of
https://github.com/gotenberg/gotenberg.git
synced 2026-08-08 08:32:16 +01:00
91 lines
2.2 KiB
Go
91 lines
2.2 KiB
Go
/*
|
|
Package main handles the application startup and shutdown.
|
|
|
|
Gotenberg is a stateless API for converting HTML files and Office documents to PDF.
|
|
|
|
For more information, go to https://github.com/gulien/gotenberg.
|
|
*/
|
|
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"time"
|
|
|
|
"github.com/gulien/gotenberg/app"
|
|
"github.com/gulien/gotenberg/app/config"
|
|
"github.com/gulien/gotenberg/app/converter/process"
|
|
"github.com/gulien/gotenberg/app/logger"
|
|
|
|
"github.com/gorilla/mux"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
// version will be set on build time.
|
|
var version = "master"
|
|
|
|
// defaultConfigurationFilePath is our default configuration file to parse.
|
|
const defaultConfigurationFilePath = "gotenberg.yml"
|
|
|
|
// main initializes the application, starts it, and handles
|
|
// graceful shutdown.
|
|
func main() {
|
|
c, err := config.NewAppConfig(defaultConfigurationFilePath)
|
|
if err != nil {
|
|
logger.SetLevel(logrus.InfoLevel)
|
|
logger.Fatal(err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
// defines our application logging.
|
|
logger.SetLevel(c.Logs.Level)
|
|
logger.SetFormatter(c.Logs.Formatter)
|
|
|
|
// defines our application router.
|
|
r := mux.NewRouter()
|
|
r.Handle("/", app.GetHandlersChain()).Methods(http.MethodPost)
|
|
|
|
// defines our server.
|
|
s := &http.Server{
|
|
Addr: fmt.Sprintf(":%s", c.Port),
|
|
Handler: r,
|
|
}
|
|
|
|
process.Load(c.CommandsConfig)
|
|
logger.Infof("Starting Gotenberg version %s", version)
|
|
logger.Infof("Listening on port %s", c.Port)
|
|
|
|
// runs our server in a goroutine so that it doesn't block.
|
|
go func() {
|
|
if err = s.ListenAndServe(); err != nil {
|
|
logger.SetLevel(logrus.InfoLevel)
|
|
logger.Panic(err)
|
|
os.Exit(1)
|
|
}
|
|
}()
|
|
|
|
quit := make(chan os.Signal, 1)
|
|
// we'll accept graceful shutdowns when quit via SIGINT (Ctrl+C)
|
|
// SIGKILL, SIGQUIT or SIGTERM (Ctrl+/) will not be caught.
|
|
signal.Notify(quit, os.Interrupt)
|
|
|
|
// blocks until we receive our signal.
|
|
<-quit
|
|
|
|
// creates a deadline to wait for.
|
|
var wait time.Duration
|
|
ctx, cancel := context.WithTimeout(context.Background(), wait)
|
|
defer cancel()
|
|
|
|
// doesn't block if no connections, but will otherwise wait
|
|
// until the timeout deadline.
|
|
s.Shutdown(ctx)
|
|
logger.SetLevel(logrus.InfoLevel)
|
|
|
|
logger.Info("Bye!")
|
|
os.Exit(0)
|
|
}
|