using System; using System.Linq; using System.Threading.Tasks; using System.Collections.Generic; using System.Collections.Immutable; using Microsoft.EntityFrameworkCore; using VNLib.Utils; using VNLib.Plugins.Extensions.Data.Abstractions; namespace VNLib.Plugins.Extensions.Data { public static class Extensions { public static int GetPageOrDefault(this IReadOnlyDictionary queryArgs, int @default, int minClamp = 0, int maxClamp = int.MaxValue) { return queryArgs.TryGetValue("page", out string? pageStr) && int.TryParse(pageStr, out int page) ? Math.Clamp(page, minClamp, maxClamp) : @default; } public static int GetLimitOrDefault(this IReadOnlyDictionary queryArgs, int @default, int minClamp = 0, int maxClamp = int.MaxValue) { return queryArgs.TryGetValue("limit", out string? limitStr) && int.TryParse(limitStr, out int limit) ? Math.Clamp(limit, minClamp, maxClamp) : @default; } public static async Task AddBulkAsync(this DbStore store, IEnumerable records, string userId, bool overwriteTime = true) where TEntity : class, IDbModel, IUserEntity { //Open context and transaction await using TransactionalDbContext database = store.NewContext(); await database.OpenTransactionAsync(); //Get the entity set DbSet set = database.Set(); //Generate random ids for the feeds and set user-id foreach (TEntity entity in records) { entity.Id = store.RecordIdBuilder; //Explicitly assign the user-id entity.UserId = userId; //If the entity has the default created time, update it, otherwise leave it as is if (overwriteTime || entity.Created == default) { entity.Created = DateTime.UtcNow; } //Update last-modified time entity.LastModified = DateTime.UtcNow; } //Add feeds to database set.AddRange(records); //Commit changes ERRNO count = await database.SaveChangesAsync(); //Commit transaction and exit await database.CommitTransactionAsync(); return count; } } }