DataSimulation/DataSimulation.Repostory/EFContext/SimulationContext.cs

using System;
using System.Collections.Generic;
using System.Data.Entity;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using SQLite.CodeFirst;
using System.Data.Entity.ModelConfiguration.Conventions;
using System.Data.Entity.Infrastructure.Interception;
using DataSimulation.Repostory.Model;
using System.IO;
using System.Data.Common;
using System.Data.Entity.Infrastructure;
using System.Reflection;
using System.Data.SQLite.EF6;
using System.Data.SQLite;
using System.Data.Entity.Core.Common;

namespace DataSimulation.Repostory.EFContext
{

    /// <summary>
    /// 基础表上下文(id为int)
    /// </summary>
    public class SimulationContext : DbContext
    {
        public SimulationContext() : base("DbCon") //配置使用的连接名
        {

        }

        protected override void OnModelCreating(DbModelBuilder modelBuilder)
        {
            this.Database.Log = msg =>
            {
            };
            modelBuilder.Conventions.Remove<PluralizingTableNameConvention>();
            modelBuilder.Configurations.AddFromAssembly(typeof(SimulationContext).Assembly);//自动加载SqliteConfiguration

            var sqliteConnectionInitializer = new SqliteCreateDatabaseIfNotExists<SimulationContext>(modelBuilder);
            Database.SetInitializer(sqliteConnectionInitializer);
            DbInterception.Add(new SqliteInterceptor());//拦截器
            base.OnModelCreating(modelBuilder);
        }

        public DbSet<SatInfo> SatInfos { get; set; }

        public DbSet<AntInfo> AntInfos { get; set; }

        public DbSet<RefInfo> RefInfos { get; set; }

        public DbSet<SysSetings> SysSetings { get; set; }


    }

    /// <summary>
    /// 分区表上下文(id为long)
    /// </summary>
    //[DbConfigurationType(typeof(SqliteConfiguration))]
    public class SimulationPartContext : DbContext
    {
        public static SimulationPartContext GetContext(string dbFile)
        {
            var connectionString = $@"Data Source={dbFile}";
            SQLiteConnection con = new SQLiteConnection(connectionString);
            return new SimulationPartContext(con);
        }

        public static SimulationPartContext GetContext(DateTime partTime)
        {
            var connectionString = $@"Data Source=|DataDirectory|\DbPart\{partTime.Year}\{partTime:MMdd}.db";
            SQLiteConnection con = new SQLiteConnection(connectionString);
            return new SimulationPartContext(con);
        }

        private SimulationPartContext(DbConnection con)
        : base(con, true)
        {
        }


        protected override void OnModelCreating(DbModelBuilder modelBuilder)
        {
            this.Database.Log = msg =>
            {
            };

            modelBuilder.Conventions.Remove<PluralizingTableNameConvention>();
            modelBuilder.Configurations.AddFromAssembly(typeof(SimulationPartContext).Assembly);

            var sqliteConnectionInitializer = new SqliteCreateDatabaseIfNotExists<SimulationPartContext>(modelBuilder);
            Database.SetInitializer(sqliteConnectionInitializer);
            DbInterception.Add(new SqliteInterceptor());//拦截器
            base.OnModelCreating(modelBuilder);
        }
        public DbSet<XlInfo> XlInfos { set; get; }

        public DbSet<SimulationInfo> SimulationInfos { get; set; }

        public DbSet<SimulationPonit> SimulationPonits { get; set; }

        public DbSet<TaskSimulation> TaskSimulations { get; set; }

        public DbSet<TaskInfo> TaskInfos { set; get; }
    }


    public class SqliteConfiguration : DbConfiguration
    {
        public SqliteConfiguration()
        {
            SetProviderFactory("System.Data.SQLite", SQLiteFactory.Instance);
            SetProviderFactory("System.Data.SQLite.EF6", SQLiteProviderFactory.Instance);
            SetProviderServices("System.Data.SQLite", (DbProviderServices)SQLiteProviderFactory.Instance.GetService(typeof(DbProviderServices)));

        }
    }
}