using SQLite.CodeFirst; using System; using System.Data.Common; using System.Data.Entity; using System.Data.Entity.Core.Common; using System.Data.Entity.Infrastructure.Interception; using System.Data.Entity.ModelConfiguration.Conventions; using System.Data.SQLite; using System.Data.SQLite.EF6; using System.IO; using System.Text.RegularExpressions; using XdCxRhDW.Entity; namespace XdCxRhDW.Repostory { /// /// 基础表上下文(id为int) /// public class RHDWContext : DbContext { public RHDWContext() : base("DbCon") //配置使用的连接名 { } protected override void OnModelCreating(DbModelBuilder modelBuilder) { this.Database.Log = msg => { }; modelBuilder.Conventions.Remove(); modelBuilder.Configurations.AddFromAssembly(typeof(RHDWContext).Assembly);//自动加载Entity-Type var sqliteConnectionInitializer = new SqliteCreateDatabaseIfNotExists(modelBuilder); Database.SetInitializer(sqliteConnectionInitializer); base.OnModelCreating(modelBuilder); } public DbSet XlInfos { set; get; } public DbSet TaskInfos { set; get; } public DbSet TaskSigs { set; get; } public DbSet TxInfos { get; set; } public DbSet SatInfos { get; set; } public DbSet SigInfos { get; set; } public DbSet SigDelays { get; set; } public DbSet TargetInfos { get; set; } public DbSet SysSetings { get; set; } } /// /// 分区表上下文(id为long) /// public class RHDWPartContext : DbContext { public static RHDWPartContext GetContext(string dbFile, bool createDb = false) { if (!File.Exists(dbFile) && !createDb) { return null; } var connectionString = $@"Data Source={dbFile}"; SQLiteConnection con = new SQLiteConnection(connectionString); return new RHDWPartContext(con); } public static RHDWPartContext GetContext(DateTime partTime, bool createDb = false, string prefix = "") { var dir = Path.Combine(AppDomain.CurrentDomain.BaseDirectory, "DbPart"); var dayFile = Path.Combine(dir, $@"{partTime.Year}\{prefix}{partTime:MMdd}.db"); if (!File.Exists(dayFile) && !createDb) { return null; } var connectionString = $@"Data Source=|DataDirectory|\DbPart\{partTime.Year}\{prefix}{partTime:MMdd}.db"; SQLiteConnection con = new SQLiteConnection(connectionString); return new RHDWPartContext(con); } private RHDWPartContext(DbConnection con) : base(con, true) { } protected override void OnModelCreating(DbModelBuilder modelBuilder) { this.Database.Log = msg => { }; modelBuilder.Conventions.Remove(); modelBuilder.Configurations.AddFromAssembly(typeof(RHDWPartContext).Assembly); var sqliteConnectionInitializer = new SqliteCreateDatabaseIfNotExists(modelBuilder); Database.SetInitializer(sqliteConnectionInitializer); base.OnModelCreating(modelBuilder); } public DbSet StationRes { get; set; } public DbSet CxRes { get; set; } public DbSet CgRes { get; set; } public DbSet CgXgfRes { get; set; } public DbSet PosRes { get; set; } public DbSet CheckRes { get; set; } } public class SqliteConfiguration : DbConfiguration { public SqliteConfiguration() { DbInterception.Add(new SqliteInterceptor());//拦截器 SetProviderFactory("System.Data.SQLite", SQLiteFactory.Instance); SetProviderFactory("System.Data.SQLite.EF6", SQLiteProviderFactory.Instance); SetProviderServices("System.Data.SQLite", (DbProviderServices)SQLiteProviderFactory.Instance.GetService(typeof(DbProviderServices))); } } }