| 246 | } |
| 247 | |
| 248 | arrow::Result<int64_t> FlightSqlClient::ExecuteIngest( |
| 249 | const FlightCallOptions& options, const std::shared_ptr<RecordBatchReader>& reader, |
| 250 | const TableDefinitionOptions& table_definition_options, const std::string& table, |
| 251 | const std::optional<std::string>& schema, const std::optional<std::string>& catalog, |
| 252 | const bool temporary, const Transaction& transaction, |
| 253 | const std::unordered_map<std::string, std::string>& ingest_options) { |
| 254 | flight_sql_pb::CommandStatementIngest command; |
| 255 | |
| 256 | flight_sql_pb::CommandStatementIngest_TableDefinitionOptions* |
| 257 | pb_table_definition_options = |
| 258 | new flight_sql_pb::CommandStatementIngest_TableDefinitionOptions(); |
| 259 | switch (table_definition_options.if_not_exist) { |
| 260 | case TableDefinitionOptionsTableNotExistOption::kUnspecified: |
| 261 | pb_table_definition_options->set_if_not_exist( |
| 262 | flight_sql_pb:: |
| 263 | CommandStatementIngest_TableDefinitionOptions_TableNotExistOption_TABLE_NOT_EXIST_OPTION_UNSPECIFIED); // NOLINT(whitespace/line_length) |
| 264 | break; |
| 265 | case TableDefinitionOptionsTableNotExistOption::kCreate: |
| 266 | pb_table_definition_options->set_if_not_exist( |
| 267 | flight_sql_pb:: |
| 268 | CommandStatementIngest_TableDefinitionOptions_TableNotExistOption_TABLE_NOT_EXIST_OPTION_CREATE); // NOLINT(whitespace/line_length) |
| 269 | break; |
| 270 | case TableDefinitionOptionsTableNotExistOption::kFail: |
| 271 | pb_table_definition_options->set_if_not_exist( |
| 272 | flight_sql_pb:: |
| 273 | CommandStatementIngest_TableDefinitionOptions_TableNotExistOption_TABLE_NOT_EXIST_OPTION_FAIL); // NOLINT(whitespace/line_length) |
| 274 | break; |
| 275 | |
| 276 | default: |
| 277 | break; |
| 278 | } |
| 279 | |
| 280 | switch (table_definition_options.if_exists) { |
| 281 | case TableDefinitionOptionsTableExistsOption::kUnspecified: |
| 282 | pb_table_definition_options->set_if_exists( |
| 283 | flight_sql_pb:: |
| 284 | CommandStatementIngest_TableDefinitionOptions_TableExistsOption_TABLE_EXISTS_OPTION_UNSPECIFIED); // NOLINT(whitespace/line_length) |
| 285 | break; |
| 286 | case TableDefinitionOptionsTableExistsOption::kFail: |
| 287 | pb_table_definition_options->set_if_exists( |
| 288 | flight_sql_pb:: |
| 289 | CommandStatementIngest_TableDefinitionOptions_TableExistsOption_TABLE_EXISTS_OPTION_FAIL); // NOLINT(whitespace/line_length) |
| 290 | break; |
| 291 | case TableDefinitionOptionsTableExistsOption::kAppend: |
| 292 | pb_table_definition_options->set_if_exists( |
| 293 | flight_sql_pb:: |
| 294 | CommandStatementIngest_TableDefinitionOptions_TableExistsOption_TABLE_EXISTS_OPTION_APPEND); // NOLINT(whitespace/line_length) |
| 295 | break; |
| 296 | case TableDefinitionOptionsTableExistsOption::kReplace: |
| 297 | pb_table_definition_options->set_if_exists( |
| 298 | flight_sql_pb:: |
| 299 | CommandStatementIngest_TableDefinitionOptions_TableExistsOption_TABLE_EXISTS_OPTION_REPLACE); // NOLINT(whitespace/line_length) |
| 300 | break; |
| 301 | |
| 302 | default: |
| 303 | break; |
| 304 | } |
| 305 |
nothing calls this directly
no test coverage detected