-
-
Notifications
You must be signed in to change notification settings - Fork 302
/
table.c
119 lines (101 loc) · 2.67 KB
/
table.c
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
116
117
118
119
/*!
* \file db/dbmi_client/table.c
*
* \brief DBMI Library (client) - table management
*
* (C) 1999-2008 by the GRASS Development Team
*
* This program is free software under the GNU General Public
* License (>=v2). Read the file COPYING that comes with GRASS
* for details.
*
* \author Joel Jones (CERL/UIUC), Radim Blazek
*/
#include <stdlib.h>
#include <string.h>
#include <grass/gis.h>
#include <grass/dbmi.h>
#include <grass/glocale.h>
/*!
\brief Check if table exists
\param drvname driver name
\param dbname database name
\param tabname table name
\return 1 exist
\return 0 doesn't exist
\return -1 error
*/
int db_table_exists(const char *drvname, const char *dbname, const char *tabname)
{
dbDriver *driver;
dbString *names;
int i, count, found = 0;
int full = 0;
char buf[1000];
char *bufp, *c;
if (strchr(tabname, '.'))
full = 1;
driver = db_start_driver_open_database(drvname, dbname);
if (driver == NULL) {
G_warning(_("Unable open database <%s> by driver <%s>"), dbname,
drvname);
return -1;
}
/* The table tabname can be either fully qualified in form table.schema,
* or it can be only table name. If the name is fully qualified, compare whole name,
* if it is not, compare only table names */
/* user tables */
if (db_list_tables(driver, &names, &count, 0) != DB_OK)
return (-1);
for (i = 0; i < count; i++) {
strcpy(buf, db_get_string(&names[i]));
bufp = buf;
if (!full && (c = strchr(buf, '.'))) {
bufp = c + 1;
}
G_debug(2, "table = %s -> %s", buf, bufp);
if (G_strcasecmp(tabname, bufp) == 0) {
found = 1;
break;
}
}
db_free_string_array(names, count);
if (!found) { /* system tables */
if (db_list_tables(driver, &names, &count, 1) != DB_OK)
return (-1);
for (i = 0; i < count; i++) {
strcpy(buf, db_get_string(&names[i]));
bufp = buf;
if (!full && (c = strchr(buf, '.'))) {
bufp = c + 1;
}
if (G_strcasecmp(tabname, bufp) == 0) {
found = 1;
break;
}
}
db_free_string_array(names, count);
}
db_close_database_shutdown_driver(driver);
return (found);
}
/*!
\brief Get number of rows of table
\param driver db driver
\param sql SQL statement
\return number of records
\return -1
*/
int db_get_table_number_of_rows(dbDriver * driver, dbString * sql)
{
int nrows;
dbCursor cursor;
if (db_open_select_cursor(driver, sql, &cursor, DB_SEQUENTIAL) != DB_OK) {
G_warning(_("Unable to open select cursor: '%s'"), db_get_string(sql));
db_close_database_shutdown_driver(driver);
return -1;
}
nrows = db_get_num_rows(&cursor);
db_close_cursor(&cursor);
return nrows;
}