def updateDataset(self, request, context):
properties = None if request.properties == {} else request.properties
name_list = request.name_list
type_list = request.type_list
if not name_list:
name_list = None
if not type_list:
data_type_list = None
else:
data_type_list = []
for data_type in type_list:
data_type_list.append(DataType(DataTypeProto.Name(data_type)))
dataset_meta = self.store.update_dataset(dataset_name=request.name,
data_format=request.data_format.value if request.HasField(
'data_format') else None,
description=request.description.value if request.HasField(
'description') else None,
uri=request.uri.value if request.HasField('uri') else None,
properties=properties,
name_list=name_list,
type_list=data_type_list,
catalog_name=request.catalog_name.value if request.HasField(
'catalog_name') else None,
catalog_type=request.catalog_type.value if request.HasField(
'catalog_type') else None,
catalog_database=request.catalog_database.value if request.HasField(
'catalog_database') else None,
catalog_connection_uri=request.catalog_connection_uri.value \
if request.HasField('catalog_connection_uri') else None,
catalog_table=request.catalog_table.value if request.HasField(
'catalog_table') else None)
return _wrap_meta_response(
MetaToProto.dataset_meta_to_proto(dataset_meta))
def registerDatasetWithCatalog(self, request, context):
dataset = transform_dataset_meta(request.dataset)
dataset_meta = self.store.register_dataset_with_catalog(
name=dataset.name,
catalog_name=dataset.catalog_name,
catalog_type=dataset.catalog_type,
catalog_database=dataset.catalog_database,
catalog_connection_uri=dataset.catalog_connection_uri,
catalog_table=dataset.catalog_table)
return _wrap_meta_response(
MetaToProto.dataset_meta_to_proto(dataset_meta))
def registerDataset(self, request, context):
dataset = transform_dataset_meta(request.dataset)
dataset_meta = self.store.register_dataset(
name=dataset.name,
data_format=dataset.data_format,
description=dataset.description,
uri=dataset.uri,
properties=dataset.properties,
name_list=dataset.schema.name_list,
type_list=dataset.schema.type_list)
return _wrap_meta_response(
MetaToProto.dataset_meta_to_proto(dataset_meta))
def getDatasetByName(self, request, context):
dataset = self.store.get_dataset_by_name(request.name)
return _wrap_meta_response(MetaToProto.dataset_meta_to_proto(dataset))
