【Unity ECS】CleanupComponent を使ってEntityが破棄された時にクリーンアップ！

using Unity.Entities;

/// <summary>
/// クリーンアップが必要な状態かどうかを判別するためのコンポーネント
/// </summary>
/// <remarks>
/// この通常のコンポーネントを一緒に追加しておき、これがなくなっていたらEntityが破棄されたと判断する
/// </remarks>
public struct StillNoCleanupComponent : IComponentData
{
}

using Unity.Entities;

/// <summary>
/// <see cref="CleanupSampleComponent"/> をRuntimeで付与するためのコンポーネント
/// </summary>
/// <remarks>
/// <see cref="ICleanupComponentData"/>を実装しているコンポーネントはBakerで付与できないのでこのコンポーネントを印として使う
/// </remarks>
public struct AddCleanupSampleComponent : IComponentData
{
}

using Unity.Entities;
using UnityEngine;

/// <summary>
/// クリーンアップコンポーネント
/// </summary>
public struct CleanupSampleComponent : ICleanupComponentData
{
    /// <summary>
    /// クリーンアップに使用するデータや、その他必要なデータ
    /// </summary>
    [field: SerializeField]
    public int SampleValue { get; set; }
}

using Unity.Entities;
using UnityEngine;

public class CleanupSampleAuthoring : MonoBehaviour
{
    private class Baker : Baker<CleanupSampleAuthoring>
    {
        public override void Bake(CleanupSampleAuthoring authoring)
        {
            var entity = this.GetEntity(TransformUsageFlags.None);

            // クリーンアップコンポーネントはBakerで追加できないので、
            // 「Runtimeでクリーンアップコンポーネントを追加するための印のコンポーネント」を追加
            this.AddComponent<AddCleanupSampleComponent>(entity);
        }
    }
}

using Unity.Burst;
using Unity.Entities;


/// <summary>
/// クリーンアップコンポーネントを付与するシステム
/// </summary>
[RequireMatchingQueriesForUpdate]
partial struct AddCleanupComponentSampleSystem : ISystem
{
    [BurstCompile]
    public void OnUpdate(ref SystemState state)
    {
        var commandBufferSystem = SystemAPI.GetSingleton<BeginSimulationEntityCommandBufferSystem.Singleton>();
        var ecb = commandBufferSystem.CreateCommandBuffer(state.WorldUnmanaged);
        foreach (var (_, entity) in SystemAPI.Query<AddCleanupSampleComponent>().WithEntityAccess())
        {
            // クリーンアップコンポーネントの追加: お試しで数字を入れておく
            ecb.AddComponent(entity, new CleanupSampleComponent() { SampleValue = 12345 });

            // 一緒に通常のコンポーネントを追加しておく
            // 通常のコンポーネントなので、Entity破棄時に消えるので、それを利用して破棄されているかを判別する
            ecb.AddComponent<StillNoCleanupComponent>(entity);

            // クリーンアップコンポーネントを追加するための印だっただけなのでRemove
            ecb.RemoveComponent<AddCleanupSampleComponent>(entity);
        }
    }
}

using Unity.Entities;
using UnityEngine;

/// <summary>
/// お試しでEntityを破棄するシステム
/// </summary>
[RequireMatchingQueriesForUpdate]
partial struct SampleDestroySystem : ISystem
{
    public void OnUpdate(ref SystemState state)
    {
        var commandBufferSystem = SystemAPI.GetSingleton<BeginSimulationEntityCommandBufferSystem.Singleton>();
        var ecb = commandBufferSystem.CreateCommandBuffer(state.WorldUnmanaged);

        // StillNoCleanupComponent もまだ一緒に追加されていれば、破棄された状態ではないと判断する
        foreach (var (cleanup, _, entity) in SystemAPI.Query<RefRW<CleanupSampleComponent>, RefRO<StillNoCleanupComponent>>().WithEntityAccess())
        {
            // お試しの値が0以下になったらEntityを破棄
            if (cleanup.ValueRO.SampleValue <= 0)
            {
                ecb.DestroyEntity(entity);
                Debug.unityLogger.Log($"destroy {entity}");
            }
            else
            {
                // デクリメントしておく
                --cleanup.ValueRW.SampleValue;
                Debug.unityLogger.Log($"decrement {entity}, {cleanup.ValueRO.SampleValue}");
            }
        }
    }
}

using Unity.Entities;
using UnityEngine;

/// <summary>
/// クリーンアップシステム
/// </summary>
[RequireMatchingQueriesForUpdate]
partial struct CleanupSampleSystem : ISystem
{
    public void OnUpdate(ref SystemState state)
    {
        // Runtimeで必要なクリーンアップ
        var commandBufferSystem = SystemAPI.GetSingleton<BeginSimulationEntityCommandBufferSystem.Singleton>();
        var ecb = commandBufferSystem.CreateCommandBuffer(state.WorldUnmanaged);

        // CleanupSampleComponentがあって、StillNoCleanupComponentがないものは破棄されたEntityと判断し、クリーンアップする
        foreach (var (cleanup, entity) in SystemAPI.Query<RefRO<CleanupSampleComponent>>().WithAbsent<StillNoCleanupComponent>().WithEntityAccess())
        {
            this.Cleanup(ref state, ref ecb, cleanup, entity);
        }
    }

    public void OnDestroy(ref SystemState state)
    {
        // システム終了時にも必要であればクリーンアップ
        var commandBufferSystem = SystemAPI.GetSingleton<BeginSimulationEntityCommandBufferSystem.Singleton>();
        var ecb = commandBufferSystem.CreateCommandBuffer(state.WorldUnmanaged);

        // 必要に応じて、システム破棄時はクリーンアップコンポーネントを「Entityが破棄されているか」に関係なくクリーンアップ
        foreach (var (cleanup, entity) in SystemAPI.Query<RefRO<CleanupSampleComponent>>().WithEntityAccess())
        {
            this.Cleanup(ref state, ref ecb, cleanup, entity);
        }
    }

    private void Cleanup(ref SystemState state, ref EntityCommandBuffer ecb, RefRO<CleanupSampleComponent> cleanup, Entity entity)
    {
        // クリーンアップを行う：今回はお試しでログを出すだけ
        Debug.unityLogger.Log($"cleanup on destroy {cleanup.ValueRO.SampleValue}");

        // クリーンアップコンポーネントをRemoveすることで、Entityが破棄される
        ecb.RemoveComponent<CleanupSampleComponent>(entity);
    }
}