bitwarden/server
1
0
Fork 0
mirror of https://github.com/bitwarden/server synced 2025-12-21 18:53:41 +00:00
Code Issues Releases Wiki Activity

update to new 2.0.0 preview for notification hub

Browse Source
This commit is contained in:
Kyle Spearrin
2018-08-06 09:04:31 -04:00
parent 58d29cc4a8
commit b2715503c3
5 changed files with 50 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
src/Core/Services/Implementations/NotificationHubPushRegistrationService.cs
View File

@@ -1,5 +1,4 @@
#if NET471
using System.Collections.Generic;
using System.Collections.Generic;
using System.Threading.Tasks;
using Microsoft.Azure.NotificationHubs;
using Bit.Core.Enums;
@@ -10,13 +9,15 @@ namespace Bit.Core.Services
{
public class NotificationHubPushRegistrationService : IPushRegistrationService
{
private readonly NotificationHubClient _client;
private readonly GlobalSettings _globalSettings;
private NotificationHubClient _client = null;
private DateTime? _clientExpires = null;
public NotificationHubPushRegistrationService(
GlobalSettings globalSettings)
{
_client = NotificationHubClient.CreateClientFromConnectionString(globalSettings.NotificationHub.ConnectionString,
globalSettings.NotificationHub.HubName);
_globalSettings = globalSettings;
}
public async Task CreateOrUpdateRegistrationAsync(string pushToken, string deviceId, string userId,
@@ -76,9 +77,11 @@ namespace Bit.Core.Services
BuildInstallationTemplate(installation, "payload", payloadTemplate, userId, identifier);
BuildInstallationTemplate(installation, "message", messageTemplate, userId, identifier);
BuildInstallationTemplate(installation, "badgeMessage", badgeMessageTemplate ?? messageTemplate, userId, identifier);
BuildInstallationTemplate(installation, "badgeMessage", badgeMessageTemplate ?? messageTemplate,
userId, identifier);
await _client.CreateOrUpdateInstallationAsync(installation);
await RenewClientAndExecuteAsync(async client =>
await client.CreateOrUpdateInstallationAsync(installation));
}
private void BuildInstallationTemplate(Installation installation, string templateId, string templateBody,
@@ -113,7 +116,7 @@ namespace Bit.Core.Services
{
try
{
await _client.DeleteInstallationAsync(deviceId);
await RenewClientAndExecuteAsync(async client => await client.DeleteInstallationAsync(deviceId));
}
catch(Exception e)
{
@@ -135,7 +138,8 @@ namespace Bit.Core.Services
$"organizationId:{organizationId}");
}
private async Task PatchTagsForUserDevicesAsync(IEnumerable<string> deviceIds, UpdateOperationType op, string tag)
private async Task PatchTagsForUserDevicesAsync(IEnumerable<string> deviceIds, UpdateOperationType op,
string tag)
{
if(!deviceIds.Any())
{
@@ -161,7 +165,8 @@ namespace Bit.Core.Services
{
try
{
await _client.PatchInstallationAsync(id, new List<PartialUpdateOperation> { operation });
await RenewClientAndExecuteAsync(async client =>
await client.PatchInstallationAsync(id, new List<PartialUpdateOperation> { operation }));
}
catch(Exception e)
{
@@ -172,6 +177,18 @@ namespace Bit.Core.Services
}
}
}
private async Task RenewClientAndExecuteAsync(Func<NotificationHubClient, Task> task)
{
var now = DateTime.UtcNow;
if(_client == null || !_clientExpires.HasValue || _clientExpires.Value < now)
{
_clientExpires = now.Add(TimeSpan.FromMinutes(30));
_client = NotificationHubClient.CreateClientFromConnectionString(
_globalSettings.NotificationHub.ConnectionString,
_globalSettings.NotificationHub.HubName);
}
await task(_client);
}
}
}
#endif