/// <summary>执行任务</summary>
/// <param name="options"></param>
static void RunTask(Options options)
{
DataDumpTask task = DataDumpConfiguration.Instance.Tasks[options.TaskName];
if (task == null)
{
Console.WriteLine("task : " + options.TaskName + " is null");
}
else
{
Console.WriteLine("task : " + options.TaskName + " running...");
IDataDumpProvider provider = (IDataDumpProvider)KernelContext.CreateObject(task.DataDumpProvider);
provider.Init(task);
var result = provider.Generate();
Console.WriteLine("result => " + result);
// 输出位置
string path = StringHelper.NullTo(options.OutputFile, task.OutputFile);
if (!string.IsNullOrEmpty(path))
{
path = Path.Combine(KernelConfigurationView.Instance.ApplicationPathRoot, path);
File.WriteAllText(path, result, Encoding.UTF8);
}
}
}
/// <summary>初始化配置</summary>
/// <param name="task">任务信息</param>
public void Init(DataDumpTask task)
{
this.task = task;
this.options = new Dictionary <string, string>();
JsonData data = ((JsonData)Json.JsonMapper.ToObject(task.Options));
foreach (string key in data.Keys)
{
if (options.ContainsKey(key))
{
options[key] = data[key].ToString();
}
else
{
options.Add(key, data[key].ToString());
}
}
}
public string Generate()
{
StringBuilder outString = new StringBuilder();
// 执行依赖任务
if (!string.IsNullOrEmpty(task.Depends))
{
string[] depends = task.Depends.Split(',');
foreach (string depend in depends)
{
// 任务不能自身依赖 避免发生循环引用
if (depend == task.Name)
{
continue;
}
DataDumpTask dependTask = DataDumpConfiguration.Instance.Tasks[depend];
// 重写依赖任务的参数选项和输出数据库类型
dependTask.Options = task.Options;
dependTask.OutputDbType = task.OutputDbType;
IDataDumpProvider dependProvider = (IDataDumpProvider)KernelContext.CreateObject(dependTask.DataDumpProvider);
dependProvider.Init(dependTask);
outString.AppendLine(dependProvider.Generate());
}
}
// 执行任务
string comment, result;
foreach (TaskStatement statement in task.Statements)
{
comment = null;
result = null;
if (!string.IsNullOrEmpty(statement.Description))
{
string[] descriptionLines = statement.Description.Split(new char[] { '\n' }, StringSplitOptions.RemoveEmptyEntries);
foreach (string descriptionLine in descriptionLines)
{
if (!string.IsNullOrEmpty(descriptionLine.Trim()))
{
comment += "-- " + descriptionLine.Trim() + Environment.NewLine;
}
}
}
GenericSqlCommand command = new GenericSqlCommand(task.DataSourceName);
string sql = statement.Sql;
foreach (KeyValuePair <string, string> option in options)
{
sql = sql.Replace("$" + option.Key + "$", option.Value);
}
DataTable table = command.ExecuteQueryForDataTable(sql);
if (table.Rows.Count == 0)
{
continue;
}
result = SqlScriptHelper.GenerateDateTableScript(task.OutputDbType, statement.DestTable, table);
outString.Append(comment);
outString.AppendLine(result);
}
return(outString.ToString());
}