in experiments/client_traffic_ctrl_tasks.go [13:60]
func (e *ExperimentClient) LoadSceneTrafficControlTasksData() {
//Load traffic control data for the production environment
prodSceneTrafficControlTasksData := make(map[string][]model.TrafficControlTask, 0)
prodOpt := &api.TrafficControlApiListTrafficControlTasksOpts{
ALL: optional.NewBool(true),
ControlTargetFilter: optional.NewString("Valid"),
Env: optional.NewString("product"),
Status: optional.NewString("Running"),
Version: optional.NewString("Released"),
}
prodResponse, err := e.APIClient.TrafficControlApi.ListTrafficControlTasks(prodOpt)
if err != nil {
e.logError(fmt.Errorf("list flow tasks error, err=%v", err))
return
}
for _, task := range prodResponse.TrafficControlTasks {
prodSceneTrafficControlTasksData[task.SceneName] = append(prodSceneTrafficControlTasksData[task.SceneName], task)
}
if len(prodSceneTrafficControlTasksData) > 0 {
e.productSceneTrafficControlTaskData = prodSceneTrafficControlTasksData
}
//Load traffic control data for the pre-load environment
prepubSceneTrafficControlTasksData := make(map[string][]model.TrafficControlTask, 0)
prePubOpt := &api.TrafficControlApiListTrafficControlTasksOpts{
ALL: optional.NewBool(true),
ControlTargetFilter: optional.NewString("Valid"),
Env: optional.NewString("prepub"),
Status: optional.NewString("Running"),
Version: optional.NewString("Released"),
}
prePubResponse, _ := e.APIClient.TrafficControlApi.ListTrafficControlTasks(prePubOpt)
if err != nil {
e.logError(fmt.Errorf("list flow tasks error,error=%v", err))
return
}
for _, task := range prePubResponse.TrafficControlTasks {
prepubSceneTrafficControlTasksData[task.SceneName] = append(prepubSceneTrafficControlTasksData[task.SceneName], task)
}
if len(prepubSceneTrafficControlTasksData) > 0 {
e.prepubSceneTrafficControlTaskData = prepubSceneTrafficControlTasksData
}
}