405 lines
17 KiB
C#
405 lines
17 KiB
C#
//////////////////////////////////////////////////////////////////////////
|
||
//
|
||
// 文件：Assets/Editor/Tools/DBTools.cs
|
||
// 作者：Xoen Xie
|
||
// 时间：2023/06/19
|
||
// 描述：db code tools
|
||
// 说明：
|
||
//
|
||
//////////////////////////////////////////////////////////////////////////
|
||
using UnityEditor;
|
||
using UnityEngine;
|
||
using System.IO;
|
||
using System.Xml;
|
||
using System.Text;
|
||
|
||
public static class DBTools
|
||
{
|
||
private static string SERVER_DIR = Application.dataPath + $"/../../../NLDServer";
|
||
private static string DB_DEF_PATH = SERVER_DIR + "/common/db/db_def.go";
|
||
|
||
|
||
[MenuItem("Tools/*配置工具/生成数据库代码", false, (int)NLDMenuID.DBCodeExport)]
|
||
public static void DBCodeExport()
|
||
{
|
||
ExportRedisCode();
|
||
ExportMongoCode();
|
||
}
|
||
|
||
private static void ExportRedisCode()
|
||
{
|
||
string redisConfig = SERVER_DIR + "/bin/scp/DBRedis.xml";
|
||
if (!File.Exists(redisConfig))
|
||
{
|
||
Debug.LogError($"{redisConfig} 不存在");
|
||
return;
|
||
}
|
||
|
||
XmlReaderSettings settings = new XmlReaderSettings();
|
||
settings.IgnoreComments = true;
|
||
|
||
string sContent = File.ReadAllText(redisConfig);
|
||
XmlReader reader = XmlReader.Create(new MemoryStream(Encoding.UTF8.GetBytes(sContent)), settings);
|
||
XmlDocument xmlDoc = new XmlDocument();
|
||
xmlDoc.Load(reader);
|
||
|
||
XmlNode RootNode = xmlDoc.SelectSingleNode("root");
|
||
XmlNodeList nodes = RootNode.ChildNodes;
|
||
|
||
string Ids = "";
|
||
string funcs = "";
|
||
//string structs = "";
|
||
|
||
for (int i = 0; i < nodes.Count; ++i)
|
||
{
|
||
XmlElement ele = nodes[i] as XmlElement;
|
||
|
||
int id = EditorUtil.GetInt(ele, "id");
|
||
string sName = EditorUtil.GetString(ele, "name");
|
||
string sDesc = EditorUtil.GetString(ele, "desc");
|
||
string sOp = EditorUtil.GetString(ele, "op");
|
||
string sKey = EditorUtil.GetString(ele, "key");
|
||
string sFmtKey = EditorUtil.GetString(ele, "fmt_key");
|
||
|
||
string currentID = $"REDIS_{sName.ToUpper()}";
|
||
Ids += $"\t{currentID}\t\t\t= {id}\t// {sDesc}\n";
|
||
|
||
string[] names = sName.Split("_");
|
||
string funcDesc = $"// {sDesc}\n";
|
||
|
||
string funcName = "RD_";
|
||
for (int J = 0; J < names.Length; ++J)
|
||
{
|
||
funcName += EditorUtil.FirstLetterToUpper(names[J].ToLower());
|
||
}
|
||
|
||
string fullFunc = funcDesc + "func " + funcName + "(keyID uint32, keyStr string, rpc *xsf_rpc.Acceptor";
|
||
|
||
string funcContent = $"\tcfg := xsf_scp.XSFSchema().DBRedis.Get({currentID})\n";
|
||
funcContent += "\tif cfg == nil {\n";
|
||
funcContent += $"\t\txsf_log.Error(\"{funcName} cfg == nil {currentID}\")\n";
|
||
funcContent += "\t\treturn\n";
|
||
funcContent += "\t}\n\n";
|
||
funcContent += "\txsf_redis.Request(keyID, keyStr, cfg, rpc";
|
||
|
||
|
||
XmlNodeList param = ele.ChildNodes;
|
||
if (param.Count > 0)
|
||
{
|
||
XmlElement xmlParam = param[0] as XmlElement;
|
||
string sParamName = EditorUtil.GetString(xmlParam, "name");
|
||
string sType = EditorUtil.GetString(xmlParam, "type");
|
||
if (sOp.Contains("SET"))
|
||
{
|
||
fullFunc += $", {sParamName} {sType}" + ") {\n";
|
||
funcContent += $", {sParamName})\n";
|
||
}
|
||
else if (sOp.Contains("GET"))
|
||
{
|
||
fullFunc += ") {\n";
|
||
funcContent += $")\n";
|
||
}
|
||
}
|
||
else
|
||
{
|
||
fullFunc += ") {\n";
|
||
funcContent += $")\n";
|
||
}
|
||
|
||
|
||
funcs += fullFunc + funcContent + "}\n\n";
|
||
}
|
||
|
||
string codePath = SERVER_DIR + "/common/db/redis_func.go";
|
||
EditorUtil.ReplaceContentByTag(DB_DEF_PATH, "REDIS_REQUEST_ID_BEGIN", "REDIS_REQUEST_ID_END", Ids);
|
||
EditorUtil.ReplaceContentByTag(codePath, "REQUEST_FUNC_BEGEIN", "REQUEST_FUNC_END", funcs);
|
||
|
||
}
|
||
|
||
|
||
private static void ExportMongoCode()
|
||
{
|
||
string redisConfig = SERVER_DIR + "/bin/scp/DBMongo.xml";
|
||
if (!File.Exists(redisConfig))
|
||
{
|
||
Debug.LogError($"{redisConfig} 不存在");
|
||
return;
|
||
}
|
||
|
||
XmlReaderSettings settings = new XmlReaderSettings();
|
||
settings.IgnoreComments = true;
|
||
|
||
string sContent = File.ReadAllText(redisConfig);
|
||
XmlReader reader = XmlReader.Create(new MemoryStream(Encoding.UTF8.GetBytes(sContent)), settings);
|
||
XmlDocument xmlDoc = new XmlDocument();
|
||
xmlDoc.Load(reader);
|
||
|
||
XmlNode RootNode = xmlDoc.SelectSingleNode("root");
|
||
XmlNodeList nodes = RootNode.ChildNodes;
|
||
|
||
string Ids = "";
|
||
string funcs = "";
|
||
//string structs = "";
|
||
|
||
for (int db = 0; db < nodes.Count; db++)
|
||
{
|
||
XmlElement eleDB= nodes[db] as XmlElement;
|
||
XmlNodeList cols = eleDB.ChildNodes;
|
||
for (int col = 0; col < cols.Count; col++)
|
||
{
|
||
XmlElement eleCol = cols[col] as XmlElement;
|
||
bool isSplit = false;
|
||
if(EditorUtil.GetInt(eleCol, "split") > 0)
|
||
{
|
||
isSplit = true;
|
||
}
|
||
|
||
XmlNodeList reqs = eleCol.ChildNodes;
|
||
for (int r = 0; r < reqs.Count; r++)
|
||
{
|
||
XmlElement ele = reqs[r] as XmlElement;
|
||
Debug.Log("name=" + ele.Name);
|
||
|
||
int id = EditorUtil.GetInt(ele, "id");
|
||
string sName = EditorUtil.GetString(ele, "name");
|
||
string sDesc = EditorUtil.GetString(ele, "desc");
|
||
string sOp = EditorUtil.GetString(ele, "op");
|
||
bool dataNotFormat = EditorUtil.GetBoolean(ele, "data_not_format");
|
||
|
||
string currentID = $"MONGO_{sName.ToUpper()}";
|
||
Ids += $"\t{currentID}\t\t\t= {id}\t// {sDesc}\n";
|
||
|
||
string[] names = sName.Split("_");
|
||
string funcDesc = $"// {sDesc}\n";
|
||
|
||
string funcName = "MG_";
|
||
for (int J = 0; J < names.Length; ++J)
|
||
{
|
||
funcName += EditorUtil.FirstLetterToUpper(names[J].ToLower());
|
||
}
|
||
|
||
string fullFunc;
|
||
if(isSplit)
|
||
{
|
||
fullFunc = funcDesc + "func " + funcName + "(keyID uint32, colID uint32, rpc *xsf_rpc.Acceptor";
|
||
}
|
||
else
|
||
{
|
||
fullFunc = funcDesc + "func " + funcName + "(keyID uint32, rpc *xsf_rpc.Acceptor";
|
||
}
|
||
|
||
|
||
string funcContent = $"\tcfg := xsf_scp.XSFSchema().DBMongo.Get({currentID})\n";
|
||
funcContent += "\tif cfg == nil {\n";
|
||
funcContent += $"\t\txsf_log.Error(\"{funcName} cfg == nil {currentID}\")\n";
|
||
funcContent += "\t\treturn\n";
|
||
funcContent += "\t}\n\n";
|
||
|
||
switch (sOp)
|
||
{
|
||
case "Insert":
|
||
{
|
||
XmlNodeList datas = ele.SelectNodes("data");
|
||
string dataM = "";
|
||
for (int z = 0; z < datas.Count; z++)
|
||
{
|
||
XmlElement eleData = datas[z] as XmlElement;
|
||
string name = EditorUtil.GetString(eleData, "name");
|
||
string type = EditorUtil.GetString(eleData, "type");
|
||
fullFunc += $", D_{name} {type}";
|
||
|
||
dataM += $"\t\t\"{name}\": D_{name},\n";
|
||
}
|
||
|
||
fullFunc += ") {\n";
|
||
|
||
funcContent += "\tdata := bson.M{\n";
|
||
funcContent += dataM;
|
||
funcContent += "\t}\n\n";
|
||
|
||
funcContent += "\txsf_mongo.Insert(keyID, cfg, rpc, data)\n";
|
||
}
|
||
break;
|
||
|
||
case "FindAll":
|
||
case "FindOneSort":
|
||
case "FindOne":
|
||
{
|
||
switch(sOp)
|
||
{
|
||
case "FindAll":
|
||
fullFunc += ", page uint32";
|
||
break;
|
||
default:
|
||
break;
|
||
}
|
||
|
||
XmlNodeList datas = ele.SelectNodes("filter");
|
||
string dataF = "";
|
||
for (int z = 0; z < datas.Count; z++)
|
||
{
|
||
XmlElement eleData = datas[z] as XmlElement;
|
||
string name = EditorUtil.GetString(eleData, "name");
|
||
string type = EditorUtil.GetString(eleData, "type");
|
||
fullFunc += $", F_{name} {type}";
|
||
|
||
dataF += $"\t\t\"{name}\": F_{name},\n";
|
||
}
|
||
|
||
fullFunc += ") {\n";
|
||
|
||
funcContent += "\tfilter := bson.M{\n";
|
||
funcContent += dataF;
|
||
funcContent += "\t}\n\n";
|
||
|
||
|
||
XmlNodeList Pdatas = ele.SelectNodes("projection");
|
||
string dataP = "";
|
||
for (int z = 0; z < Pdatas.Count; z++)
|
||
{
|
||
XmlElement eleData = Pdatas[z] as XmlElement;
|
||
string name = EditorUtil.GetString(eleData, "name");
|
||
string v = EditorUtil.GetString(eleData, "value");
|
||
|
||
dataP += $"\t\t\"{name}\": {v},\n";
|
||
}
|
||
|
||
funcContent += "\tprojection := bson.M{\n";
|
||
funcContent += dataP;
|
||
funcContent += "\t}\n\n";
|
||
|
||
|
||
switch(sOp)
|
||
{
|
||
case "FindAll":
|
||
if(isSplit)
|
||
{
|
||
funcContent += "\txsf_mongo.FindAllByColID(keyID, colID, cfg, rpc, page, filter, projection)\n";
|
||
}
|
||
else
|
||
{
|
||
funcContent += "\txsf_mongo.FindAll(keyID, cfg, rpc, page, filter, projection)\n";
|
||
}
|
||
break;
|
||
|
||
case "FindOneSort":
|
||
funcContent += "\txsf_mongo.FindOneSort(keyID, cfg, rpc, filter, projection)\n";
|
||
break;
|
||
|
||
case "FindOne":
|
||
funcContent += $"\txsf_mongo.FindOne(keyID, cfg, rpc, filter, projection)\n";
|
||
break;
|
||
default:
|
||
break;
|
||
}
|
||
|
||
}
|
||
break;
|
||
|
||
case "UpdateOne":
|
||
case "DeleteOne":
|
||
{
|
||
{
|
||
XmlNodeList datas = ele.SelectNodes("filter");
|
||
string dataF = "";
|
||
for (int z = 0; z < datas.Count; z++)
|
||
{
|
||
XmlElement eleData = datas[z] as XmlElement;
|
||
string name = EditorUtil.GetString(eleData, "name");
|
||
string type = EditorUtil.GetString(eleData, "type");
|
||
fullFunc += $", F_{name} {type}";
|
||
|
||
dataF += $"\t\t\"{name}\": F_{name},\n";
|
||
}
|
||
|
||
funcContent += "\tfilter := bson.M{\n";
|
||
funcContent += dataF;
|
||
funcContent += "\t}\n\n";
|
||
}
|
||
|
||
if (dataNotFormat)
|
||
{
|
||
if(sOp == "UpdateOne")
|
||
{
|
||
XmlNodeList datas = ele.SelectNodes("data");
|
||
XmlElement eleData = datas[0] as XmlElement;
|
||
string name = EditorUtil.GetString(eleData, "name");
|
||
string type = EditorUtil.GetString(eleData, "type");
|
||
|
||
fullFunc += $", D_{name} {type}";
|
||
|
||
fullFunc += ", IsUpsert bool) {\n";
|
||
|
||
if(isSplit)
|
||
{
|
||
funcContent += $"\txsf_mongo.UpdateOneByColID(keyID, colID, cfg, rpc, IsUpsert, filter, D_{name})\n";
|
||
}
|
||
else
|
||
{
|
||
funcContent += $"\txsf_mongo.UpdateOne(keyID, cfg, rpc, IsUpsert, filter, D_{name})\n";
|
||
}
|
||
|
||
}
|
||
else
|
||
{
|
||
fullFunc += ") {\n";
|
||
|
||
funcContent += $"\txsf_mongo.DeleteOne(keyID, cfg, rpc, filter)\n";
|
||
}
|
||
}
|
||
else
|
||
{
|
||
XmlNodeList datas = ele.SelectNodes("data");
|
||
string dataM = "";
|
||
for (int z = 0; z < datas.Count; z++)
|
||
{
|
||
XmlElement eleData = datas[z] as XmlElement;
|
||
string name = EditorUtil.GetString(eleData, "name");
|
||
string type = EditorUtil.GetString(eleData, "type");
|
||
fullFunc += $", D_{name} {type}";
|
||
|
||
dataM += $"\t\t\"{name}\": D_{name},\n";
|
||
}
|
||
|
||
if(sOp == "UpdateOne")
|
||
{
|
||
funcContent += "\tdata := bson.M{\n";
|
||
funcContent += dataM;
|
||
funcContent += "\t}\n";
|
||
|
||
fullFunc += ", IsUpsert bool) {\n";
|
||
|
||
if(isSplit)
|
||
{
|
||
funcContent += "\txsf_mongo.UpdateOneByColID(keyID, colID, cfg, rpc, IsUpsert, filter, data)\n";
|
||
}
|
||
else
|
||
{
|
||
funcContent += "\txsf_mongo.UpdateOne(keyID, cfg, rpc, IsUpsert, filter, data)\n";
|
||
}
|
||
}
|
||
else
|
||
{
|
||
fullFunc += ") {\n";
|
||
|
||
funcContent += $"\txsf_mongo.DeleteOne(keyID, cfg, rpc, filter)\n";
|
||
}
|
||
}
|
||
|
||
|
||
}
|
||
break;
|
||
}
|
||
|
||
|
||
funcs += fullFunc + funcContent + "}\n\n";
|
||
}
|
||
}
|
||
}
|
||
|
||
string codePath = SERVER_DIR + "/common/db/mongo_func.go";
|
||
EditorUtil.ReplaceContentByTag(DB_DEF_PATH, "MONGO_REQUEST_ID_BEGIN", "MONGO_REQUEST_ID_END", Ids);
|
||
EditorUtil.ReplaceContentByTag(codePath, "REQUEST_FUNC_BEGEIN", "REQUEST_FUNC_END", funcs);
|
||
|
||
}
|
||
} |