fix(RunAsset): include RunInfo in RunAsset to avoid recreating it.

pollination · Mar 8, 2024 · 9c84fa4 · 9c84fa4
1 parent ec3378b
commit 9c84fa4
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 3 deletions.
diff --git a/src/PollinationSDK/Wrapper/AssetBase.cs b/src/PollinationSDK/Wrapper/AssetBase.cs
@@ -218,6 +218,11 @@ public virtual RunInfo GetRunInfo()
             }
             else
             {
+                // check cached RunInfo
+                if (this is RunAssetBase runAsset && runAsset.RunInfo != null)
+                    return runAsset.RunInfo;
+
+                // recreate RunInfo
                 var run = GetRun();
                 var proj = GetRoject();
 

diff --git a/src/PollinationSDK/Wrapper/RunAssetBase.cs b/src/PollinationSDK/Wrapper/RunAssetBase.cs
@@ -2,7 +2,7 @@
 {
     public abstract class RunAssetBase: AssetBase
     {
-
+        internal RunInfo RunInfo { get; set; }
     }
 
 

diff --git a/src/PollinationSDK/Wrapper/RunInfo.cs b/src/PollinationSDK/Wrapper/RunInfo.cs
@@ -323,7 +323,7 @@ private static RecipeInterface GetRecipe(string owner, string name, string tag =
         public List<RunOutputAsset> GetOutputAssets(string platform)
         {
             var sourcelink = this.ToString();
-            var assets = this.GetOutputs().Select(_ => new RunOutputAsset(_, platform, sourcelink)).ToList();
+            var assets = this.GetOutputs().Select(_ => new RunOutputAsset(_, platform, sourcelink) { RunInfo = this }).ToList();
             return assets;
         }
 
@@ -335,7 +335,7 @@ public List<RunOutputAsset> GetOutputAssets(string platform)
         public List<RunInputAsset> GetInputAssets()
         {
             var sourcelink = this.ToString();
-            var assets = this.GetInputs().Select(_ => new RunInputAsset(_, sourcelink)).ToList();
+            var assets = this.GetInputs().Select(_ => new RunInputAsset(_, sourcelink) { RunInfo = this }).ToList();
             return assets;
 
         }