blob: be4597190c57bda8ded78d839208fd98e754b4ed (
plain) (
tree)
|
|
using Gee;
using Sqlite;
namespace Qlite {
public class Row {
private Map<string, string?> text_map = new HashMap<string, string?>();
private Map<string, long> int_map = new HashMap<string, long>();
private Map<string, double?> real_map = new HashMap<string, double?>();
internal Row(Statement stmt) {
for (int i = 0; i < stmt.column_count(); i++) {
switch(stmt.column_type(i)) {
case TEXT:
text_map[stmt.column_name(i)] = stmt.column_text(i);
break;
case INTEGER:
int_map[stmt.column_name(i)] = (long) stmt.column_int64(i);
break;
case FLOAT:
real_map[stmt.column_name(i)] = stmt.column_double(i);
break;
}
}
}
public T get<T>(Column<T> field) {
return field[this];
}
public string? get_text(string field) {
if (text_map.has_key(field)) {
return text_map[field];
}
return null;
}
public long get_integer(string field) {
return int_map[field];
}
public bool has_integer(string field) {
return int_map.has_key(field);
}
public double get_real(string field, double def = 0) {
return real_map[field] ?? def;
}
public bool has_real(string field) {
return real_map.has_key(field) && real_map[field] != null;
}
}
public class RowIterator {
private Database db;
private Statement stmt;
public RowIterator.from_query_builder(Database db, QueryBuilder query) {
this.db = db;
this.stmt = query.prepare();
}
public RowIterator(Database db, string sql, string[]? args = null) {
this.db = db;
this.stmt = db.prepare(sql);
if (args != null) {
for (int i = 0; i < args.length; i++) {
stmt.bind_text(i, sql, sql.length);
}
}
}
public bool next() {
int r = stmt.step();
if (r == Sqlite.ROW) return true;
if (r == Sqlite.DONE) return false;
print(@"SQLite error: $(db.errcode()) - $(db.errmsg())\n");
return false;
}
public Row get() {
return new Row(stmt);
}
public Row? get_next() {
if (next()) return get();
return null;
}
}
public class RowOption {
public Row? inner { get; private set; }
public RowOption(Row? row) {
this.inner = row;
}
public bool is_present() {
return inner != null;
}
public T get<T>(Column<T> field, T def = null) {
if (inner == null || field.is_null((!)inner)) return def;
return field[(!)inner];
}
internal long get_integer(string field, long def = 0) {
if (inner == null || !((!)inner).has_integer(field)) return def;
return ((!)inner).get_integer(field);
}
}
}
|