forked from osm2pgsql-dev/osm2pgsql
-
Notifications
You must be signed in to change notification settings - Fork 0
/
table.hpp
100 lines (82 loc) · 2.96 KB
/
table.hpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
#ifndef TABLE_H
#define TABLE_H
#include "db-copy.hpp"
#include "osmtypes.hpp"
#include "pgsql.hpp"
#include "taginfo.hpp"
#include <cstddef>
#include <string>
#include <vector>
#include <utility>
#include <memory>
#include <boost/optional.hpp>
typedef std::vector<std::string> hstores_t;
class table_t
{
public:
table_t(std::string const &name, std::string const &type,
columns_t const &columns, hstores_t const &hstore_columns,
const int srid, const bool append, const int hstore_mode,
std::shared_ptr<db_copy_thread_t> const ©_thread);
table_t(const table_t &other,
std::shared_ptr<db_copy_thread_t> const ©_thread);
~table_t();
void start(std::string const &conninfo,
boost::optional<std::string> const &table_space);
void stop(bool updateable, bool enable_hstore_index,
boost::optional<std::string> const &table_space_index);
void commit();
void write_row(osmid_t id, taglist_t const &tags, std::string const &geom);
void delete_row(const osmid_t id);
//interface from retrieving well known binary geometry from the table
class wkb_reader
{
friend table_t;
public:
const char* get_next()
{
if (m_current < m_count) {
return PQgetvalue(m_result.get(), m_current++, 0);
}
return nullptr;
}
int get_count() const { return m_count; }
void reset()
{
//NOTE: PQgetvalue doc doesn't say if you can call it
// multiple times with the same row col
m_current = 0;
}
private:
wkb_reader(pg_result_t &&result)
: m_result(std::move(result)), m_count(PQntuples(m_result.get())),
m_current(0)
{}
pg_result_t m_result;
int m_count;
int m_current;
};
wkb_reader get_wkb_reader(const osmid_t id);
protected:
void connect();
void prepare();
void teardown();
void write_columns(taglist_t const &tags, std::vector<bool> *used);
void write_tags_column(taglist_t const &tags,
std::vector<bool> const &used);
void write_hstore_columns(taglist_t const &tags);
void escape_type(std::string const &value, ColumnType flags);
void generate_copy_column_list();
std::string m_conninfo;
std::shared_ptr<db_target_descr_t> m_target;
std::string type;
pg_conn *sql_conn;
std::string srid;
bool append;
int hstore_mode;
columns_t columns;
hstores_t hstore_columns;
std::string m_table_space;
db_copy_mgr_t m_copy;
};
#endif