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 => TypeCodec.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 void ParseTypeName_Should_Parse_Collections()
{
{
var type = TypeCodec.ParseTypeName(null, null, "list<int>").Result;
Assert.NotNull(type);
Assert.AreEqual(ColumnTypeCode.List, type.TypeCode);
var subTypeInfo = (ListColumnInfo)type.TypeInfo;
Assert.AreEqual(ColumnTypeCode.Int, subTypeInfo.ValueTypeCode);
}
{
var type = TypeCodec.ParseTypeName(null, null, "set<uuid>").Result;
Assert.NotNull(type);
Assert.AreEqual(ColumnTypeCode.Set, type.TypeCode);
var subTypeInfo = (SetColumnInfo)type.TypeInfo;
Assert.AreEqual(ColumnTypeCode.Uuid, subTypeInfo.KeyTypeCode);
}
{
var type = TypeCodec.ParseTypeName(null, null, "map<text, timeuuid>").Result;
Assert.NotNull(type);
Assert.AreEqual(ColumnTypeCode.Map, type.TypeCode);
var subTypeInfo = (MapColumnInfo)type.TypeInfo;
Assert.AreEqual(ColumnTypeCode.Text, subTypeInfo.KeyTypeCode);
Assert.AreEqual(ColumnTypeCode.Timeuuid, subTypeInfo.ValueTypeCode);
}
{
var type = TypeCodec.ParseTypeName(null, null, "map<text,frozen<list<int>>>").Result;
Assert.NotNull(type);
Assert.AreEqual(ColumnTypeCode.Map, type.TypeCode);
var subTypeInfo = (MapColumnInfo)type.TypeInfo;
Assert.AreEqual(ColumnTypeCode.Text, subTypeInfo.KeyTypeCode);
Assert.AreEqual(ColumnTypeCode.List, subTypeInfo.ValueTypeCode);
var subListTypeInfo = (ListColumnInfo)subTypeInfo.ValueTypeInfo;
Assert.AreEqual(ColumnTypeCode.Int, subListTypeInfo.ValueTypeCode);
}
}
public void ParseTypeName_Should_Parse_Single_Cql_Types()
{
var cqlNames = new Dictionary <string, ColumnTypeCode>
{
{ "varchar", ColumnTypeCode.Varchar },
{ "text", ColumnTypeCode.Text },
{ "ascii", ColumnTypeCode.Ascii },
{ "uuid", ColumnTypeCode.Uuid },
{ "timeuuid", ColumnTypeCode.Timeuuid },
{ "int", ColumnTypeCode.Int },
{ "blob", ColumnTypeCode.Blob },
{ "float", ColumnTypeCode.Float },
{ "double", ColumnTypeCode.Double },
{ "boolean", ColumnTypeCode.Boolean },
{ "inet", ColumnTypeCode.Inet },
{ "date", ColumnTypeCode.Date },
{ "time", ColumnTypeCode.Time },
{ "smallint", ColumnTypeCode.SmallInt },
{ "tinyint", ColumnTypeCode.TinyInt },
{ "timestamp", ColumnTypeCode.Timestamp },
{ "bigint", ColumnTypeCode.Bigint },
{ "decimal", ColumnTypeCode.Decimal },
{ "varint", ColumnTypeCode.Varint },
{ "counter", ColumnTypeCode.Counter }
};
foreach (var kv in cqlNames)
{
var type = TypeCodec.ParseTypeName(null, null, kv.Key).Result;
Assert.NotNull(type);
Assert.AreEqual(kv.Value, type.TypeCode);
Assert.Null(type.TypeInfo);
}
}
public void ParseTypeName_Should_Parse_Frozen_Cql_Types()
{
var cqlNames = new Dictionary <string, ColumnTypeCode>
{
{ "frozen<varchar>", ColumnTypeCode.Varchar },
{ "frozen<list<int>>", ColumnTypeCode.List },
{ "frozen<map<text,frozen<list<int>>>>", ColumnTypeCode.Map }
};
foreach (var kv in cqlNames)
{
var type = TypeCodec.ParseTypeName(null, null, kv.Key).Result;
Assert.NotNull(type);
Assert.AreEqual(kv.Value, type.TypeCode);
Assert.AreEqual(true, type.IsFrozen);
}
}
public override Task <AggregateMetadata> GetAggregate(string keyspaceName, string aggregateName, string signatureString)
{
var query = string.Format(SelectAggregates, keyspaceName, aggregateName, signatureString);
return(Cc
.QueryAsync(query, true)
.Then(rs =>
{
var row = rs.FirstOrDefault();
if (row == null)
{
return TaskHelper.ToTask <AggregateMetadata>(null);
}
var argumentTypes = row.GetValue <string[]>("argument_types") ?? new string[0];
//state_type + return_type + amount of argument types
var parseTasks = new Task <ColumnDesc> [2 + argumentTypes.Length];
parseTasks[0] = TypeCodec.ParseTypeName(_udtResolver, row.GetValue <string>("keyspace_name"), row.GetValue <string>("state_type"));
parseTasks[1] = TypeCodec.ParseTypeName(_udtResolver, row.GetValue <string>("keyspace_name"), row.GetValue <string>("return_type"));
for (var i = 0; i < argumentTypes.Length; i++)
{
parseTasks[2 + i] = TypeCodec.ParseTypeName(_udtResolver, row.GetValue <string>("keyspace_name"), argumentTypes[i]);
}
return Task.Factory.ContinueWhenAll(parseTasks, tasks =>
{
var ex = tasks.Select(t => t.Exception).FirstOrDefault(e => e != null);
if (ex != null)
{
throw ex.InnerException;
}
return new AggregateMetadata
{
Name = row.GetValue <string>("aggregate_name"),
KeyspaceName = row.GetValue <string>("keyspace_name"),
StateFunction = row.GetValue <string>("state_func"),
FinalFunction = row.GetValue <string>("final_func"),
InitialCondition = row.GetValue <string>("initcond"),
Signature = argumentTypes,
StateType = tasks[0].Result,
ReturnType = tasks[1].Result,
ArgumentTypes = tasks.Skip(2).Select(t => t.Result).ToArray()
};
}, TaskContinuationOptions.ExecuteSynchronously);
}));
}
public override Task <FunctionMetadata> GetFunction(string keyspaceName, string functionName, string signatureString)
{
var query = string.Format(SelectFunctions, keyspaceName, functionName, signatureString);
return(Cc
.QueryAsync(query, true)
.Then(rs =>
{
var row = rs.FirstOrDefault();
if (row == null)
{
return TaskHelper.ToTask <FunctionMetadata>(null);
}
var argumentTypes = row.GetValue <string[]>("argument_types") ?? new string[0];
var parseTasks = new Task <ColumnDesc> [1 + argumentTypes.Length];
parseTasks[0] = TypeCodec.ParseTypeName(_udtResolver, row.GetValue <string>("keyspace_name"), row.GetValue <string>("return_type"));
for (var i = 0; i < argumentTypes.Length; i++)
{
parseTasks[1 + i] = TypeCodec.ParseTypeName(_udtResolver, row.GetValue <string>("keyspace_name"), argumentTypes[i]);
}
return Task.Factory.ContinueWhenAll(parseTasks, tasks =>
{
var ex = tasks.Select(t => t.Exception).FirstOrDefault(e => e != null);
if (ex != null)
{
throw ex.InnerException;
}
return new FunctionMetadata
{
Name = row.GetValue <string>("function_name"),
KeyspaceName = row.GetValue <string>("keyspace_name"),
Signature = argumentTypes,
ArgumentNames = row.GetValue <string[]>("argument_names") ?? new string[0],
Body = row.GetValue <string>("body"),
CalledOnNullInput = row.GetValue <bool>("called_on_null_input"),
Language = row.GetValue <string>("language"),
ReturnType = tasks[0].Result,
ArgumentTypes = tasks.Skip(1).Select(t => t.Result).ToArray()
};
});
}));
}
private Task <T> ParseTableOrView <T>(Func <Row, T> newInstance, Task <IEnumerable <Row> > getTableTask, Task <IEnumerable <Row> > getColumnsTask)
where T : DataCollectionMetadata
{
var tableMetadataRow = getTableTask.Result.FirstOrDefault();
if (tableMetadataRow == null)
{
return(TaskHelper.ToTask <T>(null));
}
var columns = new Dictionary <string, TableColumn>();
var partitionKeys = new List <Tuple <int, TableColumn> >();
var clusteringKeys = new List <Tuple <int, Tuple <TableColumn, SortOrder> > >();
//Read table options
var options = new TableOptions
{
isCompactStorage = false,
bfFpChance = tableMetadataRow.GetValue <double>("bloom_filter_fp_chance"),
caching = "{" + string.Join(",", tableMetadataRow.GetValue <IDictionary <string, string> >("caching").Select(kv => "\"" + kv.Key + "\":\"" + kv.Value + "\"")) + "}",
comment = tableMetadataRow.GetValue <string>("comment"),
gcGrace = tableMetadataRow.GetValue <int>("gc_grace_seconds"),
localReadRepair = tableMetadataRow.GetValue <double>("dclocal_read_repair_chance"),
readRepair = tableMetadataRow.GetValue <double>("read_repair_chance"),
compactionOptions = tableMetadataRow.GetValue <SortedDictionary <string, string> >("compaction"),
compressionParams =
tableMetadataRow.GetValue <SortedDictionary <string, string> >("compression")
};
var columnsMetadata = getColumnsTask.Result;
Task <Tuple <TableColumn, Row> >[] columnTasks = columnsMetadata
.Select(row =>
{
return(TypeCodec.ParseTypeName(_udtResolver, tableMetadataRow.GetValue <string>("keyspace_name"), row.GetValue <string>("type")).
ContinueSync(type => Tuple.Create(new TableColumn
{
Name = row.GetValue <string>("column_name"),
Keyspace = row.GetValue <string>("keyspace_name"),
Table = row.GetValue <string>("table_name"),
TypeCode = type.TypeCode,
TypeInfo = type.TypeInfo
}, row)));
}).ToArray();
return(Task.Factory.ContinueWhenAll(columnTasks, tasks =>
{
var ex = tasks.Select(t => t.Exception).FirstOrDefault(e => e != null);
if (ex != null)
{
throw ex.InnerException;
}
foreach (var t in tasks)
{
var col = t.Result.Item1;
var row = t.Result.Item2;
switch (row.GetValue <string>("kind"))
{
case "partition_key":
partitionKeys.Add(Tuple.Create(row.GetValue <int?>("position") ?? 0, col));
break;
case "clustering":
clusteringKeys.Add(Tuple.Create(row.GetValue <int?>("position") ?? 0,
Tuple.Create(col, row.GetValue <string>("clustering_order") == "desc" ? SortOrder.Descending : SortOrder.Ascending)));
break;
}
columns.Add(col.Name, col);
}
if (typeof(T) == typeof(TableMetadata))
{
var flags = tableMetadataRow.GetValue <string[]>("flags");
var isDense = flags.Contains("dense");
var isSuper = flags.Contains("super");
var isCompound = flags.Contains("compound");
options.isCompactStorage = isSuper || isDense || !isCompound;
//remove the columns related to Thrift
var isStaticCompact = !isSuper && !isDense && !isCompound;
if (isStaticCompact)
{
PruneStaticCompactTableColumns(clusteringKeys, columns);
}
else if (isDense)
{
PruneDenseTableColumns(columns);
}
}
var result = newInstance(tableMetadataRow);
result.SetValues(columns,
partitionKeys.OrderBy(p => p.Item1).Select(p => p.Item2).ToArray(),
clusteringKeys.OrderBy(p => p.Item1).Select(p => p.Item2).ToArray(),
options);
return result;
}));
}