iot/projects/IoTNode/Data/IoTNodeDbContext.cs

using Infrastructure.Application.Entites.Settings;
using Infrastructure.Data;
using Infrastructure.Events;
using Infrastructure.Extensions;
using Infrastructure.Security;
using IoT.Shared;
using Application.Domain.Entities;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.Hosting;
using Microsoft.Extensions.Logging;

namespace IoTNode.Data
{
    public class IoTNodeDbContext : EfDbContext, IDbConfig
    {
        private readonly IEncryptionService _encryptionService;

        public IoTNodeDbContext(DbContextOptions options,
            ILogger<EfDbContext> logger,
            IHostEnvironment env,
            IConfiguration cfg,
            IEventPublisher publisher,
            IEncryptionService encryptionService)
            : base(options, logger, env, cfg, publisher)
        {
            this._encryptionService = encryptionService;
        }

        protected override void OnModelCreating(ModelBuilder modelBuilder)
        {
            IoTSharedDbConfig.OnModelCreating(modelBuilder);
            //Setting
            modelBuilder.Entity<Setting>().HasIndex(o => o.Name).IsUnique();
            //ignore
            modelBuilder.Ignore<Area>();
            modelBuilder.Ignore<Building>();
            modelBuilder.Ignore<Organ>();
            modelBuilder.Ignore<Department>();
            modelBuilder.Ignore<DepartmentUser>();
            modelBuilder.Ignore<Job>();
            modelBuilder.Ignore<Title>();
            modelBuilder.Ignore<OrganUser>();
            modelBuilder.Ignore<OrganRole>();
            modelBuilder.Ignore<OrganRolePermission>();
            modelBuilder.Ignore<OrganUserRole>();
            modelBuilder.Ignore<PermissionCategory>();
            modelBuilder.Ignore<IoTProductCategory>();
            modelBuilder.Ignore<IoTApi>();
            modelBuilder.Ignore<IoTParameter>();
            modelBuilder.Ignore<IoTCommand>();
            modelBuilder.Ignore<IoTSceneIoTCommand>();
            modelBuilder.Ignore<IoTScene>();
            modelBuilder.Ignore<IoTTimer>();
            modelBuilder.Ignore<IoTTigger>();
            modelBuilder.Entity<IoTProduct>().Ignore(o => o.DisplayOrder);
            modelBuilder.Entity<IoTApi>().Ignore(o => o.IoTCommands);
            modelBuilder.Entity<IoTGateway>().Ignore(o => o.Building);
            modelBuilder.Entity<IoTDevice>().Ignore(o => o.IoTCommands);
            modelBuilder.Ignore<User>();
            modelBuilder.Ignore<PermissionCategory>();
            modelBuilder.Ignore<Permission>();
            //
            modelBuilder.Ignore<PermissionCategory>();
            modelBuilder.Ignore<Permission>();
            modelBuilder.Ignore<User>();
            base.OnModelCreating(modelBuilder);
        }

        public void Seed(DbContext db)
        {
            var set = db.Set<Setting>();
            set.Add(new Setting { Name = "name", Value = this._cfg.GetValue("seed:NodeName", "网关"), Type = SettingType.String });
            set.Add(new Setting { Name = "logo", Value = "/images/logo.png", Type = SettingType.ImageUrl });
            set.Add(new Setting { Name = "copyright", Value = "Copyright © {0} Company. All rights reserved", Type = SettingType.Html });
            //
            var macAddress = Helper.Instance.GetMacAddress();
            set.Add(new Setting { Name = "sn", Value = macAddress, Type = SettingType.String });
            set.Add(new Setting
            {
                Name = "code",
                Value = this._env.IsDevelopment() ? Helper.Instance.MacEncrypt(Helper.Instance.GetMacAddress()) : "根据mac生成的授权码",
                Type = SettingType.String
            });
            //
            var host = this._cfg.GetValue("seed:host", "localhost");
            var stream = this._cfg.GetValue("seed:stream", "localhost");
            var delay = this._cfg.GetValue("seed:delay", "100");
            set.Add(new Setting { Name = "debug", Value = "false", Type = SettingType.String });
            set.Add(new Setting { Name = "notify:enabled", Value = "true", Type = SettingType.String });
            set.Add(new Setting { Name = "notify:host", Value = $"http://{host}/platform", Type = SettingType.String });
            set.Add(new Setting { Name = "timer.seconds", Value = "180", Type = SettingType.String });
            set.Add(new Setting { Name = "onvif.timer", Value = "1", Type = SettingType.String });
            set.Add(new Setting { Name = "onvif.speed", Value = "0.2", Type = SettingType.String });
            set.Add(new Setting { Name = "camera.usr", Value = "admin", Type = SettingType.String });
            set.Add(new Setting { Name = "camera.pwd", Value = "dsideal123", Type = SettingType.String });
            set.Add(new Setting { Name = "stream.rtmp", Value = stream, Type = SettingType.String });
            set.Add(new Setting { Name = "ffmpeg.args", Value = " -y -threads {0} -rtsp_transport tcp -use_wallclock_as_timestamps 1 -stimeout 3000000 -i \"{1}\" -fflags +genpts -c copy -f flv \"{2}\"", Type = SettingType.String });
            set.Add(new Setting { Name = "fbee.writelist", Value = "", Type = SettingType.String });
            set.Add(new Setting { Name = "camera.writelist", Value = "", Type = SettingType.String });
            set.Add(new Setting { Name = "delay", Value = delay, Type = SettingType.String });
            db.SaveChanges();
            db.Set<IoTGateway>().Add(new IoTGateway
            {
                Id = $"nodeid-{macAddress}".ToGuid(),
                Name = $"节点({macAddress})",
                Number = macAddress,
                IsOnline = true,
                Version = Helper.Instance.GetVersion(),
                Password = "123456",
                Image = "/images/classroom.svg"
            });
            db.SaveChanges();
        }
    }
}