fix: roll back when auth failed.

leancloud · Jan 12, 2021 · 4a1937c · 4a1937c
1 parent 346dcb3
commit 4a1937c
Showing 1 changed file with 26 additions and 3 deletions.
diff --git a/Storage/Storage/LCUser.cs b/Storage/Storage/LCUser.cs
@@ -326,7 +326,7 @@ public Task DisassociateWithAuthData(string platform) {
             if (string.IsNullOrEmpty(platform)) {
                 throw new ArgumentNullException(nameof(platform));
             }
-            return LinkWithAuthData(platform, null);
+            return UnlinkWithAuthData(platform);
         }
 
         /// <summary>
@@ -523,11 +523,34 @@ public static LCQuery<LCUser> GetQuery() {
             return new LCQuery<LCUser>(CLASS_NAME);
         }
 
-        Task LinkWithAuthData(string authType, Dictionary<string, object> data) {
+        async Task LinkWithAuthData(string authType, Dictionary<string, object> data) {
+            Dictionary<string, object> oriAuthData = new Dictionary<string, object>(AuthData);
             AuthData = new Dictionary<string, object> {
                 { authType, data }
             };
-            return Save();
+            try {
+                await Save();
+                oriAuthData.Add(authType, data);
+                AuthData = oriAuthData;
+            } catch (Exception e) {
+                AuthData = oriAuthData;
+                throw e;
+            }
+        }
+
+        async Task UnlinkWithAuthData(string authType) {
+            Dictionary<string, object> oriAuthData = new Dictionary<string, object>(AuthData);
+            AuthData = new Dictionary<string, object> {
+                { authType, null }
+            };
+            try {
+                await Save();
+                oriAuthData.Remove(authType);
+                AuthData = oriAuthData;
+            } catch (Exception e) {
+                AuthData = oriAuthData;
+                throw e;
+            }
         }
 
         static async Task<LCUser> Login(Dictionary<string, object> data) {