public override Task <UdtColumnInfo> GetUdtDefinition(string keyspaceName, string typeName)
{
return(Cc
.QueryAsync(string.Format(SelectUdts, keyspaceName, typeName), true)
.Then(rs =>
{
var row = rs.FirstOrDefault();
if (row == null)
{
return TaskHelper.ToTask <UdtColumnInfo>(null);
}
var udt = new UdtColumnInfo(row.GetValue <string>("keyspace_name") + "." + row.GetValue <string>("type_name"));
var fieldTypeTasks = row.GetValue <string[]>("field_types")
.Select(name => DataTypeParser.ParseTypeName(_udtResolver, keyspaceName, name))
.ToArray();
return Task.Factory.ContinueWhenAll(fieldTypeTasks, tasks =>
{
var ex = tasks.Select(t => t.Exception).FirstOrDefault(e => e != null);
if (ex != null)
{
throw ex.InnerException;
}
var fieldNames = row.GetValue <string[]>("field_names");
for (var i = 0; i < fieldNames.Length && i < tasks.Length; i++)
{
var field = tasks[i].Result;
field.Name = fieldNames[i];
udt.Fields.Add(field);
}
return udt;
});
}));
}
public override Task <UdtColumnInfo> GetUdtDefinition(string keyspaceName, string typeName)
{
return(Cc
.QueryAsync(string.Format(SelectUdts, keyspaceName, typeName), true)
.ContinueSync(rs =>
{
var row = rs.FirstOrDefault();
if (row == null)
{
return null;
}
var udt = new UdtColumnInfo(row.GetValue <string>("keyspace_name") + "." + row.GetValue <string>("type_name"));
var fieldNames = row.GetValue <string[]>("field_names");
var fieldTypes = row.GetValue <string[]>("field_types");
for (var i = 0; i < fieldNames.Length && i < fieldTypes.Length; i++)
{
var field = DataTypeParser.ParseFqTypeName(fieldTypes[i]);
field.Name = fieldNames[i];
udt.Fields.Add(field);
}
return udt;
}));
}
private IColumnInfo GetColumnInfo(FrameReader reader, ColumnTypeCode code)
{
ColumnTypeCode innercode;
switch (code)
{
case ColumnTypeCode.List:
innercode = (ColumnTypeCode)reader.ReadUInt16();
return(new ListColumnInfo
{
ValueTypeCode = innercode,
ValueTypeInfo = GetColumnInfo(reader, innercode)
});
case ColumnTypeCode.Map:
innercode = (ColumnTypeCode)reader.ReadUInt16();
IColumnInfo kci = GetColumnInfo(reader, innercode);
var vinnercode = (ColumnTypeCode)reader.ReadUInt16();
IColumnInfo vci = GetColumnInfo(reader, vinnercode);
return(new MapColumnInfo
{
KeyTypeCode = innercode,
KeyTypeInfo = kci,
ValueTypeCode = vinnercode,
ValueTypeInfo = vci
});
case ColumnTypeCode.Set:
innercode = (ColumnTypeCode)reader.ReadUInt16();
return(new SetColumnInfo
{
KeyTypeCode = innercode,
KeyTypeInfo = GetColumnInfo(reader, innercode)
});
case ColumnTypeCode.Custom:
return(new CustomColumnInfo {
CustomTypeName = reader.ReadString()
});
case ColumnTypeCode.Udt:
var udtInfo = new UdtColumnInfo(reader.ReadString() + "." + reader.ReadString());
var fieldLength = reader.ReadInt16();
for (var i = 0; i < fieldLength; i++)
{
var dataType = new ColumnDesc
{
Name = reader.ReadString(),
TypeCode = (ColumnTypeCode)reader.ReadUInt16(),
};
dataType.TypeInfo = GetColumnInfo(reader, dataType.TypeCode);
udtInfo.Fields.Add(dataType);
}
return(udtInfo);
case ColumnTypeCode.Tuple:
{
var tupleInfo = new TupleColumnInfo();
var elementLength = reader.ReadInt16();
for (var i = 0; i < elementLength; i++)
{
var dataType = new ColumnDesc
{
TypeCode = (ColumnTypeCode)reader.ReadUInt16(),
};
dataType.TypeInfo = GetColumnInfo(reader, dataType.TypeCode);
tupleInfo.Elements.Add(dataType);
}
return(tupleInfo);
}
default:
return(null);
}
}
