go-charts/main.go
2021-12-26 10:42:16 +08:00

65 lines
1.3 KiB
Go

package main
import (
"bytes"
"embed"
"fmt"
"time"
"github.com/vicanso/elton"
"github.com/vicanso/elton/middleware"
"github.com/vicanso/go-charts"
)
//go:embed web
var webFS embed.FS
func main() {
e := elton.New()
e.Use(middleware.NewLogger(middleware.LoggerConfig{
Format: middleware.LoggerCombined,
OnLog: func(s string, _ *elton.Context) {
fmt.Println(s)
},
}))
e.Use(middleware.NewDefaultError())
e.Use(middleware.NewDefaultBodyParser())
assetFS := middleware.NewEmbedStaticFS(webFS, "web")
e.GET("/static/*", middleware.NewStaticServe(assetFS, middleware.StaticServeConfig{
// 缓存服务器缓存一个小时
SMaxAge: 5 * time.Minute,
DenyQueryString: true,
DisableLastModified: true,
EnableStrongETag: true,
}))
e.GET("/ping", func(c *elton.Context) error {
c.BodyBuffer = bytes.NewBufferString("pong")
return nil
})
e.GET("/", func(c *elton.Context) error {
buf, err := webFS.ReadFile("web/index.html")
if err != nil {
return err
}
c.SetContentTypeByExt(".html")
c.BodyBuffer = bytes.NewBuffer(buf)
return nil
})
e.POST("/", func(c *elton.Context) error {
buf, err := charts.RenderEChartsToSVG(string(c.RequestBody))
if err != nil {
return err
}
c.BodyBuffer = bytes.NewBuffer(buf)
return nil
})
err := e.ListenAndServe(":7001")
if err != nil {
panic(err)
}
}