using DllUfpEntity.Dto;
using System;
using System.Collections.Generic;
using System.Data;
using System.Linq;
using System.Text;

namespace DllUfpDal.Extension
{
    public static class EnumerableExtension
    {
        public static DataTable TransferToTable(this IEnumerable<VisitLogAnalysisDto> dataSource)
        {
            DataTable dt = null;
            if (dataSource == null || dataSource.Count() <= 0)
                return dt;
            dt = new DataTable();
            var cate = new DataColumn("日期");
            dt.Columns.Add(cate);
            var totalCol = new DataColumn("汇总");
            dt.Columns.Add(totalCol);
            var row = dt.NewRow();
            row[totalCol.ColumnName] = dataSource.Sum(c => c.Count);
            row[cate.ColumnName] = "全员";
            foreach (var item in dataSource)
            {
                var col = new DataColumn(item.Date.Value.ToString("yyyy-MM-dd"));
                dt.Columns.Add(col);
                row[col.ColumnName] = item.Count;
            }
            dt.Rows.Add(row);
            return dt;
        }
    }
}