OmniSciDB  4201147b46
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
OdbcGeospatialEncoder.h
Go to the documentation of this file.
1 /*
2  * Copyright 2022 HEAVY.AI, Inc.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #pragma once
18 
19 #include "GeospatialEncoder.h"
20 
21 namespace foreign_storage {
22 
24  public:
25  OdbcGeospatialEncoder(std::list<Chunk_NS::Chunk>& chunks,
26  std::list<std::unique_ptr<ChunkMetadata>>& chunk_metadata,
27  const RenderGroupAnalyzerMap* render_group_analyzer_map,
28  const bool is_error_tracking_enabled,
29  const SQLTypeInfo& sql_type_info)
30  : GeospatialEncoder(chunks, chunk_metadata, render_group_analyzer_map)
31  , type_info_(sql_type_info)
32  , is_error_tracking_enabled_(is_error_tracking_enabled) {}
33 
34  void appendData(const std::vector<std::string>& geo_strings,
35  std::optional<std::set<size_t>>& rejected_row_local_indices) {
37  CHECK(rejected_row_local_indices.has_value());
38  }
40  int64_t num_rows = geo_strings.size();
41  for (int64_t i = 0; i < num_rows; ++i) {
43  auto const& geo_string = geo_strings[i];
44  if (geo_string.size()) {
45  try {
46  processGeoElement(geo_string);
47  } catch (const std::runtime_error& except) {
49  rejected_row_local_indices->insert(i);
52  } else {
53  throw except;
54  }
55  }
56  } else {
59  rejected_row_local_indices->insert(i);
60  }
61  }
62  }
65  }
66 
67  private:
70 };
71 
72 } // namespace foreign_storage
void processGeoElement(std::string_view geo_string_view)
void appendBaseAndRenderGroupDataAndUpdateMetadata(const int64_t row_count)
#define CHECK(condition)
Definition: Logger.h:222
std::map< int, std::unique_ptr< import_export::RenderGroupAnalyzer >> RenderGroupAnalyzerMap
HOST DEVICE bool get_notnull() const
Definition: sqltypes.h:336
void appendData(const std::vector< std::string > &geo_strings, std::optional< std::set< size_t >> &rejected_row_local_indices)
OdbcGeospatialEncoder(std::list< Chunk_NS::Chunk > &chunks, std::list< std::unique_ptr< ChunkMetadata >> &chunk_metadata, const RenderGroupAnalyzerMap *render_group_analyzer_map, const bool is_error_tracking_enabled, const SQLTypeInfo &sql_type_info)