YG-MW-WCS/Projects/永冠/WCS.Service/WCSDB.cs

using DBHelper;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Configuration;
using System.Collections.Generic;
using System.ComponentModel.DataAnnotations;
using System.Linq;
using WCS.Entity;
using WCS.Entity.Protocol;

namespace WCS.Service
{
    public class WCSDB : DbContext
    {
        public WCSDB()
        {
        }

        protected override void OnModelCreating(ModelBuilder modelBuilder)
        {
            base.OnModelCreating(modelBuilder);
        }

        protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)
        {
            optionsBuilder.UseSqlServer(AppSettings.Config.GetConnectionString("WCSDB"));
            //optionsBuilder.UseLoggerFactory(LoggerFactory.Create(builder =>
            //{
            //    builder.AddFilter((category, level) => category == DbLoggerCategory.Database.Command.Name && level == LogLevel.Information).AddConsole();
            //}));
        }

        public override int SaveChanges()
        {
            var entities = from e in ChangeTracker.Entries()
                           where e.State == EntityState.Added
                               || e.State == EntityState.Modified
                           select e.Entity;
            List<ValidationResult> list = new List<ValidationResult>();
            foreach (var entity in entities)
            {
                var validationContext = new ValidationContext(entity);
                try
                {
                    ////当前发生变更的实体是否是WCS_TASK
                    //if (entity.GetType() == typeof(WCS.Entity.WCS_TASK))
                    //{
                    //    var task = entity as WCS.Entity.WCS_TASK;
                    //    task.CreateStatusLog(this);
                    //}
                    Validator.ValidateObject(entity, validationContext);
                }
                catch (ValidationException ex)
                {
                    list.Add(ex.ValidationResult);
                }
            }
            if (list.Count > 0)
                throw new EntityValidationException { EntityValidationErrors = list };

            return base.SaveChanges();
        }

        public DbSet<WCS_PLC> WCS_PLC { get; set; }
        public DbSet<WCS_DATABLOCK> WCS_DATABLOCK { get; set; }
        public DbSet<WCS_DEVICE> WCS_DEVICE { get; set; }
        public DbSet<WCS_DEVICEPROTOCOL> WCS_DEVICEPROTOCOL { get; set; }
        public DbSet<WCS_EXCEPTION> WCS_EXCEPTION { get; set; }
        public DbSet<WCS_STATION520> WCS_STATION520 { get; set; }
        public DbSet<WCS_STATION521> WCS_STATION521 { get; set; }
        public DbSet<WCS_STATION523> WCS_STATION523 { get; set; }

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

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

        public DbSet<WCS_RGV520> WCS_RGV520 { get; set; }
        public DbSet<WCS_RGV521> WCS_RGV521 { get; set; }
        public DbSet<WCS_RGV523> WCS_RGV523 { get; set; }

        public DbSet<WCS_SC520> WCS_SC520 { get; set; }
        public DbSet<WCS_SC521> WCS_SC521 { get; set; }
        public DbSet<WCS_SC537> WCS_SC537 { get; set; }

        public DbSet<WCS_BCR80> WCS_BCR80 { get; set; }
        public DbSet<WCS_CMD> WCS_CMD { get; set; }

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

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

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

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

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

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

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

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

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

        public DbSet<WCS_DRGV520> WCS_DRGV520 { get; set; }
        public DbSet<WCS_DRGV521> WCS_DRGV521 { get; set; }
        public DbSet<WCS_DRGV523> WCS_DRGV523 { get; set; }
    }
}