feat: receive resiliency report from scenario container logs and change Scenario struct to recieve weight and config

Signed-off-by: Abhinav Sharma <abhinavs1920bpl@gmail.com>

Author: abhinavs1920

pkg/scenarioorchestrator/common_functions.go

@@ -2,21 +2,31 @@ package scenarioorchestrator
 
 import (
 	"context"
+	"encoding/json"
 	"fmt"
-	"github.com/fatih/color"
-	"github.com/krkn-chaos/krknctl/pkg/config"
-	providermodels "github.com/krkn-chaos/krknctl/pkg/provider/models"
-	"github.com/krkn-chaos/krknctl/pkg/scenarioorchestrator/models"
-	"github.com/krkn-chaos/krknctl/pkg/scenarioorchestrator/utils"
 	"io"
 	"os"
 	"os/signal"
 	"path"
+	"regexp"
 	"sort"
+	"strings"
 	"sync"
 	"syscall"
+	"time"
+
+	"github.com/fatih/color"
+	"github.com/krkn-chaos/krknctl/pkg/config"
+	providermodels "github.com/krkn-chaos/krknctl/pkg/provider/models"
+	"github.com/krkn-chaos/krknctl/pkg/scenarioorchestrator/models"
+	"github.com/krkn-chaos/krknctl/pkg/scenarioorchestrator/utils"
 )
 
+// ResiliencyReport represents resiliency report extracted from scenario logs
+type ResiliencyReport struct {
+	Score float64 `json:"score"`
+}
+
 func CommonRunGraph(
 	scenarios models.ScenarioSet,
 	resolvedGraph models.ResolvedGraph,
@@ -29,6 +39,14 @@ func CommonRunGraph(
 	registry *providermodels.RegistryV2,
 	userID *int,
 ) {
+	// collectors initialization
+	var allResiliencyReports []ResiliencyReport
+	var resiliencyScores []float64
+	var resiliencyWeights []float64
+	runStartTime := time.Now().UTC()
+
+	reportRegex := regexp.MustCompile(`^KRKN_RESILIENCY_REPORT_JSON:(.*)$`)
+
 	for step, s := range resolvedGraph {
 		var wg sync.WaitGroup
 		for _, scID := range s {
@@ -64,6 +82,13 @@ func CommonRunGraph(
 				volumes[k] = v
 			}
 
+			// inject resiliency config as env variable if provided (Use Case: Resiliency Config is provided via env variable)
+			if scenario.ResiliencyConfigPath != "" {
+				if content, err := os.ReadFile(scenario.ResiliencyConfigPath); err == nil {
+					env["RESILIENCY_CONFIG"] = string(content)
+				}
+			}
+
 			containerName := utils.GenerateContainerName(config, scenario.Name, &scID)
 			filename := fmt.Sprintf("%s.log", containerName)
 			file, err := os.Create(path.Clean(filename))
@@ -86,7 +111,35 @@ func CommonRunGraph(
 
 		}
 		wg.Wait()
+
+		// Parse resiliency reports 
+		for _, scIDIter := range s {
+			scenarioIter := scenarios[scIDIter]
+			containerNameIter := utils.GenerateContainerName(config, scenarioIter.Name, &scIDIter)
+			logFilePath := fmt.Sprintf("%s.log", containerNameIter)
+			if data, err := os.ReadFile(logFilePath); err == nil {
+				lines := strings.Split(string(data), "\n")
+				for _, line := range lines {
+					if matches := reportRegex.FindStringSubmatch(line); len(matches) > 1 {
+						var report ResiliencyReport
+						if err := json.Unmarshal([]byte(strings.TrimSpace(matches[1])), &report); err == nil {
+							allResiliencyReports = append(allResiliencyReports, report)
+							resiliencyScores = append(resiliencyScores, report.Score)
+							resiliencyWeights = append(resiliencyWeights, scenarioIter.ResiliencyWeight)
+						}
+						break
+					}
+				}
+			}
+		}
 	}
+	runEndTime := time.Now().UTC()
+	_ = runStartTime
+	_ = runEndTime
+	_ = allResiliencyReports
+	_ = resiliencyScores
+	_ = resiliencyWeights
+
 	commChannel <- nil
 }
 

pkg/scenarioorchestrator/models/models.go

@@ -41,6 +41,8 @@ type Scenario struct {
 	Env                  map[string]string `json:"env,omitempty"`
 	Volumes              map[string]string `json:"volumes,omitempty"`
 	ResiliencyConfig     string            `json:"resiliencyConfig,omitempty"`
+	ResiliencyConfigPath string            `json:"resiliencyConfigPath,omitempty"`
+	ResiliencyWeight     float64           `json:"resiliencyWeight,omitempty"`
 }
 
 type ScenarioContainer struct {