rts-sim-testing-service/ts/test_simulation_manage.go

package ts

import (
	"strconv"
	"sync"

	"joylink.club/bj-rtsts-server/config"
	"joylink.club/bj-rtsts-server/message_server"
	"joylink.club/bj-rtsts-server/sys_error"
	"joylink.club/bj-rtsts-server/ts/simulation/wayside/memory"

	"joylink.club/bj-rtsts-server/dto"
)

// 仿真存储集合
var simulationMap sync.Map

// 创建前检查
func IsExistSimulation() bool {
	i := 0
	simulationMap.Range(func(_, _ any) bool {
		i++
		return true
	})
	return i > 0
}

// 创建仿真对象
func CreateSimulation(projectId int32, mapIds []int32, runConfig string) (string, error) {
	simulationId := createSimulationId(projectId)
	_, e := simulationMap.Load(simulationId)
	if !e && IsExistSimulation() {
		return "", sys_error.New("一套环境同时只能运行一个仿真")
	}
	if !e {
		verifySimulation, err := memory.CreateSimulation(projectId, mapIds, runConfig)
		if err != nil {
			return "", err
		}
		verifySimulation.SimulationId = simulationId
		simulationMap.Store(simulationId, verifySimulation)
		// 全部成功，启动仿真
		verifySimulation.World.StartUp()
		// 启动仿真消息服务
		message_server.Start(verifySimulation)
	}
	return simulationId, nil
}

// 删除仿真对象
func DestroySimulation(simulationId string) {
	s, e := simulationMap.Load(simulationId)
	if !e {
		return
	}
	simulationMap.Delete(simulationId)
	simulationInfo := s.(*memory.VerifySimulation)
	message_server.Close(simulationInfo)
	simulationInfo.StopSimulation()
}

func createSimulationId(projectId int32) string {
	// 当前服务器IP + 端口 + 项目
	return config.SimulationId_prefix + "_" + strconv.Itoa(config.Config.Server.Port) + "_" + strconv.Itoa(int(projectId))
}

// 获取仿真列表
func ListAllSimulations() []*dto.SimulationInfoRspDto {
	var simArr []*dto.SimulationInfoRspDto
	simulationMap.Range(func(_, v any) bool {
		s := v.(*memory.VerifySimulation)
		simArr = append(simArr, &dto.SimulationInfoRspDto{
			SimulationId: s.SimulationId,
			MapId:        s.MapIds[0],
			MapIds:       s.MapIds,
			ProjectId:    s.ProjectId,
		})
		return true
	})
	return simArr
}

// 根据仿真id查找仿真实例
func FindSimulation(simulationId string) *memory.VerifySimulation {
	m, e := simulationMap.Load(simulationId)
	if e {
		return m.(*memory.VerifySimulation)
	}
	return nil
}

// 获取普通仿真数组
func GetSimulationArr() []*memory.VerifySimulation {
	var result []*memory.VerifySimulation
	simulationMap.Range(func(_, v any) bool {
		result = append(result, v.(*memory.VerifySimulation))
		return true
	})
	return result
}