rts-sim-module/examples/test1/tstorages/model_storage.go

package tstorages

import (
	"fmt"

	"joylink.club/rtsssimulation/umi"
)

// 仿真模型数据定义
type ModelData = umi.IDeviceModel

// 模型存储、管理仓库
type ModelStorage struct {
	//key-模型id,value-模型指针
	idModelMap map[string]ModelData
	//key-设备类型，value-对应设备类型的所有模型数据的指针列表
	typeModelMap map[umi.DeviceType][]ModelData
}

func NewModelStorage() *ModelStorage {
	return &ModelStorage{idModelMap: make(map[string]ModelData, 2048), typeModelMap: make(map[umi.DeviceType][]ModelData, 128)}
}

// 添加模型数据
// m 为具体模型数据的指针
func (me *ModelStorage) AddModel(m ModelData) error {
	_, ok := me.idModelMap[m.Id()]
	if ok {
		return fmt.Errorf("模型[%s]已经存在", m.Id())
	} else {
		me.idModelMap[m.Id()] = m
		//
		_, mdOk := me.typeModelMap[m.Type()]
		if !mdOk {
			me.typeModelMap[m.Type()] = make([]ModelData, 0, 512)
		}
		me.typeModelMap[m.Type()] = append(me.typeModelMap[m.Type()], m)
		//
		return nil
	}
}

// 根据设备类型获取该类型的所有设备数据
func (me *ModelStorage) FindModelsByType(deviceType umi.DeviceType) []ModelData {
	models, ok := me.typeModelMap[deviceType]
	if ok {
		return models
	} else {
		return []ModelData{}
	}
}

// 根据设备id获取对应模型数据
// 如果不存在返回nil
func (me *ModelStorage) FindModelById(id string) ModelData {
	md, ok := me.idModelMap[id]
	if ok {
		return md
	} else {
		return nil
	}
}

// 遍历某个类型的所有设备
func (me *ModelStorage) ForEachModelsByType(deviceType umi.DeviceType, callback func(md ModelData)) {
	mds := me.FindModelsByType(deviceType)
	for _, modelData := range mds {
		callback(modelData)
	}
}

// 根据id检测设备模型数据是否存在
func (me *ModelStorage) HasModelById(id string) bool {
	_, ok := me.idModelMap[id]
	return ok
}

// 获取当前模型数据总数量
func (me *ModelStorage) AmountOfModels() int {
	return len(me.idModelMap)
}
func (me *ModelStorage) AmountOfDeviceTypes() int {
	return len(me.typeModelMap)
}