mirror of
https://github.com/Egor00f/luasqlite3-kolibrios.git
synced 2025-09-11 14:39:57 +02:00
upload luasqlite3
This commit is contained in:
349
examples/hooks_advanced.lua
Normal file
349
examples/hooks_advanced.lua
Normal file
@@ -0,0 +1,349 @@
|
||||
|
||||
local sqlite3 = require("lsqlite3")
|
||||
local db
|
||||
|
||||
optbl = { [sqlite3.UPDATE] = "UPDATE";
|
||||
[sqlite3.INSERT] = "INSERT";
|
||||
[sqlite3.DELETE] = "DELETE"
|
||||
}
|
||||
setmetatable(optbl,
|
||||
{__index=function(t,n) return string.format("Unknown op %d",n) end})
|
||||
|
||||
|
||||
function update_hook(ud, op, dname, tname, rowid)
|
||||
print("Sqlite Update Hook:", optbl[op], dname, tname, rowid)
|
||||
end
|
||||
|
||||
function commit_hook(ud)
|
||||
print("<Sqlite Commit Hook>")
|
||||
end
|
||||
|
||||
function rollback_hook(ud)
|
||||
print("<Sqlite Rollback Hook>")
|
||||
end
|
||||
|
||||
-- db:exec wrapper with result assertion
|
||||
function db_exec(stmt)
|
||||
if db:exec(stmt) ~= sqlite3.OK then
|
||||
print("Sqlite ERROR: ", db:errmsg())
|
||||
end
|
||||
end
|
||||
|
||||
-- debug output for database table
|
||||
function db_print_tables(...)
|
||||
for i = 1,select('#',...) do
|
||||
name = select(i,...)
|
||||
print(string.format("\n%s contents:", name))
|
||||
for row in db:nrows("SELECT * FROM " .. name) do
|
||||
print(row.id, row.content)
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
local tests = {}
|
||||
|
||||
-- runs specified test with all necessary setup
|
||||
-- @param string name
|
||||
function run_test(name)
|
||||
-- banner
|
||||
print(string.format([[
|
||||
|
||||
|
||||
==========================
|
||||
%s
|
||||
==========================
|
||||
]], name))
|
||||
|
||||
-- setup
|
||||
db = sqlite3.open_memory()
|
||||
local udtbl = {0, 0, 0}
|
||||
db:update_hook(update_hook, udtbl)
|
||||
db:commit_hook(commit_hook, udtbl)
|
||||
db:rollback_hook(rollback_hook, udtbl)
|
||||
|
||||
-- run test
|
||||
tests[name]()
|
||||
|
||||
-- destroy
|
||||
db:close()
|
||||
end
|
||||
|
||||
|
||||
function tests.insert_select()
|
||||
db_exec[[
|
||||
CREATE TABLE T1 ( id INTEGER PRIMARY KEY, content VARCHAR );
|
||||
CREATE TABLE T2 ( id INTEGER PRIMARY KEY, content VARCHAR );
|
||||
|
||||
INSERT INTO T1 VALUES (NULL, 'Hello World');
|
||||
INSERT INTO T1 VALUES (NULL, 'Hello Lua');
|
||||
INSERT INTO T1 VALUES (NULL, 'Hello Sqlite3');
|
||||
|
||||
INSERT INTO T2 SELECT * FROM T1;
|
||||
]]
|
||||
|
||||
db_print_tables('T1', 'T2')
|
||||
end
|
||||
|
||||
function tests.trigger_insert()
|
||||
db_exec[[
|
||||
CREATE TABLE T1 ( id INTEGER PRIMARY KEY, content VARCHAR );
|
||||
CREATE TABLE T2 ( id INTEGER PRIMARY KEY, content VARCHAR );
|
||||
|
||||
CREATE TRIGGER after_insert_T1
|
||||
AFTER INSERT ON T1
|
||||
FOR EACH ROW
|
||||
BEGIN
|
||||
INSERT INTO T2 VALUES(NEW.id, NEW.content);
|
||||
END;
|
||||
|
||||
INSERT INTO T1 VALUES (NULL, 'Hello World');
|
||||
INSERT INTO T1 VALUES (NULL, 'Hello Lua');
|
||||
INSERT INTO T1 VALUES (NULL, 'Hello Sqlite3');
|
||||
]]
|
||||
|
||||
db_print_tables('T1', 'T2')
|
||||
end
|
||||
|
||||
function tests.cascade_delete()
|
||||
db_exec[[
|
||||
PRAGMA foreign_keys = ON;
|
||||
|
||||
CREATE TABLE T1 ( id INTEGER PRIMARY KEY, content VARCHAR );
|
||||
CREATE TABLE T2 ( id INTEGER PRIMARY KEY REFERENCES T1 ON DELETE CASCADE, content VARCHAR );
|
||||
CREATE TABLE T3 ( id INTEGER PRIMARY KEY REFERENCES T2 ON DELETE CASCADE, content VARCHAR );
|
||||
|
||||
INSERT INTO T1 VALUES (NULL, 'a');
|
||||
INSERT INTO T1 VALUES (NULL, 'b');
|
||||
INSERT INTO T1 VALUES (NULL, 'c');
|
||||
INSERT INTO T1 VALUES (NULL, 'd');
|
||||
|
||||
INSERT INTO T2 SELECT * FROM T1;
|
||||
INSERT INTO T3 SELECT * FROM T2;
|
||||
|
||||
DELETE FROM T1 WHERE id < 3;
|
||||
]]
|
||||
|
||||
db_print_tables('T1', 'T2', 'T3')
|
||||
end
|
||||
|
||||
function tests.cascade_update()
|
||||
db_exec[[
|
||||
PRAGMA foreign_keys = ON;
|
||||
|
||||
CREATE TABLE T1 ( id INTEGER PRIMARY KEY, content VARCHAR );
|
||||
CREATE TABLE T2 ( id INTEGER PRIMARY KEY REFERENCES T1 ON UPDATE CASCADE, content VARCHAR );
|
||||
CREATE TABLE T3 ( id INTEGER PRIMARY KEY REFERENCES T2 ON UPDATE CASCADE, content VARCHAR );
|
||||
|
||||
INSERT INTO T1 VALUES (NULL, 'a');
|
||||
INSERT INTO T1 VALUES (NULL, 'b');
|
||||
INSERT INTO T1 VALUES (NULL, 'c');
|
||||
INSERT INTO T1 VALUES (NULL, 'd');
|
||||
|
||||
INSERT INTO T2 SELECT * FROM T1;
|
||||
INSERT INTO T3 SELECT * FROM T2;
|
||||
|
||||
UPDATE T1 SET id = id + 10 WHERE id < 3;
|
||||
]]
|
||||
|
||||
db_print_tables('T1', 'T2', 'T3')
|
||||
end
|
||||
|
||||
-- hook "anomaly"
|
||||
-- implicit one-statement transaction rollback demonstration
|
||||
function tests.cascade_update_restrict()
|
||||
db_exec[[
|
||||
PRAGMA foreign_keys = ON;
|
||||
|
||||
CREATE TABLE T1 ( id INTEGER PRIMARY KEY, content VARCHAR );
|
||||
CREATE TABLE T2 ( id INTEGER PRIMARY KEY REFERENCES T1 ON UPDATE RESTRICT, content VARCHAR );
|
||||
CREATE TABLE T3 ( id INTEGER PRIMARY KEY REFERENCES T2 ON UPDATE RESTRICT, content VARCHAR );
|
||||
|
||||
INSERT INTO T1 VALUES (NULL, 'a');
|
||||
INSERT INTO T1 VALUES (NULL, 'b');
|
||||
INSERT INTO T1 VALUES (NULL, 'c');
|
||||
INSERT INTO T1 VALUES (NULL, 'd');
|
||||
|
||||
INSERT INTO T2 SELECT * FROM T1;
|
||||
INSERT INTO T3 SELECT * FROM T2;
|
||||
|
||||
-- this update gets reverted but the update_hook with rowid=11 *DOES* get triggered
|
||||
UPDATE T1 SET id = id + 10 WHERE id < 3;
|
||||
]]
|
||||
|
||||
db_print_tables('T1', 'T2', 'T3')
|
||||
end
|
||||
|
||||
-- hook "anomaly"
|
||||
-- case is analogous to cascade_update_restrict
|
||||
function tests.cascade_delete_restrict()
|
||||
db_exec[[
|
||||
PRAGMA foreign_keys = ON;
|
||||
|
||||
CREATE TABLE T1 ( id INTEGER PRIMARY KEY, content VARCHAR );
|
||||
CREATE TABLE T2 ( id INTEGER PRIMARY KEY REFERENCES T1 ON DELETE RESTRICT, content VARCHAR );
|
||||
CREATE TABLE T3 ( id INTEGER PRIMARY KEY REFERENCES T2 ON DELETE RESTRICT, content VARCHAR );
|
||||
|
||||
INSERT INTO T1 VALUES (NULL, 'a');
|
||||
INSERT INTO T1 VALUES (NULL, 'b');
|
||||
INSERT INTO T1 VALUES (NULL, 'c');
|
||||
INSERT INTO T1 VALUES (NULL, 'd');
|
||||
|
||||
INSERT INTO T2 SELECT * FROM T1;
|
||||
INSERT INTO T3 SELECT * FROM T2;
|
||||
|
||||
DELETE FROM T1 WHERE id < 3;
|
||||
]]
|
||||
|
||||
db_print_tables('T1', 'T2', 'T3')
|
||||
end
|
||||
|
||||
-- no anomalies here
|
||||
function tests.fk_violate_insert()
|
||||
db_exec[[
|
||||
PRAGMA foreign_keys = ON;
|
||||
|
||||
CREATE TABLE T1 ( id INTEGER PRIMARY KEY, content VARCHAR );
|
||||
CREATE TABLE T2 ( id INTEGER PRIMARY KEY REFERENCES T1, content VARCHAR);
|
||||
|
||||
INSERT INTO T1 VALUES (NULL, 'a');
|
||||
INSERT INTO T1 VALUES (NULL, 'b');
|
||||
|
||||
INSERT INTO T2 VALUES(99, 'xxx');
|
||||
]]
|
||||
|
||||
db_print_tables('T1', 'T2')
|
||||
end
|
||||
|
||||
-- hook "anomaly"
|
||||
function tests.fk_violate_update()
|
||||
db_exec[[
|
||||
PRAGMA foreign_keys = ON;
|
||||
|
||||
CREATE TABLE T1 ( id INTEGER PRIMARY KEY, content VARCHAR );
|
||||
CREATE TABLE T2 ( id INTEGER PRIMARY KEY REFERENCES T1, content VARCHAR);
|
||||
|
||||
INSERT INTO T1 VALUES (NULL, 'a');
|
||||
INSERT INTO T1 VALUES (NULL, 'b');
|
||||
INSERT INTO T2 VALUES(1, 'a');
|
||||
|
||||
-- update doesn't succeed but we get a hook for tuple with rowid=99 in T2
|
||||
UPDATE T2 SET id = 99 WHERE id = 1;
|
||||
]]
|
||||
|
||||
db_print_tables('T1', 'T2')
|
||||
end
|
||||
|
||||
-- like fk_violate_update but wrapped inside an explicit transaction
|
||||
function tests.transaction_fk_violate_update()
|
||||
db_exec[[
|
||||
PRAGMA foreign_keys = ON;
|
||||
|
||||
CREATE TABLE T1 ( id INTEGER PRIMARY KEY, content VARCHAR );
|
||||
CREATE TABLE T2 ( id INTEGER PRIMARY KEY REFERENCES T1, content VARCHAR);
|
||||
|
||||
BEGIN TRANSACTION;
|
||||
INSERT INTO T1 VALUES (NULL, 'a');
|
||||
INSERT INTO T1 VALUES (NULL, 'b');
|
||||
INSERT INTO T2 VALUES(1, 'a');
|
||||
|
||||
-- Doesn't trigger rollback hook because the implicit update statement transaction
|
||||
-- is nested inside our explicit transaction. However we *do* get an error.
|
||||
UPDATE T2 SET id = 99 WHERE id = 1;
|
||||
COMMIT;
|
||||
]]
|
||||
|
||||
db_print_tables('T1', 'T2')
|
||||
end
|
||||
|
||||
function tests.cascade_update_setnull()
|
||||
db_exec[[
|
||||
PRAGMA foreign_keys = ON;
|
||||
|
||||
CREATE TABLE T1 ( id INTEGER PRIMARY KEY, content VARCHAR );
|
||||
CREATE TABLE T2 ( id INTEGER PRIMARY KEY, content INTEGER REFERENCES T1(id) ON UPDATE SET NULL);
|
||||
|
||||
INSERT INTO T1 VALUES (NULL, 'a');
|
||||
INSERT INTO T1 VALUES (NULL, 'b');
|
||||
INSERT INTO T1 VALUES (NULL, 'c');
|
||||
INSERT INTO T1 VALUES (NULL, 'd');
|
||||
|
||||
INSERT INTO T2 SELECT NULL, id FROM T1;
|
||||
|
||||
UPDATE T1 SET id = id + 10 WHERE id < 3;
|
||||
]]
|
||||
|
||||
db_print_tables('T1', 'T2')
|
||||
end
|
||||
|
||||
function tests.transaction_commit()
|
||||
db_exec[[
|
||||
CREATE TABLE T1 ( id INTEGER PRIMARY KEY, content VARCHAR );
|
||||
|
||||
BEGIN TRANSACTION;
|
||||
INSERT INTO T1 VALUES (NULL, 'Hello World');
|
||||
INSERT INTO T1 VALUES (NULL, 'Hello Lua');
|
||||
COMMIT;
|
||||
]]
|
||||
|
||||
db_print_tables('T1')
|
||||
end
|
||||
|
||||
function tests.transaction_rollback()
|
||||
db_exec[[
|
||||
CREATE TABLE T1 ( id INTEGER PRIMARY KEY, content VARCHAR );
|
||||
|
||||
BEGIN TRANSACTION;
|
||||
INSERT INTO T1 VALUES (NULL, 'Hello World');
|
||||
INSERT INTO T1 VALUES (NULL, 'Hello Lua');
|
||||
ROLLBACK;
|
||||
]]
|
||||
|
||||
db_print_tables('T1')
|
||||
end
|
||||
|
||||
function tests.savepoint_nested_commit()
|
||||
db_exec[[
|
||||
CREATE TABLE T1 ( id INTEGER PRIMARY KEY, content VARCHAR );
|
||||
|
||||
SAVEPOINT S1;
|
||||
INSERT INTO T1 VALUES (NULL, 'Hello World');
|
||||
INSERT INTO T1 VALUES (NULL, 'Hello Lua');
|
||||
|
||||
SAVEPOINT S2;
|
||||
INSERT INTO T1 VALUES (NULL, 'Hello Sqlite3');
|
||||
-- nested commit doesn't trigger commit_hook
|
||||
RELEASE S2;
|
||||
|
||||
INSERT INTO T1 VALUES (NULL, 'Hello transactions');
|
||||
RELEASE S1;
|
||||
]]
|
||||
|
||||
db_print_tables('T1')
|
||||
end
|
||||
|
||||
function tests.savepoint_nested_rollback()
|
||||
db_exec[[
|
||||
CREATE TABLE T1 ( id INTEGER PRIMARY KEY, content VARCHAR );
|
||||
|
||||
SAVEPOINT S1;
|
||||
INSERT INTO T1 VALUES (NULL, 'Hello World');
|
||||
INSERT INTO T1 VALUES (NULL, 'Hello Lua');
|
||||
|
||||
SAVEPOINT S2;
|
||||
INSERT INTO T1 VALUES (NULL, 'Hello Sqlite3');
|
||||
-- nested rollback doesn't trigger rollback_hook
|
||||
ROLLBACK TO S2;
|
||||
|
||||
INSERT INTO T1 VALUES (NULL, 'Hello transactions');
|
||||
RELEASE S1;
|
||||
]]
|
||||
|
||||
db_print_tables('T1')
|
||||
end
|
||||
|
||||
|
||||
-- run_test('fk_violate_insert')
|
||||
|
||||
for k,v in pairs(tests) do
|
||||
run_test(k)
|
||||
end
|
||||
Reference in New Issue
Block a user