CsvReader.cs 4.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169
  1. using System;
  2. using System.Collections.Generic;
  3. using System.IO;
  4. using System.Linq;
  5. using System.Text;
  6. using System.Threading.Tasks;
  7. using UnityEngine.UI;
  8. namespace Game.Config
  9. {
  10. public class CsvReader
  11. {
  12. private List<string[]> rowDatas;
  13. private Dictionary<string, int> fieldKey;
  14. private string path;
  15. private Encoding encoding;
  16. private bool isgb2312;
  17. private int fieldLine;
  18. private int typeLine;
  19. private int startLine;
  20. public string Name;
  21. public int StartLine { get { return startLine; } set { startLine = value; } }
  22. public CsvReader(string csvpath, int fl = 0,int tl = 1 ,int sl = 2,Encoding edg = null)
  23. {
  24. if (string.IsNullOrEmpty(csvpath))
  25. {
  26. throw new ArgumentNullException("path");
  27. }
  28. fieldLine = fl;
  29. typeLine = tl;
  30. startLine = sl;
  31. rowDatas = new List<string[]>();
  32. fieldKey = new Dictionary<string, int>();
  33. encoding = edg;
  34. path = csvpath;
  35. if (edg == null)
  36. {
  37. encoding = Encoding.UTF8;
  38. isgb2312 = false;
  39. }
  40. else if (edg != Encoding.UTF8)
  41. {
  42. isgb2312 = true;
  43. }
  44. Name =FileHelper.GetName(path);
  45. ReadFromCsv();
  46. InitFieldKey();
  47. }
  48. /// <summary>
  49. ///
  50. /// </summary>
  51. /// <param name="name"></param>
  52. /// <param name="bytes"></param>
  53. /// <param name="fl">字段名</param>
  54. /// <param name="tl">类型名</param>
  55. /// <param name="sl">多少行开始</param>
  56. public CsvReader(string name,byte[] bytes,int fl = 0, int tl = 1, int sl = 2)
  57. {
  58. fieldLine = fl;
  59. typeLine = tl;
  60. startLine = sl;
  61. rowDatas = new List<string[]>();
  62. fieldKey = new Dictionary<string, int>();
  63. isgb2312 = false;
  64. Name = name;
  65. ReadFromMemory(bytes);
  66. InitFieldKey();
  67. }
  68. public string[][] RowDatas
  69. {
  70. get
  71. {
  72. return rowDatas.ToArray();
  73. }
  74. }
  75. public string[] Fields()
  76. {
  77. return rowDatas[fieldLine];
  78. }
  79. public string GetDataByFieldName(string fieldName,int line)
  80. {
  81. string ret = "";
  82. if (fieldKey.ContainsKey(fieldName) && rowDatas.Count > line)
  83. {
  84. ret = rowDatas[line][fieldKey[fieldName]];
  85. }
  86. return ret;
  87. }
  88. public void SetDataByFieldName(string fieldName, string value,int line)
  89. {
  90. if (fieldKey.ContainsKey(fieldName) && rowDatas.Count > line)
  91. {
  92. rowDatas[line][fieldKey[fieldName]] = value;
  93. }
  94. }
  95. public bool HasField(string fieldName)
  96. {
  97. return fieldKey.ContainsKey(fieldName);
  98. }
  99. public string[] Types()
  100. {
  101. return rowDatas[typeLine];
  102. }
  103. public string GetString()
  104. {
  105. StringBuilder stringBuilder = new StringBuilder();
  106. for (int i = 0; i < rowDatas.Count; i++)
  107. {
  108. stringBuilder.Append(CsvWriter<GameData>.ToCsvLine(rowDatas[i]));
  109. }
  110. return stringBuilder.ToString();
  111. }
  112. private void ReadFromCsv()//改异步
  113. {
  114. string[] datas = File.ReadAllLines(path, encoding);
  115. if (datas == null || datas.Length <= 0)
  116. {
  117. return;
  118. }
  119. if (isgb2312)
  120. {
  121. for (int i = 0; i < datas.Length; i++)
  122. {
  123. datas[i] = FileHelper.GB2312ToUTF8(datas[i]);
  124. }
  125. }
  126. foreach (var item in datas)
  127. {
  128. string[] rdt = FileHelper.SpiltData(item);
  129. rowDatas.Add(rdt);
  130. }
  131. }
  132. private void ReadFromMemory(byte[] bytes)
  133. {
  134. string str = Encoding.UTF8.GetString(bytes);
  135. string[] datas = str.Split(new string[] { "\r\n" }, StringSplitOptions.RemoveEmptyEntries);
  136. foreach (var item in datas)
  137. {
  138. string[] rdt = FileHelper.SpiltData(item);
  139. rowDatas.Add(rdt);
  140. }
  141. }
  142. private void InitFieldKey()
  143. {
  144. string[] fields = rowDatas[fieldLine];
  145. for (int i = 0; i < fields.Length; i++)
  146. {
  147. fieldKey.Add(fields[i],i);
  148. }
  149. }
  150. }
  151. }