7 static void exit_mysql_err(MYSQL *mysql) {
8 const char *err = mysql_error(mysql);
10 fprintf(stderr, "013_populate_node_tags_and_remove_helper: MySQL error: %s\n", err);
12 fprintf(stderr, "013_populate_node_tags_and_remove_helper: MySQL error\n");
18 static void write_csv_col(FILE *f, const char *str, char end) {
19 char *out = (char *) malloc(2 * strlen(str) + 4);
27 } else if (*str == '\"') {
39 if (fwrite(out, len, 1, f) != 1) {
47 static void unescape(char *str) {
48 char *i = str, *o = str;
54 case 's': *o++ = ';'; break;
55 case 'e': *o++ = '='; break;
56 case '\\': *o++ = '\\'; break;
64 static int read_node_tags(char **tags, char **k, char **v) {
65 if (!**tags) return 0;
66 char *i = strchr(*tags, ';');
67 if (!i) i = *tags + strlen(*tags);
68 char *j = strchr(*tags, '=');
75 *tags = *i ? i + 1 : i;
91 static void proc_nodes(struct data *d, const char *tbl, FILE *out, FILE *out_tags, int hist) {
96 snprintf(query, sizeof(query), "SELECT id, latitude, longitude, "
97 "user_id, visible, tags, timestamp, tile FROM %s", tbl);
98 if (mysql_query(d->mysql, query))
99 exit_mysql_err(d->mysql);
101 res = mysql_use_result(d->mysql);
102 if (!res) exit_mysql_err(d->mysql);
104 while ((row = mysql_fetch_row(res))) {
105 unsigned long id = strtoul(row[0], NULL, 10);
108 if (id > d->version_size) {
109 fprintf(stderr, "preallocated nodes size exceeded");
114 version = ++(d->version[id]);
116 fprintf(out, "\"%s\",\"%s\",\"%s\",\"%s\",\"%s\",\"%s\",\"%s\",\"%u\"\n",
117 row[0], row[1], row[2], row[3], row[4], row[6], row[7], version);
119 /*fprintf(out, "\"%s\",\"%s\",\"%s\",\"%s\",\"%s\",\"%s\",\"%s\"\n",
120 row[0], row[1], row[2], row[3], row[4], row[6], row[7]);*/
123 char *tags_it = row[5], *k, *v;
124 while (read_node_tags(&tags_it, &k, &v)) {
126 fprintf(out_tags, "\"%s\",\"%u\",", row[0], version);
128 fprintf(out_tags, "\"%s\",", row[0]);
131 write_csv_col(out_tags, k, ',');
132 write_csv_col(out_tags, v, '\n');
135 if (mysql_errno(d->mysql)) exit_mysql_err(d->mysql);
137 mysql_free_result(res);
140 static size_t select_size(MYSQL *mysql, const char *q) {
145 if (mysql_query(mysql, q))
146 exit_mysql_err(mysql);
148 res = mysql_store_result(mysql);
149 if (!res) exit_mysql_err(mysql);
151 row = mysql_fetch_row(res);
152 if (!row) exit_mysql_err(mysql);
155 ret = strtoul(row[0], NULL, 10);
160 mysql_free_result(res);
165 static MYSQL *connect_to_mysql(char **argv) {
166 MYSQL *mysql = mysql_init(NULL);
167 if (!mysql) exit_mysql_err(mysql);
169 if (!mysql_real_connect(mysql, argv[1], argv[2], argv[3], argv[4],
170 argv[5][0] ? atoi(argv[5]) : 0, argv[6][0] ? argv[6] : NULL, 0))
171 exit_mysql_err(mysql);
173 if (mysql_set_character_set(mysql, "utf8"))
174 exit_mysql_err(mysql);
179 static void open_file(FILE **f, char *fn) {
180 *f = fopen(fn, "w+");
187 int main(int argc, char **argv) {
189 FILE *current_nodes, *current_node_tags, *nodes, *node_tags;
191 struct data data, *d = &data;
194 printf("Usage: 013_populate_node_tags_and_remove_helper host user passwd database port socket prefix\n");
198 d->mysql = connect_to_mysql(argv);
200 d->version_size = 1 + select_size(d->mysql, "SELECT max(id) FROM current_nodes");
201 d->version = malloc(sizeof(uint32_t) * d->version_size);
203 prefix_len = strlen(argv[7]);
204 tempfn = (char *) malloc(prefix_len + 16);
205 strcpy(tempfn, argv[7]);
207 strcpy(tempfn + prefix_len, "current_nodes");
208 open_file(¤t_nodes, tempfn);
210 strcpy(tempfn + prefix_len, "current_node_tags");
211 open_file(¤t_node_tags, tempfn);
213 strcpy(tempfn + prefix_len, "nodes");
214 open_file(&nodes, tempfn);
216 strcpy(tempfn + prefix_len, "node_tags");
217 open_file(&node_tags, tempfn);
221 proc_nodes(d, "nodes", nodes, node_tags, 1);
222 proc_nodes(d, "current_nodes", current_nodes, current_node_tags, 0);
226 mysql_close(d->mysql);
228 fclose(current_nodes);
229 fclose(current_node_tags);