public void DefineObjectShape_WhenCalledWithObject_ReturnsObjectShapeDefinition()
{
var node = new SqlTable(null, null, "products", "products", "products", new Dictionary <string, object>(),
true);
node.AddColumn("id", "id", "id", true);
node.AddColumn("name", "name", "name");
var variantsTable = node.AddTable(null, "variants", "variants", "variants", new Dictionary <string, object>(), true);
variantsTable.AddColumn("id", "id", "id", true);
variantsTable.AddColumn("name", "name", "name");
variantsTable.SortKey = new SortKey("products", "sortOrder", "sortOrder", typeof(int), SortDirection.Ascending);
var colorsTable = variantsTable.AddTable(null, "colors", "color", "color", new Dictionary <string, object>(), true);
colorsTable.AddColumn("id", "id", "id", true);
colorsTable.AddColumn("color", "color", "color");
var objectShaper = new ObjectShaper(new SqlAstValidator());
var definition = objectShaper.DefineObjectShape(node);
var json = JsonConvert.SerializeObject(definition, new JsonSerializerSettings
{
ContractResolver = new CamelCasePropertyNamesContractResolver()
});
json.Should()
.Be("{\"properties\":[{\"name\":\"id\",\"column\":\"id\",\"isId\":true},{\"name\":\"name\",\"column\":\"name\",\"isId\":false},{\"name\":\"variants\",\"properties\":[{\"name\":\"sortOrder\",\"column\":\"variants__sortOrder\",\"isId\":false},{\"name\":\"id\",\"column\":\"variants__id\",\"isId\":true},{\"name\":\"name\",\"column\":\"variants__name\",\"isId\":false},{\"name\":\"color\",\"properties\":[{\"name\":\"id\",\"column\":\"variants__color__id\",\"isId\":true},{\"name\":\"color\",\"column\":\"variants__color__color\",\"isId\":false}]}]}]}");
}
private async Task NextBatchChild(SqlTable sqlTable, object?data, DatabaseCallDelegate databaseCall,
IResolveFieldContext context, CancellationToken cancellationToken)
{
var fieldName = sqlTable.FieldName;
data = data switch
{
List <object> objList => objList.Select(x => (IDictionary <string, object?>)x),
Connection <object> connection => connection.Items.Select(x => (IDictionary <string, object?>)x),
_ => data
};
// see if any begin a new batch
if (sqlTable.Batch != null || sqlTable.Junction?.Batch != null)
{
string thisKey = null !;
string parentKey = null !;
if (sqlTable.Batch != null)
{
// if so, we know we'll need to get the key for matching with the parent key
sqlTable.AddColumn(sqlTable.Batch.ThisKey);
thisKey = sqlTable.Batch.ThisKey.FieldName;
parentKey = sqlTable.Batch.ParentKey.FieldName;
}
else if (sqlTable.Junction?.Batch != null)
{
sqlTable.AddColumn(sqlTable.Junction.Batch.ThisKey);
thisKey = sqlTable.Junction.Batch.ThisKey.FieldName;
parentKey = sqlTable.Junction.Batch.ParentKey.FieldName;
}
if (data is IEnumerable <IDictionary <string, object?> > entries)
{
// the "batch scope" is the set of values to match this key against from the previous batch
var batchScope = new List <object>();
var entryList = entries.ToList();
foreach (var entry in entryList)
{
var values = PrepareValues(entry, parentKey);
batchScope.AddRange(values);
}
if (batchScope.Count == 0)
{
return;
}
// generate the SQL, with the batch scope values incorporated in a WHERE IN clause
var sqlResult = _compiler.Compile(sqlTable, context, SqlDialect.CastArray(batchScope));
var objectShape = _objectShaper.DefineObjectShape(sqlTable);
// grab the data
var newData = await HandleDatabaseCall(databaseCall, sqlResult, objectShape, cancellationToken);
// group the rows by the key so we can match them with the previous batch
var newDataGrouped = newData.GroupBy(x => x[thisKey])
.ToDictionary(x => x.Key, x => x.ToList());
// but if we paginate, we must convert to connection type first
if (sqlTable.Paginate)
{
//TODO: implement
// foreach (var group in newDataGrouped)
// newDataGrouped[group.Key] =
// (List<Dictionary<string, object?>>)
// _arrayToConnectionConverter.Convert(group.Value, sqlTable, context);
}
// if we they want many rows, give them an array
if (sqlTable.GrabMany)
{
foreach (var entry in entryList)
{
var values = PrepareValues(entry, parentKey);
var res = new List <Dictionary <string, object?> >();
foreach (var value in values)
{
if (newDataGrouped.TryGetValue(value, out var obj))
{
res.AddRange(obj);
}
}
entry[fieldName] = res;
}
}
else
{
var matchedData = new List <object>();
foreach (var entry in entryList)
{
if (entry.TryGetValue(parentKey, out var key) == false)
{
continue;
}
if (newDataGrouped.TryGetValue(key, out var list) && list.Count > 0)
{
entry[fieldName] = _arrayToConnectionConverter.Convert(list[0], sqlTable, context);
matchedData.Add(entry);
}
else
{
entry[fieldName] = null;
}
}
data = matchedData;
}
switch (data)
{
case IEnumerable <IDictionary <string, object?> > list:
{
var nextLevelData = list
.Where(x => x.Count > 0)
.Select(x =>
{
if (x.TryGetValue(fieldName, out var value) &&
value is List <Dictionary <string, object?> > dict)
{
return(dict);
}
return(null);
})
.Where(x => x is { Count: > 0 })
.SelectMany(x => x)
.Select(x => x.ToDictionary())
.ToList();
await NextBatch(sqlTable, nextLevelData, databaseCall, context, cancellationToken);
return;
}