From e21b5f85f0d66f51e23a7c1cbf260f2981a83a49 Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Wed, 10 Feb 2021 19:17:08 +0800
Subject: ...

---
 BackEnd/Timeline/Services/DataManager.cs | 27 ++++++++++++++++++++-------
 1 file changed, 20 insertions(+), 7 deletions(-)

(limited to 'BackEnd/Timeline/Services/DataManager.cs')

diff --git a/BackEnd/Timeline/Services/DataManager.cs b/BackEnd/Timeline/Services/DataManager.cs
index d447b0d5..f24bb59b 100644
--- a/BackEnd/Timeline/Services/DataManager.cs
+++ b/BackEnd/Timeline/Services/DataManager.cs
@@ -38,13 +38,12 @@ namespace Timeline.Services
         public Task FreeEntry(string tag);
 
         /// <summary>
-        /// Retrieve the entry with given tag.
+        /// Retrieve the entry with given tag. If not exist, returns null.
         /// </summary>
         /// <param name="tag">The tag of the entry.</param>
-        /// <returns>The data of the entry.</returns>
+        /// <returns>The data of the entry. If not exist, returns null.</returns>
         /// <exception cref="ArgumentNullException">Thrown when <paramref name="tag"/> is null.</exception>
-        /// <exception cref="InvalidOperationException">Thrown when entry with given tag does not exist.</exception>
-        public Task<byte[]> GetEntry(string tag);
+        public Task<byte[]?> GetEntry(string tag);
     }
 
     public class DataManager : IDataManager
@@ -106,17 +105,31 @@ namespace Timeline.Services
             }
         }
 
-        public async Task<byte[]> GetEntry(string tag)
+        public async Task<byte[]?> GetEntry(string tag)
         {
             if (tag == null)
                 throw new ArgumentNullException(nameof(tag));
 
             var entity = await _database.Data.Where(d => d.Tag == tag).Select(d => new { d.Data }).SingleOrDefaultAsync();
 
-            if (entity == null)
-                throw new InvalidOperationException(Resources.Services.DataManager.ExceptionEntryNotExist);
+            if (entity is null)
+                return null;
 
             return entity.Data;
         }
     }
+
+    public static class DataManagerExtensions
+    {
+        /// <summary>
+        /// Try to get an entry and throw <see cref="DatabaseCorruptedException"/> if not exist.
+        /// </summary>
+        public static async Task<byte[]> GetEntryAndCheck(this IDataManager dataManager, string tag, string notExistMessage)
+        {
+            var data = await dataManager.GetEntry(tag);
+            if (data is null)
+                throw new DatabaseCorruptedException($"Can't get data of tag {tag}. {notExistMessage}");
+            return data;
+        }
+    }
 }
-- 
cgit v1.2.3