GHZMRSpaceXR/Assets/Studio/XRLib/Codes/PublicTools/XMLDataBase/TableHelper.cs

using BeinLab.Util;
using System;
using System.Collections;
using System.Collections.Generic;
using System.IO;
using UnityEngine;
using XRTool.Util;

namespace PublicTools.XMLDataBase
{
    /// <summary>
    /// 一个辅助读写的脚本
    /// 提供相关类型的读写操作
    /// </summary>
    public class TableHelper<T>
    {
        private TableInterface table;
        private string path;
        private List<T> dataList;
        public event Action readConfComplete;
        public bool isInit = false;
        private Dictionary<string, T> dataDic = new Dictionary<string, T>();
        private string priKey;
        /// <summary>
        /// 是否可写，默认可写
        /// </summary>
        private bool isWrite = true;

        public TableHelper(string path, string priKey = null)
        {
            this.path = path;
            this.priKey = priKey;
            readConfComplete += OnRreadConfComplete;
        }
        /// <summary>
        /// 读取完成之后
        /// </summary>
        private void OnRreadConfComplete()
        {
            if (DataList != null)
            {
                for (int i = 0; i < DataList.Count; i++)
                {
                    if (string.IsNullOrEmpty(priKey))
                    {
                        if (DataList[i] is TableBase)
                        {
                            var da = DataList[i] as TableBase;
                            if (!dataDic.ContainsKey(da.priKey))
                            {
                                dataDic.Add(da.priKey, DataList[i]);
                            }
                        }
                        else if (DataList[i] is UnityEngine.Object)
                        {
                            var da = DataList[i] as UnityEngine.Object;
                            if (!dataDic.ContainsKey(da.name))
                            {
                                dataDic.Add(da.name, DataList[i]);
                            }
                        }
                    }
                    else
                    {
                        string key = ClazzFactory.GetPriKey<T>(DataList[i], priKey);
                        if (!dataDic.ContainsKey(key))
                        {
                            dataDic.Add(key, DataList[i]);
                        }
                    }
                }
            }
            isInit = true;
        }

        public void Reset()
        {
            if (DataList != null)
            {
                DataList.Clear();
            }
            isInit = false;
            DataList = null;
            dataDic.Clear();
            if (table != null)
            {
                table.Close();
            }
            XSql.Instance.CloseTable(typeof(T).Name);
        }

        public List<T> DataList { get => dataList; set => dataList = value; }
        /// <summary>
        /// 打开指定的数据
        /// </summary>
        /// <param name="priKey"></param>
        public void Open(string priKey = null)
        {
            if (!string.IsNullOrEmpty(priKey))
            {
                this.priKey = priKey;
            }
            isWrite = true;
            if (path.StartsWith(Application.streamingAssetsPath))
            {
#if !UNITY_EDITOR
                isWrite = false;
#endif
            }
            if (isWrite)
            {
                table = XSql.Instance.OpenTable(path, typeof(T).Name, ".xml", true);
                if (!table.Open())
                {
                    table.Create(typeof(T).Name);
                }
                this.DataList = table.FindAllData<T>();
                readConfComplete?.Invoke();
            }
            else
            {
                if (TimerMgr.Instance)
                {
                    string fullPath = Path.Combine(path, typeof(T).Name + ".xml");
                    TimerMgr.Instance.StartCoroutine(XSql.Instance.ReadServerData(fullPath, (List<T> scrollList) =>
                    {
                        this.DataList = scrollList;
                        readConfComplete?.Invoke();
                    }));
                }
            }
        }
        /// <summary>
        /// 保存数据
        /// </summary>
        public void SaveData()
        {
            if (table != null)
            {
                table.Save();
            }
        }
        /// <summary>
        /// 添加一条配置
        /// </summary>
        /// <param name="conf"></param>
        public void AddData(T conf, bool isSave = true)
        {
            if (table == null)
            {
                Open(priKey);
            }
            if (table != null)
            {
                if (string.IsNullOrEmpty(priKey))
                {
                    if (conf is TableBase)
                    {
                        var tab = conf as TableBase;
                        if (table.InsertData(conf, tab.priKey))
                        {
                            if (isSave)
                                table.Save();
                            UnityLog.Instance.Log("Add Success!" + tab.priKey, 2);
                        }
                        else
                        {
                            UnityLog.Instance.LogError("Add DataConf Error!" + conf);
                        }
                    }
                    else if (conf is UnityEngine.Object)
                    {
                        var tab = conf as UnityEngine.Object;
                        if (table.InsertData(conf, tab.name))
                        {
                            if (isSave)
                                table.Save();
                            UnityLog.Instance.Log("Add Success!" + tab.name, 2);
                        }
                        else
                        {
                            UnityLog.Instance.LogError("Add DataConf Error!" + conf);
                        }
                    }
                    else
                    {
                        if (table.InsertData(conf))
                        {
                            if (isSave)
                                table.Save();
                            UnityLog.Instance.Log("Add Success!" + conf, 2);
                        }
                        else
                        {
                            UnityLog.Instance.LogError("Add DataConf Error!" + conf);
                        }
                    }
                }
                else
                {
                    string key = ClazzFactory.GetPriKey<T>(conf, priKey);
                    if (table.InsertData(conf, key))
                    {
                        if (isSave)
                            table.Save();
                        UnityLog.Instance.Log("Add Success!" + conf + key, 2);
                    }
                    else
                    {
                        UnityLog.Instance.LogError("Add DataConf Error!" + conf);
                    }
                }
            }
            else
            {
                UnityLog.Instance.LogError("Add DataConf Error!No table" + conf);
            }
        }
        /// <summary>
        /// 删除这条配置
        /// </summary>
        /// <param name="key"></param>
        public void DelConf(string key, bool isSave = true)
        {
            if (table != null)
            {
                table.DeleteData<T>(key);
                if (isSave)
                    table.Save();
            }
            Open(key);
        }
        /// <summary>
        /// 根据key值查找数据
        /// 如果缓存列表里没有这条记录，则去循环列表中查找
        /// </summary>
        /// <param name="name"></param>
        /// <returns></returns>
        public T FindData(string name, string priKey = null)
        {
            if (string.IsNullOrEmpty(priKey))
            {
                priKey = this.priKey;
            }
            if (!string.IsNullOrEmpty(name))
            {
                if (dataDic != null && dataDic.ContainsKey(name))
                {
                    return dataDic[name];
                }
                else
                {
                    ///存在主键值
                    if (!string.IsNullOrEmpty(priKey))
                    {
                        return DataList.Find((data) =>
                        {
                            if (name == ClazzFactory.GetPriKey<T>(data, priKey))
                            {
                                return true;
                            }
                            return false;
                        });
                    }
                    else
                    {
                        ///暴力破解法，很少使用,请尽量避免使用此分支
                        UnityLog.Instance.LogError("Use Wrong method!");
                        return DataList.Find((data) =>
                        {
                            var dic = ClazzFactory.ClazzAnalyze<T>(data);
                            if (dic != null)
                            {
                                foreach (var item in dic)
                                {
                                    if (item.Value == name)
                                    {
                                        return true;
                                    }
                                }
                            }
                            return false;
                        });
                    }
                }
            }
            return default;
        }
    }
}