iFactory.Godot/Mods/工业数据采集与分析应用分享/Scripts/IDIS_Service.cs

using System;
using System.Collections.Generic;
using System.ComponentModel.DataAnnotations;
using System.Linq;
using System.Net.Mime;
using System.Reflection.Metadata;
using BITKit;
using Cysharp.Threading.Tasks;
using Godot;
using Microsoft.EntityFrameworkCore;
using Constant = BITKit.Constant;

namespace BITFactory;
// ReSharper disable once IdentifierTypo
[Serializable]
public abstract class IDIS_Base
{
	/// <summary>
	/// 创建时间
	/// </summary>
	public DateTime CreateTime { get; set; }=DateTime.Now;
	/// <summary>
	/// 更新时间
	/// </summary>
	public DateTime UpdateTime { get; set; }=DateTime.Now;
}
// ReSharper disable once IdentifierTypo
/// <summary>
/// 标识的值
/// </summary>
[Serializable]
public class IDIS_Value:IDIS_Base
{
	/// <summary>
	/// 标识码
	/// </summary>
	[Key]
	public string Handle { get; set; }
}
// ReSharper disable once IdentifierTypo
public class IDIS_Query : IDIS_Value
{
	
	// ReSharper disable once IdentifierTypo
	/// <summary>
	/// 关联的数据
	/// </summary>
	public IDIS_Data[] Datas { get; internal set; }
	// ReSharper disable once IdentifierTypo
	/// <summary>
	/// 关联的标识引用
	/// </summary>
	public IDIS_Reference[] References { get; internal set; }
}

/// <summary>
/// 标识的格式
/// </summary>
[Serializable]
// ReSharper disable once IdentifierTypo
public class IDIS_Data:IDIS_Base
{
	/// <summary>
	/// 不需要处理这个
	/// </summary>
	[Key]
	public int Index { get; set; }
	/// <summary>
	/// 标识码
	/// </summary>
	public string Handle { get; set; }
	/// <summary>
	/// 记录的名称,通常为中文
	/// </summary>
	public string Name { get; set; }
	/// <summary>
	/// 值类型,例如string,url,site
	/// </summary>
	public string Format { get; set; }
	/// <summary>
	/// 值
	/// </summary>
	public string Value { get; set; }
	/// <summary>
	/// 类型,用于解析时对数据条目进行分类
	/// </summary>
	public string Category { get; set; }
}
/// <summary>
/// 标识的引用或关联
/// </summary>
[Serializable]
// ReSharper disable once IdentifierTypo
public class IDIS_Reference:IDIS_Base
{
	/// <summary>
	/// 不需要处理这个
	/// </summary>
	[Key]
	public int Index{ get; set; }
	/// <summary>
	/// 标识码
	/// </summary>
	public string Handle { get; set; }
	/// <summary>
	/// 相关联的标识
	/// </summary>
	public string RelatedHandle { get; set; }
}
// ReSharper disable once IdentifierTypo
public class IDIS_DBContext:DbContext
{
	/// <summary>
	/// 标识表
	/// </summary>
	private DbSet<IDIS_Value> Values { get; set; }
	/// <summary>
	/// 标识数据表
	/// </summary>
	private DbSet<IDIS_Data> Datas{ get; set; }
	/// <summary>
	/// 标识引用表
	/// </summary>
	private DbSet<IDIS_Reference> References{ get; set; }
	/// <summary>
	/// 确保创建数据库
	/// </summary>
	/// <param name="optionsBuilder"></param>
	protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)
	{
		// ReSharper disable once StringLiteralTypo
		//var sql = SQLiteContextHelper.GetConnectionString("IDIS");
		SQLiteContextHelper.GetConnectionSqlAndPath("IDIS",out var sql,out var path);
		optionsBuilder.UseSqlite(sql);
		optionsBuilder.UseQueryTrackingBehavior(QueryTrackingBehavior.TrackAll);
		BIT4Log.Log<IDIS_DBContext>($"已创建标识数据库:{path}");
	}

	/// <summary>
	/// 查询多个标识
	/// </summary>
	/// <param name="key">模糊标识码</param>
	/// <param name="queries">查询列表</param>
	/// <returns>是否查询到了内容</returns>
	public bool Query(string key, out IDIS_Query[] queries)
	{
		ChangeTracker.DetectChanges();
		ChangeTracker.Clear();
		queries = Values
			.Where(x => x.Handle.Contains(key))
			.Select(x => new IDIS_Query()
			{
				Handle = x.Handle,
				CreateTime = x.CreateTime,
				UpdateTime = x.UpdateTime,
				Datas = Datas.Where(data => data.Handle == x.Handle).ToArray(),
				References = References.Where(reference => reference.Handle == x.Handle).ToArray()
			}).ToArray();
		return queries.Any();
	}

	/// <summary>
	/// 查询单个标识
	/// </summary>
	/// <param name="key">模糊标识</param>
	/// <param name="query">标识查询结果</param>
	/// <returns>是否查询到了标识</returns>
	public bool Query(string key, out IDIS_Query query)
	{
		Query(key, out IDIS_Query[] queries);
		query = queries.FirstOrDefault();
		return queries.Any();
	}
	public bool Register(string handle)
	{
		var handleExists = Values.Any(x => x.Handle == handle);
		if (handleExists)
		{
			return false;
		}
		var value = new IDIS_Value()
		{
			Handle = handle
		};
		Values.Add(value);
		SaveChangesAsync();
		return true;
	}
	public void Register(string handle,string name,string format, string value,string category)
	{
		var handleExists = Values.Any(x => x.Handle == handle);
		if (!handleExists)
		{
			Register(handle);
		}
		var data = new IDIS_Data()
		{
			Name = name,
			Handle = handle,
			Format = format,
			Value = value,
			Category = category,
		};
		Datas.Add(data);
		SaveChangesAsync();
	}

	public void RegisterReference(string handle, string refenceHandle)
	{
		References.Add(new IDIS_Reference()
		{
			Handle = handle,
			RelatedHandle = refenceHandle,
		});
		SaveChangesAsync();
	}
	
	public bool Update(string handle, string format, string value)
	{
		var result = Datas.FirstOrDefault(x => x.Handle == handle && x.Format == format);
		if (result is null) return false;
		result.UpdateTime=DateTime.Now;
		result.Value = value;
		SaveChangesAsync();
		return true;
	}
}
// ReSharper disable once IdentifierTypo
/// <summary>
/// 标识码注册与查询服务
/// </summary>
public partial class IDIS_Service:Node
{
	public static IDIS_Service Singleton { get; private set; }
	private static IDIS_DBContext Context;
	public override void _Ready()
	{
		Context = new IDIS_DBContext();
		BIT4Log.Log<IDIS_Service>("已创建标识数据库");
		UniTask.Run(()=>Context.Database.EnsureCreatedAsync());
	}

	public override void _EnterTree()
	{
		Singleton = this;
	}

	public override void _ExitTree()
	{
		Context.Dispose();
		Context = null;
	}

	public bool Query(string word,out IDIS_Query[] queries) => Context.Query(word, out queries);
	public bool Register(string handle) => Context.Register(handle);
	public void Register(string handle,string name, string format, string value,string category) => Context.Register(handle,name, format, value,category);
	public void RegisterReference(string handle,string refenceHandle) => Context.RegisterReference(handle,refenceHandle);
	public static string GenerateHandle() => $"88.123.99/{Mathf.Abs(Guid.NewGuid().GetHashCode())}";
	public bool Query(string key, out IDIS_Query query) => Context.Query(key, out query);
	public bool Update(string handle, string format, string value) => Context.Update(handle, format, value);
}