CSVReaderBase.cs 4.8 KB
using log4net;
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Reflection;
using System.Text;
using System.Threading.Tasks;
using TSA_V.Common;

namespace TSA_V.LoadCSVLibrary
{
    public class CSVReaderBase
    {
        public static char Spilt_Char = ',';
        public static readonly ILog LOGGER = LogManager.GetLogger(MethodBase.GetCurrentMethod().DeclaringType);
        private static int LibNameType = ConfigAppSettings.GetIntValue(Setting_Init.LibNameType);
       
        public static Dictionary<string, string> getProAttributeMap(Type type)
        {
            Dictionary<string, string> proCsvMap = new Dictionary<string, string>();
            PropertyInfo[] props = type.GetProperties();
            foreach (PropertyInfo prop in props)
            {
                object[] array = prop.GetCustomAttributes(false);
                if (array.Length > 0)
                {
                    CSVAttribute att = (CSVAttribute)array[0];
                    if (att != null)
                    {
                        string csvName = att.FieldName;
                        if (LibNameType.Equals(1)&&(String.IsNullOrEmpty(att.FieldName1 ).Equals(false)))
                        {
                            csvName = att.FieldName1;
                        }
                        proCsvMap.Add(prop.Name, csvName);
                    }
                }
            }

            return proCsvMap;
        } 

        /// <summary>
        /// 读取一个csv文件的所有内容
        /// </summary> 
        protected static string[] ReadCSVFile(string filePath)
        {
            if (!File.Exists(filePath))
            {
                throw new FileNotFoundException("未找到文件:" + filePath);
            }
            string[] lines = File.ReadAllLines(filePath, Encoding.GetEncoding("gbk"));
            return lines;
        }
        /// <summary>
        /// 把数据写入csv中
        /// </summary> 
        protected static bool WriteCSVFile(string filePath, string[] lines)
        {
            try
            {
                if (File.Exists(filePath))
                {
                    File.Delete(filePath);
                }
                File.WriteAllLines(filePath, lines, Encoding.GetEncoding("gbk"));

                //备份保存
                try
                {
                    FileInfo file = new FileInfo(filePath);
                    string date = DateTime.Now.ToString("yyyy-MM-dd");
                    string targetBackPath = @"C:\configBack\" + date + @"\";
                    if (!Directory.Exists(targetBackPath))
                    {
                        Directory.CreateDirectory(targetBackPath);
                    }
                    string fileName = file.Name;
                    string backFile = targetBackPath + fileName;
                    if (File.Exists(backFile))
                    {
                        File.Delete(backFile);
                    }
                    File.WriteAllLines(backFile, lines, Encoding.GetEncoding("gbk"));
                }
                catch (Exception e)
                {
                    LOGGER.Error("出错:", e);
                }
            }
            catch (Exception ex)
            {
                LOGGER.Error(ex.ToString());
                return false;
            }
            return true;
        }
        protected static Dictionary<string, int> GetAllTitleIndex(string lineValue)
        {
            Dictionary<string, int> titleIndex = new Dictionary<string, int>();
            var array = lineValue.Split(',');

            //验证列
            for (int i = 0; i < array.Length; i++)
            {
                string title = array[i];
                titleIndex.Add(title, i);
            }
            return titleIndex;
        }
        protected static Dictionary<string, int> GetTitleIndex(string lineValue, List<string> cvsTitleList)
        {
            Dictionary<string, int> titleIndex = new Dictionary<string, int>();
            var array = lineValue.Split(',');
            //验证列
            for (int i = 0; i < array.Length; i++)
            {
                string title = array[i];
                if (cvsTitleList.Contains(title))
                {
                    titleIndex.Add(title, i);
                }
            }
            if (titleIndex.Count < cvsTitleList.Count)
            {
                foreach (string str in cvsTitleList)
                {
                    if (!titleIndex.ContainsKey(str) )
                    {
                        LOGGER.Error("未找到必须列:" + str + "!");
                        //throw new CVSFieldNotMatchingExection("未找到必须列:" + str + ",加载数据失败!");
                       
                    }
                }
            }
            return titleIndex;
        }
    }

}