Fix loki query log count (#82)

* Fix loki query log count

alert/eval/query.go

@@ -294,16 +294,13 @@ func loki(ctx *ctx.Context, datasourceId string, rule models.AlertRule) {
 		return
 	}
 
-	res, err := client.NewLokiClient(datasourceInfo).QueryRange(args)
+	res, count, err := client.NewLokiClient(datasourceInfo).QueryRange(args)
 	if err != nil {
 		global.Logger.Sugar().Errorf("查询 Loki 日志失败 %s", err.Error())
 		return
 	}
 
-	// count 用于统计日志条数
-	var count int
 	for _, v := range res {
-		count += len(v.Values)
 		event := func() models.AlertCurEvent {
 			event := process.BuildEvent(rule)
 			event.DatasourceId = datasourceId
@@ -325,6 +322,8 @@ func loki(ctx *ctx.Context, datasourceId string, rule models.AlertRule) {
 
 		// 评估告警条件
 		process.EvalCondition(ctx, event, float64(count), options)
+		// 只需要评估出第一条日志信息即可
+		break
 	}
 }
 

pkg/client/loki.go

@@ -44,12 +44,12 @@ type Result struct {
 	Values []interface{}     `json:"values"`
 }
 
-func (lc LokiClient) QueryRange(options QueryOptions) ([]Result, error) {
+func (lc LokiClient) QueryRange(options QueryOptions) ([]Result, int, error) {
 
 	curTime := time.Now()
 
 	if options.Query == "" {
-		return nil, nil
+		return nil, 0, nil
 	}
 
 	if options.Direction == "" {
@@ -73,18 +73,23 @@ func (lc LokiClient) QueryRange(options QueryOptions) ([]Result, error) {
 	requestURL := lc.BaseURL + args
 	res, err := http.Get(nil, requestURL)
 	if err != nil {
-		return nil, err
+		return nil, 0, err
 	}
 
 	body, _ := io.ReadAll(res.Body)
 	var resultData result
 	err = json.Unmarshal(body, &resultData)
 	if err != nil {
-		return nil, err
+		return nil, 0, err
 	}
 
-	return resultData.Data.Result, nil
+	// count 用于统计日志条数
+	var count int
+	for _, v := range resultData.Data.Result {
+		count += len(v.Values)
+	}
 
+	return resultData.Data.Result, count, nil
 }
 
 func (r Result) GetFingerprint() string {