SSIS 小脚本 - 时间参数验证

项目中也经常使用到时间参数的验证,例如根据某一时间范围来从Source中过滤一些数据. 在运行Package之前需要配置这些时间格式的参数,因此需要验证这些参数是否是正确的时间格式. 并且通常还有时间大小的验证,例如起始时间要小于结束时间.

/*
   Microsoft SQL Server Integration Services Script Task
   Write scripts using Microsoft Visual C# 2008.
   The ScriptMain is the entry point class of the script.
*/
using System;
using System.Data;
using Microsoft.SqlServer.Dts.Runtime;
using System.Windows.Forms;
using System.Globalization;
 
namespace ST_TEST.csproj
{
    [System.AddIn.AddIn("ScriptMain", Version = "1.0", Publisher = "", Description = "")]
    public partial class ScriptMain : Microsoft.SqlServer.Dts.Tasks.ScriptTask.VSTARTScriptObjectModelBase
    {
 
        #region VSTA generated code
        enum ScriptResults
        {
            Success = Microsoft.SqlServer.Dts.Runtime.DTSExecResult.Success,
            Failure = Microsoft.SqlServer.Dts.Runtime.DTSExecResult.Failure
        };
        #endregion
 
        public void Main()
        {
            String sPackageName = Dts.Variables["System::PackageName"].Value.ToString();
            string sSelectFromDate = Dts.Variables["User::SelectFromDate"].Value.ToString();
            string sSelectThruDate = Dts.Variables["User::SelectThruDate"].Value.ToString();

            DateTime dtSelectDateFrom = new DateTime();
            DateTime dtSelectDateThru = new DateTime();
            CultureInfo provider = CultureInfo.InvariantCulture;
 
            bool bParametersValid = true;
 
            bParametersValid &= ValidateRuntimeParameterFilled(sPackageName, sSelectFromDate, "SelectFromDate");
            bParametersValid &= ValidateRuntimeParameterFilled(sPackageName, sSelectThruDate, "SelectThruDate");
 
            bParametersValid &= FormatDateTimeParameter(sPackageName, sSelectFromDate, "SelectFromDate", ref dtSelectDateFrom, provider);
            bParametersValid &= FormatDateTimeParameter(sPackageName, sSelectThruDate, "SelectThruDate", ref dtSelectDateThru, provider);
 
            if (!bParametersValid)
            {
                return;
            }
 
            if (dtSelectDateFrom > dtSelectDateThru)
            {
                Dts.Events.FireError(0,
                                     sPackageName,
                                     "SelectFromDate: " + sSelectFromDate + " is greater than SelectThruDate: " + sSelectThruDate,
                                     "",
                                     0);
                return;
            }
 
            Dts.TaskResult = (int)ScriptResults.Success;
        }
 
        // Validate current run date was specified
        private bool ValidateRuntimeParameterFilled(string sPackageName,
                                                    string sParameterValue,
                                                    string sParameterName)
        {
            if (sParameterValue.Length == 0)
            {
                Dts.Events.FireError(0,
                                     sPackageName,
                                     string.Format("{0} was not specified", sParameterName),
                                     "",
                                     0);
                return false;
            }
 
            return true;
        }
 
        // Formatting datetime input parameter
        private bool FormatDateTimeParameter(string sPackageName,
                                             string sParameterValue,
                                             string sParameterName,
                                             ref DateTime dtFormattedDateTime,
                                             CultureInfo Culture)
        {
            if (sParameterValue.Length != 0)
            {
                try
                {
                    dtFormattedDateTime = DateTime.ParseExact(sParameterValue, "yyyy-MM-dd HH:mm:ss", Culture);
                }
                catch (System.Exception e)
                {
                    Dts.Events.FireError(0,
                                         sPackageName,
                                         string.Format("Exception occurred validating {0}: {1}, error: {2}",
                                                       sParameterName,
                                                       sParameterValue,
                                                       e.Message.ToString()),
                                         "",
                                         0);
                    return false;
                }
            }
 
            return true;
        }
    }
}