修正提升權限失敗問題

bug: #14
NTUT-SELab · Aug 18, 2020 · bf338f2 · bf338f2
1 parent dada089
commit bf338f2
Showing 1 changed file with 22 additions and 2 deletions.
diff --git a/MicrosoftGraphAPIBot/Telegram/TelegramHandler.cs b/MicrosoftGraphAPIBot/Telegram/TelegramHandler.cs
@@ -55,7 +55,17 @@ public async Task<bool> AddAdminPermissionAsync(long telegramId, string userName
                 }
 
                 logger.LogInformation($"Telegram user: {telegramId}(@{userName}) 升級管理者權限:成功");
-                await db.SaveChangesAsync();
+                if (!db.Database.IsInMemory())
+                {
+                    // https://docs.microsoft.com/zh-tw/ef/core/saving/explicit-values-generated-properties#explicit-values-into-sql-server-identity-columns
+                    await db.Database.OpenConnectionAsync();
+                    db.Database.ExecuteSqlRaw("SET IDENTITY_INSERT dbo.TelegramUsers ON");
+                    db.SaveChanges();
+                    db.Database.ExecuteSqlRaw("SET IDENTITY_INSERT dbo.TelegramUsers OFF");
+                    await db.Database.CloseConnectionAsync();
+                }
+                else
+                    await db.SaveChangesAsync();
                 return true;
             }
 
@@ -84,7 +94,17 @@ public async Task RemoveAdminPermissionAsync(long telegramId, string userName)
                 db.TelegramUsers.Update(telegramUser);
             }
 
-            await db.SaveChangesAsync();
+            if (!db.Database.IsInMemory())
+            {
+                // https://docs.microsoft.com/zh-tw/ef/core/saving/explicit-values-generated-properties#explicit-values-into-sql-server-identity-columns
+                await db.Database.OpenConnectionAsync();
+                db.Database.ExecuteSqlRaw("SET IDENTITY_INSERT dbo.TelegramUsers ON");
+                db.SaveChanges();
+                db.Database.ExecuteSqlRaw("SET IDENTITY_INSERT dbo.TelegramUsers OFF");
+                await db.Database.CloseConnectionAsync();
+            }
+            else
+                await db.SaveChangesAsync();
         }
 
         #region Azure app