/// <summary>
/// Given a set of <see cref="FieldDescription"/> and existing RecordInfo, create a new RecordInfo with specified fields replacing existing ones or being appended.
/// </summary>
/// <param name="inputRecordInfo">Existing RecordInfo object.</param>
/// <param name="fields">Set of Field Descriptions.</param>
/// <returns>A configured RecordInfo object.</returns>
public static AlteryxRecordInfoNet.RecordInfo CreateRecordInfo(
AlteryxRecordInfoNet.RecordInfo inputRecordInfo,
params FieldDescription[] fields)
{
var fieldDict = fields.ToDictionary(f => f.Name, f => false);
var output = new AlteryxRecordInfoNet.RecordInfo();
if (inputRecordInfo != null)
{
for (int i = 0; i < inputRecordInfo.NumFields(); i++)
{
var fieldInfo = inputRecordInfo[i];
var fieldName = fieldInfo.GetFieldName();
if (fieldDict.ContainsKey(fieldName))
{
fieldDict[fieldName] = true;
var descr = fields.First(f => f.Name == fieldName);
output.AddField(descr.Name, descr.FieldType, descr.Size, descr.Scale, descr.Source, descr.Description);
continue;
}
output.AddField(fieldInfo);
}
}
foreach (var descr in fields.Where(d => !fieldDict[d.Name]))
{
output.AddField(descr.Name, descr.FieldType, descr.Size, descr.Scale, descr.Source, descr.Description);
}
return(output);
}
public bool PI_PushAllRecords(long nRecordLimit)
{
if (nRecordLimit < 0)
{
nRecordLimit = long.MaxValue;
}
var success = false;
var connStr = $"Data Source={dataSource};Catalog={catalog}";
var conn = new AdomdConnection(connStr);
try
{
AdomdDataReader reader;
conn.Open();
engine.OutputMessage(toolId, AlteryxRecordInfoNet.MessageStatus.STATUS_Info, "Connected to server, processing the query...");
engine.OutputToolProgress(toolId, 0.05);
output.UpdateProgress(0.05);
var command = new AdomdCommand(mdx, conn);
if (nRecordLimit == 0)
{
reader = command.ExecuteReader(CommandBehavior.SchemaOnly);
}
else
{
reader = command.ExecuteReader();
}
engine.OutputMessage(toolId, AlteryxRecordInfoNet.MessageStatus.STATUS_Info, "Query finished executing, retrieving schema...");
engine.OutputToolProgress(toolId, 0.50);
output.UpdateProgress(0.50);
var fields = reader.GetSchemaTable();
var outputInfo = new AlteryxRecordInfoNet.RecordInfo();
foreach (DataRow field in fields.Rows)
{
var fieldName = field[0].ToString();
outputInfo.AddField(fieldName, AlteryxRecordInfoNet.FieldType.E_FT_V_WString, 1073741823, 0, $"SSAS Input ({toolId})", "");
}
output.Init(outputInfo, "Output", null, config);
engine.OutputMessage(toolId, AlteryxRecordInfoNet.MessageStatus.STATUS_Info, "Prepared output schema, retrieving data...");
engine.OutputToolProgress(toolId, 0.55);
output.UpdateProgress(0.55);
var record = outputInfo.CreateRecord();
int totalRead = 0;
while (totalRead < nRecordLimit && reader.Read())
{
record.Reset();
for (int i = 0; i < outputInfo.NumFields(); i++)
{
AlteryxRecordInfoNet.FieldBase fieldBase = outputInfo[i];
if (reader[i] == null)
{
fieldBase.SetNull(record);
}
else
{
fieldBase.SetFromString(record, reader[i].ToString());
}
}
output.PushRecord(record.GetRecord());
totalRead++;
}
reader.Close();
reader.Dispose();
engine.OutputMessage(toolId, AlteryxRecordInfoNet.MessageStatus.STATUS_Complete, $"Data retrieval completed. {totalRead} records output.");
engine.OutputToolProgress(toolId, 1.00);
output.UpdateProgress(1.00);
success = true;
}
catch (Exception ex)
{
engine.OutputMessage(toolId, AlteryxRecordInfoNet.MessageStatus.STATUS_Error, $"Error: {ex.Message} Stack trace: {ex.StackTrace}");
success = false;
}
finally
{
conn.Close();
conn.Dispose();
}
output.Close();
return(success);
}
