CSVReaderBase.cs 5.7 KB
using log4net;
using OnlineStore.Common;
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Reflection;
using System.Text;
 

namespace OnlineStore.LoadCSVLibrary
{
    public class CSVReaderBase
    {
        public static char Spilt_Char = ',';
     //   public static readonly ILog LOGGER = LogManager.GetLogger(MethodBase.GetCurrentMethod().DeclaringType);
        /// <summary>
        /// 获取一个类所有的《字段,AttributeName列名》集合
        /// </summary>
        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;
                        proCsvMap.Add(prop.Name, csvName);
                    }
                }
            }

            return proCsvMap;
        }
        /// <summary>
        /// 获取一个类所有的《字段,AttributeName列名》集合
        /// </summary>
        public static Dictionary<string, string> getConfigProAttributeMap(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)
                    {
                        ConfigProAttribute att = (ConfigProAttribute)array[0];
                        if (att != null)
                        {
                            string csvName = att.FieldName;
                            if (proCsvMap.ContainsKey(csvName))
                            {
                            }
                            else
                            {
                                proCsvMap.Add(csvName, prop.Name);
                            }
                        }
                    }
            }
            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)
                {
                    LogUtil.error("出错:", e);
                }
            }
            catch (Exception ex)
            {
                LogUtil.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))
                    {
                        LogUtil.error("未找到必须列:" + str + ",加载数据失败!");
                        throw new CVSFieldNotMatchingExection("未找到必须列:" + str + ",加载数据失败!");
                    }
                }
            }
            return titleIndex;
        }
    }

}