Removed all refresh tokens use :/

1 files changed, 11 insertions, 5 deletions

diff --git a/Software/Visual_Studio/Web/Tango.MachineService/Security/RefreshTokensManager.cs b/Software/Visual_Studio/Web/Tango.MachineService/Security/RefreshTokensManager.cs
index 5cc206c63..06796c41e 100644
--- a/Software/Visual_Studio/Web/Tango.MachineService/Security/RefreshTokensManager.cs
+++ b/Software/Visual_Studio/Web/Tango.MachineService/Security/RefreshTokensManager.cs
@@ -7,7 +7,7 @@ using Tango.Web.Storage;
 
 namespace Tango.MachineService.Security
 {
-    public class RefreshTokensManager
+    public class RefreshTokensManager<T> where T : class
     {
         private TableStorageManager _manager;
 
@@ -16,16 +16,22 @@ namespace Tango.MachineService.Security
             _manager = new TableStorageManager();
         }
 
-        public RefreshTokenEntity GetToken(String refreshToken)
+        public RefreshTokenEntity<T> GetToken(String refreshToken)
         {
             var table = _manager.GetTable(MachineServiceConfig.REFRESH_TOKENS_TABLE_NAME);
-            return table.GetEntityAsync<RefreshTokenEntity>(MachineServiceConfig.REFRESH_TOKENS_TABLE_PARTITION, refreshToken).Result;
+            return table.GetEntity<RefreshTokenEntity<T>>(MachineServiceConfig.REFRESH_TOKENS_TABLE_PARTITION, refreshToken);
         }
 
-        public RefreshTokenEntity InsertOrUpdate(RefreshTokenEntity refreshTokenEntity)
+        public RefreshTokenEntity<T> InsertOrUpdate(RefreshTokenEntity<T> refreshTokenEntity)
         {
             var table = _manager.GetTable(MachineServiceConfig.REFRESH_TOKENS_TABLE_NAME);
-            return table.InsertOrUpdateEntityAsync(refreshTokenEntity).Result;
+            return table.InsertOrUpdateEntity(refreshTokenEntity);
+        }
+
+        public void DeleteToken(RefreshTokenEntity<T> entity)
+        {
+            var table = _manager.GetTable(MachineServiceConfig.REFRESH_TOKENS_TABLE_NAME);
+            table.DeleteEntity(entity);
         }
     }
 }
\ No newline at end of file