IDisposable.AddTo returns T where T:IDisposable

Cysharp · Jul 23, 2024 · 142ef1e · 142ef1e
1 parent 91711fb
commit 142ef1e
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/src/R3/Disposable.cs b/src/R3/Disposable.cs
@@ -13,19 +13,25 @@ public static DisposableBuilder CreateBuilder()
         return new DisposableBuilder();
     }
 
-    public static void AddTo(this IDisposable disposable, ref DisposableBuilder builder)
+    public static T AddTo<T>(this T disposable, ref DisposableBuilder builder)
+        where T : IDisposable
     {
         builder.Add(disposable);
+        return disposable;
     }
 
-    public static void AddTo(this IDisposable disposable, ref DisposableBag bag)
+    public static T AddTo<T>(this T disposable, ref DisposableBag bag)
+        where T : IDisposable
     {
         bag.Add(disposable);
+        return disposable;
     }
 
-    public static void AddTo(this IDisposable disposable, ICollection<IDisposable> disposables)
+    public static T AddTo<T>(this T disposable, ICollection<IDisposable> disposables)
+        where T : IDisposable
     {
         disposables.Add(disposable);
+        return disposable;
     }
 
     // AddTo is already used in UniTask so avoid name conflict...