Lua 数据库访问方法
Lua 本身不提供内置的数据库访问功能,但可以通过第三方库实现与多种数据库的交互。以下是常见的 Lua 数据库访问方法:
使用 LuaSQL 库
LuaSQL 是一个轻量级数据库访问库,支持多种数据库后端(MySQL、PostgreSQL、SQLite、ODBC 等)。
安装方式:
bash
luarocks install luasql-mysql # MySQL
luarocks install luasql-sqlite3 # SQLite
示例代码(MySQL):
lua
local luasql = require "luasql.mysql"
local env = luasql.mysql()
local conn = env:connect("database", "username", "password", "host", port)
-- 执行查询
local cursor = conn:execute("SELECT * FROM users")
local row = cursor:fetch({}, "a")
while row do
print(row.id, row.name)
row = cursor:fetch(row, "a")
end
-- 关闭连接
cursor:close()
conn:close()
env:close()
使用 SQLite 的 lsqlite3 库
对于嵌入式数据库 SQLite,可以使用 lsqlite3
库。
安装方式:
bash
luarocks install lsqlite3
示例代码:
lua
local sqlite3 = require "lsqlite3"
local db = sqlite3.open("test.db")
-- 执行查询
db:exec[[
CREATE TABLE test (id INTEGER PRIMARY KEY, content TEXT);
INSERT INTO test VALUES (NULL, 'Hello World');
]]
-- 读取数据
for row in db:nrows("SELECT * FROM test") do
print(row.id, row.content)
end
db:close()
使用 Redis 的 lua-resty-redis 库
如果需要访问 Redis,可以使用 OpenResty 提供的 lua-resty-redis
库。
安装方式:
bash
luarocks install lua-resty-redis
示例代码:
lua
local redis = require "resty.redis"
local red = redis:new()
red:set_timeout(1000) -- 1 second
local ok, err = red:connect("127.0.0.1", 6379)
if not ok then
ngx.say("failed to connect: ", err)
return
end
-- 设置和获取值
red:set("dog", "an animal")
local res, err = red:get("dog")
ngx.say(res) -- 输出 "an animal"
-- 关闭连接
red:close()
使用通用数据库驱动(ODBC)
对于需要通过 ODBC 连接的数据库,可以使用 LuaSQL 的 ODBC 驱动。
安装方式:
bash
luarocks install luasql-odbc
示例代码:
lua
local luasql = require "luasql.odbc"
local env = luasql.odbc()
local conn = env:connect("DSN=mydsn;UID=user;PWD=pass")
-- 执行查询
local cursor = conn:execute("SELECT * FROM products")
-- 处理结果...
cursor:close()
conn:close()
env:close()
使用 ORM 框架(如 LuaORM)
对于更高级的数据库操作,可以使用 ORM 框架如 LuaORM
。
安装方式:
bash
luarocks install luaorm
示例代码:
lua
local orm = require "luaorm"
orm.init("sqlite3", "test.db")
local User = orm.define("users", {
{ name = "id", type = "integer", primary = true },
{ name = "name", type = "text" }
})
-- 创建表
User:create_table()
-- 插入数据
local new_user = User:new{ name = "Alice" }
new_user:save()
-- 查询数据
for _, user in ipairs(User:find_all()) do
print(user.id, user.name)
end
注意事项
- 连接池管理:在高并发场景下,建议使用连接池管理数据库连接,避免频繁创建和销毁连接。
- 错误处理:所有数据库操作都应包含错误处理逻辑,确保程序健壮性。
- SQL 注入:避免直接拼接 SQL 语句,使用参数化查询防止注入攻击。
- 性能优化:批量操作数据时,使用事务可以提高性能。
这些方法覆盖了 Lua 中访问常见数据库的主要方式,开发者可以根据项目需求选择合适的方案。