From 9920d21ad88132539140aba3e60969da9bb40933 Mon Sep 17 00:00:00 2001
From: Renjaya Raga Zenta <renjaya.zenta@formulatrix.com>
Date: Wed, 30 Jul 2025 23:57:51 +0700
Subject: [PATCH] fix: forgot to complete inner pipe

---
 src/Oh.My.Stitcher/Program.cs | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/Oh.My.Stitcher/Program.cs b/src/Oh.My.Stitcher/Program.cs
index cc76777..b14c5c5 100644
--- a/src/Oh.My.Stitcher/Program.cs
+++ b/src/Oh.My.Stitcher/Program.cs
@@ -67,8 +67,6 @@ app.MapPost("/api/image/generate", (Stitch request, IMemoryCache cache) =>
             if( value is string path )
               File.Delete(path);
           });
-        logger.ZLogDebug($"save cache key: {cacheKey}, file: {newCacheFile}");
-        cache.Set(cacheKey!, newCacheFile, cacheEntryOptions);
 
         await using FileStream cacheStream = new(newCacheFile, FileMode.Create, FileAccess.Write, FileShare.Read);
         while( true )
@@ -83,8 +81,11 @@ app.MapPost("/api/image/generate", (Stitch request, IMemoryCache cache) =>
           if( result.IsCompleted )
             break;
         }
+        logger.ZLogDebug($"save cache key: {cacheKey}, file: {newCacheFile}");
+        cache.Set(cacheKey, newCacheFile, cacheEntryOptions);
       });
       image?.WriteToStream(innerPipe.Writer.AsStream(), ".png");
+      await innerPipe.Writer.CompleteAsync();
     }
     catch( Exception e )
     {