src/DevHive.Data/DevHiveContext.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
47
48

using System;
using DevHive.Data.Models;
using Microsoft.AspNetCore.Identity.EntityFrameworkCore;
using Microsoft.EntityFrameworkCore;

namespace DevHive.Data
{
	public class DevHiveContext : IdentityDbContext<User, Role, Guid>
	{
		public DevHiveContext(DbContextOptions<DevHiveContext> options)
			: base(options) { }

		public DbSet<Technology> Technologies { get; set; }
		public DbSet<Language> Languages { get; set; }
		public DbSet<Post> Posts { get; set; }
		public DbSet<Comment> Comments { get; set; }

		protected override void OnModelCreating(ModelBuilder builder)
		{
			builder.Entity<User>()
				.HasIndex(x => x.UserName)
				.IsUnique();

			builder.Entity<User>()
				.HasMany(x => x.Roles);

			builder.Entity<User>()
				.HasMany(x => x.Friends);

			builder.Entity<User>()
				.HasMany(x => x.Languages);

			builder.Entity<Language>()
				.HasMany(x => x.Users);

			builder.Entity<User>()
				.HasMany(x => x.Technologies);

			builder.Entity<Technology>()
				.HasMany(x => x.Users);

			builder.Entity<User>()
				.HasChangeTrackingStrategy(ChangeTrackingStrategy.Snapshot);

			base.OnModelCreating(builder);
		}
	}
}