static int
l_schema_name(lua_State *L)
{
avro_schema_t schema = lua_avro_get_raw_schema(L, 1);
lua_pushstring(L, avro_schema_type_name(schema));
return 1;
}
int
avro_schema_union_append(const avro_schema_t union_schema,
const avro_schema_t schema)
{
check_param(EINVAL, is_avro_schema(union_schema), "union schema");
check_param(EINVAL, is_avro_union(union_schema), "union schema");
check_param(EINVAL, is_avro_schema(schema), "schema");
struct avro_union_schema_t *unionp = avro_schema_to_union(union_schema);
int new_index = unionp->branches->num_entries;
st_insert(unionp->branches, new_index, (st_data_t) schema);
const char *name = avro_schema_type_name(schema);
st_insert(unionp->branches_byname, (st_data_t) name,
(st_data_t) new_index);
avro_schema_incref(schema);
return 0;
}
const char *avro_schema_type_name(const avro_schema_t schema)
{
if (is_avro_record(schema)) {
return (avro_schema_to_record(schema))->name;
} else if (is_avro_enum(schema)) {
return (avro_schema_to_enum(schema))->name;
} else if (is_avro_fixed(schema)) {
return (avro_schema_to_fixed(schema))->name;
} else if (is_avro_union(schema)) {
return "union";
} else if (is_avro_array(schema)) {
return "array";
} else if (is_avro_map(schema)) {
return "map";
} else if (is_avro_int32(schema)) {
return "int";
} else if (is_avro_int64(schema)) {
return "long";
} else if (is_avro_float(schema)) {
return "float";
} else if (is_avro_double(schema)) {
return "double";
} else if (is_avro_boolean(schema)) {
return "boolean";
} else if (is_avro_null(schema)) {
return "null";
} else if (is_avro_string(schema)) {
return "string";
} else if (is_avro_bytes(schema)) {
return "bytes";
} else if (is_avro_link(schema)) {
avro_schema_t target = avro_schema_link_target(schema);
return avro_schema_type_name(target);
}
avro_set_error("Unknown schema type");
return NULL;
}
avro_schema_t avro_schema_get_subschema(const avro_schema_t schema,
const char *name)
{
if (is_avro_record(schema)) {
const struct avro_record_schema_t *rschema =
avro_schema_to_record(schema);
union {
st_data_t data;
struct avro_record_field_t *field;
} field;
if (st_lookup(rschema->fields_byname,
(st_data_t) name, &field.data))
{
return field.field->type;
}
avro_set_error("No record field named %s", name);
return NULL;
} else if (is_avro_union(schema)) {
const struct avro_union_schema_t *uschema =
avro_schema_to_union(schema);
long i;
for (i = 0; i < uschema->branches->num_entries; i++) {
union {
st_data_t data;
avro_schema_t schema;
} val;
st_lookup(uschema->branches, i, &val.data);
if (strcmp(avro_schema_type_name(val.schema),
name) == 0)
{
return val.schema;
}
}
avro_set_error("No union branch named %s", name);
return NULL;
} else if (is_avro_array(schema)) {
if (strcmp(name, "[]") == 0) {
const struct avro_array_schema_t *aschema =
avro_schema_to_array(schema);
return aschema->items;
}
avro_set_error("Array subschema must be called \"[]\"");
return NULL;
} else if (is_avro_map(schema)) {
if (strcmp(name, "{}") == 0) {
const struct avro_map_schema_t *mschema =
avro_schema_to_map(schema);
return mschema->values;
}
avro_set_error("Map subschema must be called \"{}\"");
return NULL;
}
avro_set_error("Can only retrieve subschemas from record, union, array, or map");
return NULL;
}
