Files
server/src/EventsProcessor/Functions.cs

70 lines
2.3 KiB
C#
Raw Normal View History

2017-12-08 15:02:54 -05:00
using System;
2017-12-08 16:03:20 -05:00
using System.Collections.Generic;
using System.Configuration;
2017-12-08 15:02:54 -05:00
using System.IO;
2017-12-08 23:09:50 -05:00
using System.Linq;
2017-12-08 15:02:54 -05:00
using System.Threading;
using System.Threading.Tasks;
2017-12-08 16:03:20 -05:00
using Bit.Core.Models.Data;
using Bit.Core.Services;
2017-12-08 15:02:54 -05:00
using Microsoft.Azure.WebJobs;
2017-12-08 16:03:20 -05:00
using Newtonsoft.Json;
using Newtonsoft.Json.Linq;
2017-12-08 15:02:54 -05:00
namespace Bit.EventsProcessor
{
public class Functions
{
2017-12-08 16:03:20 -05:00
private static IEventWriteService _eventWriteService;
static Functions()
{
var storageConnectionString = ConfigurationManager.ConnectionStrings["AzureWebJobsStorage"];
if(storageConnectionString == null || string.IsNullOrWhiteSpace(storageConnectionString.ConnectionString))
{
return;
}
var repo = new Core.Repositories.TableStorage.EventRepository(storageConnectionString.ConnectionString);
_eventWriteService = new RepositoryEventWriteService(repo);
}
public async static Task ProcessQueueMessageAsync([QueueTrigger("event")] string message,
2017-12-08 23:09:50 -05:00
TextWriter logger, CancellationToken cancellationToken)
2017-12-08 15:02:54 -05:00
{
2017-12-08 16:03:20 -05:00
if(_eventWriteService == null || message == null || message.Length == 0)
{
return;
}
try
{
2017-12-14 17:23:46 -05:00
var events = new List<IEvent>();
2017-12-08 23:09:50 -05:00
var token = JToken.Parse(message);
if(token is JArray)
2017-12-08 16:03:20 -05:00
{
2017-12-14 17:23:46 -05:00
var indexedEntities = token.ToObject<List<EventMessage>>()
.SelectMany(e => EventTableEntity.IndexEvent(e));
events.AddRange(indexedEntities);
2017-12-08 16:03:20 -05:00
}
2017-12-08 23:09:50 -05:00
else if(token is JObject)
2017-12-08 16:03:20 -05:00
{
2017-12-14 17:23:46 -05:00
var eventMessage = token.ToObject<EventMessage>();
events.AddRange(EventTableEntity.IndexEvent(eventMessage));
2017-12-08 16:03:20 -05:00
}
2017-12-14 17:23:46 -05:00
await _eventWriteService.CreateManyAsync(events);
2017-12-08 16:03:20 -05:00
}
catch(JsonReaderException)
{
await logger.WriteLineAsync("JsonReaderException: Unable to parse message.");
}
catch(JsonSerializationException)
{
await logger.WriteLineAsync("JsonSerializationException: Unable to serialize token.");
}
2017-12-08 15:02:54 -05:00
}
}
}