UnityECS/LocalPackages/com.nimin.tinyecs/Runtime/EntityDatabase.cs

using System;
using Unity.Collections;
using Unity.Mathematics;

namespace GameCore.TinyECS
{
    public struct EntityDatabase<TEnum, TData> : IDisposable
        where TEnum : unmanaged, IConvertible
        where TData : unmanaged, IEntityDataStruct
    {
        public TData dataStruct;
        public EntityGenerator<TEnum> generator;
        public EntityCollection<TEnum> collection;

        public int Count => collection.Count;

        public EntityDatabase(TEnum type, TData dataStruct, int capacity = 16)
        {
            generator = new (type, capacity, Allocator.Persistent);
            collection = new (type, capacity, Allocator.Persistent);
            this.dataStruct = dataStruct;
        }

        public Entity<TEnum> AddEntity()
        {
            var entity = generator.Create();
            collection.Add(entity);
            dataStruct.Add();
            return entity;
        }

        public void RemoveEntity(Entity<TEnum> entity)
        {
            generator.Delete(entity);
            if (collection.Remove(entity, out var index, out var _))
                dataStruct.Remove(index);
        }

        public T GetData<T>(Entity<TEnum> entity) where T : unmanaged
        {
            var index = collection.GetIndex(entity);
            return dataStruct.GetData<T>(index);
        }
        
        public ref T GetDataRef<T>(Entity<TEnum> entity) where T : unmanaged
        {
            var index = collection.GetIndex(entity);
            return ref dataStruct.GetDataRef<T>(index);
        }

        public void Dispose()
        {
            generator.Dispose();
            collection.Dispose();
            dataStruct.Dispose();
        }
    }
}