Files
ThingsGateway/src/Admin/ThingsGateway.DB/SqlSugar/SeedDataUtil.cs
2025-06-06 21:13:58 +08:00

132 lines
4.5 KiB
C#
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

//------------------------------------------------------------------------------
// 此代码版权声明为全文件覆盖,如有原作者特别声明,会在下方手动补充
// 此代码版权除特别声明外的代码归作者本人Diego所有
// 源代码使用协议遵循本仓库的开源协议及附加协议
// Gitee源代码仓库https://gitee.com/diego2098/ThingsGateway
// Github源代码仓库https://github.com/kimdiego2098/ThingsGateway
// 使用文档https://thingsgateway.cn/
// QQ群605534569
//------------------------------------------------------------------------------
using BootstrapBlazor.Components;
using Newtonsoft.Json;
using Newtonsoft.Json.Linq;
using SqlSugar;
using System.Reflection;
using System.Text.RegularExpressions;
using ThingsGateway.NewLife;
namespace ThingsGateway.Admin.Application;
/// <summary>
/// 种子数据工具类
/// </summary>
[ThingsGateway.DependencyInjection.SuppressSniffer]
public static class SeedDataUtil
{
/// <summary>
/// 获取List列表
/// </summary>
/// <typeparam name="T"></typeparam>
/// <param name="jsonName"></param>
/// <returns></returns>
public static List<T> GetSeedData<T>(string jsonName)
{
var basePath = AppContext.BaseDirectory;//获取项目目录
jsonName = basePath.CombinePathWithOs(jsonName);//获取文件路径
var dataString = FileUtil.ReadFile(jsonName);//读取文件
return GetSeedDataByJson<T>(dataString);
}
public static string GetManifestResourceStream(Assembly assembly, string path)
{
var name = $"{assembly.GetName().Name}.{path}";
using var readStream = assembly.GetManifestResourceStream(name);
return readStream?.ToStr();
}
public static List<T> GetSeedDataByJson<T>(string json)
{
var seedData = new List<T>();//种子数据结果
if (!string.IsNullOrEmpty(json))//如果有内容
{
//字段没有数据的替换成null
json = Regex.Replace(json, "\\\"[^\"]+?\\\": \\\"\\\"", match => match.Value.Replace("\"\"", "null"));
var jtoken = JToken.Parse(json);
jtoken = jtoken.SelectToken("Records") ?? jtoken.SelectToken("RECORDS");
var type = typeof(T);
foreach (var objectType in type.GetRuntimeProperties())
{
var isjson = objectType.CustomAttributes.Any(a => a.NamedArguments.Any(b => b.MemberName == nameof(SugarColumn.IsJson) && b.TypedValue.Value?.ToBoolean() == true));
if (isjson)
{
foreach (var item in jtoken)
{
var value = item[objectType.Name];
item[objectType.Name] = value?.ToString()?.IsNullOrEmpty() != false ? null : JToken.Parse(value?.ToString() ?? string.Empty);
}
}
else if (objectType.PropertyType.IsBoolean())
{
foreach (var item in jtoken)
{
var value = item[objectType.Name];
item[objectType.Name] = value?.ToBoolean();
}
}
}
var seedDataRecord = (List<T>)Newtonsoft.Json.JsonConvert.DeserializeObject(jtoken.ToString(), typeof(List<T>), new JsonSerializerSettings
{
Formatting = Formatting.Indented,// 使用缩进格式化输出
NullValueHandling = NullValueHandling.Ignore, // 忽略空值属性
Converters = new List<JsonConverter> { new ZeroAsFalseConverter() }
});
//var seedDataRecord = jtoken.ToObject<List<T>>();
seedData = seedDataRecord ?? new();
}
return seedData;
}
}
/// <summary>
/// 种子数据格式实体类,遵循Navicat导出json格式
/// </summary>
/// <typeparam name="T"></typeparam>
public class SeedDataRecords<T>
{
/// <summary>
/// 数据
/// </summary>
public List<T> Records { get; set; }
}
internal sealed class ZeroAsFalseConverter : JsonConverter
{
public override bool CanConvert(Type objectType)
{
return objectType == typeof(bool);
}
public override object ReadJson(JsonReader reader, Type objectType, object? existingValue, JsonSerializer serializer)
{
var value = reader.Value?.ToString()?.ToBoolean();
return value;
}
public override void WriteJson(JsonWriter writer, object? value, JsonSerializer serializer)
{
writer.WriteValue(value?.ToString());
}
}