76 lines
1.6 KiB
Go
76 lines
1.6 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"embed"
|
|
"encoding/base64"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/vicanso/elton"
|
|
"github.com/vicanso/elton/middleware"
|
|
"github.com/vicanso/go-charts/v2"
|
|
)
|
|
|
|
//go:embed web
|
|
var webFS embed.FS
|
|
|
|
func main() {
|
|
e := elton.New()
|
|
|
|
e.Use(middleware.NewLogger(middleware.LoggerConfig{
|
|
Format: `{real-ip} {when-iso} "{method} {uri} {proto}" {status} {size-human} "{userAgent}"`,
|
|
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 {
|
|
outputType := c.QueryParam("outputType")
|
|
fn := charts.RenderEChartsToSVG
|
|
isPNG := false
|
|
if outputType == "png" {
|
|
isPNG = true
|
|
fn = charts.RenderEChartsToPNG
|
|
}
|
|
buf, err := fn(string(c.RequestBody))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if isPNG {
|
|
buf = []byte(base64.StdEncoding.EncodeToString(buf))
|
|
}
|
|
c.BodyBuffer = bytes.NewBuffer(buf)
|
|
return nil
|
|
})
|
|
|
|
err := e.ListenAndServe(":7001")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|