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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
/* Implements the mid-layer processing for osm2pgsql
* using several PostgreSQL tables
*
* This layer stores data read in from the planet.osm file
* and is then read by the backend processing code to
* emit the final geometry-enabled output formats
*/
#ifndef MIDDLE_PGSQL_H
#define MIDDLE_PGSQL_H
#include "middle.hpp"
#include "node-ram-cache.hpp"
#include "node-persistent-cache.hpp"
#include "id-tracker.hpp"
#include <memory>
#include <vector>
struct middle_pgsql_t : public slim_middle_t {
middle_pgsql_t();
virtual ~middle_pgsql_t();
void start(const options_t *out_options_) override;
void stop(osmium::thread::Pool &pool) override;
void analyze(void) override;
void end(void) override;
void commit(void) override;
void nodes_set(osmium::Node const &node) override;
size_t nodes_get_list(osmium::WayNodeList *nodes) const override;
void nodes_delete(osmid_t id) override;
void node_changed(osmid_t id) override;
void ways_set(osmium::Way const &way) override;
bool ways_get(osmid_t id, osmium::memory::Buffer &buffer) const override;
size_t rel_way_members_get(osmium::Relation const &rel, rolelist_t *roles,
osmium::memory::Buffer &buffer) const override;
void ways_delete(osmid_t id) override;
void way_changed(osmid_t id) override;
bool relations_get(osmid_t id, osmium::memory::Buffer &buffer) const override;
void relations_set(osmium::Relation const &rel) override;
void relations_delete(osmid_t id) override;
void relation_changed(osmid_t id) override;
void iterate_ways(middle_t::pending_processor& pf) override;
void iterate_relations(pending_processor& pf) override;
size_t pending_count() const override;
idlist_t relations_using_way(osmid_t way_id) const override;
struct table_desc {
table_desc(const char *name_ = NULL,
const char *start_ = NULL,
const char *create_ = NULL,
const char *create_index_ = NULL,
const char *prepare_ = NULL,
const char *prepare_intarray_ = NULL,
const char *copy_ = NULL,
const char *analyze_ = NULL,
const char *stop_ = NULL,
const char *array_indexes_ = NULL);
const char *name;
const char *start;
const char *create;
const char *create_index;
const char *prepare;
const char *prepare_intarray;
const char *copy;
const char *analyze;
const char *stop;
const char *array_indexes;
int copyMode; /* True if we are in copy mode */
int transactionMode; /* True if we are in an extended transaction */
struct pg_conn *sql_conn;
};
std::shared_ptr<const middle_query_t> get_instance() const override;
private:
void pgsql_stop_one(table_desc *table);
/**
* Sets up sql_conn for the table
*/
void connect(table_desc& table);
void local_nodes_set(osmium::Node const &node);
size_t local_nodes_get_list(osmium::WayNodeList *nodes) const;
void local_nodes_delete(osmid_t osm_id);
std::vector<table_desc> tables;
int num_tables;
table_desc *node_table, *way_table, *rel_table;
bool append;
bool mark_pending;
std::shared_ptr<node_ram_cache> cache;
std::shared_ptr<node_persistent_cache> persistent_cache;
std::shared_ptr<id_tracker> ways_pending_tracker, rels_pending_tracker;
void buffer_store_string(std::string const &in, bool escape);
void buffer_store_tags(osmium::OSMObject const &obj, bool attrs, bool escape);
void buffer_correct_params(char const **param, size_t size);
bool build_indexes;
std::string copy_buffer;
};
#endif
|
