rts-sim-module/entity/singleton.go

package entity

import (
	"time"

	"github.com/yohamta/donburi/filter"
	"joylink.club/ecs"
	"joylink.club/rtsssimulation/component"
	"joylink.club/rtsssimulation/repository"
)

// 初始化世界数据
func LoadWorldData(w ecs.World, repo *repository.Repository) {
	if repo == nil {
		panic("repo不能为nil")
	}
	entry := w.Create(component.WorldDataType)
	component.WorldDataType.Set(entry, &component.WorldData{
		Repo:      repo,
		Time:      time.Now().UnixMilli(),
		EntityMap: make(map[string]*ecs.Entry),
	})
}

var query = ecs.NewQuery(filter.Contains(component.WorldDataType))

// 获取世界数据
func GetWorldData(w ecs.World) *component.WorldData {
	entry, ok := query.First(w)
	if ok {
		return component.WorldDataType.Get(entry)
	}
	panic("世界数据不存在")
}