8.7.8

Author: trueai-org

src/Midjourney.API/Midjourney.API.csproj

@@ -9,7 +9,7 @@
 		<IncludeAllContentForSelfExtract>true</IncludeAllContentForSelfExtract>
 		<EnableCompressionInSingleFile>true</EnableCompressionInSingleFile>
 		<SatelliteResourceLanguages>zh-Hans</SatelliteResourceLanguages>
-		<AssemblyVersion>8.7.7</AssemblyVersion>
+		<AssemblyVersion>8.7.8</AssemblyVersion>
 		<DockerfileContext>..\..</DockerfileContext>
 	</PropertyGroup>
 

src/Midjourney.Base/DrawCounter.cs

@@ -281,7 +281,7 @@ public static bool InitUserTodayCounter(string userId)
         /// <param name="item"></param>
         public static void Success(TaskInfo item)
         {
-            if (item == null || item.Status != TaskStatus.SUCCESS || item.Action == null)
+            if (item == null || item.Action == null)
             {
                 return;
             }

src/Midjourney.Base/Models/TaskInfo.cs

@@ -620,15 +620,15 @@ private void SuccessUpdate()
                 var prompt = finalPrompt.ToLower();
 
                 // 解析速度模式
-                if (prompt.Contains("--fast"))
+                if (prompt.Contains("--fast", StringComparison.OrdinalIgnoreCase))
                 {
                     Mode = GenerationSpeedMode.FAST;
                 }
-                else if (prompt.Contains("--relax"))
+                else if (prompt.Contains("--relax", StringComparison.OrdinalIgnoreCase))
                 {
                     Mode = GenerationSpeedMode.RELAX;
                 }
-                else if (prompt.Contains("--turbo"))
+                else if (prompt.Contains("--turbo", StringComparison.OrdinalIgnoreCase))
                 {
                     Mode = GenerationSpeedMode.TURBO;
                 }
@@ -655,7 +655,7 @@ private void SuccessUpdate()
                 SetProperty(Constants.TASK_PROPERTY_FINAL_PROMPT, finalPrompt);
             }
 
-            UpdateUserDrawCount();
+            UpdateUserDrawCount(true);
 
             // 最后才设置完成时间和状态
             FinishTime = DateTimeOffset.Now.ToUnixTimeMilliseconds();
@@ -674,11 +674,6 @@ public void Fail(string reason)
                 reason = reason.Substring(0, 4000); // 限制失败原因长度
             }
 
-            FinishTime = DateTimeOffset.Now.ToUnixTimeMilliseconds();
-            Status = TaskStatus.FAILURE;
-            FailReason = reason;
-            Progress = "";
-
             if (!string.IsNullOrWhiteSpace(reason))
             {
                 if (reason.Contains("Banned prompt detected", StringComparison.OrdinalIgnoreCase)
@@ -712,17 +707,26 @@ public void Fail(string reason)
                 }
             }
 
-            UpdateUserDrawCount();
+            FinishTime = DateTimeOffset.Now.ToUnixTimeMilliseconds();
+            Status = TaskStatus.FAILURE;
+            FailReason = reason;
+            Progress = "";
+
+            UpdateUserDrawCount(false);
         }
 
         /// <summary>
         /// 更新用户绘图次数。
         /// </summary>
-        public void UpdateUserDrawCount()
+        public void UpdateUserDrawCount(bool success)
         {
             try
             {
-                DrawCounter.Success(this);
+                if (success)
+                {
+                    DrawCounter.Success(this);
+                }
+
 
                 //if (!string.IsNullOrWhiteSpace(UserId))
                 //{