Fix loglist

internal/service/apikey_service.go

@@ -174,49 +174,40 @@ func (s *APIKeyService) Stop() {
 }
 
 func (s *APIKeyService) handleKeyUsageEvent(event *models.RequestFinishedEvent) {
-	if event.KeyID == 0 || event.GroupID == 0 {
+	if event.RequestLog.KeyID == nil || event.RequestLog.GroupID == nil {
 		return
 	}
-	// Handle success case: key recovery and timestamp update.
-	if event.IsSuccess {
-		mapping, err := s.keyRepo.GetMapping(event.GroupID, event.KeyID)
+	if event.RequestLog.IsSuccess {
+		mapping, err := s.keyRepo.GetMapping(*event.RequestLog.GroupID, *event.RequestLog.KeyID)
 		if err != nil {
-			// Log if mapping is not found, but don't proceed.
-			s.logger.Warnf("[%s] Could not find mapping for G:%d K:%d on successful request: %v", event.CorrelationID, event.GroupID, event.KeyID, err)
+			s.logger.Warnf("[%s] Could not find mapping for G:%d K:%d on successful request: %v", event.CorrelationID, *event.RequestLog.GroupID, *event.RequestLog.KeyID, err)
 			return
 		}
-
-		needsUpdate := false
+		statusChanged := false
 		oldStatus := mapping.Status
-
-		// If status was not active, it's a recovery.
 		if mapping.Status != models.StatusActive {
 			mapping.Status = models.StatusActive
 			mapping.ConsecutiveErrorCount = 0
 			mapping.LastError = ""
-			needsUpdate = true
+			statusChanged = true
 		}
-		// Always update LastUsedAt timestamp.
+
 		now := time.Now()
 		mapping.LastUsedAt = &now
-		needsUpdate = true
-
-		if needsUpdate {
-			if err := s.keyRepo.UpdateMapping(mapping); err != nil {
-				s.logger.Errorf("[%s] Failed to update mapping for G:%d K:%d after successful request: %v", event.CorrelationID, event.GroupID, event.KeyID, err)
-			} else if oldStatus != models.StatusActive {
-				// Only publish event if status actually changed.
-				go s.publishStatusChangeEvent(event.GroupID, event.KeyID, oldStatus, models.StatusActive, "key_recovered_after_use")
-			}
+		if err := s.keyRepo.UpdateMapping(mapping); err != nil {
+			s.logger.Errorf("[%s] Failed to update mapping for G:%d K:%d after successful request: %v", event.CorrelationID, *event.RequestLog.GroupID, *event.RequestLog.KeyID, err)
+			return
+		}
+		if statusChanged {
+			go s.publishStatusChangeEvent(*event.RequestLog.GroupID, *event.RequestLog.KeyID, oldStatus, models.StatusActive, "key_recovered_after_use")
 		}
 		return
 	}
-	// Handle failure case: delegate to the centralized judgment function.
 	if event.Error != nil {
 		s.judgeKeyErrors(
 			event.CorrelationID,
-			event.GroupID,
-			event.KeyID,
+			*event.RequestLog.GroupID,
+			*event.RequestLog.KeyID,
 			event.Error,
 			event.IsPreciseRouting,
 		)
@@ -354,6 +345,10 @@ func (s *APIKeyService) ListAPIKeys(params *models.APIKeyQueryParams) (*Paginate
 	}, nil
 }
 
+func (s *APIKeyService) GetKeysByIds(ids []uint) ([]models.APIKey, error) {
+	return s.keyRepo.GetKeysByIDs(ids)
+}
+
 func (s *APIKeyService) UpdateAPIKey(key *models.APIKey) error {
 	go func() {
 		var oldKey models.APIKey