Add read/write lock in the fault injection handler

aws · Aug 27, 2024 · af3afb4 · af3afb4
1 parent fec74a8
commit af3afb4
Show file tree

Hide file tree

Showing 4 changed files with 398 additions and 103 deletions.
diff --git a/agent/handlers/task_server_setup.go b/agent/handlers/task_server_setup.go
@@ -196,10 +196,7 @@ func registerFaultHandlers(
 	agentState *v4.TMDSAgentState,
 	metricsFactory metrics.EntryFactory,
 ) {
-	handler := fault.FaultHandler{
-		AgentState:     agentState,
-		MetricsFactory: metricsFactory,
-	}
+	handler := fault.New(agentState, metricsFactory)
 
 	if muxRouter == nil {
 		return

diff --git a/...dor/github.com/aws/amazon-ecs-agent/ecs-agent/tmds/handlers/fault/v1/handlers/handlers.go b/...dor/github.com/aws/amazon-ecs-agent/ecs-agent/tmds/handlers/fault/v1/handlers/handlers.go