Add ForNpgsqlUseXminAsConcurrencyToken extension

Closes #19
npgsql · Oct 22, 2016 · 06d34b3 · 06d34b3
1 parent 0739c3f
commit 06d34b3
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 23 deletions.
diff --git a/src/Npgsql.EntityFrameworkCore.PostgreSQL/Extensions/NpgsqlEntityTypeBuilderExtensions.cs b/src/Npgsql.EntityFrameworkCore.PostgreSQL/Extensions/NpgsqlEntityTypeBuilderExtensions.cs
@@ -71,5 +71,23 @@ public static EntityTypeBuilder<TEntity> ForNpgsqlToTable<TEntity>(
             [CanBeNull] string schema)
             where TEntity : class
             => (EntityTypeBuilder<TEntity>)ForNpgsqlToTable((EntityTypeBuilder)entityTypeBuilder, name, schema);
+
+        public static EntityTypeBuilder ForNpgsqlUseXminAsConcurrencyToken(
+            [NotNull] this EntityTypeBuilder entityTypeBuilder)
+        {
+            Check.NotNull(entityTypeBuilder, nameof(entityTypeBuilder));
+
+            entityTypeBuilder.Property<uint>("xmin")
+                .HasColumnType("xid")
+                .ValueGeneratedOnAddOrUpdate()
+                .IsConcurrencyToken();
+
+            return entityTypeBuilder;
+        }
+
+        public static EntityTypeBuilder<TEntity> ForNpgsqlUseXminAsConcurrencyToken<TEntity>(
+            [NotNull] this EntityTypeBuilder<TEntity> entityTypeBuilder)
+            where TEntity : class
+            => (EntityTypeBuilder<TEntity>)ForNpgsqlUseXminAsConcurrencyToken((EntityTypeBuilder)entityTypeBuilder);
     }
 }
diff --git a/test/Npgsql.EntityFrameworkCore.PostgreSQL.FunctionalTests/F1NpgsqlFixture.cs b/test/Npgsql.EntityFrameworkCore.PostgreSQL.FunctionalTests/F1NpgsqlFixture.cs
@@ -62,29 +62,9 @@ protected override void OnModelCreating(ModelBuilder modelBuilder)
         {
             base.OnModelCreating(modelBuilder);
 
-            modelBuilder.Entity<Chassis>(b =>
-            {
-                b.Property<uint>("xmin")
-                    .HasColumnType("xid")
-                    .ValueGeneratedOnAddOrUpdate()
-                    .IsConcurrencyToken();
-            });
-
-            modelBuilder.Entity<Driver>(b =>
-            {
-                b.Property<uint>("xmin")
-                    .HasColumnType("xid")
-                    .ValueGeneratedOnAddOrUpdate()
-                    .IsConcurrencyToken();
-            });
-
-            modelBuilder.Entity<Team>(b =>
-            {
-                b.Property<uint>("xmin")
-                    .HasColumnType("xid")
-                    .ValueGeneratedOnAddOrUpdate()
-                    .IsConcurrencyToken();
-            });
+            modelBuilder.Entity<Chassis>().ForNpgsqlUseXminAsConcurrencyToken();
+            modelBuilder.Entity<Driver>().ForNpgsqlUseXminAsConcurrencyToken();
+            modelBuilder.Entity<Team>().ForNpgsqlUseXminAsConcurrencyToken();
         }
     }
 }