fix: preserve log alert recovery events

This commit is contained in:
zxr
2026-07-21 21:21:47 +08:00
parent 0655afb5ab
commit eaca3d0ac8
3 changed files with 74 additions and 9 deletions

View File

@@ -156,14 +156,19 @@ func processOneOutbox(db *gorm.DB, row models.AlertOutbox) error {
}
return markOutboxDead(db, row, row.RetryCount, "invalid_payload: "+err.Error(), now)
}
forwardErr := forwardOutboxPayload(row.PayloadJSON, body, fmt.Sprintf("logs-outbox:%d", row.ID))
fallbackKey := fmt.Sprintf("logs-outbox:%d", row.ID)
occurredAt := row.CreatedAt.UTC()
if occurredAt.IsZero() {
occurredAt = time.Now().UTC()
}
forwardErr := forwardOutboxPayload(row.PayloadJSON, body, fallbackKey, occurredAt)
now, err := databaseNow(db)
if err != nil {
return err
}
if forwardErr != nil {
if errors.Is(forwardErr, errAlertForwardDisabled) {
return markOutboxDead(db, row, row.RetryCount, forwardErr.Error(), now)
return markOutboxWaiting(db, row, forwardErr.Error(), now)
}
return markOutboxRetry(db, row, forwardErr.Error(), now)
}
@@ -178,16 +183,35 @@ func databaseNow(db *gorm.DB) (time.Time, error) {
return now.UTC(), nil
}
func forwardOutboxPayload(payloadJSON string, legacyBody AlertReceiveBody, fallbackSourceEventKey string) error {
func forwardOutboxPayload(payloadJSON string, legacyBody AlertReceiveBody, fallbackSourceEventKey string, occurredAt time.Time) error {
var rawEvent RawEventIngestBody
if err := json.Unmarshal([]byte(payloadJSON), &rawEvent); err == nil && rawEvent.SourceType != "" && len(rawEvent.RawPayload) > 0 {
if strings.TrimSpace(rawEvent.SourceEventKey) == "" {
rawEvent.SourceEventKey = fallbackSourceEventKey
}
if rawEvent.EventTime.IsZero() {
rawEvent.EventTime = occurredAt
}
rawEvent.TraceID = ensureAlertTraceID(rawEvent.TraceID, firstNonEmpty(rawEvent.SourceEventKey, fallbackSourceEventKey))
return forwardRawEvent(rawEvent)
}
if strings.TrimSpace(legacyBody.SourceEventKey) == "" {
legacyBody.SourceEventKey = fallbackSourceEventKey
}
if legacyBody.OccurredAt.IsZero() {
legacyBody.OccurredAt = occurredAt
}
legacyBody.TraceID = ensureAlertTraceID(legacyBody.TraceID, firstNonEmpty(legacyBody.SourceEventKey, fallbackSourceEventKey))
return forwardAlert(legacyBody)
}
func markOutboxWaiting(db *gorm.DB, row models.AlertOutbox, msg string, now time.Time) error {
return updateClaimedOutbox(db, row, map[string]interface{}{
"status": outboxStatusRetrying, "next_retry_at": now.Add(30 * time.Second),
"last_error": truncateError(msg, 1024), "lease_until": nil, "lease_owner": "",
}, "retrying")
}
func forwardRawEvent(body RawEventIngestBody) error {
cfg := config.Spec.AlertForward
if cfg == nil || !cfg.Enabled || cfg.BaseURL == "" {
@@ -246,9 +270,16 @@ func markOutboxSent(db *gorm.DB, row models.AlertOutbox, now time.Time) error {
if result.RowsAffected == 0 {
return nil
}
return tx.Model(&models.LogEvent{}).Where("id = ? AND dispatch_outbox_id = ?", row.LogEventID, row.ID).Updates(map[string]interface{}{
eventResult := tx.Model(&models.LogEvent{}).Where("id = ? AND dispatch_outbox_id = ?", row.LogEventID, row.ID).Updates(map[string]interface{}{
"alert_sent": true, "dispatch_status": "sent",
}).Error
})
if eventResult.Error != nil {
return eventResult.Error
}
if eventResult.RowsAffected != 1 {
return fmt.Errorf("log event %d does not belong to outbox %d", row.LogEventID, row.ID)
}
return nil
})
}
@@ -286,7 +317,14 @@ func updateClaimedOutbox(db *gorm.DB, row models.AlertOutbox, updates map[string
if result.RowsAffected == 0 {
return nil
}
return tx.Model(&models.LogEvent{}).Where("id = ? AND dispatch_outbox_id = ?", row.LogEventID, row.ID).Update("dispatch_status", eventStatus).Error
eventResult := tx.Model(&models.LogEvent{}).Where("id = ? AND dispatch_outbox_id = ?", row.LogEventID, row.ID).Update("dispatch_status", eventStatus)
if eventResult.Error != nil {
return eventResult.Error
}
if eventResult.RowsAffected != 1 {
return fmt.Errorf("log event %d does not belong to outbox %d", row.LogEventID, row.ID)
}
return nil
})
}