1
0
Fork 0

Updated server with better handling and caching

This commit is contained in:
Marcus Noble 2021-05-11 17:48:47 +01:00
parent a9e9185b41
commit 262e4ec4d7
1 changed files with 16 additions and 5 deletions

21
main.go
View File

@ -11,8 +11,13 @@ import (
"time"
"github.com/canhlinh/svg2png"
"github.com/gofiber/fiber/v2"
"github.com/patrickmn/go-cache"
"github.com/gofiber/fiber/v2"
fiberCache "github.com/gofiber/fiber/v2/middleware/cache"
"github.com/gofiber/fiber/v2/middleware/compress"
"github.com/gofiber/fiber/v2/middleware/cors"
"github.com/gofiber/fiber/v2/middleware/logger"
)
//go:embed index.html svg.tmpl
@ -21,7 +26,12 @@ var content embed.FS
func main() {
app := fiber.New()
ch := cache.New(5*24*time.Hour, 7*24*time.Hour)
app.Use(fiberCache.New())
app.Use(compress.New())
app.Use(cors.New())
app.Use(logger.New())
ch := cache.New(6*time.Hour, 24*time.Hour)
app.Get("/", func(c *fiber.Ctx) error {
c.Type("html", "UTF8")
@ -45,13 +55,14 @@ func main() {
key := generateKey(vars)
png, found := ch.Get(key)
if !found {
if !found || len(png.([]byte)) == 0 {
var err error
png, err = generateImage(vars)
if err != nil {
return err
fmt.Println(err)
return c.SendStatus(500)
}
ch.Set(key, png, -1)
ch.Set(key, png, 0)
}
c.Type("png")