123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726 |
- using Admin.Core.Common.Attributes;
- using Admin.Core.Common.Auth;
- using Admin.Core.Common.BaseModel;
- using Admin.Core.Common.Configs;
- using Admin.Core.Common.Extensions;
- using Admin.Core.Common.Helpers;
- using Admin.Core.Model.Admin;
- using Admin.Core.Model.Personnel;
- using Admin.Core.Repository.Admin.Output;
- using Admin.Core.Repository.Admin.Permission.Output;
- using Admin.Core.Repository.Admin.View.Output;
- using Admin.Core.Repository.Personnel.Output;
- using FreeSql;
- using FreeSql.Aop;
- using FreeSql.DataAnnotations;
- using Newtonsoft.Json;
- using Newtonsoft.Json.Serialization;
- using System;
- using System.Collections.Generic;
- using System.IO;
- using System.Linq;
- using System.Reflection;
- using System.Threading.Tasks;
- using Yitter.IdGenerator;
- namespace Admin.Core.Repository
- {
- public class DbHelper
- {
- /// <summary>
- /// 偏移时间
- /// </summary>
- public static TimeSpan TimeOffset;
- /// <summary>
- /// 创建数据库
- /// </summary>
- /// <param name="dbConfig"></param>
- /// <returns></returns>
- public async static Task CreateDatabaseAsync(DbConfig dbConfig)
- {
- if (!dbConfig.CreateDb || dbConfig.Type == DataType.Sqlite)
- {
- return;
- }
- var db = new FreeSqlBuilder()
- .UseConnectionString(dbConfig.Type, dbConfig.CreateDbConnectionString)
- .Build();
- try
- {
- Console.WriteLine("\r\n create database started");
- await db.Ado.ExecuteNonQueryAsync(dbConfig.CreateDbSql);
- Console.WriteLine(" create database succeed");
- }
- catch (Exception e)
- {
- Console.WriteLine($" create database failed.\n {e.Message}");
- }
- }
- /// <summary>
- /// 获得指定程序集表实体
- /// </summary>
- /// <returns></returns>
- public static Type[] GetEntityTypes()
- {
- List<string> assemblyNames = new List<string>()
- {
- "Admin.Core.Model"
- };
- List<Type> entityTypes = new List<Type>();
- foreach (var assemblyName in assemblyNames)
- {
- foreach (Type type in Assembly.Load(assemblyName).GetExportedTypes())
- {
- foreach (Attribute attribute in type.GetCustomAttributes())
- {
- if (attribute is TableAttribute tableAttribute)
- {
- if (tableAttribute.DisableSyncStructure == false)
- {
- entityTypes.Add(type);
- }
- }
- }
- }
- }
- return entityTypes.ToArray();
- }
- /// <summary>
- /// 配置实体
- /// </summary>
- public static void ConfigEntity(IFreeSql db, AppConfig appConfig = null)
- {
- //租户生成和操作租户Id
- if (!appConfig.Tenant)
- {
- var iTenant = nameof(ITenant);
- var tenantId = nameof(ITenant.TenantId);
- //获得指定程序集表实体
- var entityTypes = GetEntityTypes();
- foreach (var entityType in entityTypes)
- {
- if (entityType.GetInterfaces().Any(a => a.Name == iTenant))
- {
- db.CodeFirst.Entity(entityType, a =>
- {
- a.Ignore(tenantId);
- });
- }
- }
- }
- }
- /// <summary>
- /// 审计数据
- /// </summary>
- /// <param name="e"></param>
- /// <param name="timeOffset"></param>
- /// <param name="user"></param>
- public static void AuditValue(AuditValueEventArgs e, TimeSpan timeOffset, IUser user)
- {
- if (e.Property.GetCustomAttribute<ServerTimeAttribute>(false) != null
- && (e.Column.CsType == typeof(DateTime) || e.Column.CsType == typeof(DateTime?))
- && (e.Value == null || (DateTime)e.Value == default || (DateTime?)e.Value == default))
- {
- e.Value = DateTime.Now.Subtract(timeOffset);
- }
- if (e.Column.CsType == typeof(long)
- && e.Property.GetCustomAttribute<SnowflakeAttribute>(false) is SnowflakeAttribute snowflakeAttribute
- && snowflakeAttribute.Enable && (e.Value == null || (long)e.Value == default || (long?)e.Value == default))
- {
- e.Value = YitIdHelper.NextId();
- }
- if (user == null || user.Id <= 0)
- {
- return;
- }
- if (e.AuditValueType == AuditValueType.Insert)
- {
- switch (e.Property.Name)
- {
- case "CreatedUserId":
- if (e.Value == null || (long)e.Value == default || (long?)e.Value == default)
- {
- e.Value = user.Id;
- }
- break;
- case "CreatedUserName":
- if (e.Value == null || ((string)e.Value).IsNull())
- {
- e.Value = user.Name;
- }
- break;
- case "TenantId":
- if (e.Value == null || (long)e.Value == default || (long?)e.Value == default)
- {
- e.Value = user.TenantId;
- }
- break;
- }
- }
- else if (e.AuditValueType == AuditValueType.Update)
- {
- switch (e.Property.Name)
- {
- case "ModifiedUserId":
- e.Value = user.Id;
- break;
- case "ModifiedUserName":
- e.Value = user.Name;
- break;
- }
- }
- }
- /// <summary>
- /// 同步结构
- /// </summary>
- public static void SyncStructure(IFreeSql db, string msg = null, DbConfig dbConfig = null, AppConfig appConfig = null)
- {
- //打印结构比对脚本
- //var dDL = db.CodeFirst.GetComparisonDDLStatements<PermissionEntity>();
- //Console.WriteLine("\r\n " + dDL);
- //打印结构同步脚本
- //db.Aop.SyncStructureAfter += (s, e) =>
- //{
- // if (e.Sql.NotNull())
- // {
- // Console.WriteLine(" sync structure sql:\n" + e.Sql);
- // }
- //};
- // 同步结构
- var dbType = dbConfig.Type.ToString();
- Console.WriteLine($"\r\n {(msg.NotNull() ? msg : $"sync {dbType} structure")} started");
- if (dbConfig.Type == DataType.Oracle)
- {
- db.CodeFirst.IsSyncStructureToUpper = true;
- }
- //获得指定程序集表实体
- var entityTypes = GetEntityTypes();
- db.CodeFirst.SyncStructure(entityTypes);
- Console.WriteLine($" {(msg.NotNull() ? msg : $"sync {dbType} structure")} succeed");
- }
- /// <summary>
- /// 检查实体属性是否为自增长
- /// </summary>
- /// <typeparam name="T"></typeparam>
- /// <returns></returns>
- private static bool CheckIdentity<T>() where T : class
- {
- var isIdentity = false;
- var properties = typeof(T).GetProperties();
- foreach (var property in properties)
- {
- if (property.GetCustomAttributes(typeof(ColumnAttribute), false).FirstOrDefault() is ColumnAttribute columnAttribute && columnAttribute.IsIdentity)
- {
- isIdentity = true;
- break;
- }
- }
- return isIdentity;
- }
- /// <summary>
- /// 初始化数据表数据
- /// </summary>
- /// <typeparam name="T"></typeparam>
- /// <param name="db"></param>
- /// <param name="unitOfWork"></param>
- /// <param name="tran"></param>
- /// <param name="data"></param>
- /// <param name="dbConfig"></param>
- /// <returns></returns>
- private static async Task InitDtDataAsync<T>(
- IFreeSql db,
- IUnitOfWork unitOfWork,
- System.Data.Common.DbTransaction tran,
- T[] data,
- DbConfig dbConfig = null
- ) where T : class, new()
- {
- var table = typeof(T).GetCustomAttributes(typeof(TableAttribute), false).FirstOrDefault() as TableAttribute;
- var tableName = table.Name;
- try
- {
- if (await db.Queryable<T>().AnyAsync())
- {
- Console.WriteLine($" table: {tableName} record already exists");
- return;
- }
- if (!(data?.Length > 0))
- {
- Console.WriteLine($" table: {tableName} import data []");
- return;
- }
- var repo = db.GetRepositoryBase<T>();
- var insert = db.Insert<T>();
- if (unitOfWork != null)
- {
- repo.UnitOfWork = unitOfWork;
- insert = insert.WithTransaction(tran);
- }
- var isIdentity = CheckIdentity<T>();
- if (isIdentity)
- {
- if (dbConfig.Type == DataType.SqlServer)
- {
- var insrtSql = insert.AppendData(data).InsertIdentity().ToSql();
- await repo.Orm.Ado.ExecuteNonQueryAsync($"SET IDENTITY_INSERT {tableName} ON\n {insrtSql} \nSET IDENTITY_INSERT {tableName} OFF");
- }
- else
- {
- await insert.AppendData(data).InsertIdentity().ExecuteAffrowsAsync();
- }
- }
- else
- {
- repo.DbContextOptions.EnableAddOrUpdateNavigateList = true;
- await repo.InsertAsync(data);
- }
- Console.WriteLine($" table: {tableName} sync data succeed");
- }
- catch (Exception ex)
- {
- Console.WriteLine($" table: {tableName} sync data failed.\n{ex.Message}");
- }
- }
- /// <summary>
- /// 同步数据审计方法
- /// </summary>
- /// <param name="s"></param>
- /// <param name="e"></param>
- private static void SyncDataAuditValue(object s, AuditValueEventArgs e)
- {
- var user = new { Id = 161223411986501, Name = "admin", TenantId = 161223412138053 };
- if (e.Property.GetCustomAttribute<ServerTimeAttribute>(false) != null
- && (e.Column.CsType == typeof(DateTime) || e.Column.CsType == typeof(DateTime?))
- && (e.Value == null || (DateTime)e.Value == default || (DateTime?)e.Value == default))
- {
- e.Value = DateTime.Now.Subtract(TimeOffset);
- }
- if (e.Column.CsType == typeof(long)
- && e.Property.GetCustomAttribute<SnowflakeAttribute>(false) != null
- && (e.Value == null || (long)e.Value == default || (long?)e.Value == default))
- {
- e.Value = YitIdHelper.NextId();
- }
- if (user == null || user.Id <= 0)
- {
- return;
- }
- if (e.AuditValueType == AuditValueType.Insert)
- {
- switch (e.Property.Name)
- {
- case "CreatedUserId":
- if (e.Value == null || (long)e.Value == default || (long?)e.Value == default)
- {
- e.Value = user.Id;
- }
- break;
- case "CreatedUserName":
- if (e.Value == null || ((string)e.Value).IsNull())
- {
- e.Value = user.Name;
- }
- break;
- case "TenantId":
- if (e.Value == null || (long)e.Value == default || (long?)e.Value == default)
- {
- e.Value = user.TenantId;
- }
- break;
- }
- }
- else if (e.AuditValueType == AuditValueType.Update)
- {
- switch (e.Property.Name)
- {
- case "ModifiedUserId":
- e.Value = user.Id;
- break;
- case "ModifiedUserName":
- e.Value = user.Name;
- break;
- }
- }
- }
- /// <summary>
- /// 同步数据
- /// </summary>
- /// <returns></returns>
- public static async Task SyncDataAsync(IFreeSql db, DbConfig dbConfig = null, AppConfig appConfig = null)
- {
- try
- {
- //db.Aop.CurdBefore += (s, e) =>
- //{
- // Console.WriteLine($"{e.Sql}\r\n");
- //};
- Console.WriteLine("\r\n sync data started");
- db.Aop.AuditValue += SyncDataAuditValue;
- var fileName = appConfig.Tenant ? "data-share.json" : "data.json";
- var filePath = Path.Combine(AppContext.BaseDirectory, $"Db/Data/{fileName}").ToPath();
- var jsonData = FileHelper.ReadFile(filePath);
- var data = JsonConvert.DeserializeObject<Data>(jsonData);
- using (var uow = db.CreateUnitOfWork())
- using (var tran = uow.GetOrBeginTransaction())
- {
- var dualRepo = db.GetRepositoryBase<DualEntity>();
- dualRepo.UnitOfWork = uow;
- if (!await dualRepo.Select.AnyAsync())
- {
- await dualRepo.InsertAsync(new DualEntity { });
- }
- //admin
- await InitDtDataAsync(db, uow, tran, data.DictionaryTypes, dbConfig);
- await InitDtDataAsync(db, uow, tran, data.Dictionaries, dbConfig);
- await InitDtDataAsync(db, uow, tran, data.ApiTree, dbConfig);
- await InitDtDataAsync(db, uow, tran, data.ViewTree, dbConfig);
- await InitDtDataAsync(db, uow, tran, data.PermissionTree, dbConfig);
- await InitDtDataAsync(db, uow, tran, data.Users, dbConfig);
- await InitDtDataAsync(db, uow, tran, data.Roles, dbConfig);
- await InitDtDataAsync(db, uow, tran, data.UserRoles, dbConfig);
- await InitDtDataAsync(db, uow, tran, data.RolePermissions, dbConfig);
- await InitDtDataAsync(db, uow, tran, data.Tenants, dbConfig);
- await InitDtDataAsync(db, uow, tran, data.TenantPermissions, dbConfig);
- await InitDtDataAsync(db, uow, tran, data.PermissionApis, dbConfig);
- //人事
- await InitDtDataAsync(db, uow, tran, data.Positions, dbConfig);
- await InitDtDataAsync(db, uow, tran, data.OrganizationTree, dbConfig);
- await InitDtDataAsync(db, uow, tran, data.Employees, dbConfig);
- uow.Commit();
- }
- db.Aop.AuditValue -= SyncDataAuditValue;
- Console.WriteLine(" sync data succeed\r\n");
- }
- catch (Exception ex)
- {
- throw new Exception($" sync data failed.\n{ex.Message}");
- }
- }
- /// <summary>
- /// 生成极简数据
- /// </summary>
- /// <param name="db"></param>
- /// <param name="appConfig"></param>
- /// <returns></returns>
- public static async Task GenerateSimpleJsonDataAsync(IFreeSql db, AppConfig appConfig = null)
- {
- try
- {
- Console.WriteLine("\r\n generate data started");
- #region 数据表
- //admin
- #region 数据字典
-
- var dictionaryTypes = await db.Queryable<DictionaryTypeEntity>().ToListAsync<DictionaryTypeDataOutput>();
- var dictionaries = await db.Queryable<DictionaryEntity>().ToListAsync<DictionaryDataOutput>();
- #endregion
- #region 接口
- var apis = await db.Queryable<ApiEntity>().ToListAsync<ApiDataOutput>();
- var apiTree = apis.ToTree((r, c) =>
- {
- return c.ParentId == 0;
- },
- (r, c) =>
- {
- return r.Id == c.ParentId;
- },
- (r, datalist) =>
- {
- r.Childs ??= new List<ApiDataOutput>();
- r.Childs.AddRange(datalist);
- });
- #endregion
- #region 视图
- var views = await db.Queryable<ViewEntity>().ToListAsync<ViewDataOutput>();
- var viewTree = views.ToTree((r, c) =>
- {
- return c.ParentId == 0;
- },
- (r, c) =>
- {
- return r.Id == c.ParentId;
- },
- (r, datalist) =>
- {
- r.Childs ??= new List<ViewDataOutput>();
- r.Childs.AddRange(datalist);
- });
- #endregion
- #region 权限
- var permissions = await db.Queryable<PermissionEntity>().ToListAsync<PermissionDataOutput>();
- var permissionTree = permissions.ToTree((r, c) =>
- {
- return c.ParentId == 0;
- },
- (r, c) =>
- {
- return r.Id == c.ParentId;
- },
- (r, datalist) =>
- {
- r.Childs ??= new List<PermissionDataOutput>();
- r.Childs.AddRange(datalist);
- });
- #endregion
- #region 用户
- var users = await db.Queryable<UserEntity>().ToListAsync<UserDataOutput>();
- #endregion
- #region 角色
- var roles = await db.Queryable<RoleEntity>().ToListAsync<RoleDataOutput>();
- #endregion
- #region 用户角色
- var userRoles = await db.Queryable<UserRoleEntity>().ToListAsync(a => new
- {
- a.Id,
- a.UserId,
- a.RoleId
- });
- #endregion
- #region 角色权限
- var rolePermissions = await db.Queryable<RolePermissionEntity>().ToListAsync(a => new
- {
- a.Id,
- a.RoleId,
- a.PermissionId
- });
- #endregion
- #region 租户
- var tenants = await db.Queryable<TenantEntity>().ToListAsync(a => new
- {
- a.Id,
- a.UserId,
- a.RoleId,
- a.Name,
- a.Code,
- a.RealName,
- a.Phone,
- a.Email,
- a.TenantType,
- a.DataIsolationType,
- a.DbType,
- a.ConnectionString,
- a.IdleTime,
- a.Description
- });
- #endregion
- #region 租户权限
- var tenantPermissions = await db.Queryable<TenantPermissionEntity>().ToListAsync(a => new
- {
- a.Id,
- a.TenantId,
- a.PermissionId
- });
- #endregion
- #region 权限接口
- var permissionApis = await db.Queryable<PermissionApiEntity>().ToListAsync(a => new
- {
- a.Id,
- a.PermissionId,
- a.ApiId
- });
- #endregion
- //人事
- #region 部门
- var organizations = await db.Queryable<OrganizationEntity>().ToListAsync<OrganizationDataOutput>();
- var organizationTree = organizations.ToTree((r, c) =>
- {
- return c.ParentId == 0;
- },
- (r, c) =>
- {
- return r.Id == c.ParentId;
- },
- (r, datalist) =>
- {
- r.Childs ??= new List<OrganizationDataOutput>();
- r.Childs.AddRange(datalist);
- });
- #endregion
- #region 岗位
- var positions = await db.Queryable<PositionEntity>().ToListAsync<PositionDataOutput>();
- #endregion
- #region 员工
- var employees = await db.Queryable<EmployeeEntity>().ToListAsync<EmployeeDataOutput>();
- #endregion
- #endregion
- if (!(users?.Count > 0))
- {
- return;
- }
- #region 生成数据
- var settings = new JsonSerializerSettings();
- settings.ContractResolver = new CamelCasePropertyNamesContractResolver();
- settings.NullValueHandling = NullValueHandling.Ignore;
- settings.DefaultValueHandling = DefaultValueHandling.Ignore;
- var jsonData = JsonConvert.SerializeObject(new
- {
- dictionaries,
- dictionaryTypes,
- apis,
- apiTree,
- viewTree,
- permissionTree,
- users,
- roles,
- userRoles,
- rolePermissions,
- tenants,
- tenantPermissions,
- permissionApis,
- organizationTree,
- positions,
- employees
- },
- //Formatting.Indented,
- settings
- );
-
- var isTenant = appConfig.Tenant;
- var fileName = isTenant ? "data-share.json" : "data.json";
- var filePath = Path.Combine(Directory.GetCurrentDirectory(), $"Db/Data/{fileName}").ToPath();
- FileHelper.WriteFile(filePath, jsonData);
- if (isTenant)
- {
- var tenantId = tenants.Where(a => a.Code.ToLower() == "zhontai").FirstOrDefault().Id;
- organizationTree = organizations.Where(a => a.TenantId == tenantId).ToList().ToTree((r, c) =>
- {
- return c.ParentId == 0;
- },
- (r, c) =>
- {
- return r.Id == c.ParentId;
- },
- (r, datalist) =>
- {
- r.Childs ??= new List<OrganizationDataOutput>();
- r.Childs.AddRange(datalist);
- });
- jsonData = JsonConvert.SerializeObject(new
- {
- dictionaries = dictionaries.Where(a=>a.TenantId == tenantId),
- dictionaryTypes = dictionaryTypes.Where(a => a.TenantId == tenantId),
- apis,
- apiTree,
- viewTree,
- permissionTree,
- users = users.Where(a => a.TenantId == tenantId),
- roles = roles.Where(a => a.TenantId == tenantId),
- userRoles,
- rolePermissions,
- tenants,
- tenantPermissions,
- permissionApis,
- organizationTree
- },
- settings
- );
- filePath = Path.Combine(Directory.GetCurrentDirectory(), "Db/Data/data.json").ToPath();
- FileHelper.WriteFile(filePath, jsonData);
- }
- #endregion
- Console.WriteLine(" generate data succeed\r\n");
- }
- catch (Exception ex)
- {
- throw new Exception($" generate data failed。\n{ex.Message}\r\n");
- }
- }
- }
- }
|