RHDWContext.cs 4.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128
  1. using System;
  2. using System.Collections.Generic;
  3. using System.Data.Entity;
  4. using System.Linq;
  5. using System.Text;
  6. using System.Threading.Tasks;
  7. using SQLite.CodeFirst;
  8. using System.Data.Entity.ModelConfiguration.Conventions;
  9. using System.Data.Entity.Infrastructure.Interception;
  10. using XdCxRhDW.Repostory.Model;
  11. using System.IO;
  12. using System.Data.Common;
  13. using System.Data.Entity.Infrastructure;
  14. using System.Reflection;
  15. using System.Data.SQLite.EF6;
  16. using System.Data.SQLite;
  17. using System.Data.Entity.Core.Common;
  18. using System.Data.Entity.Core.EntityClient;
  19. using static System.Net.WebRequestMethods;
  20. namespace XdCxRhDW.Repostory.EFContext
  21. {
  22. /// <summary>
  23. /// 基础表上下文(id为int)
  24. /// </summary>
  25. public class RHDWContext: DbContext
  26. {
  27. public RHDWContext() : base("DbCon") //配置使用的连接名
  28. {
  29. }
  30. protected override void OnModelCreating(DbModelBuilder modelBuilder)
  31. {
  32. this.Database.Log = msg =>
  33. {
  34. };
  35. modelBuilder.Conventions.Remove<PluralizingTableNameConvention>();
  36. modelBuilder.Configurations.AddFromAssembly(typeof(RHDWContext).Assembly);//自动加载SqliteConfiguration
  37. var sqliteConnectionInitializer = new SqliteCreateDatabaseIfNotExists<RHDWContext>(modelBuilder);
  38. Database.SetInitializer(sqliteConnectionInitializer);
  39. DbInterception.Add(new SqliteInterceptor());//拦截器
  40. base.OnModelCreating(modelBuilder);
  41. }
  42. public DbSet<TaskInfo> TaskInfos { set; get; }
  43. public DbSet<TxInfo> TxInfos { get; set; }
  44. public DbSet<SatInfo> SatInfos { get; set; }
  45. public DbSet<TargetInfo> TargetInfos { get; set; }
  46. //public DbSet<StationRes> StationRes { get; set; }
  47. //public DbSet<CxRes> CxRes { get; set; }
  48. //public DbSet<CgRes> CgRes { get; set; }
  49. //public DbSet<PosRes> PosRes { get; set; }
  50. public DbSet<SysSetings> SysSetings { get; set; }
  51. }
  52. /// <summary>
  53. /// 分区表上下文(id为long)
  54. /// </summary>
  55. //[DbConfigurationType(typeof(SqliteConfiguration))]
  56. public class RHDWPartContext : DbContext
  57. {
  58. public static RHDWPartContext GetContext(string dbFile)
  59. {
  60. var connectionString = $@"Data Source={dbFile}";
  61. SQLiteConnection con = new SQLiteConnection(connectionString);
  62. return new RHDWPartContext(con);
  63. }
  64. public static RHDWPartContext GetContext(DateTime partTime)
  65. {
  66. var connectionString = $@"Data Source=|DataDirectory|\DbPart\{partTime.Year}\{partTime:MMdd}.db";
  67. SQLiteConnection con = new SQLiteConnection(connectionString);
  68. return new RHDWPartContext(con);
  69. }
  70. private RHDWPartContext(DbConnection con)
  71. :base(con,true)
  72. {
  73. }
  74. protected override void OnModelCreating(DbModelBuilder modelBuilder)
  75. {
  76. this.Database.Log = msg =>
  77. {
  78. };
  79. modelBuilder.Conventions.Remove<PluralizingTableNameConvention>();
  80. modelBuilder.Configurations.AddFromAssembly(typeof(RHDWPartContext).Assembly);
  81. var sqliteConnectionInitializer = new SqliteCreateDatabaseIfNotExists<RHDWPartContext>(modelBuilder);
  82. Database.SetInitializer(sqliteConnectionInitializer);
  83. DbInterception.Add(new SqliteInterceptor());//拦截器
  84. base.OnModelCreating(modelBuilder);
  85. }
  86. public DbSet<XlInfo> XlInfos { set; get; }
  87. public DbSet<StationRes> StationRes { get; set; }
  88. public DbSet<CxRes> CxRes { get; set; }
  89. public DbSet<CgRes> CgRes { get; set; }
  90. public DbSet<CgXgfRes> CgXgfRes { get; set; }
  91. public DbSet<PosRes> PosRes { get; set; }
  92. }
  93. public class SqliteConfiguration : DbConfiguration
  94. {
  95. public SqliteConfiguration()
  96. {
  97. SetProviderFactory("System.Data.SQLite", SQLiteFactory.Instance);
  98. SetProviderFactory("System.Data.SQLite.EF6", SQLiteProviderFactory.Instance);
  99. SetProviderServices("System.Data.SQLite", (DbProviderServices)SQLiteProviderFactory.Instance.GetService(typeof(DbProviderServices)));
  100. }
  101. }
  102. }