File: sqlite_resultset.hpp

package info (click to toggle)
mapnik 4.1.4%2Bds-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 18,580 kB
  • sloc: cpp: 163,826; python: 1,265; sh: 690; xml: 161; makefile: 123; perl: 28; lisp: 13
file content (112 lines) | stat: -rw-r--r-- 3,246 bytes parent folder | download
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
/*****************************************************************************
 *
 * This file is part of Mapnik (c++ mapping toolkit)
 *
 * Copyright (C) 2025 Artem Pavlenko
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 *
 *****************************************************************************/

#ifndef MAPNIK_SQLITE_RESULTSET_HPP
#define MAPNIK_SQLITE_RESULTSET_HPP

// mapnik
#include <mapnik/debug.hpp>
#include <mapnik/datasource.hpp>
#include <mapnik/params.hpp>

// stl
#include <string.h>

// sqlite
extern "C" {
#include <sqlite3.h>
}

//==============================================================================

class sqlite_resultset
{
  public:

    sqlite_resultset(sqlite3_stmt* stmt)
        : stmt_(stmt)
    {}

    ~sqlite_resultset()
    {
        if (stmt_)
        {
            sqlite3_finalize(stmt_);
        }
    }

    bool is_valid() { return stmt_ != 0; }

    bool step_next()
    {
        int const status = sqlite3_step(stmt_);
        if (status != SQLITE_ROW && status != SQLITE_DONE)
        {
            std::ostringstream s;
            s << "SQLite Plugin: retrieving next row failed";
            std::string msg(sqlite3_errmsg(sqlite3_db_handle(stmt_)));
            if (msg != "unknown error")
            {
                s << ": " << msg;
            }

            throw mapnik::datasource_exception(s.str());
        }
        return status == SQLITE_ROW;
    }

    int column_count() { return sqlite3_column_count(stmt_); }

    int column_type(int col) { return sqlite3_column_type(stmt_, col); }

    char const* column_name(int col) { return sqlite3_column_name(stmt_, col); }

    bool column_isnull(int col) { return sqlite3_column_type(stmt_, col) == SQLITE_NULL; }

    int column_integer(int col) { return sqlite3_column_int(stmt_, col); }

    sqlite_int64 column_integer64(int col) { return sqlite3_column_int64(stmt_, col); }

    double column_double(int col) { return sqlite3_column_double(stmt_, col); }

    char const* column_text(int col, int& len)
    {
        len = sqlite3_column_bytes(stmt_, col);
        return (char const*)sqlite3_column_text(stmt_, col);
    }

    char const* column_text(int col) { return (char const*)sqlite3_column_text(stmt_, col); }

    char const* column_blob(int col, int& bytes)
    {
        bytes = sqlite3_column_bytes(stmt_, col);
        return (char const*)sqlite3_column_blob(stmt_, col);
    }

    sqlite3_stmt* get_statement() { return stmt_; }

  private:

    sqlite3_stmt* stmt_;
};

#endif // MAPNIK_SQLITE_RESULTSET_HPP