Fix CodeRabbit comments

Author: thivindu

gateway/gateway-controller/pkg/api/handlers/handlers.go

@@ -1956,6 +1956,12 @@ func (s *APIServer) GenerateAPIKey(c *gin.Context, name string, version string)
 			zap.Error(err),
 			zap.String("name", name),
 			zap.String("version", version))
+		// Rollback database save to maintain consistency
+		if s.db != nil {
+			if delErr := s.db.DeleteAPIKey(apiKey.APIKey); delErr != nil {
+				log.Error("Failed to rollback API key from database", zap.Error(delErr))
+			}
+		}
 		c.JSON(http.StatusInternalServerError, api.ErrorResponse{
 			Status:  "error",
 			Message: "Failed to store API key",

gateway/gateway-controller/pkg/storage/sqlite.go

@@ -198,10 +198,10 @@ func (s *SQLiteStorage) initSchema() error {
 			if _, err := s.db.Exec(`CREATE INDEX IF NOT EXISTS idx_api_key_expiry ON api_keys(expires_at);`); err != nil {
 				return fmt.Errorf("failed to create api_keys expiry index: %w", err)
 			}
-			if _, err := s.db.Exec("PRAGMA user_version = 4"); err != nil {
-				return fmt.Errorf("failed to set schema version to 4: %w", err)
+			if _, err := s.db.Exec("PRAGMA user_version = 5"); err != nil {
+				return fmt.Errorf("failed to set schema version to 5: %w", err)
 			}
-			s.logger.Info("Schema migrated to version 4 (api_keys table)")
+			s.logger.Info("Schema migrated to version 5 (api_keys table)")
 			version = 5
 		}
 
@@ -1164,7 +1164,7 @@ func (s *SQLiteStorage) UpdateAPIKey(apiKey *models.APIKey) error {
 	}
 
 	s.logger.Info("API key updated successfully",
-		zap.String("key", apiKey.APIKey),
+		zap.String(" id", apiKey.ID),
 		zap.String("status", string(apiKey.Status)))
 
 	return nil
@@ -1188,7 +1188,7 @@ func (s *SQLiteStorage) DeleteAPIKey(key string) error {
 		return fmt.Errorf("%w: API key not found", ErrNotFound)
 	}
 
-	s.logger.Info("API key deleted successfully", zap.String("key", key))
+	s.logger.Info("API key deleted successfully", zap.String("key_prefix", key[:min(8, len(key))]+"***"))
 
 	return nil
 }