How to get Column name and corresponding Database Type from DbContext in Entity Framework Core

前端 未结 3 1317
臣服心动
臣服心动 2020-12-01 21:10

Suppose I have this table:

How can I get the column name and database datatype from DbContext in Entity Framework Core?

Tips

相关标签:
3条回答
  • 2020-12-01 21:38

    Based on EFC3.1 answer I have created this helper to store all table names and column names into an dictionaries of a singleton populated on first use so the code doesn't have to traverse everything again and again. We use it for NPGSQL bulk copy operations and it seems to work properly. This version does not filter out any properties from entity classes so be careful about ordering of fields when doing column name lists/strings. But then again as I understand it, you will get only properties that are mapped in context so everything might be ok.

    The helper

    public class ContextHelper
      {
        private readonly ILogger<ContextHelper> logger;
        private readonly ApplicationDbContext context;
    
        private static Dictionary<Type, string> tableNames = new Dictionary<Type, string>(30);
    
        private Dictionary<Type, Dictionary<string, string>> columnNames = new Dictionary<Type, Dictionary<string, string>>(30);
    
        public ContextHelper(ILogger<ContextHelper> logger, ApplicationDbContext context)
        {
          this.logger = logger;
          this.context = context;
    
          PopulateTableNames();
          PopulateColumnNames();
        }
    
        private void PopulateTableNames()
        {
          logger.LogInformation("Populating table names in context helper");
    
          foreach (var entityType in context.Model.GetEntityTypes())
          {
            tableNames.Add(entityType.ClrType, entityType.GetTableName());
          }
        }
    
        private void PopulateColumnNames()
        {
          logger.LogInformation("Populating column names in context helper");
    
          foreach (var entityType in context.Model.GetEntityTypes())
          {
            var clrType = entityType.ClrType;
    
            if (!columnNames.ContainsKey(clrType))
            {
              columnNames.Add(clrType, new Dictionary<string, string>(30));
            }
    
            foreach (var property in entityType.GetProperties())
            {
              columnNames[clrType].Add(property.Name, property.GetColumnName());
            }
          }
        }
    
        public string GetTableName<T>()
        {
          return context.Model.FindEntityType(typeof(T)).GetTableName();
        }
    
        public string GetColumnName<T>(string propertyName)
        {
          return columnNames[typeof(T)][propertyName];
        }
    
        public List<string> GetColumnNames<T>()
        {
          return columnNames[typeof(T)].Select(x => x.Value).ToList();
        }
      }
    

    Startup registration

    services.AddSingleton<ContextHelper>();
    

    Usage, something along these lines

    var columnNames = contextHelper.GetColumnNames<OvenEventLog>().Where(x=>x != contextHelper.GetColumnName<OvenEventLog>(nameof(OvenEventLog.IdLog)));
    var separatedCN = string.Join(", ", columnNames);
    
    using (var writer = conn.BeginBinaryImport(
        $"COPY {contextHelper.GetTableName<OvenEventLog>()} ({separatedCN}) FROM STDIN (FORMAT BINARY)")
    
    
    0 讨论(0)
  • 2020-12-01 21:43

    Update (EF Core 3.x): Starting with EF Core 3.0, the metadata API has changed again - Relational() extensions have been removed, and properties have been replaced with Get and Set extension methods, so now the code looks like this:

    var entityType = dbContext.Model.FindEntityType(clrEntityType);
    
    // Table info 
    var tableName = entityType.GetTableName();
    var tableSchema = entityType.GetSchema();
    
    // Column info 
    foreach (var property in entityType.GetProperties())
    {
        var columnName = property.GetColumnName();
        var columnType = property.GetColumnType();
    };
    

    Update (EF Core 2.x): Starting with EF Core 2.0, the things have changed, so the original answer does not apply anymore. Now EF Core builds separate model for each database type, so the code is much simpler and uses directly the Relational() extensions:

    var entityType = dbContext.Model.FindEntityType(clrEntityType);
    
    // Table info 
    var tableName = entityType.Relational().TableName;
    var tableSchema = entityType.Relational().Schema;
    
    // Column info 
    foreach (var property in entityType.GetProperties())
    {
        var columnName = property.Relational().ColumnName;
        var columnType = property.Relational().ColumnType;
    };
    

    Original answer (EF Core 1.x):

    Getting the access to the associated metadata is much easier in EF Core compared to EF - you start from DbContext.Model property to get IModel, use GetEntityTypes or FindEntityType to get IEntityType, then GetProperties or FindProperty to get IProperty etc.

    However the problem is that EF Core allows you to use different setting fro different target database. In order to get the attributes corresponding to the current database used by the context, you need to get access to the IRelationalDatabaseProviderServices and use AnnotationProvider and TypeMapper properties to get the information needed.

    Here is an example:

    using System;
    using System.Collections.Generic;
    using System.Linq;
    using Microsoft.EntityFrameworkCore;
    using Microsoft.EntityFrameworkCore.Infrastructure;
    using Microsoft.EntityFrameworkCore.Internal;
    using Microsoft.EntityFrameworkCore.Storage;
    
    public class DbColumnInfo
    {
        public string Name;
        public string Type;
    }
    
    public static class RelationalDbHelpers
    {
        public static IEnumerable<DbColumnInfo> GetDbColums(this DbContext dbContext, Type clrEntityType)
        {
            var dbServices = dbContext.GetService<IDbContextServices>();
            var relationalDbServices = dbServices.DatabaseProviderServices as IRelationalDatabaseProviderServices;
            var annotationProvider = relationalDbServices.AnnotationProvider;
            var typeMapper = relationalDbServices.TypeMapper;
    
            var entityType = dbContext.Model.FindEntityType(clrEntityType);
    
            // Not needed here, just an example 
            var tableMap = annotationProvider.For(entityType);
            var tableName = tableMap.TableName;
            var tableSchema = tableMap.Schema;
    
            return from property in entityType.GetProperties()
                   let columnMap = annotationProvider.For(property)
                   let columnTypeMap = typeMapper.FindMapping(property)
                   select new DbColumnInfo
                   {
                       Name = columnMap.ColumnName,
                       Type = columnTypeMap.StoreType
                   };
        }
    }
    
    0 讨论(0)
  • 2020-12-01 22:00

    For those who arrived here, but are not using .NET CORE, like me. try this:

    public partial class MyDbContext : System.Data.Entity.DbContext
    {
        public string GetTableName(Type entityType)
        {
            var sql = Set(entityType).ToString();
            var regex = new Regex(@"FROM \[dbo\]\.\[(?<table>.*)\] AS");
            var match = regex.Match(sql);
            return match.Groups["table"].Value;
        }
    
        public string[] GetColumnName(Type entityType)
        {
            var strs = new List<string>();
            var sql = Set(entityType).ToString();
            var regex = new Regex(@"\[Extent1\]\.\[(?<columnName>.*)\] AS");
            var matches = regex.Matches(sql);
            foreach (Match item in matches)
            {
                var name = item.Groups["columnName"].Value;
                strs.Add(name);
            }
            return strs.ToArray();
        }
    }
    

    maybe redundante, but it saves time.

    antonio

    0 讨论(0)
提交回复
热议问题