65 lines
1.4 KiB
Go
65 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"os/signal"
|
|
|
|
"github.com/go-echarts/go-echarts/v2/charts"
|
|
"github.com/go-echarts/go-echarts/v2/opts"
|
|
)
|
|
|
|
func main() {
|
|
ctx, can := signal.NotifyContext(context.Background())
|
|
defer can()
|
|
if err := run(ctx); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func run(ctx context.Context) error {
|
|
config, err := newConfig()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
log.Printf("%+v", config)
|
|
|
|
x, y, data, err := config.Data.AsScatterData()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
log.Printf("(%s, %s) = %v", x, y, data)
|
|
|
|
scatter := charts.NewScatter()
|
|
scatter.SetGlobalOptions(charts.WithTitleOpts(opts.Title{
|
|
Title: config.Data.Title,
|
|
Subtitle: config.Data.Subtitle,
|
|
}))
|
|
scatter.SetGlobalOptions(charts.WithLegendOpts(opts.Legend{Show: false}))
|
|
scatter.SetGlobalOptions(charts.WithXAxisOpts(opts.XAxis{
|
|
Name: x,
|
|
NameLocation: "middle",
|
|
NameGap: config.Data.Weight.Floor + config.Data.Weight.Range,
|
|
}))
|
|
scatter.SetGlobalOptions(charts.WithYAxisOpts(opts.YAxis{
|
|
Name: y,
|
|
NameLocation: "middle",
|
|
NameGap: config.Data.Weight.Floor + config.Data.Weight.Range,
|
|
}))
|
|
scatter.AddSeries(y, data)
|
|
|
|
buff := bytes.NewBuffer(nil)
|
|
if err := scatter.Render(buff); err != nil {
|
|
return err
|
|
}
|
|
|
|
switch config.Output.Scheme {
|
|
case "file":
|
|
return os.WriteFile(config.Output.Path, buff.Bytes(), os.ModePerm)
|
|
}
|
|
return fmt.Errorf("not impl output scheme: %s", config.Output.Scheme)
|
|
}
|