2017-12-12 20:22:22 +01:00
|
|
|
|
using System;
|
|
|
|
|
using System.Collections.Generic;
|
|
|
|
|
using System.Data;
|
|
|
|
|
using System.Data.SqlClient;
|
|
|
|
|
using System.Linq;
|
|
|
|
|
using System.Threading.Tasks;
|
2022-01-11 10:40:51 +01:00
|
|
|
|
using Bit.Core.Entities;
|
2017-12-12 20:22:22 +01:00
|
|
|
|
using Bit.Core.Models.Data;
|
2022-01-11 10:40:51 +01:00
|
|
|
|
using Bit.Core.Repositories;
|
2021-02-22 22:35:16 +01:00
|
|
|
|
using Bit.Core.Settings;
|
2017-12-19 05:15:16 +01:00
|
|
|
|
using Dapper;
|
2017-12-12 20:22:22 +01:00
|
|
|
|
|
2022-01-11 10:40:51 +01:00
|
|
|
|
namespace Bit.Infrastructure.Dapper.Repositories
|
2017-12-12 20:22:22 +01:00
|
|
|
|
{
|
|
|
|
|
public class EventRepository : Repository<Event, Guid>, IEventRepository
|
|
|
|
|
{
|
|
|
|
|
public EventRepository(GlobalSettings globalSettings)
|
2018-10-09 23:21:12 +02:00
|
|
|
|
: this(globalSettings.SqlServer.ConnectionString, globalSettings.SqlServer.ReadOnlyConnectionString)
|
2017-12-12 20:22:22 +01:00
|
|
|
|
{ }
|
|
|
|
|
|
2018-10-09 23:21:12 +02:00
|
|
|
|
public EventRepository(string connectionString, string readOnlyConnectionString)
|
|
|
|
|
: base(connectionString, readOnlyConnectionString)
|
2017-12-12 20:22:22 +01:00
|
|
|
|
{ }
|
|
|
|
|
|
2017-12-19 05:15:16 +01:00
|
|
|
|
public async Task<PagedResult<IEvent>> GetManyByUserAsync(Guid userId, DateTime startDate, DateTime endDate,
|
2017-12-15 21:23:57 +01:00
|
|
|
|
PageOptions pageOptions)
|
2017-12-12 20:22:22 +01:00
|
|
|
|
{
|
2017-12-19 05:15:16 +01:00
|
|
|
|
return await GetManyAsync($"[{Schema}].[Event_ReadPageByUserId]",
|
|
|
|
|
new Dictionary<string, object>
|
|
|
|
|
{
|
|
|
|
|
["@UserId"] = userId
|
|
|
|
|
}, startDate, endDate, pageOptions);
|
2017-12-12 20:22:22 +01:00
|
|
|
|
}
|
|
|
|
|
|
2017-12-19 05:15:16 +01:00
|
|
|
|
public async Task<PagedResult<IEvent>> GetManyByOrganizationAsync(Guid organizationId,
|
2017-12-15 21:23:57 +01:00
|
|
|
|
DateTime startDate, DateTime endDate, PageOptions pageOptions)
|
2017-12-14 18:33:50 +01:00
|
|
|
|
{
|
2017-12-19 05:15:16 +01:00
|
|
|
|
return await GetManyAsync($"[{Schema}].[Event_ReadPageByOrganizationId]",
|
|
|
|
|
new Dictionary<string, object>
|
|
|
|
|
{
|
|
|
|
|
["@OrganizationId"] = organizationId
|
|
|
|
|
}, startDate, endDate, pageOptions);
|
2017-12-15 21:50:50 +01:00
|
|
|
|
}
|
|
|
|
|
|
2017-12-19 05:15:16 +01:00
|
|
|
|
public async Task<PagedResult<IEvent>> GetManyByOrganizationActingUserAsync(Guid organizationId, Guid actingUserId,
|
2017-12-15 21:50:50 +01:00
|
|
|
|
DateTime startDate, DateTime endDate, PageOptions pageOptions)
|
|
|
|
|
{
|
2017-12-19 05:15:16 +01:00
|
|
|
|
return await GetManyAsync($"[{Schema}].[Event_ReadPageByOrganizationIdActingUserId]",
|
|
|
|
|
new Dictionary<string, object>
|
|
|
|
|
{
|
|
|
|
|
["@OrganizationId"] = organizationId,
|
|
|
|
|
["@ActingUserId"] = actingUserId
|
|
|
|
|
}, startDate, endDate, pageOptions);
|
2017-12-15 21:50:50 +01:00
|
|
|
|
}
|
|
|
|
|
|
2021-07-15 16:37:27 +02:00
|
|
|
|
public async Task<PagedResult<IEvent>> GetManyByProviderAsync(Guid providerId,
|
|
|
|
|
DateTime startDate, DateTime endDate, PageOptions pageOptions)
|
|
|
|
|
{
|
|
|
|
|
return await GetManyAsync($"[{Schema}].[Event_ReadPageByProviderId]",
|
|
|
|
|
new Dictionary<string, object>
|
|
|
|
|
{
|
|
|
|
|
["@ProviderId"] = providerId
|
|
|
|
|
}, startDate, endDate, pageOptions);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public async Task<PagedResult<IEvent>> GetManyByProviderActingUserAsync(Guid providerId, Guid actingUserId,
|
|
|
|
|
DateTime startDate, DateTime endDate, PageOptions pageOptions)
|
|
|
|
|
{
|
|
|
|
|
return await GetManyAsync($"[{Schema}].[Event_ReadPageByProviderIdActingUserId]",
|
|
|
|
|
new Dictionary<string, object>
|
|
|
|
|
{
|
|
|
|
|
["@ProviderId"] = providerId,
|
|
|
|
|
["@ActingUserId"] = actingUserId
|
|
|
|
|
}, startDate, endDate, pageOptions);
|
|
|
|
|
}
|
|
|
|
|
|
2017-12-19 05:15:16 +01:00
|
|
|
|
public async Task<PagedResult<IEvent>> GetManyByCipherAsync(Cipher cipher, DateTime startDate, DateTime endDate,
|
2017-12-15 21:50:50 +01:00
|
|
|
|
PageOptions pageOptions)
|
|
|
|
|
{
|
2017-12-19 05:15:16 +01:00
|
|
|
|
return await GetManyAsync($"[{Schema}].[Event_ReadPageByCipherId]",
|
|
|
|
|
new Dictionary<string, object>
|
|
|
|
|
{
|
|
|
|
|
["@OrganizationId"] = cipher.OrganizationId,
|
|
|
|
|
["@UserId"] = cipher.UserId,
|
|
|
|
|
["@CipherId"] = cipher.Id
|
|
|
|
|
}, startDate, endDate, pageOptions);
|
2017-12-14 18:33:50 +01:00
|
|
|
|
}
|
|
|
|
|
|
2017-12-12 20:22:22 +01:00
|
|
|
|
public async Task CreateAsync(IEvent e)
|
|
|
|
|
{
|
2020-03-27 19:36:37 +01:00
|
|
|
|
if (!(e is Event ev))
|
2017-12-12 20:22:22 +01:00
|
|
|
|
{
|
|
|
|
|
ev = new Event(e);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
await base.CreateAsync(ev);
|
|
|
|
|
}
|
|
|
|
|
|
2021-05-17 16:43:02 +02:00
|
|
|
|
public async Task CreateManyAsync(IEnumerable<IEvent> entities)
|
2017-12-12 20:22:22 +01:00
|
|
|
|
{
|
2020-03-27 19:36:37 +01:00
|
|
|
|
if (!entities?.Any() ?? true)
|
2017-12-12 20:22:22 +01:00
|
|
|
|
{
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
2021-05-17 16:43:02 +02:00
|
|
|
|
if (!entities.Skip(1).Any())
|
2017-12-14 23:23:46 +01:00
|
|
|
|
{
|
|
|
|
|
await CreateAsync(entities.First());
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
2020-03-27 19:36:37 +01:00
|
|
|
|
using (var connection = new SqlConnection(ConnectionString))
|
2017-12-12 20:22:22 +01:00
|
|
|
|
{
|
|
|
|
|
connection.Open();
|
2020-03-27 19:36:37 +01:00
|
|
|
|
using (var bulkCopy = new SqlBulkCopy(connection, SqlBulkCopyOptions.KeepIdentity, null))
|
2017-12-12 20:22:22 +01:00
|
|
|
|
{
|
|
|
|
|
bulkCopy.DestinationTableName = "[dbo].[Event]";
|
2020-02-28 04:44:49 +01:00
|
|
|
|
var dataTable = BuildEventsTable(bulkCopy, entities.Select(e => e is Event ? e as Event : new Event(e)));
|
2017-12-12 20:22:22 +01:00
|
|
|
|
await bulkCopy.WriteToServerAsync(dataTable);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2017-12-19 05:15:16 +01:00
|
|
|
|
private async Task<PagedResult<IEvent>> GetManyAsync(string sprocName,
|
|
|
|
|
IDictionary<string, object> sprocParams, DateTime startDate, DateTime endDate, PageOptions pageOptions)
|
|
|
|
|
{
|
|
|
|
|
DateTime? beforeDate = null;
|
2020-03-27 19:36:37 +01:00
|
|
|
|
if (!string.IsNullOrWhiteSpace(pageOptions.ContinuationToken) &&
|
2017-12-19 05:15:16 +01:00
|
|
|
|
long.TryParse(pageOptions.ContinuationToken, out var binaryDate))
|
|
|
|
|
{
|
|
|
|
|
beforeDate = DateTime.SpecifyKind(DateTime.FromBinary(binaryDate), DateTimeKind.Utc);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
var parameters = new DynamicParameters(sprocParams);
|
|
|
|
|
parameters.Add("@PageSize", pageOptions.PageSize, DbType.Int32);
|
|
|
|
|
// Explicitly use DbType.DateTime2 for proper precision.
|
|
|
|
|
// ref: https://github.com/StackExchange/Dapper/issues/229
|
|
|
|
|
parameters.Add("@StartDate", startDate.ToUniversalTime(), DbType.DateTime2, null, 7);
|
|
|
|
|
parameters.Add("@EndDate", endDate.ToUniversalTime(), DbType.DateTime2, null, 7);
|
|
|
|
|
parameters.Add("@BeforeDate", beforeDate, DbType.DateTime2, null, 7);
|
|
|
|
|
|
2020-03-27 19:36:37 +01:00
|
|
|
|
using (var connection = new SqlConnection(ConnectionString))
|
2017-12-19 05:15:16 +01:00
|
|
|
|
{
|
|
|
|
|
var events = (await connection.QueryAsync<Event>(sprocName, parameters,
|
|
|
|
|
commandType: CommandType.StoredProcedure)).ToList();
|
|
|
|
|
|
|
|
|
|
var result = new PagedResult<IEvent>();
|
2020-03-27 19:36:37 +01:00
|
|
|
|
if (events.Any() && events.Count >= pageOptions.PageSize)
|
2017-12-19 05:15:16 +01:00
|
|
|
|
{
|
|
|
|
|
result.ContinuationToken = events.Last().Date.ToBinary().ToString();
|
|
|
|
|
}
|
|
|
|
|
result.Data.AddRange(events);
|
|
|
|
|
return result;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2020-02-28 04:44:49 +01:00
|
|
|
|
private DataTable BuildEventsTable(SqlBulkCopy bulkCopy, IEnumerable<Event> events)
|
2017-12-12 20:22:22 +01:00
|
|
|
|
{
|
|
|
|
|
var e = events.FirstOrDefault();
|
2020-03-27 19:36:37 +01:00
|
|
|
|
if (e == null)
|
2017-12-12 20:22:22 +01:00
|
|
|
|
{
|
|
|
|
|
throw new ApplicationException("Must have some events to bulk import.");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
var eventsTable = new DataTable("EventDataTable");
|
|
|
|
|
|
|
|
|
|
var idColumn = new DataColumn(nameof(e.Id), e.Id.GetType());
|
|
|
|
|
eventsTable.Columns.Add(idColumn);
|
|
|
|
|
var typeColumn = new DataColumn(nameof(e.Type), typeof(int));
|
|
|
|
|
eventsTable.Columns.Add(typeColumn);
|
|
|
|
|
var userIdColumn = new DataColumn(nameof(e.UserId), typeof(Guid));
|
|
|
|
|
eventsTable.Columns.Add(userIdColumn);
|
|
|
|
|
var organizationIdColumn = new DataColumn(nameof(e.OrganizationId), typeof(Guid));
|
|
|
|
|
eventsTable.Columns.Add(organizationIdColumn);
|
|
|
|
|
var cipherIdColumn = new DataColumn(nameof(e.CipherId), typeof(Guid));
|
|
|
|
|
eventsTable.Columns.Add(cipherIdColumn);
|
|
|
|
|
var collectionIdColumn = new DataColumn(nameof(e.CollectionId), typeof(Guid));
|
|
|
|
|
eventsTable.Columns.Add(collectionIdColumn);
|
2020-01-15 15:43:49 +01:00
|
|
|
|
var policyIdColumn = new DataColumn(nameof(e.PolicyId), typeof(Guid));
|
|
|
|
|
eventsTable.Columns.Add(policyIdColumn);
|
2017-12-12 21:04:14 +01:00
|
|
|
|
var groupIdColumn = new DataColumn(nameof(e.GroupId), typeof(Guid));
|
|
|
|
|
eventsTable.Columns.Add(groupIdColumn);
|
2017-12-12 20:22:22 +01:00
|
|
|
|
var organizationUserIdColumn = new DataColumn(nameof(e.OrganizationUserId), typeof(Guid));
|
|
|
|
|
eventsTable.Columns.Add(organizationUserIdColumn);
|
2017-12-19 05:15:16 +01:00
|
|
|
|
var actingUserIdColumn = new DataColumn(nameof(e.ActingUserId), typeof(Guid));
|
|
|
|
|
eventsTable.Columns.Add(actingUserIdColumn);
|
|
|
|
|
var deviceTypeColumn = new DataColumn(nameof(e.DeviceType), typeof(int));
|
|
|
|
|
eventsTable.Columns.Add(deviceTypeColumn);
|
2021-06-24 21:31:03 +02:00
|
|
|
|
var ipAddressColumn = new DataColumn(nameof(e.IpAddress), typeof(string));
|
2017-12-19 05:15:16 +01:00
|
|
|
|
eventsTable.Columns.Add(ipAddressColumn);
|
2021-06-24 21:31:03 +02:00
|
|
|
|
var dateColumn = new DataColumn(nameof(e.Date), typeof(DateTime));
|
2017-12-12 21:04:14 +01:00
|
|
|
|
eventsTable.Columns.Add(dateColumn);
|
2017-12-12 20:22:22 +01:00
|
|
|
|
|
2020-03-27 19:36:37 +01:00
|
|
|
|
foreach (DataColumn col in eventsTable.Columns)
|
2020-02-28 04:44:49 +01:00
|
|
|
|
{
|
|
|
|
|
bulkCopy.ColumnMappings.Add(col.ColumnName, col.ColumnName);
|
|
|
|
|
}
|
|
|
|
|
|
2017-12-12 20:22:22 +01:00
|
|
|
|
var keys = new DataColumn[1];
|
|
|
|
|
keys[0] = idColumn;
|
|
|
|
|
eventsTable.PrimaryKey = keys;
|
|
|
|
|
|
2020-03-27 19:36:37 +01:00
|
|
|
|
foreach (var ev in events)
|
2017-12-12 20:22:22 +01:00
|
|
|
|
{
|
|
|
|
|
ev.SetNewId();
|
|
|
|
|
|
|
|
|
|
var row = eventsTable.NewRow();
|
|
|
|
|
|
|
|
|
|
row[idColumn] = ev.Id;
|
2017-12-12 21:04:14 +01:00
|
|
|
|
row[typeColumn] = (int)ev.Type;
|
|
|
|
|
row[userIdColumn] = ev.UserId.HasValue ? (object)ev.UserId.Value : DBNull.Value;
|
|
|
|
|
row[organizationIdColumn] = ev.OrganizationId.HasValue ? (object)ev.OrganizationId.Value : DBNull.Value;
|
|
|
|
|
row[cipherIdColumn] = ev.CipherId.HasValue ? (object)ev.CipherId.Value : DBNull.Value;
|
|
|
|
|
row[collectionIdColumn] = ev.CollectionId.HasValue ? (object)ev.CollectionId.Value : DBNull.Value;
|
2020-01-15 15:43:49 +01:00
|
|
|
|
row[policyIdColumn] = ev.PolicyId.HasValue ? (object)ev.PolicyId.Value : DBNull.Value;
|
2017-12-19 05:15:16 +01:00
|
|
|
|
row[groupIdColumn] = ev.GroupId.HasValue ? (object)ev.GroupId.Value : DBNull.Value;
|
2017-12-12 21:04:14 +01:00
|
|
|
|
row[organizationUserIdColumn] = ev.OrganizationUserId.HasValue ?
|
|
|
|
|
(object)ev.OrganizationUserId.Value : DBNull.Value;
|
2017-12-19 05:15:16 +01:00
|
|
|
|
row[actingUserIdColumn] = ev.ActingUserId.HasValue ? (object)ev.ActingUserId.Value : DBNull.Value;
|
|
|
|
|
row[deviceTypeColumn] = ev.DeviceType.HasValue ? (object)ev.DeviceType.Value : DBNull.Value;
|
|
|
|
|
row[ipAddressColumn] = ev.IpAddress != null ? (object)ev.IpAddress : DBNull.Value;
|
|
|
|
|
row[dateColumn] = ev.Date;
|
2017-12-12 20:22:22 +01:00
|
|
|
|
|
|
|
|
|
eventsTable.Rows.Add(row);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return eventsTable;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|