按特定条件批量导出 Collection 中的数据
使用前请先授权 VikingDB 跨服务访问 TOS:去授权
public CreateVikingdbTaskResponse createVikingdbTask(CreateVikingdbTaskRequest body) throws ApiException
参数 | 子参数 | 类型 | 是否必填 | 描述 |
|---|---|---|---|---|
ProjectName | String | 否 | 项目名称 | |
CollectionName | String | 2选1 | 数据集名称 | |
ResourceId | String | 数据集资源ID。请求必须指定ResourceId和CollectionName其中之一。 | ||
TaskType | String | 是 | data_export | |
TaskConfig | TaskConfigForCreateVikingdbTaskInput | 是 | 任务具体配置 | |
FileType | String | 是 | 文件类型:json 或 parquet,必填 | |
FilterConds | List[Object] | 否 | 过滤条件。使用参考:https://www.volcengine.com/docs/84313/1791133
| |
TosPath | String | 是 | TOS 路径,格式:{桶名}/{路径},注意不是域名。必填 | |
ExportAll | Boolean | 否 | 是否导出全部数据,此时 FilterConds 不生效,默认为 false |
参数 | 类型 | 描述 |
|---|---|---|
taskId | String | 任务ID |
message | String | 操作结果信息 |
运行示例前,请完成以下准备:
- 按“安装与 client 初始化”文档在 Java 项目中引入 VikingDB V2 SDK,并将示例放入可执行的
main方法后编译运行。- AK/SK 在火山引擎控制台“密钥管理”中创建;Host 和 Region 取自地域和访问域名,并按代码读取方式配置相应环境变量。
- Collection、Index 名称从 VikingDB 控制台对应资源详情获取,Task ID 使用创建或查询任务接口返回的 ID。请将代码中的
<YOUR_...>占位符全部替换为真实值。- 若示例包含 TOS 路径,请从 TOS 控制台确认桶名和对象路径,并替换为真实的
<YOUR_BUCKET_NAME>/<YOUR_TOS_PATH>。
package org.example.newsubproduct.console.task; import com.volcengine.ApiClient; import com.volcengine.ApiException; import com.volcengine.sign.Credentials; import com.volcengine.vikingdb.VikingdbApi; import com.volcengine.vikingdb.model.*; public class CreateVikingdbTaskDataExport { public static void main(String[] args) { String ak = System.getenv("AK"); // ak String sk = System.getenv("SK"); // sk String endpoint = "vikingdb.cn-beijing.volcengineapi.com"; // 填写向量库控制面 V2 的域名:https://www.volcengine.com/docs/84313/1792715 String region = "cn-beijing"; // 服务区域 ApiClient apiClient = new ApiClient() .setEndpoint(endpoint) .setCredentials(Credentials.getCredentials(ak, sk)) .setRegion(region); VikingdbApi api = new VikingdbApi(apiClient); CreateVikingdbTaskRequest request = new CreateVikingdbTaskRequest() .collectionName("test_collection_for_sdk_with_vector") .taskType(CreateVikingdbTaskRequest.TaskTypeEnum.DATA_EXPORT) .taskConfig( new TaskConfigForCreateVikingdbTaskInput() .fileType(TaskConfigForCreateVikingdbTaskInput.FileTypeEnum.JSON) .tosPath("test-doc1-tos/output.json") .ignoreError(true) ); try { CreateVikingdbTaskResponse response = api.createVikingdbTask(request); System.out.println("response body: " + response); System.out.println(); System.out.println("response meta RequestId: " + response.getResponseMetadata().getRequestId()); System.out.println("response meta Service: " + response.getResponseMetadata().getService()); System.out.println("response meta Region: " + response.getResponseMetadata().getRegion()); System.out.println("response meta Action: " + response.getResponseMetadata().getAction()); System.out.println("response meta Version: " + response.getResponseMetadata().getVersion()); } catch (ApiException e) { System.out.println("exception http code: " + e.getCode()); System.out.println("exception response body: " + e.getResponseBody()); System.out.println(); System.out.println("exception RequestId: " + e.getResponseMetadata().getRequestId()); System.out.println("exception Action: " + e.getResponseMetadata().getAction()); System.out.println("exception Region: " + e.getResponseMetadata().getRegion()); System.out.println("exception Service: " + e.getResponseMetadata().getService()); System.out.println("exception Error.Code: " + e.getResponseMetadata().getError().getCode()); System.out.println("exception Error.Message: " + e.getResponseMetadata().getError().getMessage()); } } }
请在火山引擎控制台的密钥管理页面获取 AK 和 SK,并将示例中的 <YOUR_AK>、<YOUR_SK> 等占位符替换为真实配置。
运行示例前,请将其中的 TOS Bucket、对象路径和本地路径占位符替换为实际值。
import tos DOMAIN = "api-vikingdb.volces.com" TOS_ENDPOINT = "tos-cn-beijing.ivolces.com" REGION = "cn-beijing" AK = "****" SK = "****" COLLECTION_NAME = "example" BUCKET_NAME = "bucket_name" TOS_DIR = "tos_dir" def download(client, bucket_name, object_key, local_path): file_path = "{}/{}".format(local_path, object_key) try: client.get_object_to_file(bucket_name, object_key, file_path) except tos.exceptions.TosClientError as e: # 操作失败,捕获客户端异常,一般情况为非法请求参数或网络异常 return 'fail with client error, message:{}, cause: {}'.format(e.message, e.cause) except tos.exceptions.TosServerError as e: return 'fail with server error : {}'.format(e) except Exception as e: return 'fail with unknown error: {}'.format(e) return '' client = tos.TosClientV2(AK, SK, TOS_ENDPOINT, REGION) download(client, BUCKET_NAME, TOS_DIR, "./")
import pyarrow.parquet as pq def process_parquet(file_path): parquet_file = pq.ParquetFile(file_path) file_data_count = sum(parquet_file.metadata.row_group(i).num_rows for i in range(parquet_file.num_row_groups)) schema = parquet_file.schema.to_arrow_schema() row_iterator = parquet_file.iter_batches(batch_size=100) # 迭代读取数据 for batch in row_iterator: df = batch.to_pandas() for index, row in df.iterrows(): """ do what you want """ print(row) return '' process_parquet("1.parquet")