up

server/src/st/date-histogram.go

@@ -0,0 +1,53 @@
+package st
+
+import (
+	"fmt"
+	"project/es"
+	"project/pb"
+	"project/zj"
+
+	_ "embed" //
+
+	"github.com/zhengkai/zu"
+)
+
+//go:embed tpl/date-histogram.json
+var queryDateHistogramTpl string
+
+// DateHistogram ...
+func DateHistogram() {
+
+	query := fmt.Sprintf(
+		queryDateHistogramTpl,
+		`now-24h`,
+		`now`,
+	)
+
+	d := &pb.EsResultDateHistogram{}
+	err := es.Search(query, d)
+	if err != nil {
+		zj.W(`fail`, err)
+		return
+	}
+
+	li := d.GetAggregations().GetBytesSum().GetBuckets()
+	if len(li) == 0 {
+		return
+	}
+
+	re := make([]*pb.EsDateHistogram, len(li))
+	for i, v := range li {
+		re[i] = &pb.EsDateHistogram{
+			Ts:              uint32(v.GetKey() / 1000),
+			ReqBytes:        v.GetReqBytes().GetValue(),
+			RspBytes:        v.GetRspBytes().GetValue(),
+			TokenTotal:      v.GetTokenTotal().GetValue(),
+			TokenCompletion: v.GetTokenCompletion().GetValue(),
+			TokenPrompt:     v.GetTokenPrompt().GetValue(),
+		}
+	}
+
+	j := zu.JSONPretty(re)
+	// util.WriteFile(`date-histogram.json`, []byte(j))
+	zj.J(j)
+}