up

server/src/core/metrics.go

@@ -0,0 +1,32 @@
+package core
+
+import (
+	"encoding/json"
+	"net/http"
+	"project/metrics"
+	"project/pb"
+	"project/util"
+)
+
+func doMetrics(ab []byte, cached bool, r *http.Request) {
+
+	metrics.RspBytes(len(ab))
+
+	o := &pb.Rsp{}
+	json.Unmarshal(ab, o)
+
+	u := o.GetUsage()
+	if u == nil {
+		metrics.RspJSONFail()
+		return
+	}
+
+	metrics.RspToken(u.PromptTokens, u.TotalTokens, cached)
+
+	ip, err := util.GetIP(r)
+	sip := ip.String()
+	if err != nil {
+		sip = `unknown`
+	}
+	metrics.RspTokenByIP(sip, u.TotalTokens)
+}