From 84387fc2a59b2eda77ae8454f90775c31ac6e598 Mon Sep 17 00:00:00 2001 From: ewezy Date: Fri, 2 May 2025 15:41:31 +0800 Subject: [PATCH] Fix missing message queue config due to incorrect parsing --- plugins/turing/manager/experiment_manager.go | 19 +++++++++++++------ 1 file changed, 13 insertions(+), 6 deletions(-) diff --git a/plugins/turing/manager/experiment_manager.go b/plugins/turing/manager/experiment_manager.go index 68351b0..1695ae5 100644 --- a/plugins/turing/manager/experiment_manager.go +++ b/plugins/turing/manager/experiment_manager.go @@ -4,6 +4,7 @@ import ( "context" "encoding/json" "fmt" + common_mq_config "github.com/caraml-dev/xp/common/messagequeue" "net/http" "strconv" "time" @@ -142,12 +143,18 @@ func (em *experimentManager) MakeTreatmentServicePluginConfig( projectID int, ) (*config.Config, error) { pluginConfig := &config.Config{ - Port: em.TreatmentServicePluginConfig.Port, - ProjectIds: []string{strconv.Itoa(projectID)}, - AssignedTreatmentLogger: em.TreatmentServicePluginConfig.AssignedTreatmentLogger, - DebugConfig: em.TreatmentServicePluginConfig.DebugConfig, - DeploymentConfig: em.TreatmentServicePluginConfig.DeploymentConfig, - MessageQueueConfig: em.TreatmentServicePluginConfig.MessageQueueConfig, + Port: em.TreatmentServicePluginConfig.Port, + ProjectIds: []string{strconv.Itoa(projectID)}, + AssignedTreatmentLogger: em.TreatmentServicePluginConfig.AssignedTreatmentLogger, + DebugConfig: em.TreatmentServicePluginConfig.DebugConfig, + DeploymentConfig: em.TreatmentServicePluginConfig.DeploymentConfig, + MessageQueueConfig: common_mq_config.MessageQueueConfig{ + Kind: common_mq_config.MessageQueueKind(*treatmentServiceConfig.MessageQueueConfig.Kind), + PubSubConfig: &common_mq_config.PubSubConfig{ + Project: *treatmentServiceConfig.MessageQueueConfig.PubSub.Project, + TopicName: *treatmentServiceConfig.MessageQueueConfig.PubSub.TopicName, + }, + }, ManagementService: em.TreatmentServicePluginConfig.ManagementService, MonitoringConfig: em.TreatmentServicePluginConfig.MonitoringConfig, SwaggerConfig: em.TreatmentServicePluginConfig.SwaggerConfig,