-
Notifications
You must be signed in to change notification settings - Fork 1
/
Context.cs
46 lines (32 loc) · 1.2 KB
/
Context.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
using kuro_desserts.Models;
using Microsoft.EntityFrameworkCore;
namespace kuro_desserts.Data;
public class Context : DbContext
{
public DbSet<Address> Addresses { get; set; } = null!;
public DbSet<Cart> Carts { get; set; } = null!;
public DbSet<Dessert> Desserts { get; set; } = null!;
public DbSet<Flavor> Flavors { get; set; } = null!;
public DbSet<Order> Orders { get; set; } = null!;
public DbSet<OrderTopping> OrderToppings { get; set; } = null!;
public DbSet<Topping> Toppings { get; set; } = null!;
public DbSet<User> Users { get; set; } = null!;
public Context(DbContextOptions<Context> options) : base(options)
{
}
public override int SaveChanges()
{
var entries = ChangeTracker
.Entries()
.Where(e => e.Entity is Auditable && e.State is EntityState.Added or EntityState.Modified);
foreach (var entityEntry in entries)
{
((Auditable)entityEntry.Entity).UpdatedAt = DateTime.Now;
if (entityEntry.State == EntityState.Added)
{
((Auditable)entityEntry.Entity).CreatedAt = DateTime.Now;
}
}
return base.SaveChanges();
}
}