using System;
using System.Data;
using System.Linq;
using System.Threading.Tasks;
using Microsoft.Extensions.Logging;
using SmartCode.Configuration;
using SmartSql;

namespace SmartCode.ETL
{
    public class ExtractDataSource : AbstractExtractDataSource
    {
        public ExtractDataSource(Project project, ILoggerFactory loggerFactory, ILogger<ExtractDataSource> logger,
            IProjectBuilder projectBuilder, IPluginManager pluginManager) : base(project, loggerFactory, logger,
            projectBuilder, pluginManager)
        {
        }

        public override string Name => "Extract";
        public DataSet DataSet { get; set; }
        public DataTable TransformData { get; set; }
        protected override async Task LoadData(RequestContext requestContext)
        {
            DataSet = await SqlMapper.GetDataSetAsync(requestContext);
            TransformData = DataSet.Tables[0];
        }
        public override long GetMaxId(string pkColumn)
        {
            return TransformData.Rows.Cast<DataRow>().AsParallel().Max(dr => Convert.ToInt64(dr[pkColumn]));
        }

        public override DateTime GetMaxModifyTime(string modifyTime)
        {
            return TransformData.Rows.Cast<DataRow>().AsParallel().Max(dr => Convert.ToDateTime(dr[modifyTime]));
        }

        public override int GetCount()
        {
            return TransformData.Rows.Count;
        }


    }
}