@rocicorp/zero-sqlite3 1.0.8 → 1.0.10

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -0,0 +1,113 @@
1
+ StatementIterator::StatementIterator(Statement* stmt, bool bound) :
2
+ node::ObjectWrap(),
3
+ stmt(stmt),
4
+ handle(stmt->handle),
5
+ db_state(stmt->db->GetState()),
6
+ bound(bound),
7
+ safe_ints(stmt->safe_ints),
8
+ mode(stmt->mode),
9
+ alive(true),
10
+ logged(!db_state->has_logger) {
11
+ assert(stmt != NULL);
12
+ assert(handle != NULL);
13
+ assert(stmt->bound == bound);
14
+ assert(stmt->alive == true);
15
+ assert(stmt->locked == false);
16
+ assert(db_state->iterators < USHRT_MAX);
17
+ stmt->locked = true;
18
+ db_state->iterators += 1;
19
+ }
20
+
21
+ // The ~Statement destructor currently covers any state this object creates.
22
+ // Additionally, we actually DON'T want to revert stmt->locked or db_state
23
+ // ->iterators in this destructor, to ensure deterministic database access.
24
+ StatementIterator::~StatementIterator() {}
25
+
26
+ void StatementIterator::Next(NODE_ARGUMENTS info) {
27
+ assert(alive == true);
28
+ db_state->busy = true;
29
+ if (!logged) {
30
+ logged = true;
31
+ if (stmt->db->Log(OnlyIsolate, handle)) {
32
+ db_state->busy = false;
33
+ Throw();
34
+ return;
35
+ }
36
+ }
37
+ int status = sqlite3_step(handle);
38
+ db_state->busy = false;
39
+ if (status == SQLITE_ROW) {
40
+ UseIsolate;
41
+ UseContext;
42
+ info.GetReturnValue().Set(
43
+ NewRecord(isolate, ctx, Data::GetRowJS(isolate, ctx, handle, safe_ints, mode), db_state->addon, false)
44
+ );
45
+ } else {
46
+ if (status == SQLITE_DONE) Return(info);
47
+ else Throw();
48
+ }
49
+ }
50
+
51
+ void StatementIterator::Return(NODE_ARGUMENTS info) {
52
+ Cleanup();
53
+ STATEMENT_RETURN_LOGIC(DoneRecord(OnlyIsolate, db_state->addon));
54
+ }
55
+
56
+ void StatementIterator::Throw() {
57
+ Cleanup();
58
+ Database* db = stmt->db;
59
+ STATEMENT_THROW_LOGIC();
60
+ }
61
+
62
+ void StatementIterator::Cleanup() {
63
+ assert(alive == true);
64
+ alive = false;
65
+ stmt->locked = false;
66
+ db_state->iterators -= 1;
67
+ sqlite3_reset(handle);
68
+ }
69
+
70
+ INIT(StatementIterator::Init) {
71
+ v8::Local<v8::FunctionTemplate> t = NewConstructorTemplate(isolate, data, JS_new, "StatementIterator");
72
+ SetPrototypeMethod(isolate, data, t, "next", JS_next);
73
+ SetPrototypeMethod(isolate, data, t, "return", JS_return);
74
+ SetPrototypeSymbolMethod(isolate, data, t, v8::Symbol::GetIterator(isolate), JS_symbolIterator);
75
+ return t->GetFunction(OnlyContext).ToLocalChecked();
76
+ }
77
+
78
+ NODE_METHOD(StatementIterator::JS_new) {
79
+ UseAddon;
80
+ if (!addon->privileged_info) return ThrowTypeError("Disabled constructor");
81
+ assert(info.IsConstructCall());
82
+
83
+ StatementIterator* iter;
84
+ {
85
+ NODE_ARGUMENTS info = *addon->privileged_info;
86
+ STATEMENT_START_LOGIC(REQUIRE_STATEMENT_RETURNS_DATA, DOES_ADD_ITERATOR);
87
+ iter = new StatementIterator(stmt, bound);
88
+ }
89
+ UseIsolate;
90
+ UseContext;
91
+ iter->Wrap(info.This());
92
+ SetFrozen(isolate, ctx, info.This(), addon->cs.statement, addon->privileged_info->This());
93
+
94
+ info.GetReturnValue().Set(info.This());
95
+ }
96
+
97
+ NODE_METHOD(StatementIterator::JS_next) {
98
+ StatementIterator* iter = Unwrap<StatementIterator>(info.This());
99
+ REQUIRE_DATABASE_NOT_BUSY(iter->db_state);
100
+ if (iter->alive) iter->Next(info);
101
+ else info.GetReturnValue().Set(DoneRecord(OnlyIsolate, iter->db_state->addon));
102
+ }
103
+
104
+ NODE_METHOD(StatementIterator::JS_return) {
105
+ StatementIterator* iter = Unwrap<StatementIterator>(info.This());
106
+ REQUIRE_DATABASE_NOT_BUSY(iter->db_state);
107
+ if (iter->alive) iter->Return(info);
108
+ else info.GetReturnValue().Set(DoneRecord(OnlyIsolate, iter->db_state->addon));
109
+ }
110
+
111
+ NODE_METHOD(StatementIterator::JS_symbolIterator) {
112
+ info.GetReturnValue().Set(info.This());
113
+ }
@@ -0,0 +1,50 @@
1
+ class StatementIterator : public node::ObjectWrap {
2
+ public:
3
+
4
+ // The ~Statement destructor currently covers any state this object creates.
5
+ // Additionally, we actually DON'T want to revert stmt->locked or db_state
6
+ // ->iterators in this destructor, to ensure deterministic database access.
7
+ ~StatementIterator();
8
+
9
+ static INIT(Init);
10
+
11
+ private:
12
+
13
+ explicit StatementIterator(Statement* stmt, bool bound);
14
+
15
+ void Next(NODE_ARGUMENTS info);
16
+ void Return(NODE_ARGUMENTS info);
17
+ void Throw();
18
+ void Cleanup();
19
+
20
+ static inline v8::Local<v8::Object> NewRecord(
21
+ v8::Isolate* isolate,
22
+ v8::Local<v8::Context> ctx,
23
+ v8::Local<v8::Value> value,
24
+ Addon* addon,
25
+ bool done
26
+ ) {
27
+ v8::Local<v8::Object> record = v8::Object::New(isolate);
28
+ record->Set(ctx, addon->cs.value.Get(isolate), value).FromJust();
29
+ record->Set(ctx, addon->cs.done.Get(isolate), v8::Boolean::New(isolate, done)).FromJust();
30
+ return record;
31
+ }
32
+
33
+ static inline v8::Local<v8::Object> DoneRecord(v8::Isolate* isolate, Addon* addon) {
34
+ return NewRecord(isolate, OnlyContext, v8::Undefined(isolate), addon, true);
35
+ }
36
+
37
+ static NODE_METHOD(JS_new);
38
+ static NODE_METHOD(JS_next);
39
+ static NODE_METHOD(JS_return);
40
+ static NODE_METHOD(JS_symbolIterator);
41
+
42
+ Statement* const stmt;
43
+ sqlite3_stmt* const handle;
44
+ Database::State* const db_state;
45
+ const bool bound;
46
+ const bool safe_ints;
47
+ const char mode;
48
+ bool alive;
49
+ bool logged;
50
+ };
@@ -0,0 +1,446 @@
1
+ Statement::Statement(
2
+ Database* db,
3
+ sqlite3_stmt* handle,
4
+ sqlite3_uint64 id,
5
+ bool returns_data
6
+ ) :
7
+ node::ObjectWrap(),
8
+ db(db),
9
+ handle(handle),
10
+ extras(new Extras(id)),
11
+ alive(true),
12
+ locked(false),
13
+ bound(false),
14
+ has_bind_map(false),
15
+ safe_ints(db->GetState()->safe_ints),
16
+ mode(Data::FLAT),
17
+ returns_data(returns_data) {
18
+ assert(db != NULL);
19
+ assert(handle != NULL);
20
+ assert(db->GetState()->open);
21
+ assert(!db->GetState()->busy);
22
+ db->AddStatement(this);
23
+ }
24
+
25
+ Statement::~Statement() {
26
+ if (alive) db->RemoveStatement(this);
27
+ CloseHandles();
28
+ delete extras;
29
+ }
30
+
31
+ // Whenever this is used, db->RemoveStatement must be invoked beforehand.
32
+ void Statement::CloseHandles() {
33
+ if (alive) {
34
+ alive = false;
35
+ sqlite3_finalize(handle);
36
+ }
37
+ }
38
+
39
+ // Returns the Statement's bind map (creates it upon first execution).
40
+ BindMap* Statement::GetBindMap(v8::Isolate* isolate) {
41
+ if (has_bind_map) return &extras->bind_map;
42
+ BindMap* bind_map = &extras->bind_map;
43
+ int param_count = sqlite3_bind_parameter_count(handle);
44
+ for (int i = 1; i <= param_count; ++i) {
45
+ const char* name = sqlite3_bind_parameter_name(handle, i);
46
+ if (name != NULL) bind_map->Add(isolate, name + 1, i);
47
+ }
48
+ has_bind_map = true;
49
+ return bind_map;
50
+ }
51
+
52
+ Statement::Extras::Extras(sqlite3_uint64 id)
53
+ : bind_map(0), id(id) {}
54
+
55
+ INIT(Statement::Init) {
56
+ v8::Local<v8::FunctionTemplate> t = NewConstructorTemplate(isolate, data, JS_new, "Statement");
57
+ SetPrototypeMethod(isolate, data, t, "run", JS_run);
58
+ SetPrototypeMethod(isolate, data, t, "get", JS_get);
59
+ SetPrototypeMethod(isolate, data, t, "all", JS_all);
60
+ SetPrototypeMethod(isolate, data, t, "iterate", JS_iterate);
61
+ SetPrototypeMethod(isolate, data, t, "bind", JS_bind);
62
+ SetPrototypeMethod(isolate, data, t, "pluck", JS_pluck);
63
+ SetPrototypeMethod(isolate, data, t, "expand", JS_expand);
64
+ SetPrototypeMethod(isolate, data, t, "raw", JS_raw);
65
+ SetPrototypeMethod(isolate, data, t, "safeIntegers", JS_safeIntegers);
66
+ SetPrototypeMethod(isolate, data, t, "columns", JS_columns);
67
+ SetPrototypeMethod(isolate, data, t, "scanStatusV2", JS_scanStatusV2);
68
+ SetPrototypeGetter(isolate, data, t, "busy", JS_busy);
69
+ return t->GetFunction(OnlyContext).ToLocalChecked();
70
+ }
71
+
72
+ NODE_METHOD(Statement::JS_new) {
73
+ UseAddon;
74
+ if (!addon->privileged_info) {
75
+ return ThrowTypeError("Statements can only be constructed by the db.prepare() method");
76
+ }
77
+ assert(info.IsConstructCall());
78
+ Database* db = Unwrap<Database>(addon->privileged_info->This());
79
+ REQUIRE_DATABASE_OPEN(db->GetState());
80
+ REQUIRE_DATABASE_NOT_BUSY(db->GetState());
81
+
82
+ v8::Local<v8::String> source = (*addon->privileged_info)[0].As<v8::String>();
83
+ v8::Local<v8::Object> database = (*addon->privileged_info)[1].As<v8::Object>();
84
+ bool pragmaMode = (*addon->privileged_info)[2].As<v8::Boolean>()->Value();
85
+ int flags = SQLITE_PREPARE_PERSISTENT;
86
+
87
+ if (pragmaMode) {
88
+ REQUIRE_DATABASE_NO_ITERATORS_UNLESS_UNSAFE(db->GetState());
89
+ flags = 0;
90
+ }
91
+
92
+ UseIsolate;
93
+ v8::String::Utf8Value utf8(isolate, source);
94
+ sqlite3_stmt* handle;
95
+ const char* tail;
96
+
97
+ if (sqlite3_prepare_v3(db->GetHandle(), *utf8, utf8.length() + 1, flags, &handle, &tail) != SQLITE_OK) {
98
+ return db->ThrowDatabaseError();
99
+ }
100
+ if (handle == NULL) {
101
+ return ThrowRangeError("The supplied SQL string contains no statements");
102
+ }
103
+ // https://github.com/WiseLibs/better-sqlite3/issues/975#issuecomment-1520934678
104
+ for (char c; (c = *tail); ) {
105
+ if (IS_SKIPPED(c)) {
106
+ ++tail;
107
+ continue;
108
+ }
109
+ if (c == '/' && tail[1] == '*') {
110
+ tail += 2;
111
+ for (char c; (c = *tail); ++tail) {
112
+ if (c == '*' && tail[1] == '/') {
113
+ tail += 2;
114
+ break;
115
+ }
116
+ }
117
+ } else if (c == '-' && tail[1] == '-') {
118
+ tail += 2;
119
+ for (char c; (c = *tail); ++tail) {
120
+ if (c == '\n') {
121
+ ++tail;
122
+ break;
123
+ }
124
+ }
125
+ } else {
126
+ sqlite3_finalize(handle);
127
+ return ThrowRangeError("The supplied SQL string contains more than one statement");
128
+ }
129
+ }
130
+
131
+ UseContext;
132
+ bool returns_data = sqlite3_column_count(handle) >= 1 || pragmaMode;
133
+ Statement* stmt = new Statement(db, handle, addon->NextId(), returns_data);
134
+ stmt->Wrap(info.This());
135
+ SetFrozen(isolate, ctx, info.This(), addon->cs.reader, v8::Boolean::New(isolate, returns_data));
136
+ SetFrozen(isolate, ctx, info.This(), addon->cs.readonly, v8::Boolean::New(isolate, sqlite3_stmt_readonly(handle) != 0));
137
+ SetFrozen(isolate, ctx, info.This(), addon->cs.source, source);
138
+ SetFrozen(isolate, ctx, info.This(), addon->cs.database, database);
139
+
140
+ info.GetReturnValue().Set(info.This());
141
+ }
142
+
143
+ NODE_METHOD(Statement::JS_run) {
144
+ STATEMENT_START(ALLOW_ANY_STATEMENT, DOES_MUTATE);
145
+ sqlite3* db_handle = db->GetHandle();
146
+ int total_changes_before = sqlite3_total_changes(db_handle);
147
+
148
+ sqlite3_step(handle);
149
+ if (sqlite3_reset(handle) == SQLITE_OK) {
150
+ int changes = sqlite3_total_changes(db_handle) == total_changes_before ? 0 : sqlite3_changes(db_handle);
151
+ sqlite3_int64 id = sqlite3_last_insert_rowid(db_handle);
152
+ Addon* addon = db->GetAddon();
153
+ UseContext;
154
+ v8::Local<v8::Object> result = v8::Object::New(isolate);
155
+ result->Set(ctx, addon->cs.changes.Get(isolate), v8::Int32::New(isolate, changes)).FromJust();
156
+ result->Set(ctx, addon->cs.lastInsertRowid.Get(isolate),
157
+ stmt->safe_ints
158
+ ? v8::BigInt::New(isolate, id).As<v8::Value>()
159
+ : v8::Number::New(isolate, (double)id).As<v8::Value>()
160
+ ).FromJust();
161
+ STATEMENT_RETURN(result);
162
+ }
163
+ STATEMENT_THROW();
164
+ }
165
+
166
+ NODE_METHOD(Statement::JS_get) {
167
+ STATEMENT_START(REQUIRE_STATEMENT_RETURNS_DATA, DOES_NOT_MUTATE);
168
+ int status = sqlite3_step(handle);
169
+ if (status == SQLITE_ROW) {
170
+ v8::Local<v8::Value> result = Data::GetRowJS(isolate, OnlyContext, handle, stmt->safe_ints, stmt->mode);
171
+ sqlite3_reset(handle);
172
+ STATEMENT_RETURN(result);
173
+ } else if (status == SQLITE_DONE) {
174
+ sqlite3_reset(handle);
175
+ STATEMENT_RETURN(v8::Undefined(isolate));
176
+ }
177
+ sqlite3_reset(handle);
178
+ STATEMENT_THROW();
179
+ }
180
+
181
+ NODE_METHOD(Statement::JS_all) {
182
+ STATEMENT_START(REQUIRE_STATEMENT_RETURNS_DATA, DOES_NOT_MUTATE);
183
+ UseContext;
184
+ const bool safe_ints = stmt->safe_ints;
185
+ const char mode = stmt->mode;
186
+
187
+ #if !defined(NODE_MODULE_VERSION) || NODE_MODULE_VERSION < 127
188
+ bool js_error = false;
189
+ uint32_t row_count = 0;
190
+ v8::Local<v8::Array> result = v8::Array::New(isolate, 0);
191
+
192
+ while (sqlite3_step(handle) == SQLITE_ROW) {
193
+ if (row_count == 0xffffffff) { ThrowRangeError("Array overflow (too many rows returned)"); js_error = true; break; }
194
+ result->Set(ctx, row_count++, Data::GetRowJS(isolate, ctx, handle, safe_ints, mode)).FromJust();
195
+ }
196
+
197
+ if (sqlite3_reset(handle) == SQLITE_OK && !js_error) {
198
+ STATEMENT_RETURN(result);
199
+ }
200
+ if (js_error) db->GetState()->was_js_error = true;
201
+ STATEMENT_THROW();
202
+ #else
203
+ v8::LocalVector<v8::Value> rows(isolate);
204
+ rows.reserve(8);
205
+
206
+ if (mode == Data::FLAT) {
207
+ RowBuilder rowBuilder(isolate, handle, safe_ints);
208
+ while (sqlite3_step(handle) == SQLITE_ROW) {
209
+ rows.emplace_back(rowBuilder.GetRowJS());
210
+ }
211
+ } else {
212
+ while (sqlite3_step(handle) == SQLITE_ROW) {
213
+ rows.emplace_back(Data::GetRowJS(isolate, ctx, handle, safe_ints, mode));
214
+ }
215
+ }
216
+
217
+ if (sqlite3_reset(handle) == SQLITE_OK) {
218
+ if (rows.size() > 0xffffffff) {
219
+ ThrowRangeError("Array overflow (too many rows returned)");
220
+ db->GetState()->was_js_error = true;
221
+ } else {
222
+ STATEMENT_RETURN(v8::Array::New(isolate, rows.data(), rows.size()));
223
+ }
224
+ }
225
+ STATEMENT_THROW();
226
+ #endif
227
+ }
228
+
229
+ NODE_METHOD(Statement::JS_iterate) {
230
+ UseAddon;
231
+ UseIsolate;
232
+ v8::Local<v8::Function> c = addon->StatementIterator.Get(isolate);
233
+ addon->privileged_info = &info;
234
+ v8::MaybeLocal<v8::Object> maybeIterator = c->NewInstance(OnlyContext, 0, NULL);
235
+ addon->privileged_info = NULL;
236
+ if (!maybeIterator.IsEmpty()) info.GetReturnValue().Set(maybeIterator.ToLocalChecked());
237
+ }
238
+
239
+ NODE_METHOD(Statement::JS_bind) {
240
+ Statement* stmt = Unwrap<Statement>(info.This());
241
+ if (stmt->bound) return ThrowTypeError("The bind() method can only be invoked once per statement object");
242
+ REQUIRE_DATABASE_OPEN(stmt->db->GetState());
243
+ REQUIRE_DATABASE_NOT_BUSY(stmt->db->GetState());
244
+ REQUIRE_STATEMENT_NOT_LOCKED(stmt);
245
+ STATEMENT_BIND(stmt->handle);
246
+ stmt->bound = true;
247
+ info.GetReturnValue().Set(info.This());
248
+ }
249
+
250
+ NODE_METHOD(Statement::JS_pluck) {
251
+ Statement* stmt = Unwrap<Statement>(info.This());
252
+ if (!stmt->returns_data) return ThrowTypeError("The pluck() method is only for statements that return data");
253
+ REQUIRE_DATABASE_NOT_BUSY(stmt->db->GetState());
254
+ REQUIRE_STATEMENT_NOT_LOCKED(stmt);
255
+ bool use = true;
256
+ if (info.Length() != 0) { REQUIRE_ARGUMENT_BOOLEAN(first, use); }
257
+ stmt->mode = use ? Data::PLUCK : stmt->mode == Data::PLUCK ? Data::FLAT : stmt->mode;
258
+ info.GetReturnValue().Set(info.This());
259
+ }
260
+
261
+ NODE_METHOD(Statement::JS_expand) {
262
+ Statement* stmt = Unwrap<Statement>(info.This());
263
+ if (!stmt->returns_data) return ThrowTypeError("The expand() method is only for statements that return data");
264
+ REQUIRE_DATABASE_NOT_BUSY(stmt->db->GetState());
265
+ REQUIRE_STATEMENT_NOT_LOCKED(stmt);
266
+ bool use = true;
267
+ if (info.Length() != 0) { REQUIRE_ARGUMENT_BOOLEAN(first, use); }
268
+ stmt->mode = use ? Data::EXPAND : stmt->mode == Data::EXPAND ? Data::FLAT : stmt->mode;
269
+ info.GetReturnValue().Set(info.This());
270
+ }
271
+
272
+ NODE_METHOD(Statement::JS_raw) {
273
+ Statement* stmt = Unwrap<Statement>(info.This());
274
+ if (!stmt->returns_data) return ThrowTypeError("The raw() method is only for statements that return data");
275
+ REQUIRE_DATABASE_NOT_BUSY(stmt->db->GetState());
276
+ REQUIRE_STATEMENT_NOT_LOCKED(stmt);
277
+ bool use = true;
278
+ if (info.Length() != 0) { REQUIRE_ARGUMENT_BOOLEAN(first, use); }
279
+ stmt->mode = use ? Data::RAW : stmt->mode == Data::RAW ? Data::FLAT : stmt->mode;
280
+ info.GetReturnValue().Set(info.This());
281
+ }
282
+
283
+ NODE_METHOD(Statement::JS_safeIntegers) {
284
+ Statement* stmt = Unwrap<Statement>(info.This());
285
+ REQUIRE_DATABASE_NOT_BUSY(stmt->db->GetState());
286
+ REQUIRE_STATEMENT_NOT_LOCKED(stmt);
287
+ if (info.Length() == 0) stmt->safe_ints = true;
288
+ else { REQUIRE_ARGUMENT_BOOLEAN(first, stmt->safe_ints); }
289
+ info.GetReturnValue().Set(info.This());
290
+ }
291
+
292
+ NODE_METHOD(Statement::JS_columns) {
293
+ Statement* stmt = Unwrap<Statement>(info.This());
294
+ if (!stmt->returns_data) return ThrowTypeError("The columns() method is only for statements that return data");
295
+ REQUIRE_DATABASE_OPEN(stmt->db->GetState());
296
+ REQUIRE_DATABASE_NOT_BUSY(stmt->db->GetState());
297
+ Addon* addon = stmt->db->GetAddon();
298
+ UseIsolate;
299
+
300
+ #if !defined(NODE_MODULE_VERSION) || NODE_MODULE_VERSION < 127
301
+ UseContext;
302
+ int column_count = sqlite3_column_count(stmt->handle);
303
+ v8::Local<v8::Array> columns = v8::Array::New(isolate);
304
+
305
+ v8::Local<v8::String> name = addon->cs.name.Get(isolate);
306
+ v8::Local<v8::String> columnName = addon->cs.column.Get(isolate);
307
+ v8::Local<v8::String> tableName = addon->cs.table.Get(isolate);
308
+ v8::Local<v8::String> databaseName = addon->cs.database.Get(isolate);
309
+ v8::Local<v8::String> typeName = addon->cs.type.Get(isolate);
310
+
311
+ for (int i = 0; i < column_count; ++i) {
312
+ v8::Local<v8::Object> column = v8::Object::New(isolate);
313
+
314
+ column->Set(ctx, name,
315
+ InternalizedFromUtf8OrNull(isolate, sqlite3_column_name(stmt->handle, i), -1)
316
+ ).FromJust();
317
+ column->Set(ctx, columnName,
318
+ InternalizedFromUtf8OrNull(isolate, sqlite3_column_origin_name(stmt->handle, i), -1)
319
+ ).FromJust();
320
+ column->Set(ctx, tableName,
321
+ InternalizedFromUtf8OrNull(isolate, sqlite3_column_table_name(stmt->handle, i), -1)
322
+ ).FromJust();
323
+ column->Set(ctx, databaseName,
324
+ InternalizedFromUtf8OrNull(isolate, sqlite3_column_database_name(stmt->handle, i), -1)
325
+ ).FromJust();
326
+ column->Set(ctx, typeName,
327
+ InternalizedFromUtf8OrNull(isolate, sqlite3_column_decltype(stmt->handle, i), -1)
328
+ ).FromJust();
329
+
330
+ columns->Set(ctx, i, column).FromJust();
331
+ }
332
+
333
+ info.GetReturnValue().Set(columns);
334
+ #else
335
+ v8::LocalVector<v8::Name> keys(isolate);
336
+ keys.reserve(5);
337
+ keys.emplace_back(addon->cs.name.Get(isolate).As<v8::Name>());
338
+ keys.emplace_back(addon->cs.column.Get(isolate).As<v8::Name>());
339
+ keys.emplace_back(addon->cs.table.Get(isolate).As<v8::Name>());
340
+ keys.emplace_back(addon->cs.database.Get(isolate).As<v8::Name>());
341
+ keys.emplace_back(addon->cs.type.Get(isolate).As<v8::Name>());
342
+
343
+ int column_count = sqlite3_column_count(stmt->handle);
344
+ v8::LocalVector<v8::Value> columns(isolate);
345
+ columns.reserve(column_count);
346
+
347
+ for (int i = 0; i < column_count; ++i) {
348
+ v8::LocalVector<v8::Value> values(isolate);
349
+ keys.reserve(5);
350
+ values.emplace_back(
351
+ InternalizedFromUtf8OrNull(isolate, sqlite3_column_name(stmt->handle, i), -1)
352
+ );
353
+ values.emplace_back(
354
+ InternalizedFromUtf8OrNull(isolate, sqlite3_column_origin_name(stmt->handle, i), -1)
355
+ );
356
+ values.emplace_back(
357
+ InternalizedFromUtf8OrNull(isolate, sqlite3_column_table_name(stmt->handle, i), -1)
358
+ );
359
+ values.emplace_back(
360
+ InternalizedFromUtf8OrNull(isolate, sqlite3_column_database_name(stmt->handle, i), -1)
361
+ );
362
+ values.emplace_back(
363
+ InternalizedFromUtf8OrNull(isolate, sqlite3_column_decltype(stmt->handle, i), -1)
364
+ );
365
+ columns.emplace_back(
366
+ v8::Object::New(isolate,
367
+ v8::Object::New(isolate)->GetPrototype(),
368
+ keys.data(),
369
+ values.data(),
370
+ keys.size()
371
+ )
372
+ );
373
+ }
374
+
375
+ info.GetReturnValue().Set(
376
+ v8::Array::New(isolate, columns.data(), columns.size())
377
+ );
378
+ #endif
379
+ }
380
+
381
+ NODE_METHOD(Statement::JS_scanStatusV2) {
382
+ Statement* stmt = Unwrap<Statement>(info.This());
383
+ REQUIRE_DATABASE_OPEN(stmt->db->GetState());
384
+ REQUIRE_DATABASE_NOT_BUSY(stmt->db->GetState());
385
+
386
+ // Validate arguments
387
+ REQUIRE_ARGUMENT_INT32(first, int idx);
388
+ REQUIRE_ARGUMENT_INT32(second, int opcode);
389
+ REQUIRE_ARGUMENT_INT32(third, int flags);
390
+
391
+ UseIsolate;
392
+ int rc;
393
+
394
+ // Call sqlite3_stmt_scanstatus_v2 with appropriate output type based on opcode
395
+ switch (opcode) {
396
+ case SQLITE_SCANSTAT_NAME:
397
+ case SQLITE_SCANSTAT_EXPLAIN: {
398
+ // String opcodes
399
+ const char* pOut;
400
+ rc = sqlite3_stmt_scanstatus_v2(stmt->handle, idx, opcode, flags, (void*)&pOut);
401
+ if (rc == SQLITE_OK && pOut != NULL) {
402
+ info.GetReturnValue().Set(InternalizedFromUtf8(isolate, pOut, -1));
403
+ } else {
404
+ info.GetReturnValue().Set(v8::Undefined(isolate));
405
+ }
406
+ return;
407
+ }
408
+ case SQLITE_SCANSTAT_EST: {
409
+ // Double opcode
410
+ double pOut;
411
+ rc = sqlite3_stmt_scanstatus_v2(stmt->handle, idx, opcode, flags, (void*)&pOut);
412
+ if (rc == SQLITE_OK) {
413
+ info.GetReturnValue().Set(v8::Number::New(isolate, pOut));
414
+ } else {
415
+ info.GetReturnValue().Set(v8::Undefined(isolate));
416
+ }
417
+ return;
418
+ }
419
+ case SQLITE_SCANSTAT_NLOOP:
420
+ case SQLITE_SCANSTAT_NVISIT:
421
+ case SQLITE_SCANSTAT_SELECTID:
422
+ case SQLITE_SCANSTAT_PARENTID:
423
+ case SQLITE_SCANSTAT_NCYCLE: {
424
+ // Integer opcodes
425
+ sqlite3_int64 pOut;
426
+ rc = sqlite3_stmt_scanstatus_v2(stmt->handle, idx, opcode, flags, (void*)&pOut);
427
+ if (rc == SQLITE_OK) {
428
+ if (stmt->safe_ints) {
429
+ info.GetReturnValue().Set(v8::BigInt::New(isolate, pOut));
430
+ } else {
431
+ info.GetReturnValue().Set(v8::Number::New(isolate, (double)pOut));
432
+ }
433
+ } else {
434
+ info.GetReturnValue().Set(v8::Undefined(isolate));
435
+ }
436
+ return;
437
+ }
438
+ default:
439
+ return ThrowRangeError("Invalid scanstatus opcode");
440
+ }
441
+ }
442
+
443
+ NODE_GETTER(Statement::JS_busy) {
444
+ Statement* stmt = Unwrap<Statement>(info.This());
445
+ info.GetReturnValue().Set(stmt->alive && stmt->locked);
446
+ }
@@ -0,0 +1,59 @@
1
+ class Statement : public node::ObjectWrap { friend class StatementIterator;
2
+ public:
3
+
4
+ ~Statement();
5
+
6
+ // Whenever this is used, db->RemoveStatement must be invoked beforehand.
7
+ void CloseHandles();
8
+
9
+ // Used to support ordered containers.
10
+ static inline bool Compare(Statement const * const a, Statement const * const b) {
11
+ return a->extras->id < b->extras->id;
12
+ }
13
+
14
+ // Returns the Statement's bind map (creates it upon first execution).
15
+ BindMap* GetBindMap(v8::Isolate* isolate);
16
+
17
+ static INIT(Init);
18
+
19
+ private:
20
+
21
+ // A class for holding values that are less often used.
22
+ class Extras { friend class Statement;
23
+ explicit Extras(sqlite3_uint64 id);
24
+ BindMap bind_map;
25
+ const sqlite3_uint64 id;
26
+ };
27
+
28
+ explicit Statement(
29
+ Database* db,
30
+ sqlite3_stmt* handle,
31
+ sqlite3_uint64 id,
32
+ bool returns_data
33
+ );
34
+
35
+ static NODE_METHOD(JS_new);
36
+ static NODE_METHOD(JS_run);
37
+ static NODE_METHOD(JS_get);
38
+ static NODE_METHOD(JS_all);
39
+ static NODE_METHOD(JS_iterate);
40
+ static NODE_METHOD(JS_bind);
41
+ static NODE_METHOD(JS_pluck);
42
+ static NODE_METHOD(JS_expand);
43
+ static NODE_METHOD(JS_raw);
44
+ static NODE_METHOD(JS_safeIntegers);
45
+ static NODE_METHOD(JS_columns);
46
+ static NODE_METHOD(JS_scanStatusV2);
47
+ static NODE_GETTER(JS_busy);
48
+
49
+ Database* const db;
50
+ sqlite3_stmt* const handle;
51
+ Extras* const extras;
52
+ bool alive;
53
+ bool locked;
54
+ bool bound;
55
+ bool has_bind_map;
56
+ bool safe_ints;
57
+ char mode;
58
+ const bool returns_data;
59
+ };