Skip to content

Commit fa59754

Browse files
committed
Code cleanup
1 parent 0ad5472 commit fa59754

File tree

2 files changed

+17
-11
lines changed

2 files changed

+17
-11
lines changed

src/SpatialFocus.EntityFrameworkCore.Extensions/EnumLookupExtension.cs

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -64,10 +64,16 @@ public static void ConfigureEnumLookup(this ModelBuilder modelBuilder, EnumLooku
6464
string tableName = enumOptions.NamingFunction(typeName);
6565
enumLookupBuilder.ToTable(tableName);
6666

67-
string keyName = enumOptions.UseNumberLookup ? nameof(EnumWithNumberLookup<Enum>.Id)
67+
string keyName = enumOptions.UseNumberLookup
68+
? nameof(EnumWithNumberLookup<Enum>.Id)
6869
: nameof(EnumWithStringLookup<Enum>.Id);
6970

70-
modelBuilder.Entity(entityType.Name).HasOne(concreteType).WithMany().HasPrincipalKey(keyName).HasForeignKey(property.Name).OnDelete(enumOptions.DeleteBehavior);
71+
modelBuilder.Entity(entityType.Name)
72+
.HasOne(concreteType)
73+
.WithMany()
74+
.HasPrincipalKey(keyName)
75+
.HasForeignKey(property.Name)
76+
.OnDelete(enumOptions.DeleteBehavior);
7177

7278
if (enumOptions.UseNumberLookup)
7379
{

src/SpatialFocus.EntityFrameworkCore.Extensions/EnumLookupOptions.cs

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -31,19 +31,26 @@ public static EnumLookupOptions Default
3131

3232
internal Func<string, string> NamingFunction => name => this.postProcessingTableNamingFunction(this.namingFunction(name));
3333

34+
internal DeleteBehavior DeleteBehavior { get; private set; }
35+
3436
internal bool UseEnumsWithAttributesOnly { get; private set; }
3537

3638
internal bool UseNumberLookup { get; private set; }
3739

38-
internal DeleteBehavior DeleteBehavior { get; private set; }
39-
4040
public EnumLookupOptions Pluralize()
4141
{
4242
this.postProcessingTableNamingFunction = name => name.Pluralize(false);
4343

4444
return this;
4545
}
4646

47+
public EnumLookupOptions SetDeleteBehavior(DeleteBehavior deleteBehavior)
48+
{
49+
DeleteBehavior = deleteBehavior;
50+
51+
return this;
52+
}
53+
4754
public EnumLookupOptions SetNamingScheme(Func<string, string> namingFunc)
4855
{
4956
this.namingFunction = namingFunc;
@@ -78,12 +85,5 @@ public EnumLookupOptions UseStringAsIdentifier()
7885

7986
return this;
8087
}
81-
82-
public EnumLookupOptions SetDeleteBehavior(DeleteBehavior deleteBehavior)
83-
{
84-
DeleteBehavior = deleteBehavior;
85-
86-
return this;
87-
}
8888
}
8989
}

0 commit comments

Comments
 (0)