メインコンテンツまでスキップ

Job & Awaitable ブリッジ

Valkarn Tasksは、UnityのJob SystemおよびAwaitable APIをValkarnTaskパイプラインに接続するブリッジ型のセットを提供します。各ブリッジは薄いアロケーション最小化レイヤーであり、隠れた魔法は存在しません。

ここで説明するすべての型は名前空間UnaPartidaMas.Valkarn.Tasks.Bridgeにあり、#if UNITY_5_3_OR_NEWER(Awaitableサポートは#if UNITY_2023_1_OR_NEWER)でガードされています。


JobHandleExtensions — 単一のJobHandleをawait

最もシンプルなブリッジです。任意のJobHandle.ToValkarnTask()を呼び出すと、ジョブが完了したときに完了するValkarnTaskを返します。

public static ValkarnTask ToValkarnTask(
this JobHandle handle,
PlayerLoopTiming timing = PlayerLoopTiming.Update,
CancellationToken cancellationToken = default)

動作の仕組み

  1. ファストパス。 handle.IsCompletedがすでにtrueの場合、handle.Complete()を即座に呼び出し、ValkarnTask.CompletedTaskを返します — ゼロアロケーション、PlayerLoop登録なし。
  2. 通常パス。 プールされたJobHandlePromiseをレンタルし、指定されたtimingでPlayerLoopに登録し、ValkarnTaskにラップして返します。各フレームでMoveNext()JobHandle.ScheduleBatchedJobs()(エディットモードとバッチモードでジョブキューをフラッシュするため)を呼び出し、次にhandle.IsCompletedを確認します。ハンドルが完了すると、プロミスがタスクを完了し、自身をプールに返却します。
  3. キャンセル。 CancellationTokenが発火すると、ハンドルを強制的に完了し(ジョブシステムのリークを防ぐためhandle.Complete()は常に呼び出されます)、タスクはキャンセル状態に遷移します。

基本的な使用法

using UnaPartidaMas.Valkarn.Tasks;
using UnaPartidaMas.Valkarn.Tasks.Bridge;
using Unity.Jobs;

// ジョブをスケジュールして即座にawait
var handle = myJob.Schedule();
await handle.ToValkarnTask();

// デフォルト以外のタイミングとキャンセルを使用
await handle.ToValkarnTask(PlayerLoopTiming.PreUpdate, destroyCancellationToken);

JobHandleWhenAll — 複数のJobHandleを並行してawait

独立した複数のジョブをスケジュールし、すべて完了した後にのみ再開する場合はJobHandleExtensions.WhenAllを使用します。

// 最もシンプルなオーバーロード: Updateタイミングですべてのハンドルを待機
public static ValkarnTask WhenAll(params JobHandle[] handles)

// フルオーバーロード: タイミングとキャンセルを設定可能
public static ValkarnTask WhenAll(
JobHandle[] handles,
PlayerLoopTiming timing,
CancellationToken cancellationToken = default)

// フルオーバーロードの拡張メソッドエイリアス
public static ValkarnTask ToValkarnTask(
this JobHandle[] handles,
PlayerLoopTiming timing = PlayerLoopTiming.Update,
CancellationToken cancellationToken = default)

動作の仕組み

  • ファストパス。 配列内のすべてのハンドルがすでに完了している場合、すべてのハンドルをファイナライズし、ValkarnTask.CompletedTaskを即座に返します。
  • 空の配列。 ValkarnTask.CompletedTaskを返します。
  • 通常パス。 プールされたJobHandleArrayPromiseが作成されます。内部的にArrayPool<JobHandle>.SharedからJobHandle[]をレンタルし(呼び出しごとのヒープアロケーションを回避)、入力ハンドルをコピーし、PlayerLoopに登録します。各フレームで、コンパクトなスワップアンドシュリンクループを使用して保留中のハンドルのみを反復し、JobHandle.ScheduleBatchedJobs()を呼び出してワーカーを実行し続けます。
  • キャンセル。 残りのすべてのハンドルを強制完了し、タスクをキャンセルします。

使用法

var handleA = jobA.Schedule();
var handleB = jobB.Schedule();
var handleC = jobC.Schedule();

// 3つすべてを待機
await JobHandleExtensions.WhenAll(handleA, handleB, handleC);

// または配列の拡張メソッドを使用
JobHandle[] handles = { handleA, handleB, handleC };
await handles.ToValkarnTask();

TempNativeArrayScope — awaitポイントをまたぐNativeArrayのライフタイム

問題

Allocator.TempJobで確保されたNativeArray<T>は短いライフタイムを持ちます。配列を確保してジョブをスケジュールし、ジョブハンドルをawaitして配列のdisposeを忘れると、Unityのセーフティシステムがメモリリークを報告します。プレーンなtry/finallyは機能しますが、長い非同期メソッドでは誤りやすいです。

TempNativeArrayScope<T>NativeArray<T>をラップし、usingステートメントを使用してスコープ終了時にdisposeする構造体です — ネイティブメモリに適用されたRAIIパターンです。

API

public struct TempNativeArrayScope<T> : IDisposable where T : struct
{
// ラップされた配列にアクセス。すでにdisposeされている場合はObjectDisposedExceptionをスロー
public NativeArray<T> Array { get; }

// スコープがdisposeされておらず配列が作成されている場合にtrue
public bool IsCreated { get; }

// Allocator.TempJobで新しいNativeArray<T>を確保して所有権を取得
public static TempNativeArrayScope<T> Create(int length);

// すでに確保されたNativeArray<T>の所有権を取得
public static TempNativeArrayScope<T> Wrap(NativeArray<T> existing);

// 配列をdisposeする。べき等: 複数回呼び出しても安全
public void Dispose();
}

// 非ジェネリックヘルパー(型推論の便宜のため)
public static class TempNativeArrayScope
{
public static TempNativeArrayScope<T> Create<T>(int length) where T : struct;
public static TempNativeArrayScope<T> Wrap<T>(NativeArray<T> existing) where T : struct;
}

DisposeInterlockedではなくプレーンなintフラグを使用します。スコープはusing varによるメインスレッドのシングルスレッド使用を前提としているためです。

使用法

using UnaPartidaMas.Valkarn.Tasks.Bridge;
using Unity.Collections;
using Unity.Jobs;

async ValkarnTask ProcessDataAsync(int count, CancellationToken ct)
{
// usingステートメントにより、通常完了・例外・キャンセルのいずれでも
// スコープ終了時にDispose()が呼ばれることが保証される
using var inputScope = TempNativeArrayScope.Create<float>(count);
using var outputScope = TempNativeArrayScope.Create<float>(count);

NativeArray<float> input = inputScope.Array;
NativeArray<float> output = outputScope.Array;

// 入力を設定してジョブをスケジュール
var job = new MyProcessingJob { Input = input, Output = output };
var handle = job.Schedule(count, 64);

// メインスレッドをブロックせずにawait
// NativeArrayは有効なまま — ジョブはまだ実行中
await handle.ToValkarnTask(cancellationToken: ct);

// ジョブが完了した。ここで結果を読み取る
float total = 0f;
for (int i = 0; i < count; i++)
total += output[i];

UnityEngine.Debug.Log($"Sum: {total}");

// inputScope.Dispose()とoutputScope.Dispose()がここで自動的に実行される
}

すでに確保した配列をラップすることもできます:

var existing = new NativeArray<int>(1024, Allocator.TempJob);
using var scope = TempNativeArrayScope.Wrap(existing);
// scopeはexistingを所有し、disposeする

よくある落とし穴: スコープなしのNativeArrayライフタイム

このパターンは壊れており、セーフティシステムエラーを引き起こします:

// 誤り: 例外が発生した場合、配列がジョブより長生きするか、リークする可能性がある
var array = new NativeArray<float>(1024, Allocator.TempJob);
var handle = job.Schedule(1024, 64);
await handle.ToValkarnTask(); // サスペンドポイント — 配列は生き続けなければならない
array.Dispose(); // 上でが例外が発生した場合は到達しない

すべてのコードパスでdisposeを保証するにはTempNativeArrayScopeまたはtry/finallyを使用してください。


AwaitableBridge — Unity AwaitableをValkarnTaskに変換

AwaitableBridgeは、UnityのAwaitableAwaitable<T>型(Unity 2023.1以降から利用可能)をValkarnTask互換のawaiterに変換する拡張メソッドを提供します。

注意: Awaitableは独自のGetAwaiter()も持っています。C#のオーバーロード解決はインスタンスメソッドを拡張メソッドより優先するため、async ValkarnTaskメソッド内でawait myAwaitableと書くだけで正しく機能します — UnityのawaiterはICriticalNotifyCompletionを実装しており、ValkarnTaskビルダーはそれを受け入れます。.AsValkarnTask()拡張メソッドが必要なのは、Awaitableをコンビネーター(ValkarnTask.WhenAllValkarnTask.WhenAny)に渡したり、ValkarnTask変数として格納したい場合のみです。

このファイルは#if UNITY_2023_1_OR_NEWERでガードされています。

API

// AwaitableをValkarnTask互換のawaiterに変換
public static AwaitableValkarnTaskAwaiter AsValkarnTask(this Awaitable awaitable)

// Awaitable<T>をValkarnTask互換のawaiterに変換
public static AwaitableValkarnTaskAwaiter<T> AsValkarnTask<T>(this Awaitable<T> awaitable)

両方のawaiterはICriticalNotifyCompletionを実装しており、ExecutionContextのキャプチャをスキップします。IsCompletedGetResultOnCompletedをラップされたUnity awaiterに直接委譲します。

使用法

using UnaPartidaMas.Valkarn.Tasks.Bridge;
using UnityEngine;

// 直接await — async ValkarnTaskメソッド内では変換不要
async ValkarnTask DirectExample()
{
await Awaitable.NextFrameAsync(); // 変換不要
await Awaitable.WaitForSecondsAsync(1f);
}

// 明示的変換 — コンビネーターと格納に必要
async ValkarnTask CombinatorExample()
{
ValkarnTask a = Awaitable.NextFrameAsync().AsValkarnTask();
ValkarnTask b = Awaitable.WaitForSecondsAsync(2f).AsValkarnTask();
await ValkarnTask.WhenAll(a, b);
}

// 結果型を持つジェネリック版
async ValkarnTask<Texture2D> LoadTextureExample(string path)
{
Awaitable<Texture2D> loadOp = LoadTextureAsync(path);
return await loadOp.AsValkarnTask();
}

JobBridge — ソース生成ラッパー

JobBridge.csはソースジェネレーターが使用するジェネリックプールドプロミス型JobPromise<TJob>を定義します。これは実装の詳細であり、通常は自分でインスタンス化しません。

// 各フレームでJobHandleをポーリング。ソース生成されたScheduleAsyncメソッドで使用される
public sealed class JobPromise<TJob> : ValkarnTask.ISource, IPlayerLoopItem, IPoolNode<JobPromise<TJob>>
where TJob : struct
{
public static JobPromise<TJob> Create(JobHandle handle, CancellationToken ct, out uint token);
}

動作はJobHandlePromiseJobHandleExtensions参照)と同一ですが、プール分離のためにジョブ型に対してジェネリック化されています — 各ジョブ型が独自のプールを持ちます。


ソースジェネレーター: JobBridgeGenerator

JobBridgeGeneratorはRoslynインクリメンタルソースジェネレーター(クラスUnaPartidaMas.Valkarn.Tasks.SourceGen.Generators.JobBridgeGenerator)で、ジョブ型のScheduleAsync拡張メソッドを自動的に生成します。

検出対象

ジェネレーターはコンパイル内のすべてのパブリック構造体をスキャンして、以下のいずれかを実装しているものを検出します:

  • Unity.Jobs.IJob
  • Unity.Jobs.IJobParallelFor
  • Unity.Jobs.IJobFor

プライベートおよびインターナルのジョブ構造体はスキップされます。非パブリック型の内部にネストされた構造体もスキップされます。

UnaPartidaMas.Valkarn.Tasks.ValkarnTaskがコンパイル内に見つからない場合、ジェネレーターは何もしません。Valkarn Tasksを参照しないアセンブリでも安全です。

生成内容

出力ファイルはValkarnTask.JobBridge.Generated.g.csです。検出された各ジョブ型に対してpublic static class __<TypeName>_AsyncExtを生成します:

ジョブインターフェース生成されるメソッドシグネチャ
IJobpublic static ValkarnTask ScheduleAsync(this ref MyJob job, CancellationToken ct = default)
IJobParallelForpublic static ValkarnTask ScheduleAsync(this ref MyJob job, int arrayLength, int innerLoopBatchCount, CancellationToken ct = default)
IJobForpublic static ValkarnTask ScheduleAsync(this ref MyJob job, int arrayLength, CancellationToken ct = default)
IJobForpublic static ValkarnTask ScheduleParallelAsync(this ref MyJob job, int arrayLength, int innerLoopBatchCount, CancellationToken ct = default)

各生成メソッドは標準のUnity拡張メソッドを使用してジョブをスケジュールし、結果のJobHandleJobPromise<TJob>にラップしてValkarnTaskを返します。

ネスト型(外部クラス内のジョブ構造体など)の場合、生成されるクラス名にはアンダースコアが使用されます:__Outer_Inner_AsyncExt

生成メソッドの使用法

// IJobの例
public struct MyCalculationJob : IJob
{
public NativeArray<float> Data;
public void Execute() { /* ... */ }
}

// ジェネレーターが生成する:
// public static ValkarnTask ScheduleAsync(this ref MyCalculationJob job, CancellationToken ct = default)

async ValkarnTask RunCalculation(CancellationToken ct)
{
using var scope = TempNativeArrayScope.Create<float>(1024);
var job = new MyCalculationJob { Data = scope.Array };
await job.ScheduleAsync(ct); // 生成された拡張メソッド
// ここでscope.Arrayから結果を読み取る
}

// IJobParallelForの例
public struct MyParallelJob : IJobParallelFor
{
public NativeArray<float> Input;
public NativeArray<float> Output;
public void Execute(int index) { Output[index] = Input[index] * 2f; }
}

async ValkarnTask RunParallel(int length, CancellationToken ct)
{
using var inputScope = TempNativeArrayScope.Create<float>(length);
using var outputScope = TempNativeArrayScope.Create<float>(length);
var job = new MyParallelJob { Input = inputScope.Array, Output = outputScope.Array };
await job.ScheduleAsync(length, innerLoopBatchCount: 64, ct);
}

ソースジェネレーター: AwaitableBridgeGenerator

AwaitableBridgeGeneratorはコンパイル内にUnityEngine.AwaitableUnityEngine.Awaitable<T>が存在するかどうかを検出し、存在する場合はAsValkarnTask()拡張メソッドを生成します。

出力ファイルはValkarnTask.AwaitableBridge.Generated.g.csです。生成されたコードはnamespace UnaPartidaMas.Valkarn.Tasks.BridgeAwaitableBridgeExtensionsクラスに格納されます。

生成メソッド:

// UnityEngine.Awaitableが見つかった場合に生成:
public static async ValkarnTask AsValkarnTask(this Awaitable awaitable)
{
await awaitable;
}

// UnityEngine.Awaitable<T>が見つかった場合に生成:
public static async ValkarnTask<T> AsValkarnTask<T>(this Awaitable<T> awaitable)
{
return await awaitable;
}

これらはasync ValkarnTaskメソッドなので、Valkarn Tasksのプールされた非同期ビルダーを通ります — ウォームプールではゼロアロケーションです。

ジェネレーターはガードされています:ValkarnTaskがコンパイル内になければコードは生成されません。これにより、Unityを参照するがValkarn Tasksを参照しないアセンブリでのCS0246エラーを防ぎます。


動作するフルの例: ECSシステムでのジョブブリッジ

以下はSamples~/ECS/JobBridgeExample.csからの例です。ISystemからBurstパラレルジョブをスケジュールし、ブロックせずにawaitし、結果を書き戻す完全なパターンを示しています。

#if UNITY_5_3_OR_NEWER && VTASKS_HAS_ENTITIES
using System.Threading;
using Unity.Burst;
using Unity.Collections;
using Unity.Entities;
using Unity.Jobs;
using UnaPartidaMas.Valkarn.Tasks;
using UnaPartidaMas.Valkarn.Tasks.Bridge;
using UnaPartidaMas.Valkarn.Tasks.ECS;

public partial struct JobBridgeExample : ISystem
{
public void OnCreate(ref SystemState state)
{
state.RequireForUpdate<HealthData>();
}

public void OnUpdate(ref SystemState state)
{
var worldCt = state.World.GetWorldCancellationToken();

// OnUpdate内でエンティティデータを同期的に抽出する
// 非同期メソッドはrefパラメーターを持てないため(CS1988)、
// データをここでコピーして値渡しで非同期メソッドに渡す必要がある
var query = SystemAPI.QueryBuilder().WithAll<HealthData>().Build();
var entityCount = query.CalculateEntityCount();
if (entityCount == 0) return;

var entities = query.ToEntityArray(Allocator.TempJob);
var healthArray = query.ToComponentDataArray<HealthData>(Allocator.TempJob);
var results = new NativeArray<float>(entityCount, Allocator.TempJob);

// 非同期メソッドがNativeArrayの所有権を取得してdisposeする
ProcessHealthAsync(state.EntityManager, entities, healthArray, results, worldCt).Forget();
state.Enabled = false;
}

public void OnDestroy(ref SystemState state) { }

static async ValkarnTask ProcessHealthAsync(
EntityManager entityManager,
NativeArray<Entity> entities,
NativeArray<HealthData> healthArray,
NativeArray<float> results,
CancellationToken ct)
{
try
{
// フェーズ1: Burstジョブをスケジュール
var job = new HealthProcessingJob
{
HealthInputs = healthArray,
ProcessedOutputs = results,
};
var handle = job.Schedule(entities.Length, batchSize: 64);

// フェーズ2: メインスレッドをブロックせずに完了を待機
await handle.ToValkarnTask(cancellationToken: ct);

// フェーズ3: 結果を適用。メインスレッドに戻っている
ct.ThrowIfCancellationRequested();

for (int i = 0; i < entities.Length; i++)
{
// ジョブ実行中にエンティティが破棄された可能性がある
if (!entityManager.SafeEntityExists(entities[i]))
continue;

entityManager.SetComponentData(entities[i], new HealthData
{
CurrentHealth = results[i],
});
}
}
finally
{
// 常にNativeArrayをdisposeする — 成功・例外・キャンセルいずれでも実行
if (entities.IsCreated) entities.Dispose();
if (healthArray.IsCreated) healthArray.Dispose();
if (results.IsCreated) results.Dispose();
}
}

[BurstCompile]
struct HealthProcessingJob : IJobParallelFor
{
[ReadOnly] public NativeArray<HealthData> HealthInputs;
[WriteOnly] public NativeArray<float> ProcessedOutputs;

public void Execute(int index)
{
var h = HealthInputs[index];
var newHealth = h.CurrentHealth + h.RegenRate;
if (newHealth > h.MaxHealth) newHealth = h.MaxHealth;
ProcessedOutputs[index] = newHealth;
}
}

struct HealthData : IComponentData
{
public float CurrentHealth;
public float MaxHealth;
public float RegenRate;
}
}
#endif

ブリッジ型のまとめ

目的アロケーション
JobHandleExtensions.ToValkarnTask()単一のJobHandleをawaitファストパスはゼロ;それ以外はプールされたプロミス
JobHandleExtensions.WhenAll()複数のJobHandleを並行してawaitファストパスはゼロ;それ以外はプールされたプロミス + ArrayPoolレンタル
TempNativeArrayScope<T>NativeArrayのRAIIライフタイム管理なし(構造体)
AwaitableBridge.AsValkarnTask()Awaitable/Awaitable<T>をValkarnTaskに変換なし(構造体awaiter)
生成されたScheduleAsync()型付きジョブを直接awaitプールされたJobPromise<TJob>