diff --git a/goCharts/draw.go b/goCharts/draw.go index bf006db..3367943 100644 --- a/goCharts/draw.go +++ b/goCharts/draw.go @@ -10,48 +10,25 @@ package goCharts import ( "github.com/vicanso/go-charts/v2" + "math/rand" + "time" ) +func getRand(len int) []float64 { + var res []float64 + for i := 0; i < len; i++ { + randFloat := float64(rand.Intn(1000)) + res = append(res, randFloat) + } + return res +} + func Draw() { values := [][]float64{ - { - 120, - 132, - 101, - // 134, - //charts.GetNullValue(), - 134, - 90, - 230, - 210, - }, - { - 220, - 182, - 191, - 234, - 290, - 330, - 310, - }, - { - 150, - 232, - 201, - 154, - 190, - 330, - 410, - }, - { - 320, - 332, - 301, - 334, - 390, - 330, - 320, - }, + getRand(7), + getRand(7), + getRand(7), + getRand(7), } p, err := charts.LineRender( values, @@ -60,15 +37,7 @@ func Draw() { charts.WidthOptionFunc(720), charts.HeightOptionFunc(1280), charts.TitleTextOptionFunc("symbol price"), - charts.XAxisDataOptionFunc([]string{ - "Mon", - "Tue", - "Wed", - "Thu", - "Fri", - "Sat", - "Sun", - }), + charts.XAxisDataOptionFunc(getLastDays()), charts.LegendLabelsOptionFunc([]string{ "BTC", "ETH", @@ -98,3 +67,19 @@ func Draw() { panic(err) } } + +func getLastDays() []string { + now := time.Now() + getDay := func(t time.Time, dif int) string { + return t.AddDate(0, 0, dif).Format("01/02") + } + return []string{ + getDay(now, -6), + getDay(now, -5), + getDay(now, -4), + getDay(now, -3), + getDay(now, -2), + getDay(now, -1), + getDay(now, 0), + } +} diff --git a/goCharts/draw_test.go b/goCharts/draw_test.go index e69d148..4f4b011 100644 --- a/goCharts/draw_test.go +++ b/goCharts/draw_test.go @@ -1,6 +1,9 @@ package goCharts -import "testing" +import ( + "fmt" + "testing" +) /* * @@ -12,3 +15,8 @@ import "testing" func TestDraw(t *testing.T) { Draw() } + +func TestGetrand(t *testing.T) { + fmt.Println(getRand(7)) + +}