mirror of https://github.com/Superioz/aqua.git
57 lines
1.4 KiB
Go
57 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/go-co-op/gocron"
|
|
"github.com/joho/godotenv"
|
|
"github.com/superioz/aqua/internal/handler"
|
|
"github.com/superioz/aqua/internal/metrics"
|
|
"github.com/superioz/aqua/pkg/env"
|
|
"github.com/superioz/aqua/pkg/middleware"
|
|
"k8s.io/klog"
|
|
"time"
|
|
)
|
|
|
|
func main() {
|
|
err := godotenv.Load()
|
|
if err != nil {
|
|
klog.Warningf("Error loading .env file: %v", err)
|
|
}
|
|
klog.Infoln("Hello World!")
|
|
|
|
r := gin.New()
|
|
r.Use(middleware.Logger(3 * time.Second))
|
|
// restrict to max 100mb
|
|
r.Use(middleware.RestrictBodySize(int64(env.IntOrDefault("FILE_MAX_SIZE", 100)) * handler.SizeMegaByte))
|
|
r.Use(gin.Recovery())
|
|
|
|
// handler for receiving uploaded files
|
|
uh := handler.NewUploadHandler()
|
|
r.POST("/upload", uh.Upload)
|
|
|
|
// scheduler to do the cleanup every x minutes
|
|
s := gocron.NewScheduler(time.UTC)
|
|
_, err = s.Every(env.IntOrDefault("FILE_EXPIRATION_CYCLE", 15)).Minutes().StartImmediately().Do(func() {
|
|
err = uh.FileStorage.Cleanup()
|
|
if err != nil {
|
|
klog.Errorln(err)
|
|
}
|
|
})
|
|
if err != nil {
|
|
klog.Fatalf("could not start cleanup scheduler: %v", err)
|
|
}
|
|
s.StartAsync()
|
|
|
|
if env.BoolOrDefault("FILE_SERVING_ENABLED", true) {
|
|
r.GET("/:file", handler.HandleStaticFiles())
|
|
r.HEAD("/:file", handler.HandleStaticFiles())
|
|
}
|
|
|
|
// finally, start the metrics server as well
|
|
if env.BoolOrDefault("METRICS_ENABLED", true) {
|
|
go metrics.StartMetricsServer()
|
|
}
|
|
|
|
_ = r.Run(":8765")
|
|
}
|